47 lines
1.2 KiB
C
47 lines
1.2 KiB
C
|
#include "wsdef.h"
|
||
|
|
|
||
|
|
#include <string.h>
|
||
|
|
|
||
|
|
#include "sha1.h"
|
||
|
|
#include "base64.h"
|
||
|
|
|
||
|
|
#include "websocket_parser.h"
|
||
|
|
|
||
|
|
// base64_encode( SHA1(key + magic) )
|
||
|
|
void ws_encode_key(const char* key, char accept[]) {
|
||
|
|
char magic[] = "258EAFA5-E914-47DA-95CA-C5AB0DC85B11";
|
||
|
|
unsigned char digest[20] = {0};
|
||
|
|
HV_SHA1_CTX ctx;
|
||
|
|
HV_SHA1Init(&ctx);
|
||
|
|
HV_SHA1Update(&ctx, (unsigned char*)key, (uint32_t)strlen(key));
|
||
|
|
HV_SHA1Update(&ctx, (unsigned char*)magic, (uint32_t)strlen(magic));
|
||
|
|
HV_SHA1Final(digest, &ctx);
|
||
|
|
hv_base64_encode(digest, 20, accept);
|
||
|
|
}
|
||
|
|
|
||
|
|
// fix-header[2] + var-length[2/8] + mask[4] + data[data_len]
|
||
|
|
int ws_calc_frame_size(int data_len, bool has_mask) {
|
||
|
|
int size = data_len + 2;
|
||
|
|
if (data_len >=126) {
|
||
|
|
if (data_len > 0xFFFF) {
|
||
|
|
size += 8;
|
||
|
|
} else {
|
||
|
|
size += 2;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
if (has_mask) size += 4;
|
||
|
|
return size;
|
||
|
|
}
|
||
|
|
|
||
|
|
int ws_build_frame(
|
||
|
|
char* out,
|
||
|
|
const char* data, int data_len,
|
||
|
|
const char mask[4], bool has_mask,
|
||
|
|
enum ws_opcode opcode,
|
||
|
|
bool fin) {
|
||
|
|
int flags = opcode;
|
||
|
|
if (fin) flags |= WS_FIN;
|
||
|
|
if (has_mask) flags |= WS_HAS_MASK;
|
||
|
|
return (int)websocket_build_frame(out, (websocket_flags)flags, mask, data, data_len);
|
||
|
|
}
|