From bdd968b726362f55454853799ec7869cd14b9a4c Mon Sep 17 00:00:00 2001 From: oroborus Date: Thu, 31 Jul 2008 20:03:29 +0000 Subject: - priprava kodu na podporu UDP a TCP sucastne - upravy v kode, ktory spusta funkcie proto.c na zaklade prijatych packetov. ( na strane servera aj clienta ) git-svn-id: http://opensvn.csie.org/tuxanci_ng@185 0ee4d065-81f0-4d1e-b06c-ff20ad07cc3e --- src/net/tcp.c | 368 ++++++++++++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 368 insertions(+) create mode 100755 src/net/tcp.c (limited to 'src/net/tcp.c') diff --git a/src/net/tcp.c b/src/net/tcp.c new file mode 100755 index 0000000..2df17b9 --- /dev/null +++ b/src/net/tcp.c @@ -0,0 +1,368 @@ + +#include +#include +#include + +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include + +#include +#include +#include + +#include "tcp.h" + +sock_tcp_t* newSockTcp(int proto) +{ + sock_tcp_t *new; + + new = malloc( sizeof(sock_tcp_t) ); + memset(new, 0, sizeof(sock_tcp_t)); + new->proto = proto; + + return new; +} + +void destroySockTcp(sock_tcp_t *p) +{ + assert( p != NULL ); + free(p); +} + +sock_tcp_t* bindTcpSocket(char *address, int port, int proto) +{ + sock_tcp_t *new; + int len; + int ret; + + assert( port > 0 && port < 65535 ); + + new = newSockTcp(proto); + + assert( new != NULL ); + + + if( new->proto == PROTO_TCPv4 ) + { + new->sock = socket(AF_INET, SOCK_STREAM, 0); + } + +#ifdef SUPPORT_IPv6 + if( new->proto == PROTO_TCPv6 ) + { + new->sock = socket(AF_INET6, SOCK_STREAM, 0); + } +#endif + + if( new->sock < 0 ) + { + fprintf(stderr, "Nemozem vytvorit sietovy socket\n"); + destroySockTcp(new); + return NULL; + } + + if( new->proto == PROTO_TCPv4 ) + { + new->sockAddr.sin_family = AF_INET; + inet_pton(AF_INET, address, &(new->sockAddr.sin_addr)); + new->sockAddr.sin_port = htons(port); + + len = sizeof(new->sockAddr); + ret = bind(new->sock, (struct sockaddr *)&new->sockAddr, len); + } + +#ifdef SUPPORT_IPv6 + if( new->proto == PROTO_TCPv6 ) + { + new->sockAddr6.sin6_family = AF_INET6; + //new->sockAddr.sin_addr.s_addr = htonl(INADDR_ANY); + inet_pton(AF_INET6, address, &(new->sockAddr6.sin6_addr)); + new->sockAddr6.sin6_port = htons(port); + + len = sizeof(new->sockAddr6); + ret = bind(new->sock, (struct sockaddr *)&new->sockAddr6, len); + } +#endif + + if( ret < 0 ) + { + fprintf(stderr, "Nemozem obsadit sietovy port %d\n", port); + destroySockTcp(new); + return NULL; + } + + listen(new->sock, 5); + + return new; +} + +sock_tcp_t* getTcpNewClient(sock_tcp_t *p) +{ + sock_tcp_t *new; + int client_len; + + assert( p != NULL ); + assert( p->sock >= 0 ); + + new = newSockTcp(p->proto); + + if( new->proto == PROTO_TCPv4 ) + { + client_len = sizeof(new->sockAddr); + + new->sock = accept(p->sock, (struct sockaddr *)&new->sockAddr, + (socklen_t *)&client_len); + } + +#ifdef SUPPORT_IPv6 + if( new->proto == PROTO_TCPv6 ) + { + client_len = sizeof(new->sockAddr6); + + new->sock = accept(p->sock, (struct sockaddr *)&new->sockAddr6, + (socklen_t *)&client_len); + } +#endif + + if( new->sock < 0 ) + { + //printf("XXX\n"); + destroySockTcp(new); + return NULL; + } + + return new; +} + +void getSockTcpIp(sock_tcp_t *p, char *str_ip, int len) +{ + assert( p != NULL ); + assert( str_ip != NULL ); + + + if( p->proto == PROTO_TCPv4 ) + { + inet_ntop(AF_INET, &(p->sockAddr.sin_addr), str_ip, len); + strcpy(str_ip, inet_ntoa(p->sockAddr.sin_addr)); + //printf("strcpy(str_ip, inet_ntoa(p->sockAddr.sin_addr));\n"); + } + +#ifdef SUPPORT_IPv6 + if( p->proto == PROTO_TCPv6 ) + { + inet_ntop(AF_INET6, &(p->sockAddr6.sin6_addr), str_ip, len); + } +#endif +} + +int getSockTcpPort(sock_tcp_t *p) +{ + assert( p != NULL ); + + if( p->proto == PROTO_TCPv4 ) + { + return htons(p->sockAddr.sin_port); + } + +#ifdef SUPPORT_IPv6 + if( p->proto == PROTO_TCPv6 ) + { + return htons(p->sockAddr6.sin6_port); + } +#endif + + assert( ! "bad proto !" ); + + return -1; +} + +sock_tcp_t* connectTcpSocket(char *ip, int port, int proto) +{ + sock_tcp_t *new; + int len; + int ret; + + assert( ip != NULL ); + assert( port > 0 && port < 65535 ); + + new = newSockTcp(proto); + + if( new->proto == PROTO_TCPv4 ) + { + new->sock = socket(AF_INET, SOCK_STREAM, 0); + } + +#ifdef SUPPORT_IPv6 + if( new->proto == PROTO_TCPv6 ) + { + new->sock = socket(AF_INET6, SOCK_STREAM, 0); + } +#endif + + if( new->sock < 0 ) + { + fprintf(stderr, "Nemozem vytvorit sietovy socket\n"); + destroySockTcp(new); + return NULL; + } + + if( new->proto == PROTO_TCPv4 ) + { + new->sockAddr.sin_family = AF_INET; + new->sockAddr.sin_addr.s_addr = inet_addr(ip); + new->sockAddr.sin_port = htons(port); + + len = sizeof(new->sockAddr); + ret = connect(new->sock, (struct sockaddr *)&new->sockAddr, len); + } + +#ifdef SUPPORT_IPv6 + if( new->proto == PROTO_TCPv6 ) + { + new->sockAddr6.sin6_family = AF_INET6; + inet_pton(AF_INET6, ip, &(new->sockAddr6.sin6_addr)); + new->sockAddr6.sin6_port = htons(port); + + len = sizeof(new->sockAddr6); + ret = connect(new->sock, (struct sockaddr *)&new->sockAddr6, len); + } +#endif + + if( ret < 0 ) + { + fprintf(stderr, "Nemozem sa pripojit na %s %d\n", ip, port); + destroySockTcp(new); + return NULL; + } + + + return new; +} + +int disableNagle(sock_tcp_t *p) +{ + int flag = 1; + + int result; + + result = setsockopt(p->sock, /* socket affected */ + IPPROTO_TCP, /* set option at TCP level */ + TCP_NODELAY, /* name of option */ + (char *) &flag, /* the cast is historical cruft */ + sizeof(int)); /* length of option value */ + + if( result < 0 ) + { + printf("disabled nagle error\n"); + } + else + { + printf("disabled nagle OK\n"); + } + + return result; +} + +int readTcpSocket(sock_tcp_t *p, void *address, int len) +{ + assert( p != NULL ); + assert( address != NULL ); + + return read(p->sock, address, len); +} + +int writeTcpSocket(sock_tcp_t *p, void *address, int len) +{ + assert( p != NULL ); + assert( address != NULL ); + + return write(p->sock, address, len); +} + +void closeTcpSocket(sock_tcp_t *p) +{ + assert( p != NULL ); + + close(p->sock); + destroySockTcp(p); +} + +#if 0 +#define MAX_CLIENTS 32 +#define BUF_SIZE 512 + +int main(int argc, char **argv) +{ + sock_tcp_t *sock; + sock_tcp_t *client[MAX_CLIENTS]; + int count_client; + int changed, i; + struct pollfd *tmp; + + sock = bindTcpSocket( "127.0.0.1", 2200, PROTO_TCPv4 ); + disableNagle(sock); + count_client = 0; + + for(;;) + { + tmp = malloc( (count_client+1) * sizeof(struct pollfd) ); + + tmp[0].fd = sock->sock; + tmp[0].events = POLLIN; + tmp[0].revents = 0; + + for( i = 0 ; i < count_client ; i++ ) + { + tmp[1+i].fd = client[i]->sock; + tmp[1+i].events = POLLIN; + tmp[1+i].revents = 0; + } + + changed = poll(tmp, count_client+1, 1000); + + if( tmp[0].revents & POLLIN != 0 ) + { + client[count_client] = getTcpNewClient(sock); + disableNagle(client[count_client]); + count_client++; + } + + for( i = 0 ; i < count_client ; i++ ) + { + if( tmp[1+i].revents & POLLIN != 0 ) + { + char buf[BUF_SIZE]; + int ret; + + ret = read(client[i]->sock, buf, BUF_SIZE); + + if( ret <= 0 ) + { + closeTcpSocket(client[i]); + + memmove(client + i, + client + (i+1), + ((count_client-1) - i) * sizeof(sock_tcp_t *) ); + + count_client--; + continue; + } + + write(client[i]->sock, buf, ret); + } + } + + free(tmp); + } +} +#endif + -- cgit v1.2.3