This repository has been archived by the owner on Apr 1, 2024. It is now read-only.
/
tls.c
88 lines (75 loc) · 2.25 KB
/
tls.c
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
#include <tls.h>
#include <syslog.h>
#include <string.h>
#include <stdlib.h>
#include <jansson.h>
#include "tls.h"
void sendOverTLS(struct tls* ctx, const char *buf) {
size_t sent;
// send Length of buf
size_t length = strlen(buf);
size_t len = sizeof(length);
while (len > 0) {
int ret = tls_write(ctx, &length, len, &sent);
if (ret == TLS_READ_AGAIN || ret == TLS_WRITE_AGAIN) {
syslog(LOG_DEBUG, "READ/WRITE AGAIN\n");
} else if (ret < 0) {
syslog(LOG_ERR, "%s\n", tls_error(ctx));
break;
} else {
len -= sent;
}
}
// send actual buf
size_t toSend = length;
while (toSend > 0) {
int ret = tls_write(ctx, buf, toSend, &sent);
if (ret == TLS_READ_AGAIN || ret == TLS_WRITE_AGAIN) {
syslog(LOG_DEBUG, "READ/WRITE AGAIN\n");
} else if (ret < 0) {
syslog(LOG_ERR, "%s\n", tls_error(ctx));
break;
} else {
buf += sent;
toSend -= sent;
}
}
}
json_t* recvOverTLS(struct tls*ctx) {
json_error_t error;
size_t getSize, size;
size_t len = sizeof(getSize);
// read length
while (len > 0) {
int ret = tls_read(ctx, &getSize, len, &size);
if (ret == TLS_READ_AGAIN || ret == TLS_WRITE_AGAIN) {
/* retry. May use select to wait for nonblocking */
} else if (ret < 0) {
syslog(LOG_ERR, "%s\n", tls_error(ctx));
break;
} else {
len -= size;
}
}
// create buffer
char* buffer = (char*)malloc((getSize +1) *sizeof(char));
char* buf = buffer;
while (getSize > 0) {
int ret = tls_read(ctx, buf, getSize, &size);
if (ret == TLS_READ_AGAIN || ret == TLS_WRITE_AGAIN) {
/* retry. May use select to wait for nonblocking */
} else if (ret < 0) {
syslog(LOG_ERR, "%s\n", tls_error(ctx));
break;
} else {
buf += size;
getSize -= size;
}
}
// Process buffer to json
json_t *json = json_loads(buffer, JSON_DISABLE_EOF_CHECK, &error);
if (verbose_flag) {
syslog(LOG_DEBUG, "%s\n", buffer);
}
return json;
}