From 593ac1e25c2053f06cb1b7e29785a695d3c57ad3 Mon Sep 17 00:00:00 2001 From: oroborus Date: Fri, 14 Mar 2008 19:31:47 +0000 Subject: - podpora UDP git-svn-id: http://opensvn.csie.org/tuxanci_ng@15 0ee4d065-81f0-4d1e-b06c-ff20ad07cc3e --- src/tcp.c | 20 +++++++++++++++++++- 1 file changed, 19 insertions(+), 1 deletion(-) (limited to 'src/tcp.c') diff --git a/src/tcp.c b/src/tcp.c index bcd7764..4558095 100755 --- a/src/tcp.c +++ b/src/tcp.c @@ -27,6 +27,7 @@ sock_tcp_t* newSockTcp() void destroySockTcp(sock_tcp_t *p) { + assert( p != NULL ); free(p); } @@ -39,6 +40,8 @@ sock_tcp_t* bindTcpSocket(int port) new = newSockTcp(); + assert( new != NULL ); + new->sock = socket(AF_INET, SOCK_STREAM, 0); if( new->sock < 0 ) @@ -71,7 +74,8 @@ sock_tcp_t* getTcpNewClient(sock_tcp_t *p) sock_tcp_t *new; int client_len; - assert( new->sock >= 0 ); + assert( p != NULL ); + assert( p->sock >= 0 ); new = newSockTcp(); @@ -85,6 +89,9 @@ sock_tcp_t* getTcpNewClient(sock_tcp_t *p) void getSockTcpIp(sock_tcp_t *p, char *str_ip) { + assert( p != NULL ); + assert( str_ip != NULL ); + strcpy(str_ip, inet_ntoa(p->sockAddr.sin_addr)); } @@ -93,6 +100,7 @@ sock_tcp_t* connectTcpSocket(char *ip, int port) sock_tcp_t *new; int len; + assert( ip != NULL ); assert( port > 0 && port < 65535 ); new = newSockTcp(); @@ -124,16 +132,24 @@ sock_tcp_t* connectTcpSocket(char *ip, int port) 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); } @@ -207,4 +223,6 @@ int test_tcp_main(int argc, char **argv) closeTcpSocket(sock); } + + return 0; } -- cgit v1.2.3