121 lines
3.9 KiB
C
121 lines
3.9 KiB
C
#pragma once
|
|
#include <netdb.h>
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
#include <unistd.h>
|
|
#include <arpa/inet.h>
|
|
#include <netinet/in.h>
|
|
|
|
#define CC_ADDRLEN (INET6_ADDRSTRLEN + 8) // [...:12345]
|
|
|
|
static int sockaddr2str(const struct sockaddr *sa, char *s, const size_t maxlen) {
|
|
if (sa->sa_family == AF_INET) {
|
|
const struct sockaddr_in *sin = (struct sockaddr_in *)sa;
|
|
inet_ntop(AF_INET, &sin->sin_addr, s, maxlen);
|
|
const size_t len = strnlen(s, maxlen);
|
|
return len + snprintf(s + len, maxlen - len, ":%u", ntohs(sin->sin_port));
|
|
}
|
|
|
|
if (sa->sa_family == AF_INET6) {
|
|
const struct sockaddr_in6 *sin6 = (struct sockaddr_in6 *)sa;
|
|
s[0] = '[';
|
|
inet_ntop(AF_INET6, &sin6->sin6_addr, s + 1, maxlen - 1);
|
|
const size_t len = strnlen(s, maxlen);
|
|
return len + snprintf(s + len, maxlen - len, "]:%u", ntohs(sin6->sin6_port));
|
|
}
|
|
|
|
strncpy(s, "unknown", maxlen);
|
|
return 7;
|
|
}
|
|
|
|
#define MAX_MESSAGE_LENGTH 1024
|
|
|
|
#define streq(a, b) (strcmp((a), (b)) == 0)
|
|
|
|
#define exit_error(...) do { \
|
|
fprintf(stderr, __VA_ARGS__); \
|
|
exit(EXIT_FAILURE); \
|
|
} while (0)
|
|
|
|
#define exit_perror(s) do { \
|
|
perror(s); \
|
|
exit(EXIT_FAILURE); \
|
|
} while (0)
|
|
|
|
static int resolve_domain_to_ipv6(const char *hostname, struct in6_addr *server_addr) {
|
|
struct addrinfo hints, *res;
|
|
int found = 0;
|
|
|
|
memset(&hints, 0, sizeof hints);
|
|
hints.ai_family = AF_UNSPEC;
|
|
hints.ai_socktype = SOCK_STREAM;
|
|
|
|
if (getaddrinfo(hostname, NULL, &hints, &res) != 0) return 0;
|
|
|
|
for (struct addrinfo *p = res; p != NULL; p = p->ai_next) {
|
|
if (p->ai_family == AF_INET6) {
|
|
struct sockaddr_in6 *ipv6 = (struct sockaddr_in6 *)p->ai_addr;
|
|
memcpy(server_addr, &ipv6->sin6_addr, sizeof(struct in6_addr));
|
|
found = 1;
|
|
break;
|
|
}
|
|
if (p->ai_family == AF_INET && !found) {
|
|
struct sockaddr_in *ipv4 = (struct sockaddr_in *)p->ai_addr;
|
|
memset(server_addr, 0, sizeof(struct in6_addr));
|
|
server_addr->s6_addr[10] = 0xff;
|
|
server_addr->s6_addr[11] = 0xff;
|
|
memcpy(&server_addr->s6_addr[12], &(ipv4->sin_addr), 4);
|
|
found = 1;
|
|
}
|
|
}
|
|
|
|
freeaddrinfo(res);
|
|
return found;
|
|
}
|
|
|
|
static int cc_client_parse_args_and_connect(int argc, char* argv[], struct sockaddr_in6* server_addr, int* server_fd, FILE** reader, FILE** writer) {
|
|
struct in6_addr bind_addr = in6addr_any;
|
|
int port = 5000;
|
|
if (argc >= 2) {
|
|
if (inet_pton(AF_INET6, argv[1], &bind_addr) != 1) {
|
|
struct in_addr v4;
|
|
if (inet_pton(AF_INET, argv[1], &v4) != 1) {
|
|
if (!resolve_domain_to_ipv6(argv[1], &bind_addr)) {
|
|
exit_error("invalid address or domain: %s\n", argv[1]);
|
|
}
|
|
}
|
|
else {
|
|
memset(&bind_addr, 0, sizeof(bind_addr));
|
|
bind_addr.s6_addr[10] = 0xff;
|
|
bind_addr.s6_addr[11] = 0xff;
|
|
memcpy(&bind_addr.s6_addr[12], &v4, 4);
|
|
}
|
|
}
|
|
}
|
|
if (argc >= 3) {
|
|
port = atoi(argv[2]);
|
|
if (port <= 0) exit_error("unknown argument: %s\n", argv[2]);
|
|
}
|
|
|
|
*server_fd = socket(AF_INET6, SOCK_STREAM, 0);
|
|
if (*server_fd < 0) exit_perror("socket");
|
|
const int server_fd2 = dup(*server_fd);
|
|
if (server_fd2 < 0) exit_perror("dup");
|
|
|
|
server_addr->sin6_family = AF_INET6;
|
|
server_addr->sin6_addr = bind_addr;
|
|
server_addr->sin6_port = htons(port);
|
|
|
|
int opt = 0;
|
|
if (setsockopt(*server_fd, IPPROTO_IPV6, IPV6_V6ONLY, &opt, sizeof(opt)) < 0) {
|
|
// i guess so bro
|
|
}
|
|
|
|
if (connect(*server_fd, (struct sockaddr *)server_addr, sizeof(*server_addr)) < 0) exit_perror("connect");
|
|
|
|
if ((*reader = fdopen(*server_fd, "r")) == NULL) exit_perror("fdopen reader");
|
|
if ((*writer = fdopen(server_fd2, "w")) == NULL) exit_perror("fdopen writer");
|
|
return 0;
|
|
}
|