Initial import
This commit is contained in:
Artem Prilutskiy
commit
12ffb10726
7 changed files with 795 additions and 0 deletions
74
Makefile
Normal file
74
Makefile
Normal file
|
|
@ -0,0 +1,74 @@
|
||||||
|
BUILD := $(shell date -u +%Y%m%d-%H%M%S)
|
||||||
|
OS := $(shell uname -s)
|
||||||
|
|
||||||
|
PREFIX = $(DESTDIR)/opt/TellusAgent
|
||||||
|
TOOLKIT = ../..
|
||||||
|
|
||||||
|
DIRECTORIES = \
|
||||||
|
$(TOOLKIT)/Common \
|
||||||
|
$(TOOLKIT)/Rewind
|
||||||
|
|
||||||
|
ifeq ($(OS), Linux)
|
||||||
|
FLAGS += -rdynamic
|
||||||
|
KIND := $(shell grep -E "^6.0" /etc/debian_version > /dev/null ; echo $?)
|
||||||
|
ifneq ($(KIND), 0)
|
||||||
|
LIBRARIES += rt
|
||||||
|
endif
|
||||||
|
endif
|
||||||
|
|
||||||
|
ifeq ($(OS), Darwin)
|
||||||
|
FLAGS += -Wno-deprecated-declarations
|
||||||
|
endif
|
||||||
|
|
||||||
|
OBJECTS = \
|
||||||
|
TellusAgent.o
|
||||||
|
|
||||||
|
FLAGS += -g -fno-omit-frame-pointer -O3 -MMD $(foreach directory, $(DIRECTORIES), -I$(directory)) -DBUILD=\"$(BUILD)\"
|
||||||
|
LIBS += $(foreach library, $(LIBRARIES), -l$(library))
|
||||||
|
|
||||||
|
CC = gcc
|
||||||
|
CFLAGS += $(FLAGS) -std=gnu99
|
||||||
|
|
||||||
|
ifneq ($(strip $(DEPENDENCIES)),)
|
||||||
|
FLAGS += $(shell pkg-config --cflags $(DEPENDENCIES))
|
||||||
|
LIBS += $(shell pkg-config --libs $(DEPENDENCIES))
|
||||||
|
endif
|
||||||
|
|
||||||
|
all: build
|
||||||
|
|
||||||
|
build: $(PREREQUISITES) $(OBJECTS)
|
||||||
|
$(CC) $(OBJECTS) $(FLAGS) $(LIBS) -o tellusagent
|
||||||
|
|
||||||
|
install:
|
||||||
|
install -D -d $(PREFIX)
|
||||||
|
install -o root -g root tellusagent $(PREFIX)
|
||||||
|
install -o root -g root tellusagent.sh $(PREFIX)
|
||||||
|
install -o root -g root tellusagent.conf $(PREFIX)
|
||||||
|
ifeq ($(OS), Linux)
|
||||||
|
install -o root -g root tellusagent-init $(PREFIX)
|
||||||
|
install -m 644 -o root -g root tellusagent-monit $(PREFIX)
|
||||||
|
install -m 644 -o root -g root tellusagent.service $(PREFIX)
|
||||||
|
endif
|
||||||
|
ifeq ($(OS), Darwin)
|
||||||
|
install -m 644 -o root -g root tellusagent.plist $(DESTDIR)/Library/LaunchDaemons
|
||||||
|
endif
|
||||||
|
|
||||||
|
clean:
|
||||||
|
rm -f $(PREREQUISITES) $(OBJECTS) tellusagent
|
||||||
|
rm -f *.d $(TOOLKIT)/*/*.d
|
||||||
|
|
||||||
|
version:
|
||||||
|
echo "#define VERSION $(shell date -u +%Y%m%d)" > Version.h
|
||||||
|
|
||||||
|
debian-package:
|
||||||
|
./UpdateLog.sh
|
||||||
|
ifdef ARCH
|
||||||
|
dpkg-buildpackage -b -a$(ARCH) -tc
|
||||||
|
else
|
||||||
|
dpkg-buildpackage -b -tc
|
||||||
|
endif
|
||||||
|
|
||||||
|
macos-archive: build
|
||||||
|
zip ../TellusAgent-macOS.zip tellusagent tellusagent.sh tellusagent.conf tellusagent.plist
|
||||||
|
|
||||||
|
.PHONY: all build clean install
|
||||||
103
Rewind.h
Normal file
103
Rewind.h
Normal file
|
|
@ -0,0 +1,103 @@
|
||||||
|
#ifndef REWIND_H
|
||||||
|
#define REWIND_H
|
||||||
|
|
||||||
|
#include <stdint.h>
|
||||||
|
#include <netinet/in.h>
|
||||||
|
|
||||||
|
#ifdef __cplusplus
|
||||||
|
extern "C"
|
||||||
|
{
|
||||||
|
#endif
|
||||||
|
|
||||||
|
#pragma pack(push, 1)
|
||||||
|
|
||||||
|
#define REWIND_KEEP_ALIVE_INTERVAL 5
|
||||||
|
|
||||||
|
#define REWIND_SIGN_LENGTH 8
|
||||||
|
#define REWIND_PROTOCOL_SIGN "REWIND01"
|
||||||
|
|
||||||
|
#define REWIND_CLASS_REWIND_CONTROL 0x0000
|
||||||
|
#define REWIND_CLASS_SYSTEM_CONSOLE 0x0100
|
||||||
|
#define REWIND_CLASS_SERVER_NOTICE 0x0200
|
||||||
|
#define REWIND_CLASS_DEVICE_DATA 0x0800
|
||||||
|
#define REWIND_CLASS_APPLICATION 0x0900
|
||||||
|
|
||||||
|
#define REWIND_CLASS_KAIROS_DATA (REWIND_CLASS_DEVICE_DATA + 0x00)
|
||||||
|
#define REWIND_CLASS_TELLUS_DATA (REWIND_CLASS_DEVICE_DATA + 0x10)
|
||||||
|
|
||||||
|
#define REWIND_TYPE_KEEP_ALIVE (REWIND_CLASS_REWIND_CONTROL + 0)
|
||||||
|
#define REWIND_TYPE_CLOSE (REWIND_CLASS_REWIND_CONTROL + 1)
|
||||||
|
#define REWIND_TYPE_CHALLENGE (REWIND_CLASS_REWIND_CONTROL + 2)
|
||||||
|
#define REWIND_TYPE_AUTHENTICATION (REWIND_CLASS_REWIND_CONTROL + 3)
|
||||||
|
|
||||||
|
#define REWIND_TYPE_REPORT (REWIND_CLASS_SYSTEM_CONSOLE + 0)
|
||||||
|
|
||||||
|
#define REWIND_TYPE_BUSY_NOTICE (REWIND_CLASS_SERVER_NOTICE + 0)
|
||||||
|
#define REWIND_TYPE_ADDRESS_NOTICE (REWIND_CLASS_SERVER_NOTICE + 1)
|
||||||
|
#define REWIND_TYPE_BINDING_NOTICE (REWIND_CLASS_SERVER_NOTICE + 2)
|
||||||
|
|
||||||
|
#define REWIND_TYPE_EXTERNAL_SERVER (REWIND_CLASS_KAIROS_DATA + 0)
|
||||||
|
#define REWIND_TYPE_REMOTE_CONTROL (REWIND_CLASS_KAIROS_DATA + 1)
|
||||||
|
#define REWIND_TYPE_SNMP_TRAP (REWIND_CLASS_KAIROS_DATA + 2)
|
||||||
|
|
||||||
|
#define REWIND_TYPE_PEER_DATA (REWIND_CLASS_TELLUS_DATA + 0)
|
||||||
|
#define REWIND_TYPE_RDAC_DATA (REWIND_CLASS_TELLUS_DATA + 1)
|
||||||
|
#define REWIND_TYPE_MEDIA_DATA (REWIND_CLASS_TELLUS_DATA + 2)
|
||||||
|
|
||||||
|
#define REWIND_TYPE_SUBSCRIPTION (REWIND_CLASS_APPLICATION + 0x00)
|
||||||
|
#define REWIND_TYPE_DMR_DATA_BASE (REWIND_CLASS_APPLICATION + 0x10)
|
||||||
|
#define REWIND_TYPE_DMR_AUDIO_FRAME (REWIND_CLASS_APPLICATION + 0x20)
|
||||||
|
|
||||||
|
#define REWIND_FLAG_NONE 0
|
||||||
|
#define REWIND_FLAG_REAL_TIME_1 (1 << 0)
|
||||||
|
#define REWIND_FLAG_REAL_TIME_2 (1 << 1)
|
||||||
|
#define REWIND_FLAG_DEFAULT_SET REWIND_FLAG_NONE
|
||||||
|
|
||||||
|
#define REWIND_ROLE_REPEATER_AGENT 0x10
|
||||||
|
#define REWIND_ROLE_APPLICATION 0x20
|
||||||
|
|
||||||
|
#define REWIND_SERVICE_CRONOS_AGENT (REWIND_ROLE_REPEATER_AGENT + 0)
|
||||||
|
#define REWIND_SERVICE_TELLUS_AGENT (REWIND_ROLE_REPEATER_AGENT + 1)
|
||||||
|
#define REWIND_SERVICE_SIMPLE_APPLICATION (REWIND_ROLE_APPLICATION + 0)
|
||||||
|
|
||||||
|
struct RewindVersionData
|
||||||
|
{
|
||||||
|
uint32_t number; // Remote ID
|
||||||
|
uint8_t service; // REWIND_SERVICE_*
|
||||||
|
char description[0]; // Software name and version
|
||||||
|
};
|
||||||
|
|
||||||
|
struct RewindAddressData
|
||||||
|
{
|
||||||
|
struct in_addr address;
|
||||||
|
uint16_t port;
|
||||||
|
};
|
||||||
|
|
||||||
|
struct RewindBindingData
|
||||||
|
{
|
||||||
|
uint16_t ports[0];
|
||||||
|
};
|
||||||
|
|
||||||
|
struct RewindSubscriptionData
|
||||||
|
{
|
||||||
|
uint32_t type;
|
||||||
|
uint32_t number;
|
||||||
|
};
|
||||||
|
|
||||||
|
struct RewindData
|
||||||
|
{
|
||||||
|
char sign[REWIND_SIGN_LENGTH];
|
||||||
|
uint16_t type; // REWIND_TYPE_*
|
||||||
|
uint16_t flags; // REWIND_FLAG_*
|
||||||
|
uint32_t number; // Packet sequence number
|
||||||
|
uint16_t length; // Length of following data
|
||||||
|
uint8_t data[0]; //
|
||||||
|
};
|
||||||
|
|
||||||
|
#pragma pack(pop)
|
||||||
|
|
||||||
|
#ifdef __cplusplus
|
||||||
|
}
|
||||||
|
#endif
|
||||||
|
|
||||||
|
#endif
|
||||||
574
TellusAgent.c
Normal file
574
TellusAgent.c
Normal file
|
|
@ -0,0 +1,574 @@
|
||||||
|
#include <stddef.h>
|
||||||
|
#include <stdint.h>
|
||||||
|
|
||||||
|
#include <stdlib.h>
|
||||||
|
#include <stdarg.h>
|
||||||
|
#include <getopt.h>
|
||||||
|
#include <string.h>
|
||||||
|
#include <errno.h>
|
||||||
|
#include <stdio.h>
|
||||||
|
|
||||||
|
#include <netdb.h>
|
||||||
|
#include <arpa/inet.h>
|
||||||
|
#include <netinet/ip.h>
|
||||||
|
|
||||||
|
#include <unistd.h>
|
||||||
|
#include <syslog.h>
|
||||||
|
#include <signal.h>
|
||||||
|
#include <sys/time.h>
|
||||||
|
#include <sys/socket.h>
|
||||||
|
#include <sys/utsname.h>
|
||||||
|
|
||||||
|
#ifdef __linux__
|
||||||
|
#include <endian.h>
|
||||||
|
#include <byteswap.h>
|
||||||
|
#include <sys/epoll.h>
|
||||||
|
#include <sys/signalfd.h>
|
||||||
|
#endif
|
||||||
|
|
||||||
|
#ifdef __MACH__
|
||||||
|
#include <launch.h>
|
||||||
|
#include <stdbool.h>
|
||||||
|
#include <sys/event.h>
|
||||||
|
#include <mach/mach.h>
|
||||||
|
#include <machine/endian.h>
|
||||||
|
#endif
|
||||||
|
|
||||||
|
#include "Rewind.h"
|
||||||
|
#include "Version.h"
|
||||||
|
|
||||||
|
#define HELPER(value) #value
|
||||||
|
#define STRING(value) HELPER(value)
|
||||||
|
|
||||||
|
#define COUNT(array) sizeof(array) / sizeof(array[0])
|
||||||
|
|
||||||
|
#ifdef __cplusplus
|
||||||
|
#define CAST(type, value) const_cast<type>(value)
|
||||||
|
#else
|
||||||
|
#define CAST(type, value) (type)value
|
||||||
|
|
||||||
|
#ifndef bool
|
||||||
|
#define bool int
|
||||||
|
#define true 1
|
||||||
|
#define false 0
|
||||||
|
#endif
|
||||||
|
#endif
|
||||||
|
|
||||||
|
#ifdef __linux__
|
||||||
|
#define CHECK(event, handle) (event->data.fd == handle)
|
||||||
|
#endif
|
||||||
|
|
||||||
|
#ifdef __MACH__
|
||||||
|
#define CHECK(event, handle) (event->ident == handle) && (event->filter == EVFILT_READ)
|
||||||
|
|
||||||
|
#define htobe16(value) OSSwapHostToBigInt16(value)
|
||||||
|
#define be16toh(value) OSSwapBigToHostInt16(value)
|
||||||
|
#define htobe32(value) OSSwapHostToBigInt32(value)
|
||||||
|
#define be32toh(value) OSSwapBigToHostInt32(value)
|
||||||
|
|
||||||
|
#define htole16(value) OSSwapHostToLittleInt16(value)
|
||||||
|
#define le16toh(value) OSSwapLittleToHostInt16(value)
|
||||||
|
#define htole32(value) OSSwapHostToLittleInt32(value)
|
||||||
|
#define le32toh(value) OSSwapLittleToHostInt32(value)
|
||||||
|
|
||||||
|
#define __bswap_16(value) OSSwapConstInt16(value)
|
||||||
|
#define __bswap_32(value) OSSwapConstInt32(value)
|
||||||
|
#endif
|
||||||
|
|
||||||
|
|
||||||
|
#define MODE_CONSOLE (1 << 0)
|
||||||
|
#define MODE_SYSLOG (1 << 1)
|
||||||
|
#define MODE_DAEMON (1 << 2)
|
||||||
|
|
||||||
|
#define EVENT_LIST_LENGTH (4 + 4)
|
||||||
|
|
||||||
|
#define BUFFER_SIZE 2048
|
||||||
|
#define PROXY_PORT_COUNT 3
|
||||||
|
|
||||||
|
|
||||||
|
int serviceMode = MODE_CONSOLE;
|
||||||
|
|
||||||
|
void print(const char* format, ...)
|
||||||
|
{
|
||||||
|
va_list arguments;
|
||||||
|
va_start(arguments, format);
|
||||||
|
if (serviceMode & MODE_CONSOLE)
|
||||||
|
vprintf(format, arguments);
|
||||||
|
if (serviceMode & MODE_SYSLOG)
|
||||||
|
vsyslog(LOG_INFO, format, arguments);
|
||||||
|
va_end(arguments);
|
||||||
|
}
|
||||||
|
|
||||||
|
int main(int argc, const char* argv[])
|
||||||
|
{
|
||||||
|
print("\n");
|
||||||
|
print("TellusAgent for BrandMeister DMR Master Server\n");
|
||||||
|
print("Copyright 2016 Artem Prilutskiy (R3ABM, cyanide.burnout@gmail.com)\n");
|
||||||
|
print("Software revision " STRING(VERSION) " build " BUILD "\n");
|
||||||
|
print("\n");
|
||||||
|
|
||||||
|
// Parameters for server
|
||||||
|
|
||||||
|
const char* serverPort = "54003";
|
||||||
|
const char* serverLocation = NULL;
|
||||||
|
struct addrinfo* serverAddress = NULL;
|
||||||
|
|
||||||
|
// Parameters for repeater
|
||||||
|
|
||||||
|
uint16_t proxyPorts[PROXY_PORT_COUNT] =
|
||||||
|
{
|
||||||
|
htobe16(50000),
|
||||||
|
htobe16(50001),
|
||||||
|
htobe16(50002)
|
||||||
|
};
|
||||||
|
|
||||||
|
// Start up
|
||||||
|
|
||||||
|
struct option options[] =
|
||||||
|
{
|
||||||
|
{ "connect-port", required_argument, NULL, 'c' },
|
||||||
|
{ "control-port", required_argument, NULL, 'r' },
|
||||||
|
{ "media-port", required_argument, NULL, 'd' },
|
||||||
|
{ "server-address", required_argument, NULL, 's' },
|
||||||
|
{ "server-port", required_argument, NULL, 'p' },
|
||||||
|
{ "service-mode", required_argument, NULL, 'm' },
|
||||||
|
{ NULL, 0, NULL, 0 }
|
||||||
|
};
|
||||||
|
|
||||||
|
int selection = 0;
|
||||||
|
while ((selection = getopt_long(argc, CAST(char* const*, argv), "c:r:d:s:p:m:", options, NULL)) != EOF)
|
||||||
|
switch (selection)
|
||||||
|
{
|
||||||
|
case 'c':
|
||||||
|
proxyPorts[0] = htons(strtol(optarg, NULL, 10));
|
||||||
|
break;
|
||||||
|
|
||||||
|
case 'r':
|
||||||
|
proxyPorts[1] = htons(strtol(optarg, NULL, 10));
|
||||||
|
break;
|
||||||
|
|
||||||
|
case 'd':
|
||||||
|
proxyPorts[2] = htons(strtol(optarg, NULL, 10));
|
||||||
|
break;
|
||||||
|
|
||||||
|
case 's':
|
||||||
|
serverLocation = optarg;
|
||||||
|
break;
|
||||||
|
|
||||||
|
case 'p':
|
||||||
|
serverPort = optarg;
|
||||||
|
break;
|
||||||
|
|
||||||
|
case 'm':
|
||||||
|
serviceMode = strtol(optarg, NULL, 10);
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (serverLocation == NULL)
|
||||||
|
{
|
||||||
|
print(
|
||||||
|
"Usage:\n"
|
||||||
|
" %s\n"
|
||||||
|
" --connect-port <local port for P2P service>\n"
|
||||||
|
" --control-port <local port for RDAC service>\n"
|
||||||
|
" --media-port <local port for DMR service>\n"
|
||||||
|
" --server-address <domain name of BrandMeister DMR Server>\n"
|
||||||
|
" --server-port <service port of BrandMeister DMR Server>\n"
|
||||||
|
" --service-mode <set of bits>\n"
|
||||||
|
" bit 0 - print to standard output\n"
|
||||||
|
" bit 1 - print to system log\n"
|
||||||
|
" bit 2 - run as daemon\n"
|
||||||
|
"\n",
|
||||||
|
argv[0]);
|
||||||
|
return EXIT_FAILURE;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (serviceMode & MODE_SYSLOG)
|
||||||
|
{
|
||||||
|
// Set proper origin for syslog (required by OpenWRT)
|
||||||
|
openlog("TellusAgent", LOG_NOWAIT | LOG_PID, LOG_USER);
|
||||||
|
}
|
||||||
|
|
||||||
|
#ifdef __linux__
|
||||||
|
if ((serviceMode & MODE_DAEMON) &&
|
||||||
|
(daemon(-1, -1) < 0))
|
||||||
|
{
|
||||||
|
print("Error launching daemon");
|
||||||
|
return EXIT_FAILURE;
|
||||||
|
}
|
||||||
|
#endif
|
||||||
|
#ifdef __MACH__
|
||||||
|
if (serviceMode & MODE_DAEMON)
|
||||||
|
{
|
||||||
|
launch_data_t request = launch_data_new_string(LAUNCH_KEY_CHECKIN);
|
||||||
|
launch_data_t response = launch_msg(request);
|
||||||
|
launch_data_free(request);
|
||||||
|
if (response == NULL)
|
||||||
|
{
|
||||||
|
print("Error calling launchd");
|
||||||
|
return EXIT_FAILURE;
|
||||||
|
}
|
||||||
|
launch_data_type_t type = launch_data_get_type(response);
|
||||||
|
launch_data_free(response);
|
||||||
|
if (type == LAUNCH_DATA_ERRNO)
|
||||||
|
{
|
||||||
|
print("launchd returned error %d\n", launch_data_get_errno(response));
|
||||||
|
return EXIT_FAILURE;
|
||||||
|
}
|
||||||
|
// launchd will return dictionary of job for successful check-in
|
||||||
|
if (type != LAUNCH_DATA_DICTIONARY)
|
||||||
|
{
|
||||||
|
print("Error launching daemon");
|
||||||
|
return EXIT_FAILURE;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
#endif
|
||||||
|
|
||||||
|
// Resolve server address
|
||||||
|
|
||||||
|
struct addrinfo hints;
|
||||||
|
|
||||||
|
memset(&hints, 0, sizeof(hints));
|
||||||
|
hints.ai_socktype = SOCK_DGRAM;
|
||||||
|
|
||||||
|
#ifdef __linux__
|
||||||
|
hints.ai_flags = AI_ADDRCONFIG;
|
||||||
|
hints.ai_family = AF_UNSPEC;
|
||||||
|
#endif
|
||||||
|
#ifdef __MACH__
|
||||||
|
hints.ai_flags = AI_V4MAPPED;
|
||||||
|
hints.ai_family = AF_INET6;
|
||||||
|
#endif
|
||||||
|
|
||||||
|
if (getaddrinfo(serverLocation, serverPort, &hints, &serverAddress) != 0)
|
||||||
|
{
|
||||||
|
print("Error resolving server address %s\n", serverLocation);
|
||||||
|
return EXIT_FAILURE;
|
||||||
|
}
|
||||||
|
|
||||||
|
// Initialize proxy sockets
|
||||||
|
|
||||||
|
int proxyHandles[PROXY_PORT_COUNT];
|
||||||
|
|
||||||
|
struct sockaddr_in proxySocketAddress;
|
||||||
|
socklen_t proxySocketLength = sizeof(proxySocketAddress);
|
||||||
|
|
||||||
|
proxySocketAddress.sin_family = AF_INET;
|
||||||
|
proxySocketAddress.sin_addr.s_addr = htonl(INADDR_ANY);
|
||||||
|
|
||||||
|
for (selection = 0; selection < PROXY_PORT_COUNT; selection ++)
|
||||||
|
{
|
||||||
|
int handle;
|
||||||
|
|
||||||
|
proxySocketAddress.sin_port = proxyPorts[selection];
|
||||||
|
handle = socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP);
|
||||||
|
|
||||||
|
if((handle < 0) ||
|
||||||
|
(bind(handle, (struct sockaddr*)&proxySocketAddress, proxySocketLength) < 0))
|
||||||
|
{
|
||||||
|
print("Error opening port for Multi-Site Connect\n");
|
||||||
|
return EXIT_FAILURE;
|
||||||
|
}
|
||||||
|
|
||||||
|
proxyHandles[selection] = handle;
|
||||||
|
}
|
||||||
|
|
||||||
|
// Initialize uplink socket
|
||||||
|
|
||||||
|
int uplinkHandle;
|
||||||
|
struct sockaddr_in6 uplinkSocketAddress;
|
||||||
|
|
||||||
|
uplinkSocketAddress.sin6_family = AF_INET6;
|
||||||
|
uplinkSocketAddress.sin6_addr = in6addr_any;
|
||||||
|
uplinkSocketAddress.sin6_port = 0;
|
||||||
|
uplinkSocketAddress.sin6_scope_id = 0;
|
||||||
|
|
||||||
|
uplinkHandle = socket(AF_INET6, SOCK_DGRAM, IPPROTO_UDP);
|
||||||
|
if((uplinkHandle < 0) ||
|
||||||
|
(bind(uplinkHandle, (struct sockaddr*)&uplinkSocketAddress, sizeof(uplinkSocketAddress)) < 0))
|
||||||
|
{
|
||||||
|
print("Error opening port for Rewind Uplink\n");
|
||||||
|
return EXIT_FAILURE;
|
||||||
|
}
|
||||||
|
|
||||||
|
#ifdef __linux__
|
||||||
|
|
||||||
|
// Initialize signal handle
|
||||||
|
|
||||||
|
int signalHandle;
|
||||||
|
sigset_t signalMask;
|
||||||
|
|
||||||
|
sigemptyset(&signalMask);
|
||||||
|
sigaddset(&signalMask, SIGINT);
|
||||||
|
sigaddset(&signalMask, SIGHUP);
|
||||||
|
sigaddset(&signalMask, SIGTERM);
|
||||||
|
sigaddset(&signalMask, SIGQUIT);
|
||||||
|
|
||||||
|
sigprocmask(SIG_BLOCK, &signalMask, NULL);
|
||||||
|
signalHandle = signalfd(-1, &signalMask, 0);
|
||||||
|
|
||||||
|
// Initialize ePoll
|
||||||
|
|
||||||
|
int pollHandle;
|
||||||
|
struct epoll_event event;
|
||||||
|
|
||||||
|
pollHandle = epoll_create(EVENT_LIST_LENGTH);
|
||||||
|
|
||||||
|
event.events = EPOLLIN;
|
||||||
|
event.data.fd = proxyHandles[0];
|
||||||
|
epoll_ctl(pollHandle, EPOLL_CTL_ADD, event.data.fd, &event);
|
||||||
|
|
||||||
|
event.events = EPOLLIN;
|
||||||
|
event.data.fd = proxyHandles[1];
|
||||||
|
epoll_ctl(pollHandle, EPOLL_CTL_ADD, event.data.fd, &event);
|
||||||
|
|
||||||
|
event.events = EPOLLIN;
|
||||||
|
event.data.fd = proxyHandles[2];
|
||||||
|
epoll_ctl(pollHandle, EPOLL_CTL_ADD, event.data.fd, &event);
|
||||||
|
|
||||||
|
event.events = EPOLLIN;
|
||||||
|
event.data.fd = uplinkHandle;
|
||||||
|
epoll_ctl(pollHandle, EPOLL_CTL_ADD, event.data.fd, &event);
|
||||||
|
|
||||||
|
event.events = EPOLLIN;
|
||||||
|
event.data.fd = signalHandle;
|
||||||
|
epoll_ctl(pollHandle, EPOLL_CTL_ADD, event.data.fd, &event);
|
||||||
|
|
||||||
|
#endif
|
||||||
|
#ifdef __MACH__
|
||||||
|
|
||||||
|
// Prepare signal handlers
|
||||||
|
|
||||||
|
signal(SIGINT, SIG_IGN);
|
||||||
|
signal(SIGHUP, SIG_IGN);
|
||||||
|
signal(SIGTERM, SIG_IGN);
|
||||||
|
signal(SIGQUIT, SIG_IGN);
|
||||||
|
|
||||||
|
// Initialize KQueue
|
||||||
|
|
||||||
|
int queueHandle;
|
||||||
|
|
||||||
|
struct kevent changes[EVENT_LIST_LENGTH];
|
||||||
|
struct kevent* change = changes;
|
||||||
|
|
||||||
|
queueHandle = kqueue();
|
||||||
|
|
||||||
|
EV_SET(change, proxyHandles[0], EVFILT_READ, EV_ADD | EV_ENABLE, 0, 0, 0);
|
||||||
|
change ++;
|
||||||
|
|
||||||
|
EV_SET(change, proxyHandles[1], EVFILT_READ, EV_ADD | EV_ENABLE, 0, 0, 0);
|
||||||
|
change ++;
|
||||||
|
|
||||||
|
EV_SET(change, proxyHandles[2], EVFILT_READ, EV_ADD | EV_ENABLE, 0, 0, 0);
|
||||||
|
change ++;
|
||||||
|
|
||||||
|
EV_SET(change, uplinkHandle, EVFILT_READ, EV_ADD | EV_ENABLE, 0, 0, 0);
|
||||||
|
change ++;
|
||||||
|
|
||||||
|
EV_SET(change, SIGINT, EVFILT_SIGNAL, EV_ADD | EV_ENABLE, 0, 0, 0);
|
||||||
|
change ++;
|
||||||
|
|
||||||
|
EV_SET(change, SIGHUP, EVFILT_SIGNAL, EV_ADD | EV_ENABLE, 0, 0, 0);
|
||||||
|
change ++;
|
||||||
|
|
||||||
|
EV_SET(change, SIGTERM, EVFILT_SIGNAL, EV_ADD | EV_ENABLE, 0, 0, 0);
|
||||||
|
change ++;
|
||||||
|
|
||||||
|
EV_SET(change, SIGQUIT, EVFILT_SIGNAL, EV_ADD | EV_ENABLE, 0, 0, 0);
|
||||||
|
change ++;
|
||||||
|
|
||||||
|
#endif
|
||||||
|
|
||||||
|
// Prepare buffers
|
||||||
|
|
||||||
|
void* controlBuffer = alloca(BUFFER_SIZE);
|
||||||
|
|
||||||
|
struct RewindData* incomingBuffer = (struct RewindData*)alloca(sizeof(struct RewindData) + BUFFER_SIZE);
|
||||||
|
struct RewindData* outgoingBuffer = (struct RewindData*)alloca(sizeof(struct RewindData) + BUFFER_SIZE);
|
||||||
|
|
||||||
|
memset(outgoingBuffer, 0, sizeof(struct RewindData));
|
||||||
|
memcpy(outgoingBuffer, REWIND_PROTOCOL_SIGN, REWIND_SIGN_LENGTH);
|
||||||
|
|
||||||
|
struct utsname systemName;
|
||||||
|
uname(&systemName);
|
||||||
|
|
||||||
|
struct sockaddr_in repeaterSocketAddresses[3];
|
||||||
|
memset(repeaterSocketAddresses, 0, sizeof(repeaterSocketAddresses));
|
||||||
|
|
||||||
|
// Main loop
|
||||||
|
|
||||||
|
bool running = true;
|
||||||
|
|
||||||
|
print("Server started\n");
|
||||||
|
|
||||||
|
while (running)
|
||||||
|
{
|
||||||
|
#ifdef __linux__
|
||||||
|
struct epoll_event events[EVENT_LIST_LENGTH];
|
||||||
|
int count = epoll_wait(pollHandle, events, EVENT_LIST_LENGTH, -1);
|
||||||
|
#endif
|
||||||
|
#ifdef __MACH__
|
||||||
|
struct kevent events[EVENT_LIST_LENGTH];
|
||||||
|
int count = kevent(queueHandle, changes, EVENT_LIST_LENGTH, events, EVENT_LIST_LENGTH, NULL);
|
||||||
|
#endif
|
||||||
|
|
||||||
|
if (count < 0)
|
||||||
|
{
|
||||||
|
int error = errno;
|
||||||
|
print("Error processing handles: %s (%d)\n", strerror(error), error);
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
|
||||||
|
for (size_t index = 0; index < count; index ++)
|
||||||
|
{
|
||||||
|
#ifdef __linux__
|
||||||
|
struct epoll_event* event = events + index;
|
||||||
|
#endif
|
||||||
|
#ifdef __MACH__
|
||||||
|
struct kevent* event = events + index;
|
||||||
|
#endif
|
||||||
|
|
||||||
|
// Handle packet from the uplink
|
||||||
|
|
||||||
|
if (CHECK(event, uplinkHandle))
|
||||||
|
{
|
||||||
|
struct sockaddr_in6 address;
|
||||||
|
socklen_t size = sizeof(address);
|
||||||
|
size_t length = recvfrom(uplinkHandle, incomingBuffer, BUFFER_SIZE, 0, (struct sockaddr*)&address, &size);
|
||||||
|
|
||||||
|
if ((length >= sizeof(struct RewindData)) &&
|
||||||
|
((serverAddress->ai_addr->sa_family == AF_INET) || /* Work-around for Linux */
|
||||||
|
(memcmp(&address, serverAddress->ai_addr, serverAddress->ai_addrlen) == 0)) &&
|
||||||
|
(memcmp(incomingBuffer->sign, REWIND_PROTOCOL_SIGN, REWIND_SIGN_LENGTH) == 0))
|
||||||
|
{
|
||||||
|
uint16_t type = le16toh(incomingBuffer->type);
|
||||||
|
size_t length = le16toh(incomingBuffer->length);
|
||||||
|
|
||||||
|
if ((type <= REWIND_TYPE_PEER_DATA) &&
|
||||||
|
(type >= REWIND_TYPE_MEDIA_DATA))
|
||||||
|
{
|
||||||
|
int handle;
|
||||||
|
struct sockaddr_in* address;
|
||||||
|
|
||||||
|
selection = type - REWIND_TYPE_PEER_DATA;
|
||||||
|
address = repeaterSocketAddresses + selection;
|
||||||
|
handle = proxyHandles[selection];
|
||||||
|
|
||||||
|
sendto(handle, incomingBuffer->data, length, 0, (struct sockaddr*)address, sizeof(struct sockaddr_in));
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (type == REWIND_TYPE_REPORT)
|
||||||
|
{
|
||||||
|
incomingBuffer->data[length] = '\0';
|
||||||
|
print("Server message: %s\n", incomingBuffer->data);
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (type == REWIND_TYPE_CLOSE)
|
||||||
|
{
|
||||||
|
print("Disconnect request received\n");
|
||||||
|
running = false;
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Handle packet from the repeater
|
||||||
|
|
||||||
|
if (CHECK(event, proxyHandles[0]))
|
||||||
|
{
|
||||||
|
struct msghdr message;
|
||||||
|
struct iovec vectors[2];
|
||||||
|
|
||||||
|
outgoingBuffer->type = htole16(REWIND_TYPE_BINDING_NOTICE);
|
||||||
|
outgoingBuffer->length = htole16(sizeof(proxyPorts));
|
||||||
|
|
||||||
|
vectors[0].iov_base = outgoingBuffer;
|
||||||
|
vectors[0].iov_len = sizeof(struct RewindData);
|
||||||
|
vectors[1].iov_base = proxyPorts;
|
||||||
|
vectors[1].iov_len = sizeof(proxyPorts);
|
||||||
|
message.msg_name = serverAddress->ai_addr;
|
||||||
|
message.msg_namelen = serverAddress->ai_addrlen;
|
||||||
|
message.msg_iov = vectors;
|
||||||
|
message.msg_iovlen = 2;
|
||||||
|
message.msg_control = NULL;
|
||||||
|
message.msg_controllen = 0;
|
||||||
|
message.msg_flags = 0;
|
||||||
|
|
||||||
|
sendmsg(uplinkHandle, &message, 0);
|
||||||
|
}
|
||||||
|
|
||||||
|
for (selection = 0; selection < PROXY_PORT_COUNT; selection ++)
|
||||||
|
{
|
||||||
|
int handle = proxyHandles[selection];
|
||||||
|
if (CHECK(event, handle))
|
||||||
|
{
|
||||||
|
socklen_t size = sizeof(struct sockaddr_in);
|
||||||
|
uint8_t* buffer = (uint8_t*)outgoingBuffer->data;
|
||||||
|
struct sockaddr_in* address = repeaterSocketAddresses + selection;
|
||||||
|
size_t length = recvfrom(handle, buffer, BUFFER_SIZE, 0, (struct sockaddr*)address, &size);
|
||||||
|
|
||||||
|
outgoingBuffer->type = htole16(REWIND_CLASS_TELLUS_DATA + selection);
|
||||||
|
outgoingBuffer->length = htole16(length);
|
||||||
|
|
||||||
|
length += sizeof(struct RewindData);
|
||||||
|
sendto(uplinkHandle, outgoingBuffer, length, 0, serverAddress->ai_addr, serverAddress->ai_addrlen);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Handle signal from the kernel
|
||||||
|
|
||||||
|
#ifdef __linux__
|
||||||
|
if (event->data.fd == signalHandle)
|
||||||
|
{
|
||||||
|
struct signalfd_siginfo information;
|
||||||
|
read(signalHandle, &information, sizeof(information));
|
||||||
|
#endif
|
||||||
|
#ifdef __MACH__
|
||||||
|
if (event->filter == EVFILT_SIGNAL)
|
||||||
|
{
|
||||||
|
#endif
|
||||||
|
outgoingBuffer->type = htole16(REWIND_TYPE_CLOSE);
|
||||||
|
outgoingBuffer->length = 0;
|
||||||
|
sendto(uplinkHandle, outgoingBuffer, sizeof(struct RewindData), 0, serverAddress->ai_addr, serverAddress->ai_addrlen);
|
||||||
|
#ifdef __linux__
|
||||||
|
if ((information.ssi_signo == SIGINT) ||
|
||||||
|
(information.ssi_signo == SIGTERM) ||
|
||||||
|
(information.ssi_signo == SIGQUIT))
|
||||||
|
#endif
|
||||||
|
#ifdef __MACH__
|
||||||
|
if ((event->ident == SIGINT) ||
|
||||||
|
(event->ident == SIGTERM) ||
|
||||||
|
(event->ident == SIGQUIT))
|
||||||
|
#endif
|
||||||
|
{
|
||||||
|
running = false;
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
print("Server stopped\n");
|
||||||
|
|
||||||
|
// Clean up
|
||||||
|
|
||||||
|
close(uplinkHandle);
|
||||||
|
close(proxyHandles[0]);
|
||||||
|
close(proxyHandles[1]);
|
||||||
|
close(proxyHandles[2]);
|
||||||
|
|
||||||
|
#ifdef __linux__
|
||||||
|
close(pollHandle);
|
||||||
|
close(signalHandle);
|
||||||
|
#endif
|
||||||
|
#ifdef __MACH__
|
||||||
|
close(queueHandle);
|
||||||
|
#endif
|
||||||
|
|
||||||
|
freeaddrinfo(serverAddress);
|
||||||
|
|
||||||
|
return EXIT_SUCCESS;
|
||||||
|
};
|
||||||
1
Version.h
Normal file
1
Version.h
Normal file
|
|
@ -0,0 +1 @@
|
||||||
|
#define VERSION 20161101
|
||||||
6
tellusagent.conf
Normal file
6
tellusagent.conf
Normal file
|
|
@ -0,0 +1,6 @@
|
||||||
|
|
||||||
|
CONNECT_PORT=50000
|
||||||
|
CONTROL_PORT=50001
|
||||||
|
MEDIA_PORT=50002
|
||||||
|
SERVER_ADDRESS=aesyle.dstar.su
|
||||||
|
SERVICE_MODE=1
|
||||||
33
tellusagent.sh
Executable file
33
tellusagent.sh
Executable file
|
|
@ -0,0 +1,33 @@
|
||||||
|
#!/bin/sh
|
||||||
|
|
||||||
|
DIRECTORY=$(dirname $0)
|
||||||
|
|
||||||
|
if [ -f $DIRECTORY/tellusagent.conf ]
|
||||||
|
then
|
||||||
|
# Read configuration from text file
|
||||||
|
# Dot is equivalent of bash's source
|
||||||
|
. $DIRECTORY/tellusagent.conf
|
||||||
|
fi
|
||||||
|
|
||||||
|
if [ -f /etc/config/tellusagent ]
|
||||||
|
then
|
||||||
|
# Read configuration from UCI/LuCI on OpenWRT
|
||||||
|
CONNECT_PORT=$(uci get tellusagent.@tellusagent[0].connectPort)
|
||||||
|
CONTROL_PORT=$(uci get tellusagent.@tellusagent[0].controlPort)
|
||||||
|
MEDIA_PORT=$(uci get tellusagent.@tellusagent[0].mediaPort)
|
||||||
|
fi
|
||||||
|
|
||||||
|
if [ -n "$1" ]
|
||||||
|
then
|
||||||
|
# 1 - print to standard output
|
||||||
|
# 2 - print to system log
|
||||||
|
# 6 - run as daemon and print to system log
|
||||||
|
SERVICE_MODE=$1
|
||||||
|
fi
|
||||||
|
|
||||||
|
$DIRECTORY/tellusagent \
|
||||||
|
--connect-port ${CONNECT_PORT} \
|
||||||
|
--control-port ${CONTROL_PORT} \
|
||||||
|
--media-port ${MEDIA_PORT} \
|
||||||
|
--server-address ${SERVER_ADDRESS} \
|
||||||
|
--service-mode ${SERVICE_MODE}
|
||||||
4
test.sh
Executable file
4
test.sh
Executable file
|
|
@ -0,0 +1,4 @@
|
||||||
|
#/bin/bash
|
||||||
|
make
|
||||||
|
./tellusagent.sh
|
||||||
|
make clean
|
||||||
Loading…
Add table
Reference in a new issue