* extract common functions as library.

git-svn-id: http://opensvn.csie.org/pttbbs/trunk/pttbbs@3673 63ad8ddf-47c3-0310-b6dd-a9e9d9715204

1 files changed, 114 insertions, 0 deletions

diff --git a/src/libbbsutil/net.c b/src/libbbsutil/net.c
new file mode 100644
index 00000000..60208a65
--- /dev/null
+++ b/src/libbbsutil/net.c
@@ -0,0 +1,114 @@
+#include <stdio.h>
+#include <string.h>
+#include <stdlib.h>
+#include <sys/types.h>
+#include <sys/socket.h>
+#include <netinet/in.h>
+#include <arpa/inet.h>
+#include <unistd.h>
+
+#include "libbbsutil.h"
+
+unsigned int
+ipstr2int(const char *ip)
+{
+    unsigned int i, val = 0;
+    char buf[32];
+    char *nil, *p;
+
+    strlcpy(buf, ip, sizeof(buf));
+    p = buf;
+    for (i = 0; i < 4; i++) {
+	nil = strchr(p, '.');
+	if (nil != NULL)
+	    *nil = 0;
+	val *= 256;
+	val += atoi(p);
+	if (nil != NULL)
+	    p = nil + 1;
+    }
+    return val;
+}
+
+int tobind(const char * host, int port)
+{
+    int     sockfd, val = 1;
+    struct  sockaddr_in     servaddr;
+
+    if( (sockfd = socket(AF_INET, SOCK_STREAM, 0)) < 0 ) {
+	perror("socket()");
+	exit(1);
+    }
+    setsockopt(sockfd, SOL_SOCKET, SO_REUSEADDR,
+	       (char *)&val, sizeof(val));
+    bzero(&servaddr, sizeof(servaddr));
+    servaddr.sin_family = AF_INET;
+    if (host == NULL || host[0] == 0)
+	servaddr.sin_addr.s_addr = htonl(INADDR_ANY);
+    else if (inet_aton(host, &servaddr.sin_addr) == 0) {
+	perror("inet_aton()");
+	exit(1);
+    }
+    servaddr.sin_port = htons(port);
+    if( bind(sockfd, (struct sockaddr *)&servaddr, sizeof(servaddr)) < 0 ) {
+	perror("bind()");
+	exit(1);
+    }
+    if( listen(sockfd, 5) < 0 ) {
+	perror("listen()");
+	exit(1);
+    }
+
+    return sockfd;
+}
+
+int toconnect(const char *host, int port)
+{
+    int    sock;
+    struct sockaddr_in serv_name;
+    if( (sock = socket(AF_INET, SOCK_STREAM, 0)) < 0 ){
+        perror("socket");
+        return -1;
+    }
+
+    serv_name.sin_family = AF_INET;
+    serv_name.sin_addr.s_addr = inet_addr(host);
+    serv_name.sin_port = htons(port);
+    if( connect(sock, (struct sockaddr*)&serv_name, sizeof(serv_name)) < 0 ){
+        close(sock);
+        return -1;
+    }
+    return sock;
+}
+
+/**
+ * same as read(2), but read until exactly size len 
+ */
+int toread(int fd, void *buf, int len)
+{
+    int     l;
+    for( l = 0 ; len > 0 ; )
+	if( (l = read(fd, buf, len)) <= 0 )
+	    return -1;
+	else{
+	    buf += l;
+	    len -= l;
+	}
+    return l;
+}
+
+/**
+ * same as write(2), but write until exactly size len 
+ */
+int towrite(int fd, const void *buf, int len)
+{
+    int     l;
+    for( l = 0 ; len > 0 ; )
+	if( (l = write(fd, buf, len)) <= 0 )
+	    return -1;
+	else{
+	    buf += l;
+	    len -= l;
+	}
+    return l;
+}