diff options
author | Toni Uhlig <matzeton@googlemail.com> | 2020-11-08 15:13:43 +0100 |
---|---|---|
committer | Toni Uhlig <matzeton@googlemail.com> | 2020-11-08 15:14:08 +0100 |
commit | ca28fd9a0357da62267a7b5549bdd0dc6ef55d96 (patch) | |
tree | 571619fb4c53995b977187d75fc468e024112e8b /examples/c-captured | |
parent | 54f6ab21925f990a2f7ab49e6b1dbb1ebb2ae65a (diff) |
Make sure that DAEMON_EVENT_SHUTDOWN gets transmitted before close() (hacky, but works).
Signed-off-by: Toni Uhlig <matzeton@googlemail.com>
Diffstat (limited to 'examples/c-captured')
-rw-r--r-- | examples/c-captured/c-captured.c | 134 |
1 files changed, 134 insertions, 0 deletions
diff --git a/examples/c-captured/c-captured.c b/examples/c-captured/c-captured.c new file mode 100644 index 000000000..eb56470e7 --- /dev/null +++ b/examples/c-captured/c-captured.c @@ -0,0 +1,134 @@ +#include <arpa/inet.h> +#include <errno.h> +#include <stdio.h> +#include <stdlib.h> +#include <string.h> +#include <sys/socket.h> +#include <sys/types.h> +#include <unistd.h> + +#include "config.h" +#include "jsmn.h" + +static char serv_listen_addr[INET_ADDRSTRLEN] = DISTRIBUTOR_HOST; +static uint16_t serv_listen_port = DISTRIBUTOR_PORT; + +int main(void) +{ + int sockfd = socket(AF_INET, SOCK_STREAM, 0); + struct sockaddr_in remote_addr = {}; + socklen_t remote_addrlen = sizeof(remote_addr); + uint8_t buf[NETWORK_BUFFER_MAX_SIZE]; + size_t buf_used = 0; + size_t json_start = 0; + unsigned long long int json_bytes = 0; + jsmn_parser parser; + jsmntok_t tokens[128]; + + if (sockfd < 0) + { + perror("socket"); + return 1; + } + + remote_addr.sin_family = AF_INET; + if (inet_pton(AF_INET, &serv_listen_addr[0], &remote_addr.sin_addr) != 1) + { + perror("inet_pton"); + return 1; + } + remote_addr.sin_port = htons(serv_listen_port); + + if (connect(sockfd, (struct sockaddr *)&remote_addr, remote_addrlen) != 0) + { + perror("connect"); + return 1; + } + + while (1) + { + errno = 0; + ssize_t bytes_read = read(sockfd, buf + buf_used, sizeof(buf) - buf_used); + + if (bytes_read <= 0 || errno != 0) + { + fprintf(stderr, "Remote end disconnected.\n"); + break; + } + + buf_used += bytes_read; + while (buf_used >= nDPIsrvd_JSON_BYTES + 1) + { + if (buf[nDPIsrvd_JSON_BYTES] != '{') + { + fprintf(stderr, "BUG: JSON invalid opening character: '%c'\n", buf[nDPIsrvd_JSON_BYTES]); + exit(1); + } + + char * json_str_start = NULL; + json_bytes = strtoull((char *)buf, &json_str_start, 10); + json_bytes += (uint8_t *)json_str_start - buf; + json_start = (uint8_t *)json_str_start - buf; + + if (errno == ERANGE) + { + fprintf(stderr, "BUG: Size of JSON exceeds limit\n"); + exit(1); + } + if ((uint8_t *)json_str_start == buf) + { + fprintf(stderr, "BUG: Missing size before JSON string: \"%.*s\"\n", nDPIsrvd_JSON_BYTES, buf); + exit(1); + } + if (json_bytes > sizeof(buf)) + { + fprintf(stderr, "BUG: JSON string too big: %llu > %zu\n", json_bytes, sizeof(buf)); + exit(1); + } + if (json_bytes > buf_used) + { + break; + } + + if (buf[json_bytes - 1] != '}') + { + fprintf(stderr, "BUG: Invalid JSON string: %.*s\n", (int)json_bytes, buf); + exit(1); + } + + int r; + jsmn_init(&parser); + r = jsmn_parse(&parser, + (char *)(buf + json_start), + json_bytes - json_start, + tokens, + sizeof(tokens) / sizeof(tokens[0])); + if (r < 0 || tokens[0].type != JSMN_OBJECT) + { + fprintf(stderr, "JSON parsing failed with return value %d at position %u\n", r, parser.pos); + fprintf(stderr, "JSON string: '%.*s'\n", (int)(json_bytes - json_start), (char *)(buf + json_start)); + exit(1); + } + + for (int i = 1; i < r; i++) + { + if (i % 2 == 1) + { + printf("[%.*s : ", tokens[i].end - tokens[i].start, (char *)(buf + json_start) + tokens[i].start); + } + else + { + printf("%.*s] ", tokens[i].end - tokens[i].start, (char *)(buf + json_start) + tokens[i].start); + } + } + printf("EoF\n"); + + memmove(buf, buf + json_bytes, buf_used - json_bytes); + buf_used -= json_bytes; + json_bytes = 0; + json_start = 0; + } + } + + return 0; +} |