Made socket() accept an udp or tcp protocol number

diff --git a/lib/ip/socket.c b/lib/ip/socket.c
index 70a812e1efa204f84696aa8e4d8bd075bdfb0018..7900ce341e84b24f1e915e6f2349a88e5ef04baf 100644 (file)
--- a/lib/ip/socket.c
+++ b/lib/ip/socket.c
@@ -5,6 +5,7 @@
 #include <sys/socket.h>
 
 #include <net/netlib.h>
+#include <netinet/in.h>
 
 #define DEBUG 0
 
@@ -42,7 +43,7 @@ int socket(int domain, int type, int protocol)
 static int _tcp_socket(int protocol)
 {
        int fd;
-       if (protocol != 0)
+       if (protocol != 0 && protocol != IPPROTO_TCP)
        {
 #if DEBUG
                fprintf(stderr, "socket(tcp): bad protocol %d\n", protocol);
@@ -58,7 +59,7 @@ static int _udp_socket(int protocol)
 {
        int fd;
 
-       if (protocol != 0)
+       if (protocol != 0 && protocol != IPPROTO_UDP)
        {
 #if DEBUG
                fprintf(stderr, "socket(udp): bad protocol %d\n", protocol);