Remove astobj.h from some places where it wasn't needed
[asterisk/asterisk.git] / main / netsock.c
1 /*
2  * Asterisk -- An open source telephony toolkit.
3  *
4  * Copyright (C) 1999 - 2005, Digium, Inc.
5  *
6  * Kevin P. Fleming <kpfleming@digium.com>
7  * Mark Spencer <markster@digium.com>
8  *
9  * See http://www.asterisk.org for more information about
10  * the Asterisk project. Please do not directly contact
11  * any of the maintainers of this project for assistance;
12  * the project provides a web site, mailing lists and IRC
13  * channels for your use.
14  *
15  * This program is free software, distributed under the terms of
16  * the GNU General Public License Version 2. See the LICENSE file
17  * at the top of the source tree.
18  */
19
20 /*! \file
21  *
22  * \brief Network socket handling
23  *
24  * \author Kevin P. Fleming <kpfleming@digium.com>
25  * \author Mark Spencer <markster@digium.com>
26  */
27
28 #include "asterisk.h"
29
30 ASTERISK_FILE_VERSION(__FILE__, "$Revision$")
31
32 #if defined (SOLARIS)
33 #include <sys/sockio.h>
34 #endif
35
36 #include "asterisk/netsock.h"
37 #include "asterisk/utils.h"
38 #include "asterisk/astobj.h"
39
40 struct ast_netsock {
41         ASTOBJ_COMPONENTS(struct ast_netsock);
42         struct sockaddr_in bindaddr;
43         int sockfd;
44         int *ioref;
45         struct io_context *ioc;
46         void *data;
47 };
48
49 struct ast_netsock_list {
50         ASTOBJ_CONTAINER_COMPONENTS(struct ast_netsock);
51         struct io_context *ioc;
52 };
53
54 static void ast_netsock_destroy(struct ast_netsock *netsock)
55 {
56         ast_io_remove(netsock->ioc, netsock->ioref);
57         close(netsock->sockfd);
58         ast_free(netsock);
59 }
60
61 struct ast_netsock_list *ast_netsock_list_alloc(void)
62 {
63         return ast_calloc(1, sizeof(struct ast_netsock_list));
64 }
65
66 int ast_netsock_init(struct ast_netsock_list *list)
67 {
68         memset(list, 0, sizeof(*list));
69         ASTOBJ_CONTAINER_INIT(list);
70
71         return 0;
72 }
73
74 int ast_netsock_release(struct ast_netsock_list *list)
75 {
76         ASTOBJ_CONTAINER_DESTROYALL(list, ast_netsock_destroy);
77         ASTOBJ_CONTAINER_DESTROY(list);
78
79         return 0;
80 }
81
82 struct ast_netsock *ast_netsock_find(struct ast_netsock_list *list,
83                                      struct sockaddr_in *sa)
84 {
85         struct ast_netsock *sock = NULL;
86
87         ASTOBJ_CONTAINER_TRAVERSE(list, !sock, {
88                 ASTOBJ_RDLOCK(iterator);
89                 if (!inaddrcmp(&iterator->bindaddr, sa))
90                         sock = iterator;
91                 ASTOBJ_UNLOCK(iterator);
92         });
93
94         return sock;
95 }
96
97 struct ast_netsock *ast_netsock_bindaddr(struct ast_netsock_list *list, struct io_context *ioc, struct sockaddr_in *bindaddr, int tos, int cos, ast_io_cb callback, void *data)
98 {
99         int netsocket = -1;
100         int *ioref;
101         
102         struct ast_netsock *ns;
103         const int reuseFlag = 1;
104         
105         /* Make a UDP socket */
106         netsocket = socket(AF_INET, SOCK_DGRAM, IPPROTO_IP);
107         
108         if (netsocket < 0) {
109                 ast_log(LOG_ERROR, "Unable to create network socket: %s\n", strerror(errno));
110                 return NULL;
111         }
112         if (setsockopt(netsocket, SOL_SOCKET, SO_REUSEADDR, (char *)&reuseFlag, sizeof reuseFlag) < 0) {
113                         ast_log(LOG_WARNING, "Error setting SO_REUSEADDR on sockfd '%d'\n", netsocket);
114         }
115         if (bind(netsocket,(struct sockaddr *)bindaddr, sizeof(struct sockaddr_in))) {
116                 ast_log(LOG_ERROR, "Unable to bind to %s port %d: %s\n", ast_inet_ntoa(bindaddr->sin_addr), ntohs(bindaddr->sin_port), strerror(errno));
117                 close(netsocket);
118                 return NULL;
119         }
120
121         ast_netsock_set_qos(netsocket, tos, cos, "IAX2");
122                 
123         ast_enable_packet_fragmentation(netsocket);
124
125         if (!(ns = ast_calloc(1, sizeof(*ns)))) {
126                 close(netsocket);
127                 return NULL;
128         }
129         
130         /* Establish I/O callback for socket read */
131         if (!(ioref = ast_io_add(ioc, netsocket, callback, AST_IO_IN, ns))) {
132                 close(netsocket);
133                 ast_free(ns);
134                 return NULL;
135         }       
136         ASTOBJ_INIT(ns);
137         ns->ioref = ioref;
138         ns->ioc = ioc;
139         ns->sockfd = netsocket;
140         ns->data = data;
141         memcpy(&ns->bindaddr, bindaddr, sizeof(ns->bindaddr));
142         ASTOBJ_CONTAINER_LINK(list, ns);
143
144         return ns;
145 }
146
147 int ast_netsock_set_qos(int netsocket, int tos, int cos, const char *desc)
148 {
149         int res;
150         
151         if ((res = setsockopt(netsocket, IPPROTO_IP, IP_TOS, &tos, sizeof(tos))))
152                 ast_log(LOG_WARNING, "Unable to set %s TOS to %d, may be you have no root privileges\n", desc, tos);
153         else if (tos)
154                 ast_verb(2, "Using %s TOS bits %d\n", desc, tos);
155
156 #if defined(linux)                                                              
157         if (setsockopt(netsocket, SOL_SOCKET, SO_PRIORITY, &cos, sizeof(cos)))
158                 ast_log(LOG_WARNING, "Unable to set %s CoS to %d\n", desc, cos);
159         else if (cos)
160                 ast_verb(2, "Using %s CoS mark %d\n", desc, cos);
161 #endif
162                                                         
163         return res;
164 }
165                                                                                                         
166
167 struct ast_netsock *ast_netsock_bind(struct ast_netsock_list *list, struct io_context *ioc, const char *bindinfo, int defaultport, int tos, int cos, ast_io_cb callback, void *data)
168 {
169         struct sockaddr_in sin;
170         char *tmp;
171         char *host;
172         char *port;
173         int portno;
174
175         memset(&sin, 0, sizeof(sin));
176         sin.sin_family = AF_INET;
177         sin.sin_port = htons(defaultport);
178         tmp = ast_strdupa(bindinfo);
179
180         host = strsep(&tmp, ":");
181         port = tmp;
182
183         if (port && ((portno = atoi(port)) > 0))
184                 sin.sin_port = htons(portno);
185
186         inet_aton(host, &sin.sin_addr);
187
188         return ast_netsock_bindaddr(list, ioc, &sin, tos, cos, callback, data);
189 }
190
191 int ast_netsock_sockfd(const struct ast_netsock *ns)
192 {
193         return ns ? ns-> sockfd : -1;
194 }
195
196 const struct sockaddr_in *ast_netsock_boundaddr(const struct ast_netsock *ns)
197 {
198         return &(ns->bindaddr);
199 }
200
201 void *ast_netsock_data(const struct ast_netsock *ns)
202 {
203         return ns->data;
204 }
205
206 void ast_netsock_unref(struct ast_netsock *ns)
207 {
208         ASTOBJ_UNREF(ns, ast_netsock_destroy);
209 }