Fixed recv function in the driver to process more than one packet per interrupt.
[akaros.git] / kern / include / socket.h
1 #ifndef ROS_SOCKET_H
2 #define ROS_SOCKET_H
3
4 #include <ros/common.h>
5 #include <sys/queue.h>
6 #include <atomic.h>
7 #include <net/pbuf.h>
8 #include <kthread.h>
9 // Just a couple of AF types that we might support
10 #define AF_UNSPEC       0
11 #define AF_UNIX         1       /* Unix domain sockets          */
12 #define AF_LOCAL        1       /* POSIX name for AF_UNIX       */
13 #define AF_INET         2       /* Internet IP Protocol         */
14
15 #define PF_UNSPEC       AF_UNSPEC
16 #define PF_UNIX         AF_UNIX
17 #define PF_LOCAL        AF_LOCAL
18 #define PF_INET         AF_INET
19
20 #define SS_NOFDREF              0x0001  /* no file table ref any more */
21 #define SS_ISCONNECTED          0x0002  /* socket connected to a peer */
22 #define SS_ISCONNECTING         0x0004  /* in process of connecting to peer */
23 #define SS_ISDISCONNECTING      0x0008  /* in process of disconnecting */
24 #define SS_NBIO                 0x0100  /* non-blocking ops */
25 #define SS_ASYNC                0x0200  /* async i/o notify */
26 #define SS_ISCONFIRMING         0x0400  /* deciding to accept connection req */
27 #define SS_ISDISCONNECTED       0x2000  /* socket disconnected from peer */
28
29 /* Define an range for automatic port assignment */
30 #define SOCKET_PORT_START 4096
31 #define SOCKET_PORT_END  0x7fff
32
33 struct socket;
34 struct proc;
35 // These are probably defined elsewhere too..
36 #ifndef socklen_t
37 typedef int socklen_t;
38 typedef int sa_family_t;
39 #endif
40 #define inet_addr_to_ipaddr_p(target_ipaddr_p, source_inaddr)   ((target_ipaddr_p) = (ip_addr_t*)&((source_inaddr)->s_addr))
41 enum sock_type {
42     SOCK_STREAM = 1,
43     SOCK_DGRAM  = 2,
44     SOCK_RAW    = 3,
45     SOCK_RDM    = 4,
46     SOCK_SEQPACKET  = 5,
47     SOCK_DCCP   = 6,
48     SOCK_PACKET = 10,
49 };
50 struct socket{
51   //int so_count;       /* (b) reference count */
52   short   so_type;        /* (a) generic type, see socket.h */
53         short   so_family;
54         int     so_protocol;
55   short   so_options;     /* from socket call, see socket.h */
56   //short   so_linger;      /* time to linger while closing */
57   short   so_state;       /* (b) internal state flags SS_* */
58         //int so_qstate;      /* (e) internal state flags SQ_* */
59         void    *so_pcb;        /* protocol control block */
60         struct pbuf_head recv_buff;
61         struct pbuf_head send_buff;
62         struct semaphore sem;   /* semaphone to for a process to sleep on */
63         
64         //struct  vnet *so_vnet;      /* network stack instance */
65         //struct  protosw *so_proto;  /* (a) protocol handle */
66 };
67
68 struct in_addr {
69     uint32_t s_addr;
70 };
71
72 /* members are in network byte order */
73 struct sockaddr_in {
74     // uint8_t sin_len; -- bsd only field
75     uint8_t sin_family;
76     uint16_t sin_port;
77     struct in_addr sin_addr;
78     char sin_zero[8];
79 };
80
81
82 struct sockaddr {
83         unsigned char   sa_len;         /* bsd only total length */
84         sa_family_t     sa_family;      /* address family */
85         char            sa_data[14];    /* actually longer; address value */
86 };
87
88 /*
89  * Message header for recvmsg and sendmsg calls.
90  * Used value-result for recvmsg, value only for sendmsg.
91  */ 
92 struct msghdr {
93     void        *msg_name;      /* optional address */
94     socklen_t    msg_namelen;       /* size of address */
95     struct iovec    *msg_iov;       /* scatter/gather array */
96     int      msg_iovlen;        /* # elements in msg_iov */
97     void        *msg_control;       /* ancillary data, see below */
98     socklen_t    msg_controllen;    /* ancillary data buffer len */
99     int      msg_flags;     /* flags on received message */
100 };
101
102
103
104 extern struct kmem_cache *sock_kcache;
105 extern struct kmem_cache *mbuf_kcache;
106 extern struct kmem_cache *udp_pcb_kcache;
107
108 void socket_init();
109 intreg_t send_iov(struct socket* sock, struct iovec* iov, int flags);
110 int send_datagram(struct socket* sock, struct iovec* iov, int flags);
111
112 intreg_t sys_socket(struct proc *p, int socket_family, int socket_type, int protocol);
113 intreg_t sys_sendto(struct proc *p, int socket, const void *buffer, size_t length, int flags, const struct sockaddr *dest_addr, socklen_t dest_len);
114 intreg_t sys_recvfrom(struct proc *p, int socket, void *restrict buffer, size_t length, int flags, struct sockaddr *restrict address, socklen_t *restrict address_len);
115
116
117 #endif /* ROS_SOCKET_H */
118