mirror of
https://github.com/kasmtech/KasmVNC.git
synced 2024-12-01 04:23:22 +01:00
150 lines
4.3 KiB
C
150 lines
4.3 KiB
C
#include <openssl/ssl.h>
|
|
#include <stdint.h>
|
|
|
|
#define BUFSIZE 65536
|
|
#define DBUFSIZE (BUFSIZE * 3) / 4 - 20
|
|
|
|
#define SERVER_HANDSHAKE_HIXIE "HTTP/1.1 101 Web Socket Protocol Handshake\r\n\
|
|
Upgrade: WebSocket\r\n\
|
|
Connection: Upgrade\r\n\
|
|
%sWebSocket-Origin: %s\r\n\
|
|
%sWebSocket-Location: %s://%s%s\r\n\
|
|
%sWebSocket-Protocol: %s\r\n\
|
|
\r\n%s"
|
|
|
|
#define SERVER_HANDSHAKE_HYBI "HTTP/1.1 101 Switching Protocols\r\n\
|
|
Upgrade: websocket\r\n\
|
|
Connection: Upgrade\r\n\
|
|
Sec-WebSocket-Accept: %s\r\n\
|
|
Sec-WebSocket-Protocol: %s\r\n\
|
|
\r\n"
|
|
|
|
#define HYBI_GUID "258EAFA5-E914-47DA-95CA-C5AB0DC85B11"
|
|
|
|
#define HYBI10_ACCEPTHDRLEN 29
|
|
|
|
#define HIXIE_MD5_DIGEST_LENGTH 16
|
|
|
|
#define POLICY_RESPONSE "<cross-domain-policy><allow-access-from domain=\"*\" to-ports=\"*\" /></cross-domain-policy>\n"
|
|
|
|
#define OPCODE_TEXT 0x01
|
|
#define OPCODE_BINARY 0x02
|
|
|
|
typedef struct {
|
|
char path[1024+1];
|
|
char host[1024+1];
|
|
char origin[1024+1];
|
|
char version[1024+1];
|
|
char connection[1024+1];
|
|
char protocols[1024+1];
|
|
char key1[1024+1];
|
|
char key2[1024+1];
|
|
char key3[8+1];
|
|
} headers_t;
|
|
|
|
typedef struct {
|
|
int sockfd;
|
|
SSL_CTX *ssl_ctx;
|
|
SSL *ssl;
|
|
int hixie;
|
|
int hybi;
|
|
int opcode;
|
|
headers_t *headers;
|
|
char *cin_buf;
|
|
char *cout_buf;
|
|
char *tin_buf;
|
|
char *tout_buf;
|
|
|
|
char user[32];
|
|
char ip[64];
|
|
} ws_ctx_t;
|
|
|
|
struct wspass_t {
|
|
int csock;
|
|
unsigned id;
|
|
char ip[64];
|
|
};
|
|
|
|
typedef struct {
|
|
int verbose;
|
|
int listen_sock;
|
|
unsigned int handler_id;
|
|
const char *cert;
|
|
const char *key;
|
|
uint8_t disablebasicauth;
|
|
const char *passwdfile;
|
|
int ssl_only;
|
|
const char *httpdir;
|
|
|
|
void *messager;
|
|
uint8_t *(*screenshotCb)(void *messager, uint16_t w, uint16_t h, const uint8_t q,
|
|
const uint8_t dedup,
|
|
uint32_t *len, uint8_t *staging);
|
|
uint8_t (*adduserCb)(void *messager, const char name[], const char pw[],
|
|
const uint8_t write);
|
|
uint8_t (*removeCb)(void *messager, const char name[]);
|
|
uint8_t (*givecontrolCb)(void *messager, const char name[]);
|
|
void (*bottleneckStatsCb)(void *messager, char *buf, uint32_t len);
|
|
void (*frameStatsCb)(void *messager, char *buf, uint32_t len);
|
|
|
|
uint8_t (*requestFrameStatsNoneCb)(void *messager);
|
|
uint8_t (*requestFrameStatsOwnerCb)(void *messager);
|
|
uint8_t (*requestFrameStatsAllCb)(void *messager);
|
|
uint8_t (*requestFrameStatsOneCb)(void *messager, const char *client);
|
|
|
|
uint8_t (*ownerConnectedCb)(void *messager);
|
|
uint8_t (*numActiveUsersCb)(void *messager);
|
|
uint8_t (*getClientFrameStatsNumCb)(void *messager);
|
|
uint8_t (*serverFrameStatsReadyCb)(void *messager);
|
|
} settings_t;
|
|
|
|
#ifdef __cplusplus
|
|
extern "C" {
|
|
#endif
|
|
|
|
int resolve_host(struct in_addr *sin_addr, const char *hostname);
|
|
|
|
ssize_t ws_recv(ws_ctx_t *ctx, void *buf, size_t len);
|
|
|
|
ssize_t ws_send(ws_ctx_t *ctx, const void *buf, size_t len);
|
|
|
|
/* base64.c declarations */
|
|
//int b64_ntop(u_char const *src, size_t srclength, char *target, size_t targsize);
|
|
//int b64_pton(char const *src, u_char *target, size_t targsize);
|
|
|
|
extern __thread unsigned wsthread_handler_id;
|
|
|
|
#define gen_handler_msg(stream, ...) \
|
|
if (settings.verbose) { \
|
|
fprintf(stream, " websocket %d: ", wsthread_handler_id); \
|
|
fprintf(stream, __VA_ARGS__); \
|
|
}
|
|
|
|
#define wserr(...) \
|
|
{ \
|
|
fprintf(stderr, " websocket %d: ", wsthread_handler_id); \
|
|
fprintf(stderr, __VA_ARGS__); \
|
|
}
|
|
|
|
#define handler_msg(...) gen_handler_msg(stderr, __VA_ARGS__);
|
|
#define handler_emsg(...) gen_handler_msg(stderr, __VA_ARGS__);
|
|
|
|
void traffic(const char * token);
|
|
|
|
int encode_hixie(u_char const *src, size_t srclength,
|
|
char *target, size_t targsize);
|
|
int decode_hixie(char *src, size_t srclength,
|
|
u_char *target, size_t targsize,
|
|
unsigned int *opcode, unsigned int *left);
|
|
int encode_hybi(u_char const *src, size_t srclength,
|
|
char *target, size_t targsize, unsigned int opcode);
|
|
int decode_hybi(unsigned char *src, size_t srclength,
|
|
u_char *target, size_t targsize,
|
|
unsigned int *opcode, unsigned int *left);
|
|
|
|
void *start_server(void *unused);
|
|
|
|
#ifdef __cplusplus
|
|
} // extern C
|
|
#endif
|