diff options
author | Matthias (think) <matthias@mpb.li> | 2012-07-11 11:49:12 +0200 |
---|---|---|
committer | Matthias (think) <matthias@mpb.li> | 2012-07-11 11:49:12 +0200 |
commit | bbab73a63b8c7b50e8a8cb228999d45024fad984 (patch) | |
tree | 81a822aa9bd33cae8a5fb8ade4faa83b652316d9 /src/dabOutput.cpp | |
download | dabmux-bbab73a63b8c7b50e8a8cb228999d45024fad984.tar.gz dabmux-bbab73a63b8c7b50e8a8cb228999d45024fad984.tar.bz2 dabmux-bbab73a63b8c7b50e8a8cb228999d45024fad984.zip |
added unmodified mmbtools
Diffstat (limited to 'src/dabOutput.cpp')
-rw-r--r-- | src/dabOutput.cpp | 1112 |
1 files changed, 1112 insertions, 0 deletions
diff --git a/src/dabOutput.cpp b/src/dabOutput.cpp new file mode 100644 index 0000000..5351501 --- /dev/null +++ b/src/dabOutput.cpp @@ -0,0 +1,1112 @@ +/* + Copyright (C) 2004, 2005, 2006, 2007, 2008, 2009 Her Majesty the Queen in + Right of Canada (Communications Research Center Canada) + */ +/* + This file is part of CRC-DabMux. + + CRC-DabMux is free software: you can redistribute it and/or modify + it under the terms of the GNU General Public License as + published by the Free Software Foundation, either version 3 of the + License, or (at your option) any later version. + + CRC-DabMux is distributed in the hope that it will be useful, + but WITHOUT ANY WARRANTY; without even the implied warranty of + MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + GNU General Public License for more details. + + You should have received a copy of the GNU General Public License + along with CRC-DabMux. If not, see <http://www.gnu.org/licenses/>. + */ + +#include "dabOutput.h" +#include "UdpSocket.h" + +#include <string.h> +#include <stdio.h> +#include <fcntl.h> +#include <limits.h> + +#ifdef _WIN32 +# include <io.h> +# ifdef __MINGW32__ +# define FS_DECLARE_CFG_ARRAYS +# include <winioctl.h> +# endif +# include <sdci.h> +#else +# include <farsync.h> +# include <unistd.h> +# include <sys/time.h> +# ifndef O_BINARY +# define O_BINARY 0 +# endif // O_BINARY +#endif + + +int dabOutputDefaultInit(void** args) { + return -1; +} + + +int dabOutputDefaultOpen(void* args, const char* filename) +{ + return -1; +} + + +int dabOutputDefaultWrite(void* args, void* buffer, int size) +{ + return -1; +} + + +int dabOutputDefaultClose(void* args) +{ + return -1; +} + + +int dabOutputDefaultClean(void** args) +{ + return -1; +} + + +struct dabOutputOperations dabOutputDefaultOperations = { + dabOutputDefaultInit, + dabOutputDefaultOpen, + dabOutputDefaultWrite, + dabOutputDefaultClose, + dabOutputDefaultClean +}; + + +enum EtiFileType { + ETI_FILE_TYPE_NONE = 0, + ETI_FILE_TYPE_RAW, + ETI_FILE_TYPE_STREAMED, + ETI_FILE_TYPE_FRAMED +}; + + +struct dabOutputFifoData { + int file; + EtiFileType type; + unsigned long nbFrames; +}; + + +int dabOutputFifoInit(void** args) +{ + dabOutputFifoData* data = new dabOutputFifoData; + + *args = data; + data->file = -1; + data->type = ETI_FILE_TYPE_STREAMED; + data->nbFrames = 0; + return 0; +} + + +int dabOutputFifoOpen(void* args, const char* filename) +{ + dabOutputFifoData* data = (dabOutputFifoData*)args; + + char* token = strchr((char*)filename, '?'); + if (token != NULL) { + *(token++) = 0; + char* nextPair; + char* key; + char* value; + do { + nextPair = strchr(token, '&'); + if (nextPair != NULL) { + *nextPair = 0; + } + key = token; + value = strchr(token, '='); + if (value != NULL) { + *(value++) = 0; + if (strcmp(key, "type") == 0) { + if (strcmp(value, "raw") == 0) { + data->type = ETI_FILE_TYPE_RAW; + break; + } else if (strcmp(value, "framed") == 0) { + data->type = ETI_FILE_TYPE_FRAMED; + break; + } else if (strcmp(value, "streamed") == 0) { + data->type = ETI_FILE_TYPE_STREAMED; + break; + } else { + etiLog.printHeader(TcpLog::ERR, + "File type '%s' is not supported.\n", value); + return -1; + } + } + } + } while (nextPair != NULL); + } + + data->file = open(filename, O_WRONLY | O_CREAT | O_TRUNC | O_BINARY, 0666); + if (data->file == -1) { + perror(filename); + return -1; + } + return 0; +} + + +int dabOutputFifoWrite(void* args, void* buffer, int size) +{ + dabOutputFifoData* data = (dabOutputFifoData*)args; + uint8_t padding[6144]; + + switch (data->type) { + case ETI_FILE_TYPE_FRAMED: + if (data->nbFrames == 0) { + uint32_t nbFrames = (uint32_t)-1; + // Writting nb frame + if (write(data->file, &nbFrames, 4) == -1) goto FIFO_WRITE_ERROR; + } + case ETI_FILE_TYPE_STREAMED: + // Writting frame length + if (write(data->file, &size, 2) == -1) goto FIFO_WRITE_ERROR; + // Appending data + if (write(data->file, buffer, size) == -1) goto FIFO_WRITE_ERROR; + break; + case ETI_FILE_TYPE_RAW: + // Appending data + if (write(data->file, buffer, size) == -1) goto FIFO_WRITE_ERROR; + // Appending padding + memset(padding, 0x55, 6144 - size); + if (write(data->file, padding, 6144 - size) == -1) goto FIFO_WRITE_ERROR; + break; + default: + etiLog.printHeader(TcpLog::ERR, "File type is not supported.\n"); + return -1; + } + + + return size; + +FIFO_WRITE_ERROR: + perror("Error while writting to file"); + return -1; +} + + +int dabOutputFifoClose(void* args) +{ + dabOutputFifoData* data = (dabOutputFifoData*)args; + + if (close(data->file) == 0) { + data->file = -1; + return 0; + } + perror("Can't close file"); + return -1; +} + + +int dabOutputFifoClean(void** args) +{ + delete (dabOutputFifoData*)(*args); + return 0; +} + + +struct dabOutputOperations dabOutputFifoOperations = { + dabOutputFifoInit, + dabOutputFifoOpen, + dabOutputFifoWrite, + dabOutputFifoClose, + dabOutputFifoClean +}; + + +struct dabOutputFileData { + int file; + EtiFileType type; + unsigned long nbFrames; +}; + + +int dabOutputFileInit(void** args) +{ + dabOutputFileData* data = new dabOutputFileData; + + *args = data; + data->file = -1; + data->type = ETI_FILE_TYPE_FRAMED; + data->nbFrames = 0; + return 0; +} + + +int dabOutputFileWrite(void* args, void* buffer, int size) +{ + dabOutputFileData* data = (dabOutputFileData*)args; + + uint8_t padding[6144]; + ++data->nbFrames; + + switch (data->type) { + case ETI_FILE_TYPE_FRAMED: + // Writting nb of frames at beginning of file + if (lseek(data->file, 0, SEEK_SET) == -1) goto FILE_WRITE_ERROR; + if (write(data->file, &data->nbFrames, 4) == -1) goto FILE_WRITE_ERROR; + + // Writting nb frame length at end of file + if (lseek(data->file, 0, SEEK_END) == -1) goto FILE_WRITE_ERROR; + if (write(data->file, &size, 2) == -1) goto FILE_WRITE_ERROR; + + // Appending data + if (write(data->file, buffer, size) == -1) goto FILE_WRITE_ERROR; + break; + case ETI_FILE_TYPE_STREAMED: + // Writting nb frame length at end of file + if (write(data->file, &size, 2) == -1) goto FILE_WRITE_ERROR; + + // Appending data + if (write(data->file, buffer, size) == -1) goto FILE_WRITE_ERROR; + break; + case ETI_FILE_TYPE_RAW: + // Appending data + if (write(data->file, buffer, size) == -1) goto FILE_WRITE_ERROR; + + // Appending padding + memset(padding, 0x55, 6144 - size); + if (write(data->file, padding, 6144 - size) == -1) goto FILE_WRITE_ERROR; + break; + default: + etiLog.printHeader(TcpLog::ERR, "File type is not supported.\n"); + return -1; + } + + return size; + +FILE_WRITE_ERROR: + perror("Error while writting to file"); + return -1; +} + + +int dabOutputFileClean(void** args) +{ + delete (dabOutputFileData*)(*args); + return 0; +} + + +struct dabOutputOperations dabOutputFileOperations = { + dabOutputFileInit, + dabOutputFifoOpen, + dabOutputFileWrite, + dabOutputFifoClose, + dabOutputFileClean +}; + + +struct dabOutputRawData { +#ifdef _WIN32 + HANDLE socket; +#else + int socket; + bool isCyclades; +#endif + unsigned char* buffer; +}; + + +const unsigned char revTable[] = { + 0x00, 0x80, 0x40, 0xc0, 0x20, 0xa0, 0x60, 0xe0, + 0x10, 0x90, 0x50, 0xd0, 0x30, 0xb0, 0x70, 0xf0, + 0x08, 0x88, 0x48, 0xc8, 0x28, 0xa8, 0x68, 0xe8, + 0x18, 0x98, 0x58, 0xd8, 0x38, 0xb8, 0x78, 0xf8, + 0x04, 0x84, 0x44, 0xc4, 0x24, 0xa4, 0x64, 0xe4, + 0x14, 0x94, 0x54, 0xd4, 0x34, 0xb4, 0x74, 0xf4, + 0x0c, 0x8c, 0x4c, 0xcc, 0x2c, 0xac, 0x6c, 0xec, + 0x1c, 0x9c, 0x5c, 0xdc, 0x3c, 0xbc, 0x7c, 0xfc, + 0x02, 0x82, 0x42, 0xc2, 0x22, 0xa2, 0x62, 0xe2, + 0x12, 0x92, 0x52, 0xd2, 0x32, 0xb2, 0x72, 0xf2, + 0x0a, 0x8a, 0x4a, 0xca, 0x2a, 0xaa, 0x6a, 0xea, + 0x1a, 0x9a, 0x5a, 0xda, 0x3a, 0xba, 0x7a, 0xfa, + 0x06, 0x86, 0x46, 0xc6, 0x26, 0xa6, 0x66, 0xe6, + 0x16, 0x96, 0x56, 0xd6, 0x36, 0xb6, 0x76, 0xf6, + 0x0e, 0x8e, 0x4e, 0xce, 0x2e, 0xae, 0x6e, 0xee, + 0x1e, 0x9e, 0x5e, 0xde, 0x3e, 0xbe, 0x7e, 0xfe, + 0x01, 0x81, 0x41, 0xc1, 0x21, 0xa1, 0x61, 0xe1, + 0x11, 0x91, 0x51, 0xd1, 0x31, 0xb1, 0x71, 0xf1, + 0x09, 0x89, 0x49, 0xc9, 0x29, 0xa9, 0x69, 0xe9, + 0x19, 0x99, 0x59, 0xd9, 0x39, 0xb9, 0x79, 0xf9, + 0x05, 0x85, 0x45, 0xc5, 0x25, 0xa5, 0x65, 0xe5, + 0x15, 0x95, 0x55, 0xd5, 0x35, 0xb5, 0x75, 0xf5, + 0x0d, 0x8d, 0x4d, 0xcd, 0x2d, 0xad, 0x6d, 0xed, + 0x1d, 0x9d, 0x5d, 0xdd, 0x3d, 0xbd, 0x7d, 0xfd, + 0x03, 0x83, 0x43, 0xc3, 0x23, 0xa3, 0x63, 0xe3, + 0x13, 0x93, 0x53, 0xd3, 0x33, 0xb3, 0x73, 0xf3, + 0x0b, 0x8b, 0x4b, 0xcb, 0x2b, 0xab, 0x6b, 0xeb, + 0x1b, 0x9b, 0x5b, 0xdb, 0x3b, 0xbb, 0x7b, 0xfb, + 0x07, 0x87, 0x47, 0xc7, 0x27, 0xa7, 0x67, 0xe7, + 0x17, 0x97, 0x57, 0xd7, 0x37, 0xb7, 0x77, 0xf7, + 0x0f, 0x8f, 0x4f, 0xcf, 0x2f, 0xaf, 0x6f, 0xef, + 0x1f, 0x9f, 0x5f, 0xdf, 0x3f, 0xbf, 0x7f, 0xff +}; + + +int dabOutputRawInit(void** args) +{ + dabOutputRawData* data = new dabOutputRawData; + + *args = data; +#ifdef _WIN32 + data->socket = INVALID_HANDLE_VALUE; +#else + data->socket = -1; + data->isCyclades = false; +#endif + data->buffer = new unsigned char[6144]; + return 0; +} + + +#ifdef _WIN32 +# include <fscfg.h> +# include <sdci.h> +#else +# include <netinet/in.h> +# include <sys/types.h> +# include <sys/socket.h> +# include <sys/ioctl.h> +# include <linux/if_packet.h> +# include <linux/netdevice.h> +# include <net/if_arp.h> +#endif +int dabOutputRawOpen(void* args, const char* filename) +{ + dabOutputRawData* data = (dabOutputRawData*)args; + + if (filename == NULL) { + etiLog.printHeader(TcpLog::ERR, "Socket name must be provided!\n"); + return -1; + } + +#ifdef _WIN32 + // Opening device + data->socket = CreateFile(filename, GENERIC_WRITE, 0, NULL, OPEN_EXISTING, + FILE_ATTRIBUTE_NORMAL, NULL); + if (data->socket == INVALID_HANDLE_VALUE) { + etiLog.printHeader(TcpLog::ERR, "Can't open raw device '%s': %i\n", + filename, GetLastError()); + return -1; + } + + // Configuring device + DWORD result; + FS_TE1_CONFIG config; + if (!DeviceIoControl(data->socket, IoctlCodeFarSyncGetTE1Config, NULL, 0, + &config, sizeof(config), &result, NULL)) { + etiLog.printHeader(TcpLog::ERR, + "Can't get raw device '%s' config: %i\n", + filename, GetLastError()); + return -1; + } + config.dataRate = 2048000; + config.clocking = CLOCKING_MASTER; + config.framing = FRAMING_E1; + config.structure = STRUCTURE_UNFRAMED; + config.iface = INTERFACE_BNC; + config.coding = CODING_HDB3; + config.lineBuildOut = LBO_0dB; + config.equalizer = EQUALIZER_SHORT; + config.transparentMode = TRUE; + config.loopMode = LOOP_NONE; + config.range = RANGE_0_40_M; + config.txBufferMode = BUFFER_2_FRAME; + config.rxBufferMode = BUFFER_2_FRAME; + config.startingTimeSlot = 0; + config.losThreshold = 2; + config.enableIdleCode = TRUE; + config.idleCode = 0xff; + if (!DeviceIoControl(data->socket, IoctlCodeFarSyncSetTE1Config, + &config, sizeof(config), NULL, 0, &result, NULL)) { + etiLog.printHeader(TcpLog::ERR, + "Can't set raw device '%s' config: %i\n", + filename, GetLastError()); + return -1; + } + + // Starting device + if (!DeviceIoControl(data->socket, IoctlCodeFarSyncQuickStart, NULL, 0, + NULL, 0, &result, NULL)) { + etiLog.printHeader(TcpLog::ERR, "Can't start raw device '%s': %i\n", + filename, GetLastError()); + return -1; + } +#else + data->socket = socket(PF_INET, SOCK_DGRAM, IPPROTO_IP); + if (data->socket == -1) { + etiLog.printHeader(TcpLog::ERR, "Are you logged as root?\n"); + perror(filename); + return -1; + } + + struct ifreq ifr; + struct sockaddr_ll saddr; + memset(&ifr, 0, sizeof(struct ifreq)); + (void)strncpy(ifr.ifr_name, filename, sizeof(ifr.ifr_name)); + + // Get current Farsync configuration + struct fstioc_info info; + memset(&info, 0, sizeof(info)); + ifr.ifr_data = (char*)&info; + if (ioctl(data->socket, FSTGETCONF, &ifr) == -1) { + etiLog.printHeader(TcpLog::DBG, "Cyclades card identified.\n"); + data->isCyclades = true; + + // Set the interface MTU if needed + if (ioctl(data->socket, SIOCGIFMTU, &ifr) == -1) { + etiLog.printHeader(TcpLog::ERR, "Can't get raw device MTU!\n"); + perror(filename); + return -1; + } else { + if (ifr.ifr_mtu != 6143) { + ifr.ifr_mtu = 6143; + if (ioctl(data->socket, SIOCSIFMTU, &ifr) == -1) { + etiLog.printHeader(TcpLog::ERR, + "Can't Cyclades device MTU!\n"); + perror(filename); + return -1; + } + } + } + } else { + etiLog.printHeader(TcpLog::DBG, "Farsync card identified.\n"); + data->isCyclades = false; + + info.lineInterface = E1; + info.proto = FST_RAW; + info.internalClock = EXTCLK; + info.lineSpeed = 2048000; + //info.debug = DBG_INIT | DBG_OPEN | DBG_PCI | DBG_IOCTL | DBG_TX; + info.transparentMode = 1; + info.ignoreCarrier = 1; + info.numTxBuffers = 8; + info.numRxBuffers = 8; + info.txBufferSize = 6144; + info.rxBufferSize = 6144; + // E1 specific config + info.clockSource = CLOCKING_SLAVE; + info.structure = STRUCTURE_UNFRAMED; + info.interface = INTERFACE_BNC; //RJ48C; + info.coding = CODING_HDB3; + info.txBufferMode = BUFFER_2_FRAME; + info.idleCode = 0xff; + info.valid = FSTVAL_ALL; + + // Setting configuration + etiLog.printHeader(TcpLog::DBG, "Set configuration.\n"); + ifr.ifr_data = (char*)&info; + if (ioctl(data->socket, FSTSETCONF, &ifr) == -1) { + etiLog.printHeader(TcpLog::ERR, + "Can't set Farsync configurationi!\n"); + perror(filename); + return -1; + } + + // Disabling notify + etiLog.printHeader(TcpLog::DBG, "Disable notify.\n"); + int notify = 0; + ifr.ifr_data = (char*)¬ify; + if (ioctl(data->socket, FSTSNOTIFY, &ifr) == -1) { + etiLog.printHeader(TcpLog::ERR, "Can't disable Farsync notify!\n"); + perror(filename); + return -1; + } + + // Apply the new configuration + // Set the interface down if needed + etiLog.printHeader(TcpLog::DBG, "Get flags.\n"); + if (ioctl(data->socket, SIOCGIFFLAGS, &ifr) == -1) { + etiLog.printHeader(TcpLog::ERR, "Can't get Farsync flags!\n"); + perror(filename); + return -1; + } else { + if (ifr.ifr_flags & IFF_UP) { + etiLog.printHeader(TcpLog::DBG, "Set flags.\n"); + ifr.ifr_flags &= ~IFF_UP; + if (ioctl(data->socket, SIOCSIFFLAGS, &ifr) == -1) { + etiLog.printHeader(TcpLog::ERR, + "Can't turn down Farsync device!\n"); + perror(filename); + return -1; + } + } + } + + // Set the interface MTU if needed + etiLog.printHeader(TcpLog::DBG, "Get MTU.\n"); + if (ioctl(data->socket, SIOCGIFMTU, &ifr) == -1) { + etiLog.printHeader(TcpLog::ERR, "Can't get Farsync MTU!\n"); + perror(filename); + return -1; + } else { + if (ifr.ifr_mtu != 6144) { + etiLog.printHeader(TcpLog::DBG, "Set MTU.\n"); + ifr.ifr_mtu = 6144; + if (ioctl(data->socket, SIOCSIFMTU, &ifr) == -1) { + etiLog.printHeader(TcpLog::ERR, "Can't set Farsync MTU!\n"); + perror(filename); + return -1; + } + } + } + } + + // Set the interface up if needed + etiLog.printHeader(TcpLog::DBG, "Get flags.\n"); + if (ioctl(data->socket, SIOCGIFFLAGS, &ifr) == -1) { + etiLog.printHeader(TcpLog::ERR, "Can't get raw device flags!\n"); + perror(filename); + return -1; + } else { + if (!(ifr.ifr_flags & IFF_UP)) { + ifr.ifr_flags |= IFF_UP; + etiLog.printHeader(TcpLog::DBG, "Set flags.\n"); + if (ioctl(data->socket, SIOCSIFFLAGS, &ifr) == -1) { + etiLog.printHeader(TcpLog::ERR, "Can't turn up raw device!\n"); + perror(filename); + return -1; + } + } + } + + close(data->socket); + + //////////////////// + // Opening device // + //////////////////// + + if ((data->socket = socket(PF_PACKET, SOCK_RAW, htons(ETH_P_CUST))) == -1) { + etiLog.printHeader(TcpLog::ERR, "Are you logged as root?\n"); + perror(filename); + return -1; + } + + // ioctl to read the interface number + etiLog.printHeader(TcpLog::DBG, "Get index.\n"); + memset(&ifr, 0, sizeof(struct ifreq)); + strncpy(ifr.ifr_name, filename, sizeof(ifr.ifr_name)); + if (ioctl(data->socket, SIOCGIFINDEX, (char *) &ifr) == -1) { + perror(filename); + return -1; + } + + // Bind to the interface name + etiLog.printHeader(TcpLog::DBG, "Bind interface.\n"); + memset(&saddr, 0, sizeof(struct sockaddr_ll)); + saddr.sll_family = AF_PACKET; + saddr.sll_protocol = ARPHRD_RAWHDLC; + saddr.sll_ifindex = ifr.ifr_ifindex; + if (bind(data->socket, (struct sockaddr *) &saddr, sizeof(saddr)) == -1) { + etiLog.printHeader(TcpLog::ERR, "Can't bind raw device!\n"); + perror(filename); + return -1; + } +#endif + + return 0; +} + + +int dabOutputRawWrite(void* args, void* buffer, int size) +{ + dabOutputRawData* data = (dabOutputRawData*)args; + + // Encoding data + memcpy(data->buffer, buffer, size); + memset(data->buffer + size, 0x55, 6144 - size); + for (int i = 0; i < 6144; ++i) { + data->buffer[i] = revTable[data->buffer[i]]; + } + + // Writting data +#ifdef _WIN32 + DWORD result; + if(!DeviceIoControl(data->socket, IoctlCodeTxFrame, data->buffer, 6144, + NULL, 0, &result, NULL)) { + goto RAW_WRITE_ERROR; + } +#else + /* + if (write(data->socket, data->buffer + 1, 6143) != 6143) { + goto RAW_WRITE_ERROR; + } + */ + if (data->isCyclades) { + if (write(data->socket, data->buffer + 1, 6143) != 6143) { + goto RAW_WRITE_ERROR; + } + } else { + int ret = send(data->socket, data->buffer, 6144, 0); + if (ret != 6144) { + fprintf(stderr, "%i/6144 bytes written\n", ret); + goto RAW_WRITE_ERROR; + } + } +#endif + + return size; + +RAW_WRITE_ERROR: +#ifdef _WIN32 + DWORD err = GetLastError(); + LPSTR errMsg; + if(FormatMessage( + FORMAT_MESSAGE_ALLOCATE_BUFFER | FORMAT_MESSAGE_FROM_SYSTEM, + NULL, + err, + 0, + (LPTSTR)&errMsg, + 0, + NULL) == 0) { + fprintf(stderr, "Error while writting to raw socket: %i\n", err); + } else { + fprintf(stderr, "Error while writting to raw socket: %s\n", errMsg); + LocalFree(errMsg); + } +#else + perror("Error while writting to raw socket"); +#endif + + return -1; +} + + +int dabOutputRawClose(void* args) +{ + dabOutputRawData* data = (dabOutputRawData*)args; + +#ifdef _WIN32 + CancelIo(data->socket); + CloseHandle(data->socket); + return 0; +#else + if (close(data->socket) == 0) { + data->socket = -1; + return 0; + } + perror("Can't close raw socket"); +#endif + + return -1; +} + + +int dabOutputRawClean(void** args) +{ + dabOutputRawData* data = *((dabOutputRawData**)args); + + delete []data->buffer; + delete data; + return 0; +} + + +struct dabOutputOperations dabOutputRawOperations = { + dabOutputRawInit, + dabOutputRawOpen, + dabOutputRawWrite, + dabOutputRawClose, + dabOutputRawClean +}; + + +struct dabOutputUdpData { + UdpSocket* socket; + UdpPacket* packet; +}; + + +int dabOutputUdpInit(void** args) +{ + dabOutputUdpData* data = new dabOutputUdpData; + + *args = data; + UdpSocket::init(); + data->packet = new UdpPacket(6144); + data->socket = new UdpSocket(); + + return 0; +} + + +int dabOutputUdpOpen(void* args, const char* filename) +{ + dabOutputUdpData* data = (dabOutputUdpData*)args; + filename = strdup(filename); + + char* address; + long port; + address = strchr((char*)filename, ':'); + if (address == NULL) { + etiLog.printHeader(TcpLog::ERR, + "\"%s\" is an invalid format for udp address: " + "should be [address]:port - > aborting\n", + filename); + return -1; + } + *(address++) = 0; + port = strtol(address, (char **)NULL, 10); + if ((port == LONG_MIN) || (port == LONG_MAX)) { + etiLog.printHeader(TcpLog::ERR, + "can't convert port number in udp address %s\n", address); + return -1; + } + if (port == 0) { + etiLog.printHeader(TcpLog::ERR, + "can't use port number 0 in udp address\n"); + return -1; + } + address = (char*)filename; + if (strlen(address) > 0) { + if (data->packet->getAddress().setAddress(address) == -1) { + etiLog.printHeader(TcpLog::ERR, "can't set address %s (%s: %s) " + "-> aborting\n", address, inetErrDesc, inetErrMsg); + return -1; + } + } + data->packet->getAddress().setPort(port); + + if (data->socket->create() == -1) { + etiLog.printHeader(TcpLog::ERR, "can't create Udp socket (%s: %s) " + "-> aborting\n)", inetErrDesc, inetErrMsg); + return -1; + } + + //sprintf(filename, "%s:%i", data->packet->getAddress().getHostAddress(), + // data->packet->getAddress().getPort()); + return 0; +} + + +int dabOutputUdpWrite(void* args, void* buffer, int size) +{ + dabOutputUdpData* data = (dabOutputUdpData*)args; + + data->packet->setLength(0); + data->packet->addData(buffer, size); + return data->socket->send(*data->packet); +} + + +int dabOutputUdpClose(void* args) +{ + //dabOutputUdpData* data = (dabOutputUdpData*)args; + + return 0; +} + + +int dabOutputUdpClean(void** args) +{ + dabOutputUdpData* data = *((dabOutputUdpData**)args); + + delete data->socket; + delete data->packet; + delete data; + + return 0; +} + + +struct dabOutputOperations dabOutputUdpOperations = { + dabOutputUdpInit, + dabOutputUdpOpen, + dabOutputUdpWrite, + dabOutputUdpClose, + dabOutputUdpClean +}; + + +#include "TcpServer.h" +struct dabOutputTcpData { + TcpServer* server; + TcpSocket* client; + pthread_t thread; +}; + + +int dabOutputTcpInit(void** args) +{ + dabOutputTcpData* data = new dabOutputTcpData; + + *args = data; + TcpSocket::init(); + data->server = new TcpServer(); + data->client = NULL; + + return 0; +} + + +void* dabOutputTcpThread(void* args) +{ + dabOutputTcpData* data = (dabOutputTcpData*)args; + TcpSocket* client; + + while ((client = data->server->accept()) != NULL) { + etiLog.print(TcpLog::INFO, "TCP server got a new client.\n"); + if (data->client != NULL) { + delete data->client; + } + data->client = client; + } + etiLog.print(TcpLog::ERR, "TCP thread can't accept new client (%s)\n", + inetErrDesc, inetErrMsg); + + return NULL; +} + + +int dabOutputTcpOpen(void* args, const char* filename) +{ + dabOutputTcpData* data = (dabOutputTcpData*)args; + filename = strdup(filename); + + char* address; + long port; + address = strchr((char*)filename, ':'); + if (address == NULL) { + etiLog.printHeader(TcpLog::ERR, + "\"%s\" is an invalid format for tcp address: " + "should be [address]:port - > aborting\n", + filename); + return -1; + } + *(address++) = 0; + port = strtol(address, (char **)NULL, 10); + if ((port == LONG_MIN) || (port == LONG_MAX)) { + etiLog.printHeader(TcpLog::ERR, + "can't convert port number in tcp address %s\n", address); + return -1; + } + if (port == 0) { + etiLog.printHeader(TcpLog::ERR, + "can't use port number 0 in tcp address\n"); + return -1; + } + address = (char*)filename; + if (strlen(address) > 0) { + if (data->server->create(port, address) == -1) { + etiLog.printHeader(TcpLog::ERR, "Can't create Tcp server on %s:%i " + "(%s: %s) -> aborting\n", + address, port, inetErrDesc, inetErrMsg); + return -1; + } + } else { + if (data->server->create(port) == -1) { + etiLog.printHeader(TcpLog::ERR, "Can't create Tcp server on :%i " + "(%s: %s) -> aborting\n", + port, inetErrDesc, inetErrMsg); + return -1; + } + } + + //sprintf(filename, "%s:%i", data->packet->getAddress().getHostAddress(), + // data->packet->getAddress().getPort()); + + if (data->server->listen() == -1) { + etiLog.printHeader(TcpLog::ERR, "Can't listen on Tcp socket (%s: %s)\n", + inetErrDesc, inetErrMsg); + return -1; + } +#ifdef _WIN32 + data->thread = CreateThread(NULL, 0, (LPTHREAD_START_ROUTINE)dabOutputTcpThread, data, 0, NULL); + if (data->thread == NULL) { + fprintf(stderr, "Can't create TCP child"); + return -1; + } +#else + if (pthread_create(&data->thread, NULL, dabOutputTcpThread, data)) { + perror("Can't create TCP child"); + return -1; + } +#endif + + return 0; +} + + +int dabOutputTcpWrite(void* args, void* buffer, int size) +{ + dabOutputTcpData* data = (dabOutputTcpData*)args; + + if (data->client != NULL) { + if (data->client->write(&size, 2) == 2) { + if (data->client->write(buffer, size) != size) { + return size; + } + } + else { + etiLog.print(TcpLog::INFO, "TCP server client disconnected.\n"); + delete data->client; + data->client = NULL; + } + } + return size; +} + + +#include <signal.h> +int dabOutputTcpClose(void* args) +{ + dabOutputTcpData* data = (dabOutputTcpData*)args; + + data->server->close(); + if( data->client != NULL ) + data->client->close(); +#ifdef WIN32 + DWORD status; + for (int i = 0; i < 5; ++i) { + if (GetExitCodeThread(data->thread, &status)) { + break; + } + Sleep(100); + } + TerminateThread(data->thread, 1); +#else + pthread_kill(data->thread, SIGPIPE); +#endif + + return 0; +} + + +int dabOutputTcpClean(void** args) +{ + dabOutputTcpData* data = *((dabOutputTcpData**)args); + +#ifdef _WIN32 + CloseHandle(data->thread); +#endif + + delete data->server; + delete data->client; + delete data; + + return 0; +} + + +struct dabOutputOperations dabOutputTcpOperations = { + dabOutputTcpInit, + dabOutputTcpOpen, + dabOutputTcpWrite, + dabOutputTcpClose, + dabOutputTcpClean +}; + + +struct dabOutputSimulData { +#ifdef _WIN32 + DWORD startTime; +#else + timeval startTime; +#endif +}; + + +int dabOutputSimulInit(void** args) { + dabOutputSimulData* data = new dabOutputSimulData; + + *args = data; + + return 0; +} + + +int dabOutputSimulOpen(void* args, const char* filename) +{ + dabOutputSimulData* data = (dabOutputSimulData*)args; + +#ifdef _WIN32 + data->startTime = GetTickCount(); +#else + gettimeofday(&data->startTime, NULL); +#endif + + return 0; +} + + +int dabOutputSimulWrite(void* args, void* buffer, int size) +{ + dabOutputSimulData* data = (dabOutputSimulData*)args; + + unsigned long current; + unsigned long start; + unsigned long waiting; + +#ifdef _WIN32 + current = GetTickCount(); + start = data->startTime; + if (current < start) { + waiting = start - current + 24; + Sleep(waiting); + } else { + waiting = 24 - (current - start); + if ((current - start) < 24) { + Sleep(waiting); + } + } + data->startTime += 24; +#else + timeval curTime; + gettimeofday(&curTime, NULL); + current = (1000000ul * curTime.tv_sec) + curTime.tv_usec; + start = (1000000ul * data->startTime.tv_sec) + data->startTime.tv_usec; + waiting = 24000ul - (current - start); + if ((current - start) < 24000ul) { + usleep(waiting); + } + + data->startTime.tv_usec += 24000; + if (data->startTime.tv_usec >= 1000000) { + data->startTime.tv_usec -= 1000000; + ++data->startTime.tv_sec; + } +#endif + + return size; +} + + +int dabOutputSimulClose(void* args) +{ + //dabOutputSimulData* data = (dabOutputSimulData*)args); + + return 0; +} + + +int dabOutputSimulClean(void** args) +{ + dabOutputSimulData* data = *((dabOutputSimulData**)args); + + delete data; + + return 0; +} + + +struct dabOutputOperations dabOutputSimulOperations = { + dabOutputSimulInit, + dabOutputSimulOpen, + dabOutputSimulWrite, + dabOutputSimulClose, + dabOutputSimulClean +}; + + |