diff options
author | Jay Sorg <jay.sorg@gmail.com> | 2012-01-26 00:31:17 -0600 |
---|---|---|
committer | Jay Sorg <jay.sorg@gmail.com> | 2012-01-26 00:31:17 -0600 |
commit | 012ed7541c4e0fe299c668a475595c4ac7cd31fc (patch) | |
tree | e7c95906cddae0a600b3f355050ec39732d89103 | |
parent | bffdbe1d4b8bad17bf688806e33521c59e5982d7 (diff) | |
download | xrdp-proprietary-012ed7541c4e0fe299c668a475595c4ac7cd31fc.tar.gz xrdp-proprietary-012ed7541c4e0fe299c668a475595c4ac7cd31fc.zip |
added tcp_proxy
-rw-r--r-- | tests/tcp_proxy/arch.h | 80 | ||||
-rw-r--r-- | tests/tcp_proxy/main.c | 265 | ||||
-rw-r--r-- | tests/tcp_proxy/os_calls.c | 1449 | ||||
-rw-r--r-- | tests/tcp_proxy/os_calls.h | 208 |
4 files changed, 2002 insertions, 0 deletions
diff --git a/tests/tcp_proxy/arch.h b/tests/tcp_proxy/arch.h new file mode 100644 index 00000000..2f3c51a2 --- /dev/null +++ b/tests/tcp_proxy/arch.h @@ -0,0 +1,80 @@ +/* + Copyright (c) 2004-2007 Jay Sorg + + Permission is hereby granted, free of charge, to any person obtaining a + copy of this software and associated documentation files (the "Software"), + to deal in the Software without restriction, including without limitation + the rights to use, copy, modify, merge, publish, distribute, sublicense, + and/or sell copies of the Software, and to permit persons to whom the + Software is furnished to do so, subject to the following conditions: + + The above copyright notice and this permission notice shall be included + in all copies or substantial portions of the Software. + + THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS + OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, + FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE + AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER + LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING + FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER + DEALINGS IN THE SOFTWARE. + +*/ + +#if !defined(ARCH_H) +#define ARCH_H + +/* check endianess */ +#if defined(__sparc__) || defined(__PPC__) || defined(__ppc__) +#define B_ENDIAN +#elif __BYTE_ORDER == __LITTLE_ENDIAN +#define L_ENDIAN +#elif __BYTE_ORDER == __BIG_ENDIAN +#define B_ENDIAN +#endif +/* check if we need to align data */ +#if defined(__sparc__) || defined(__alpha__) || defined(__hppa__) || \ + defined(__AIX__) || defined(__PPC__) || defined(__mips__) || \ + defined(__ia64__) || defined(__ppc__) +#define NEED_ALIGN +#endif + +/* defines for thread creation factory functions */ +#if defined(_WIN32) +#define THREAD_RV unsigned long +#define THREAD_CC __stdcall +#else +#define THREAD_RV void* +#define THREAD_CC +#endif + +#if defined(__BORLANDC__) || defined(_WIN32) +#define APP_CC __fastcall +#define DEFAULT_CC __cdecl +#else +#define APP_CC +#define DEFAULT_CC +#endif + +#if defined(_WIN32) +#if defined(__BORLANDC__) +#define EXPORT_CC _export __cdecl +#else +#define EXPORT_CC +#endif +#else +#define EXPORT_CC +#endif + +typedef char ti8; +typedef unsigned char tui8; +typedef signed char tsi8; +typedef short ti16; +typedef unsigned short tui16; +typedef signed short tsi16; +typedef int ti32; +typedef unsigned int tui32; +typedef signed int tsi32; +typedef long tbus; + +#endif diff --git a/tests/tcp_proxy/main.c b/tests/tcp_proxy/main.c new file mode 100644 index 00000000..7bfc747e --- /dev/null +++ b/tests/tcp_proxy/main.c @@ -0,0 +1,265 @@ + +#include "os_calls.h" + +int g_loc_io_count = 0; // bytes read from local port +int g_rem_io_count = 0; // bytes read from remote port + +static int g_terminated = 0; +static char g_buf[1024 * 32]; + +/*****************************************************************************/ +static int +main_loop(char* local_port, char* remote_ip, char* remote_port, int hexdump) +{ + int lis_sck; + int acc_sck; + int con_sck; + int sel; + int count; + int sent; + int error; + int i; + int acc_to_con; + int con_to_acc; + + acc_to_con = 0; + con_to_acc = 0; + acc_sck = 0; + + /* create the listening socket and setup options */ + lis_sck = g_tcp_socket(); + g_tcp_set_non_blocking(lis_sck); + error = g_tcp_bind(lis_sck, local_port); + if (error != 0) + { + g_writeln("bind failed"); + } + + /* listen for an incomming connection */ + if (error == 0) + { + error = g_tcp_listen(lis_sck); + if (error == 0) + { + g_writeln("listening for connection"); + } + } + + /* accept an incomming connection */ + if (error == 0) + { + while ((!g_terminated) && (error == 0)) + { + acc_sck = g_tcp_accept(lis_sck); + if ((acc_sck == -1) && g_tcp_last_error_would_block(lis_sck)) + { + g_sleep(100); + } + else if (acc_sck == -1) + { + error = 1; + } + else + { + break; + } + } + if (error == 0) + { + error = g_terminated; + } + + /* stop listening */ + g_tcp_close(lis_sck); + lis_sck = 0; + if (error == 0) + { + g_writeln("got connection"); + } + } + + /* connect outgoing socket */ + con_sck = 0; + if (error == 0) + { + con_sck = g_tcp_socket(); + g_tcp_set_non_blocking(con_sck); + error = g_tcp_connect(con_sck, remote_ip, remote_port); + if ((error == -1) && g_tcp_last_error_would_block(con_sck)) + { + error = 0; + i = 0; + while ((!g_tcp_can_send(con_sck, 100)) && (!g_terminated) && (i < 100)) + { + g_sleep(100); + i++; + } + if (i > 99) + { + g_writeln("timout connecting"); + error = 1; + } + if (g_terminated) + { + error = 1; + } + } + if ((error != 0) && (!g_terminated)) + { + g_writeln("error connecting to remote\r\n"); + } + } + while ((!g_terminated) && (error == 0)) + { + sel = g_tcp_select(con_sck, acc_sck); + if (sel == 0) + { + g_sleep(10); + continue; + } + if (sel & 1) + { + // can read from con_sck w/o blocking + count = g_tcp_recv(con_sck, g_buf, 1024 * 16, 0); + error = count < 1; + if (error == 0) + { + g_loc_io_count += count; + con_to_acc += count; + if (hexdump) + { + g_writeln("from remove, the socket from connect"); + g_hexdump(g_buf, count); + } +#if 0 + g_writeln("local_io_count: %d\tremote_io_count: %d", + g_loc_io_count, g_rem_io_count); +#endif + sent = 0; + while ((sent < count) && (error == 0) && (!g_terminated)) + { + i = g_tcp_send(acc_sck, g_buf + sent, count - sent, 0); + if ((i == -1) && g_tcp_last_error_would_block(acc_sck)) + { + g_tcp_can_send(acc_sck, 1000); + } + else if (i < 1) + { + error = 1; + } + else + { + sent += i; + } + } + } + } + if (sel & 2) + { + // can read from acc_sck w/o blocking + count = g_tcp_recv(acc_sck, g_buf, 1024 * 16, 0); + error = count < 1; + if (error == 0) + { + g_rem_io_count += count; + acc_to_con += count; + if (hexdump) + { + g_writeln("from accepted, the socket from accept"); + g_hexdump(g_buf, count); + } +#if 0 + g_writeln("local_io_count: %d\tremote_io_count: %d", + g_loc_io_count, g_rem_io_count); +#endif + sent = 0; + while ((sent < count) && (error == 0) && (!g_terminated)) + { + i = g_tcp_send(con_sck, g_buf + sent, count - sent, 0); + if ((i == -1) && g_tcp_last_error_would_block(con_sck)) + { + g_tcp_can_send(con_sck, 1000); + } + else if (i < 1) + { + error = 1; + } + else + { + sent += i; + } + } + } + } + } + g_tcp_close(lis_sck); + g_tcp_close(con_sck); + g_tcp_close(acc_sck); + g_writeln("acc_to_con %d", acc_to_con); + g_writeln("con_to_acc %d", con_to_acc); + return 0; +} + + +/*****************************************************************************/ +static int +usage(void) +{ + g_writeln("tcp_proxy <local-port> <remote-ip> <remote-port> [dump]"); + return 0; +} + + +/*****************************************************************************/ +void +proxy_shutdown(int sig) +{ + g_writeln("shutting down"); + g_terminated = 1; +} + +void +clear_counters(int sig) +{ + g_writeln("cleared counters at: local_io_count: %d remote_io_count: %d", + g_loc_io_count, g_rem_io_count); + g_loc_io_count = 0; + g_rem_io_count = 0; +} + +/*****************************************************************************/ +int +main(int argc, char** argv) +{ + int dump; + + if (argc < 4) + { + usage(); + return 0; + } + g_init(); + g_signal(2, proxy_shutdown); /* SIGINT */ + g_signal(9, proxy_shutdown); /* SIGKILL */ + g_signal(10, clear_counters);/* SIGUSR1*/ + g_signal(15, proxy_shutdown);/* SIGTERM */ + if (argc < 5) + { + while (!g_terminated) + { + g_loc_io_count = 0; + g_rem_io_count = 0; + main_loop(argv[1], argv[2], argv[3], 0); + } + } + else + { + dump = g_strcasecmp(argv[4], "dump") == 0; + while (!g_terminated) + { + main_loop(argv[1], argv[2], argv[3], dump); + } + } + g_deinit(); + return 0; +} diff --git a/tests/tcp_proxy/os_calls.c b/tests/tcp_proxy/os_calls.c new file mode 100644 index 00000000..b17856bc --- /dev/null +++ b/tests/tcp_proxy/os_calls.c @@ -0,0 +1,1449 @@ +/* + Copyright (c) 2004-2007 Jay Sorg + + Permission is hereby granted, free of charge, to any person obtaining a + copy of this software and associated documentation files (the "Software"), + to deal in the Software without restriction, including without limitation + the rights to use, copy, modify, merge, publish, distribute, sublicense, + and/or sell copies of the Software, and to permit persons to whom the + Software is furnished to do so, subject to the following conditions: + + The above copyright notice and this permission notice shall be included + in all copies or substantial portions of the Software. + + THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS + OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, + FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE + AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER + LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING + FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER + DEALINGS IN THE SOFTWARE. + + generic operating system calls + + put all the os / arch define in here you want +*/ + +#if defined(_WIN32) +#include <windows.h> +#include <winsock.h> +#else +/* fix for solaris 10 with gcc 3.3.2 problem */ +#if defined(sun) || defined(__sun) +#define ctid_t id_t +#endif +#include <unistd.h> +#include <errno.h> +#include <netinet/in.h> +#include <netinet/tcp.h> +#include <sys/socket.h> +#include <sys/un.h> +#include <sys/time.h> +#include <sys/types.h> +#include <sys/wait.h> +#include <sys/stat.h> +#include <dlfcn.h> +#include <arpa/inet.h> +#include <netdb.h> +#include <signal.h> +#include <fcntl.h> +#include <pwd.h> +#include <time.h> +#include <grp.h> +#endif + +#include <stdlib.h> +#include <string.h> +#include <stdarg.h> +#include <stdio.h> + +#include "os_calls.h" +#include "arch.h" + +/* for clearenv() */ +#if defined(_WIN32) +#else +extern char** environ; +#endif + +/* for solaris */ +#if !defined(PF_LOCAL) +#define PF_LOCAL AF_UNIX +#endif +#if !defined(INADDR_NONE) +#define INADDR_NONE ((unsigned long)-1) +#endif + +/*****************************************************************************/ +void APP_CC +g_init(void) +{ + #if defined(_WIN32) + WSADATA wsadata; + + WSAStartup(2, &wsadata); + #endif +} + + +/*****************************************************************************/ +void APP_CC +g_deinit(void) +{ + #if defined(_WIN32) + WSACleanup(); + #endif +} + + +/*****************************************************************************/ +/* allocate memory, returns a pointer to it, size bytes are allocated, + if zero is non zero, each byte will be set to zero */ +void* APP_CC +g_malloc(int size, int zero) +{ + char* rv; + + rv = (char*)malloc(size); + if (zero) + { + memset(rv, 0, size); + } + return rv; +} + + +/*****************************************************************************/ +/* free the memory pointed to by ptr, ptr can be zero */ +void APP_CC +g_free(void* ptr) +{ + if (ptr != 0) + { + free(ptr); + } +} + + +/*****************************************************************************/ +/* output text to stdout, try to use g_write / g_writeln instead to avoid + linux / windows EOL problems */ +void DEFAULT_CC +g_printf(const char* format, ...) +{ + va_list ap; + + va_start(ap, format); + vfprintf(stdout, format, ap); + va_end(ap); +} + + +/*****************************************************************************/ +void DEFAULT_CC +g_sprintf(char* dest, const char* format, ...) +{ + va_list ap; + + va_start(ap, format); + vsprintf(dest, format, ap); + va_end(ap); +} + + +/*****************************************************************************/ +void DEFAULT_CC +g_snprintf(char* dest, int len, const char* format, ...) +{ + va_list ap; + + va_start(ap, format); + vsnprintf(dest, len, format, ap); + va_end(ap); +} + + +/*****************************************************************************/ +void DEFAULT_CC +g_writeln(const char* format, ...) +{ + va_list ap; + + va_start(ap, format); + vfprintf(stdout, format, ap); + va_end(ap); + #if defined(_WIN32) + g_printf("\r\n"); + #else + g_printf("\n"); + #endif +} + + +/*****************************************************************************/ +void DEFAULT_CC +g_write(const char* format, ...) +{ + va_list ap; + + va_start(ap, format); + vfprintf(stdout, format, ap); + va_end(ap); +} + + +/*****************************************************************************/ +/* produce a hex dump */ +void APP_CC +g_hexdump(char* p, int len) +{ + unsigned char* line; + int i; + int thisline; + int offset; + + line = (unsigned char*)p; + offset = 0; + while (offset < len) + { + g_printf("%04x ", offset); + thisline = len - offset; + if (thisline > 16) + { + thisline = 16; + } + for (i = 0; i < thisline; i++) + { + g_printf("%02x ", line[i]); + } + for (; i < 16; i++) + { + g_printf(" "); + } + for (i = 0; i < thisline; i++) + { + g_printf("%c", (line[i] >= 0x20 && line[i] < 0x7f) ? line[i] : '.'); + } + g_writeln(""); + offset += thisline; + line += thisline; + } +} + + +/*****************************************************************************/ +void APP_CC +g_memset(void* ptr, int val, int size) +{ + memset(ptr, val, size); +} + + +/*****************************************************************************/ +void APP_CC +g_memcpy(void* d_ptr, const void* s_ptr, int size) +{ + memcpy(d_ptr, s_ptr, size); +} + + +/*****************************************************************************/ +int APP_CC +g_getchar(void) +{ + return getchar(); +} + + +/*****************************************************************************/ +int APP_CC +g_tcp_set_no_delay(int sck) +{ + int i; + + i = 1; + #if defined(_WIN32) + setsockopt(sck, IPPROTO_TCP, TCP_NODELAY, (char*)&i, sizeof(i)); + #else + setsockopt(sck, IPPROTO_TCP, TCP_NODELAY, (void*)&i, sizeof(i)); + #endif + return 0; +} + + +/*****************************************************************************/ +int APP_CC +g_tcp_socket(void) +{ + int rv; + int i; + + rv = socket(PF_INET, SOCK_STREAM, 0); + #if defined(_WIN32) + i = 1; + setsockopt(rv, IPPROTO_TCP, TCP_NODELAY, (char*)&i, sizeof(i)); + i = 1; + setsockopt(rv, SOL_SOCKET, SO_REUSEADDR, (char*)&i, sizeof(i)); + i = 8192 * 2; + setsockopt(rv, SOL_SOCKET, SO_SNDBUF, (char*)&i, sizeof(i)); + #else + i = 1; + setsockopt(rv, IPPROTO_TCP, TCP_NODELAY, (void*)&i, sizeof(i)); + i = 1; + setsockopt(rv, SOL_SOCKET, SO_REUSEADDR, (void*)&i, sizeof(i)); + i = 8192 * 2; + setsockopt(rv, SOL_SOCKET, SO_SNDBUF, (void*)&i, sizeof(i)); + #endif + return rv; +} + + +/*****************************************************************************/ +int APP_CC +g_tcp_local_socket(void) +{ + #if defined(_WIN32) + return 0; + #else + return socket(PF_LOCAL, SOCK_STREAM, 0); + #endif +} + + +/*****************************************************************************/ +void APP_CC +g_tcp_close(int sck) +{ + if (sck == 0) + { + return; + } + shutdown(sck, 2); + #if defined(_WIN32) + closesocket(sck); + #else + close(sck); + #endif +} + + +/*****************************************************************************/ +int APP_CC +g_tcp_connect(int sck, const char* address, const char* port) +{ + struct sockaddr_in s; + struct hostent* h; + + g_memset(&s, 0, sizeof(struct sockaddr_in)); + s.sin_family = AF_INET; + s.sin_port = htons(atoi(port)); + s.sin_addr.s_addr = inet_addr(address); + if (s.sin_addr.s_addr == INADDR_NONE) + { + h = gethostbyname(address); + if (h != 0) + { + if (h->h_name != 0) + { + if (h->h_addr_list != 0) + { + if ((*(h->h_addr_list)) != 0) + { + s.sin_addr.s_addr = *((int*)(*(h->h_addr_list))); + } + } + } + } + } + return connect(sck, (struct sockaddr*)&s, sizeof(struct sockaddr_in)); +} + + +/*****************************************************************************/ +int APP_CC +g_tcp_set_non_blocking(int sck) +{ + unsigned long i; + + #if defined(_WIN32) + i = 1; + ioctlsocket(sck, FIONBIO, &i); + #else + i = fcntl(sck, F_GETFL); + i = i | O_NONBLOCK; + fcntl(sck, F_SETFL, i); + #endif + return 0; +} + + +/*****************************************************************************/ +int APP_CC +g_tcp_bind(int sck, char* port) +{ + struct sockaddr_in s; + + memset(&s, 0, sizeof(struct sockaddr_in)); + s.sin_family = AF_INET; + s.sin_port = htons(atoi(port)); + s.sin_addr.s_addr = INADDR_ANY; + return bind(sck, (struct sockaddr*)&s, sizeof(struct sockaddr_in)); +} + + +/*****************************************************************************/ +int APP_CC +g_tcp_local_bind(int sck, char* port) +{ + #if defined(_WIN32) + return -1; + #else + struct sockaddr_un s; + + memset(&s, 0, sizeof(struct sockaddr_un)); + s.sun_family = AF_UNIX; + strcpy(s.sun_path, port); + return bind(sck, (struct sockaddr*)&s, sizeof(struct sockaddr_un)); + #endif +} + + +/*****************************************************************************/ +int APP_CC +g_tcp_listen(int sck) +{ + return listen(sck, 2); +} + + +/*****************************************************************************/ +int APP_CC +g_tcp_accept(int sck) +{ + struct sockaddr_in s; + #if defined(_WIN32) + signed int i; + #else + unsigned int i; + #endif + + i = sizeof(struct sockaddr_in); + memset(&s, 0, i); + return accept(sck, (struct sockaddr*)&s, &i); +} + + +/*****************************************************************************/ +void APP_CC +g_sleep(int msecs) +{ + #if defined(_WIN32) + Sleep(msecs); + #else + usleep(msecs * 1000); + #endif +} + + +/*****************************************************************************/ +int APP_CC +g_tcp_last_error_would_block(int sck) +{ + #if defined(_WIN32) + return WSAGetLastError() == WSAEWOULDBLOCK; + #else + return (errno == EWOULDBLOCK) || (errno == EINPROGRESS); + #endif +} + + +/*****************************************************************************/ +int APP_CC +g_tcp_recv(int sck, void* ptr, int len, int flags) +{ + #if defined(_WIN32) + return recv(sck, (char*)ptr, len, flags); + #else + return recv(sck, ptr, len, flags); + #endif +} + + +/*****************************************************************************/ +int APP_CC +g_tcp_send(int sck, const void* ptr, int len, int flags) +{ + #if defined(_WIN32) + return send(sck, (const char*)ptr, len, flags); + #else + return send(sck, ptr, len, flags); + #endif +} + + +/*****************************************************************************/ +/* wait 'millis' milliseconds for the socket to be able to write */ +/* returns boolean */ +int APP_CC +g_tcp_can_send(int sck, int millis) +{ + fd_set wfds; + struct timeval time; + int rv; + + time.tv_sec = millis / 1000; + time.tv_usec = (millis * 1000) % 1000000; + FD_ZERO(&wfds); + if (sck > 0) + { + FD_SET(((unsigned int)sck), &wfds); + rv = select(sck + 1, 0, &wfds, 0, &time); + if (rv > 0) + { + return 1; + } + } + return 0; +} + + +/*****************************************************************************/ +/* wait 'millis' milliseconds for the socket to be able to receive */ +/* returns boolean */ +int APP_CC +g_tcp_can_recv(int sck, int millis) +{ + fd_set rfds; + struct timeval time; + int rv; + + time.tv_sec = millis / 1000; + time.tv_usec = (millis * 1000) % 1000000; + FD_ZERO(&rfds); + if (sck > 0) + { + FD_SET(((unsigned int)sck), &rfds); + rv = select(sck + 1, &rfds, 0, 0, &time); + if (rv > 0) + { + return 1; + } + } + return 0; +} + + +/*****************************************************************************/ +int APP_CC +g_tcp_select(int sck1, int sck2) +{ + fd_set rfds; + struct timeval time; + int max; + int rv; + + time.tv_sec = 0; + time.tv_usec = 0; + FD_ZERO(&rfds); + if (sck1 > 0) + { + FD_SET(((unsigned int)sck1), &rfds); + } + if (sck2 > 0) + { + FD_SET(((unsigned int)sck2), &rfds); + } + max = sck1; + if (sck2 > max) + { + max = sck2; + } + rv = select(max + 1, &rfds, 0, 0, &time); + if (rv > 0) + { + rv = 0; + if (FD_ISSET(((unsigned int)sck1), &rfds)) + { + rv = rv | 1; + } + if (FD_ISSET(((unsigned int)sck2), &rfds)) + { + rv = rv | 2; + } + } + else + { + rv = 0; + } + return rv; +} + + +/*****************************************************************************/ +void APP_CC +g_random(char* data, int len) +{ + #if defined(_WIN32) + memset(data, 0x44, len); + #else + int fd; + + memset(data, 0x44, len); + fd = open("/dev/urandom", O_RDONLY); + if (fd == -1) + { + fd = open("/dev/random", O_RDONLY); + } + if (fd != -1) + { + read(fd, data, len); + close(fd); + } + #endif +} + + +/*****************************************************************************/ +int APP_CC +g_abs(int i) +{ + return abs(i); +} + + +/*****************************************************************************/ +int APP_CC +g_memcmp(const void* s1, const void* s2, int len) +{ + return memcmp(s1, s2, len); +} + + +/*****************************************************************************/ +/* returns -1 on error, else return handle or file descriptor */ +int APP_CC +g_file_open(const char* file_name) +{ + #if defined(_WIN32) + return (int)CreateFile(file_name, GENERIC_READ | GENERIC_WRITE, + FILE_SHARE_READ | FILE_SHARE_WRITE, + 0, OPEN_ALWAYS, FILE_ATTRIBUTE_NORMAL, 0); + #else + int rv; + + rv = open(file_name, O_RDWR | O_CREAT, S_IRUSR | S_IWUSR); + if (rv == -1) + { + /* can't open read / write, try to open read only */ + rv = open(file_name, O_RDONLY); + } + return rv; + #endif +} + + +/*****************************************************************************/ +/* returns error, always 0 */ +int APP_CC +g_file_close(int fd) +{ + #if defined(_WIN32) + CloseHandle((HANDLE)fd); + #else + close(fd); + #endif + return 0; +} + + +/*****************************************************************************/ +/* read from file, returns the number of bytes read or -1 on error */ +int APP_CC +g_file_read(int fd, char* ptr, int len) +{ + #if defined(_WIN32) + if (ReadFile((HANDLE)fd, (LPVOID)ptr, (DWORD)len, (LPDWORD)&len, 0)) + { + return len; + } + else + { + return -1; + } + #else + return read(fd, ptr, len); + #endif +} + + +/*****************************************************************************/ +/* write to file, returns the number of bytes writen or -1 on error */ +int APP_CC +g_file_write(int fd, char* ptr, int len) +{ + #if defined(_WIN32) + if (WriteFile((HANDLE)fd, (LPVOID)ptr, (DWORD)len, (LPDWORD)&len, 0)) + { + return len; + } + else + { + return -1; + } + #else + return write(fd, ptr, len); + #endif +} + + +/*****************************************************************************/ +/* move file pointer, returns offset on success, -1 on failure */ +int APP_CC +g_file_seek(int fd, int offset) +{ + #if defined(_WIN32) + int rv; + + rv = (int)SetFilePointer((HANDLE)fd, offset, 0, FILE_BEGIN); + if (rv == (int)INVALID_SET_FILE_POINTER) + { + return -1; + } + else + { + return rv; + } + #else + return (int)lseek(fd, offset, SEEK_SET); + #endif +} + + +/*****************************************************************************/ +/* do a write lock on a file */ +/* return boolean */ +int APP_CC +g_file_lock(int fd, int start, int len) +{ + #if defined(_WIN32) + return LockFile((HANDLE)fd, start, 0, len, 0); + #else + struct flock lock; + + lock.l_type = F_WRLCK; + lock.l_whence = SEEK_SET; + lock.l_start = start; + lock.l_len = len; + if (fcntl(fd, F_SETLK, &lock) == -1) + { + return 0; + } + return 1; + #endif +} + + +/*****************************************************************************/ +/* returns error, always zero */ +int APP_CC +g_set_file_rights(const char* filename, int read, int write) +{ + #if defined(_WIN32) + return 0; + #else + int flags; + + flags = read ? S_IRUSR : 0; + flags |= write ? S_IWUSR : 0; + chmod(filename, flags); + return 0; + #endif +} + + +/*****************************************************************************/ +/* returns error */ +int APP_CC +g_chmod_hex(const char* filename, int flags) +{ + #if defined(_WIN32) + return 0; + #else + int fl; + + fl = 0; + fl |= (flags & 0x4000) ? S_ISUID : 0; + fl |= (flags & 0x2000) ? S_ISGID : 0; + fl |= (flags & 0x1000) ? S_ISVTX : 0; + fl |= (flags & 0x0400) ? S_IRUSR : 0; + fl |= (flags & 0x0200) ? S_IWUSR : 0; + fl |= (flags & 0x0100) ? S_IXUSR : 0; + fl |= (flags & 0x0040) ? S_IRGRP : 0; + fl |= (flags & 0x0020) ? S_IWGRP : 0; + fl |= (flags & 0x0010) ? S_IXGRP : 0; + fl |= (flags & 0x0004) ? S_IROTH : 0; + fl |= (flags & 0x0002) ? S_IWOTH : 0; + fl |= (flags & 0x0001) ? S_IXOTH : 0; + return chmod(filename, fl); + #endif +} + + +/*****************************************************************************/ +/* returns error, always zero */ +int APP_CC +g_mkdir(const char* dirname) +{ + #if defined(_WIN32) + return 0; + #else + mkdir(dirname, S_IRWXU); + return 0; + #endif +} + + +/*****************************************************************************/ +/* gets the current working directory and puts up to maxlen chars in + dirname + always returns 0 */ +char* APP_CC +g_get_current_dir(char* dirname, int maxlen) +{ + #if defined(_WIN32) + GetCurrentDirectory(maxlen, dirname); + return 0; + #else + getcwd(dirname, maxlen); + return 0; + #endif +} + + +/*****************************************************************************/ +/* returns error, zero on success and -1 on failure */ +int APP_CC +g_set_current_dir(char* dirname) +{ + #if defined(_WIN32) + if (SetCurrentDirectory(dirname)) + { + return 0; + } + else + { + return -1; + } + #else + return chdir(dirname); + #endif +} + + +/*****************************************************************************/ +/* returns boolean, non zero if the file exists */ +int APP_CC +g_file_exist(const char* filename) +{ + #if defined(_WIN32) + return 0; // use FileAge(filename) <> -1 + #else + return access(filename, F_OK) == 0; + #endif +} + + +/*****************************************************************************/ +/* returns boolean, non zero if the directory exists */ +int APP_CC +g_directory_exist(const char* dirname) +{ + #if defined(_WIN32) + return 0; // use GetFileAttributes and check return value + // is not -1 and FILE_ATTRIBUT_DIRECTORY bit is set + #else + struct stat st; + + if (stat(dirname, &st) == 0) + { + return S_ISDIR(st.st_mode); + } + else + { + return 0; + } + #endif +} + + +/*****************************************************************************/ +/* returns boolean */ +int APP_CC +g_create_dir(const char* dirname) +{ + #if defined(_WIN32) + // test this + return CreateDirectory(dirname, 0); + #else + return mkdir(dirname, (mode_t)-1) == 0; + #endif +} + + +/*****************************************************************************/ +/* returns boolean */ +int APP_CC +g_remove_dir(const char* dirname) +{ + #if defined(_WIN32) + // test this + return RemoveDirectory(dirname); + #else + return rmdir(dirname) == 0; + #endif +} + + +/*****************************************************************************/ +/* returns non zero if the file was deleted */ +int APP_CC +g_file_delete(const char* filename) +{ + #if defined(_WIN32) + return DeleteFile(filename); + #else + return unlink(filename) != -1; + #endif +} + + +/*****************************************************************************/ +/* returns length of text */ +int APP_CC +g_strlen(const char* text) +{ + if (text == 0) + { + return 0; + } + return strlen(text); +} + + +/*****************************************************************************/ +/* returns dest */ +char* APP_CC +g_strcpy(char* dest, const char* src) +{ + if (src == 0 && dest != 0) + { + dest[0] = 0; + return dest; + } + if (dest == 0 || src == 0) + { + return 0; + } + return strcpy(dest, src); +} + + +/*****************************************************************************/ +/* returns dest */ +char* APP_CC +g_strncpy(char* dest, const char* src, int len) +{ + char* rv; + + if (src == 0 && dest != 0) + { + dest[0] = 0; + return dest; + } + if (dest == 0 || src == 0) + { + return 0; + } + rv = strncpy(dest, src, len); + dest[len] = 0; + return rv; +} + + +/*****************************************************************************/ +/* returns dest */ +char* APP_CC +g_strcat(char* dest, const char* src) +{ + if (dest == 0 || src == 0) + { + return dest; + } + return strcat(dest, src); +} + + +/*****************************************************************************/ +/* if in = 0, return 0 else return newly alloced copy of in */ +char* APP_CC +g_strdup(const char* in) +{ + int len; + char* p; + + if (in == 0) + { + return 0; + } + len = g_strlen(in); + p = (char*)g_malloc(len + 1, 0); + g_strcpy(p, in); + return p; +} + + +/*****************************************************************************/ +int APP_CC +g_strcmp(const char* c1, const char* c2) +{ + return strcmp(c1, c2); +} + + +/*****************************************************************************/ +int APP_CC +g_strncmp(const char* c1, const char* c2, int len) +{ + return strncmp(c1, c2, len); +} + + +/*****************************************************************************/ +int APP_CC +g_strcasecmp(const char* c1, const char* c2) +{ + #if defined(_WIN32) + return stricmp(c1, c2); + #else + return strcasecmp(c1, c2); + #endif +} + + +/*****************************************************************************/ +int APP_CC +g_strncasecmp(const char* c1, const char* c2, int len) +{ + #if defined(_WIN32) + return strnicmp(c1, c2, len); + #else + return strncasecmp(c1, c2, len); + #endif +} + + +/*****************************************************************************/ +int APP_CC +g_atoi(char* str) +{ + return atoi(str); +} + + +/*****************************************************************************/ +int APP_CC +g_pos(char* str, const char* to_find) +{ + char* pp; + + pp = strstr(str, to_find); + if (pp == 0) + { + return -1; + } + return (pp - str); +} + + +/*****************************************************************************/ +long APP_CC +g_load_library(char* in) +{ + #if defined(_WIN32) + return (long)LoadLibrary(in); + #else + return (long)dlopen(in, RTLD_LOCAL | RTLD_LAZY); + #endif +} + + +/*****************************************************************************/ +int APP_CC +g_free_library(long lib) +{ + if (lib == 0) + { + return 0; + } + #if defined(_WIN32) + return FreeLibrary((HMODULE)lib); + #else + return dlclose((void*)lib); + #endif +} + + +/*****************************************************************************/ +/* returns NULL if not found */ +void* APP_CC +g_get_proc_address(long lib, const char* name) +{ + if (lib == 0) + { + return 0; + } + #if defined(_WIN32) + return GetProcAddress((HMODULE)lib, name); + #else + return dlsym((void*)lib, name); + #endif +} + + +/*****************************************************************************/ +int APP_CC +g_system(char* aexec) +{ + #if defined(_WIN32) + return 0; + #else + return system(aexec); + #endif +} + + +/*****************************************************************************/ +char* APP_CC +g_get_strerror(void) +{ + #if defined(_WIN32) + return 0; + #else + return strerror(errno); + #endif +} + + +/*****************************************************************************/ +int APP_CC +g_execvp(const char* p1, char* args[]) +{ + #if defined(_WIN32) + return 0; + #else + return execvp(p1, args); + #endif +} + + +/*****************************************************************************/ +int APP_CC +g_execlp3(const char* a1, const char* a2, const char* a3) +{ + #if defined(_WIN32) + return 0; + #else + return execlp(a1, a2, a3, (void*)0); + #endif +} + + +/*****************************************************************************/ +void APP_CC +g_signal(int sig_num, void (*func)(int)) +{ + #if defined(_WIN32) + #else + signal(sig_num, func); + #endif +} + + +/*****************************************************************************/ +void APP_CC +g_signal_child_stop(void (*func)(int)) +{ + #if defined(_WIN32) + #else + signal(SIGCHLD, func); + #endif +} + + +/*****************************************************************************/ +void APP_CC +g_unset_signals(void) +{ + #if defined(_WIN32) + #else + sigset_t mask; + + sigemptyset(&mask); + sigprocmask(SIG_SETMASK, &mask, NULL); + #endif +} + + +/*****************************************************************************/ +int APP_CC +g_fork(void) +{ + #if defined(_WIN32) + return 0; + #else + return fork(); + #endif +} + + +/*****************************************************************************/ +int APP_CC +g_setgid(int pid) +{ + #if defined(_WIN32) + return 0; + #else + return setgid(pid); + #endif +} + + +/*****************************************************************************/ +/* returns error, zero is success, non zero is error */ +int APP_CC +g_initgroups(const char* user, int gid) +{ + #if defined(_WIN32) + return 0; + #else + return initgroups(user ,gid); + #endif +} + + +/*****************************************************************************/ +int APP_CC +g_setuid(int pid) +{ + #if defined(_WIN32) + return 0; + #else + return setuid(pid); + #endif +} + + +/*****************************************************************************/ +int APP_CC +g_waitchild(void) +{ + #if defined(_WIN32) + return 0; + #else + int wstat; + + return waitpid(0, &wstat, WNOHANG); + #endif +} + + +/*****************************************************************************/ +int APP_CC +g_waitpid(int pid) +{ + #if defined(_WIN32) + return 0; + #else + return waitpid(pid, 0, 0); + #endif +} + + +/*****************************************************************************/ +void APP_CC +g_clearenv(void) +{ + #if defined(_WIN32) + #else + environ = 0; + #endif +} + + +/*****************************************************************************/ +int APP_CC +g_setenv(const char* name, const char* value, int rewrite) +{ + #if defined(_WIN32) + return 0; + #else + return setenv(name, value, rewrite); + #endif +} + + +/*****************************************************************************/ +char* APP_CC +g_getenv(const char* name) +{ + #if defined(_WIN32) + return 0; + #else + return getenv(name); + #endif +} + + +/*****************************************************************************/ +int APP_CC +g_exit(int exit_code) +{ + _exit(exit_code); + return 0; +} + + +/*****************************************************************************/ +int APP_CC +g_getpid(void) +{ + #if defined(_WIN32) + return 0; + #else + return getpid(); + #endif +} + + +/*****************************************************************************/ +int APP_CC +g_sigterm(int pid) +{ + #if defined(_WIN32) + return 0; + #else + return kill(pid, SIGTERM); + #endif +} + + +/*****************************************************************************/ +/* returns 0 if ok */ +int APP_CC +g_getuser_info(const char* username, int* gid, int* uid, char* shell, +char* dir, char* gecos) +{ + #if defined(_WIN32) + return 1; + #else + struct passwd* pwd_1; + + pwd_1 = getpwnam(username); + if (pwd_1 != 0) + { + if (gid != 0) + { + *gid = pwd_1->pw_gid; + } + if (uid != 0) + { + *uid = pwd_1->pw_uid; + } + if (dir != 0) + { + g_strcpy(dir, pwd_1->pw_dir); + } + if (shell != 0) + { + g_strcpy(shell, pwd_1->pw_shell); + } + if (gecos != 0) + { + g_strcpy(gecos, pwd_1->pw_gecos); + } + return 0; + } + return 1; + #endif +} + + +/*****************************************************************************/ +/* returns 0 if ok */ +int APP_CC +g_getgroup_info(const char* groupname, int* gid) +{ + #if defined(_WIN32) + return 1; + #else + struct group* g; + + g = getgrnam(groupname); + if (g != 0) + { + if (gid != 0) + { + *gid = g->gr_gid; + } + return 0; + } + return 1; + #endif +} + + +/*****************************************************************************/ +/* returns error */ +/* if zero is returned, then ok is set */ +int APP_CC +g_check_user_in_group(const char* username, int gid, int* ok) +{ + #if defined(_WIN32) + return 1; + #else + struct group* groups; + int i; + + groups = getgrgid(gid); + if (groups == 0) + { + return 1; + } + *ok = 0; + i = 0; + while (0 != groups->gr_mem[i]) + { + if (0 == g_strcmp(groups->gr_mem[i], username)) + { + *ok = 1; + break; + } + i++; + } + return 0; + #endif +} + + +/*****************************************************************************/ +/* returns the time since the Epoch (00:00:00 UTC, January 1, 1970), + measured in seconds. */ +int APP_CC +g_time1(void) +{ + #if defined(_WIN32) + return GetTickCount() / 1000; + #else + return time(0); + #endif +} diff --git a/tests/tcp_proxy/os_calls.h b/tests/tcp_proxy/os_calls.h new file mode 100644 index 00000000..768cc38c --- /dev/null +++ b/tests/tcp_proxy/os_calls.h @@ -0,0 +1,208 @@ +/* + Copyright (c) 2004-2007 Jay Sorg + + Permission is hereby granted, free of charge, to any person obtaining a + copy of this software and associated documentation files (the "Software"), + to deal in the Software without restriction, including without limitation + the rights to use, copy, modify, merge, publish, distribute, sublicense, + and/or sell copies of the Software, and to permit persons to whom the + Software is furnished to do so, subject to the following conditions: + + The above copyright notice and this permission notice shall be included + in all copies or substantial portions of the Software. + + THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS + OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, + FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE + AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER + LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING + FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER + DEALINGS IN THE SOFTWARE. + + generic operating system calls +*/ + +#if !defined(OS_CALLS_H) +#define OS_CALLS_H + +#include "arch.h" + +void APP_CC +g_init(void); +void APP_CC +g_deinit(void); +void* APP_CC +g_malloc(int size, int zero); +void APP_CC +g_free(void* ptr); +void DEFAULT_CC +g_printf(const char *format, ...); +void DEFAULT_CC +g_sprintf(char* dest, const char* format, ...); +void DEFAULT_CC +g_snprintf(char* dest, int len, const char* format, ...); +void DEFAULT_CC +g_writeln(const char* format, ...); +void DEFAULT_CC +g_write(const char* format, ...); +void APP_CC +g_hexdump(char* p, int len); +void APP_CC +g_memset(void* ptr, int val, int size); +void APP_CC +g_memcpy(void* d_ptr, const void* s_ptr, int size); +int APP_CC +g_getchar(void); +int APP_CC +g_tcp_set_no_delay(int sck); +int APP_CC +g_tcp_socket(void); +int APP_CC +g_tcp_local_socket(void); +void APP_CC +g_tcp_close(int sck); +int APP_CC +g_tcp_connect(int sck, const char* address, const char* port); +int APP_CC +g_tcp_force_send(int sck, char* data, int len); +int APP_CC +g_tcp_force_recv(int sck, char* data, int len); +int APP_CC +g_tcp_set_non_blocking(int sck); +int APP_CC +g_tcp_bind(int sck, char* port); +int APP_CC +g_tcp_local_bind(int sck, char* port); +int APP_CC +g_tcp_listen(int sck); +int APP_CC +g_tcp_accept(int sck); +int APP_CC +g_tcp_recv(int sck, void* ptr, int len, int flags); +int APP_CC +g_tcp_send(int sck, const void* ptr, int len, int flags); +int APP_CC +g_tcp_last_error_would_block(int sck); +int APP_CC +g_tcp_can_send(int sck, int millis); +int APP_CC +g_tcp_can_recv(int sck, int millis); +int APP_CC +g_tcp_select(int sck1, int sck2); +void APP_CC +g_sleep(int msecs); +void APP_CC +g_random(char* data, int len); +int APP_CC +g_abs(int i); +int APP_CC +g_memcmp(const void* s1, const void* s2, int len); +int APP_CC +g_file_open(const char* file_name); +int APP_CC +g_file_close(int fd); +int APP_CC +g_file_read(int fd, char* ptr, int len); +int APP_CC +g_file_write(int fd, char* ptr, int len); +int APP_CC +g_file_seek(int fd, int offset); +int APP_CC +g_file_lock(int fd, int start, int len); +int APP_CC +g_set_file_rights(const char* filename, int read, int write); +int APP_CC +g_chmod_hex(const char* filename, int flags); +int APP_CC +g_mkdir(const char* dirname); +char* APP_CC +g_get_current_dir(char* dirname, int maxlen); +int APP_CC +g_set_current_dir(char* dirname); +int APP_CC +g_file_exist(const char* filename); +int APP_CC +g_directory_exist(const char* dirname); +int APP_CC +g_create_dir(const char* dirname); +int APP_CC +g_remove_dir(const char* dirname); +int APP_CC +g_file_delete(const char* filename); +int APP_CC +g_strlen(const char* text); +char* APP_CC +g_strcpy(char* dest, const char* src); +char* APP_CC +g_strncpy(char* dest, const char* src, int len); +char* APP_CC +g_strcat(char* dest, const char* src); +char* APP_CC +g_strdup(const char* in); +int APP_CC +g_strcmp(const char* c1, const char* c2); +int APP_CC +g_strncmp(const char* c1, const char* c2, int len); +int APP_CC +g_strcasecmp(const char* c1, const char* c2); +int APP_CC +g_strncasecmp(const char* c1, const char* c2, int len); +int APP_CC +g_atoi(char* str); +int APP_CC +g_pos(char* str, const char* to_find); +long APP_CC +g_load_library(char* in); +int APP_CC +g_free_library(long lib); +void* APP_CC +g_get_proc_address(long lib, const char* name); +int APP_CC +g_system(char* aexec); +char* APP_CC +g_get_strerror(void); +int APP_CC +g_execvp(const char* p1, char* args[]); +int APP_CC +g_execlp3(const char* a1, const char* a2, const char* a3); +void APP_CC +g_signal(int sig_num, void (*func)(int)); +void APP_CC +g_signal_child_stop(void (*func)(int)); +void APP_CC +g_unset_signals(void); +int APP_CC +g_fork(void); +int APP_CC +g_setgid(int pid); +int APP_CC +g_initgroups(const char* user, int gid); +int APP_CC +g_setuid(int pid); +int APP_CC +g_waitchild(void); +int APP_CC +g_waitpid(int pid); +void APP_CC +g_clearenv(void); +int APP_CC +g_setenv(const char* name, const char* value, int rewrite); +char* APP_CC +g_getenv(const char* name); +int APP_CC +g_exit(int exit_code); +int APP_CC +g_getpid(void); +int APP_CC +g_sigterm(int pid); +int APP_CC +g_getuser_info(const char* username, int* gid, int* uid, char* shell, + char* dir, char* gecos); +int APP_CC +g_getgroup_info(const char* groupname, int* gid); +int APP_CC +g_check_user_in_group(const char* username, int gid, int* ok); +int APP_CC +g_time1(void); + +#endif |