00001
00002
00003
00004
00005
00006
00007
00008
00009
00010
00011
00012
00013
00014
00015
00016
00017
00018
00019
00020
00021
00022
00023
00024
00025
00026
00027
00028 #include "asterisk.h"
00029
00030 ASTERISK_FILE_VERSION(__FILE__, "$Revision$")
00031
00032 #include <stdio.h>
00033 #include <stdlib.h>
00034 #include <string.h>
00035 #include <sys/time.h>
00036 #include <signal.h>
00037 #include <errno.h>
00038 #include <unistd.h>
00039 #include <netinet/in.h>
00040 #include <arpa/inet.h>
00041 #include <sys/socket.h>
00042 #include <netdb.h>
00043 #include <net/if.h>
00044 #include <netinet/in_systm.h>
00045 #include <netinet/ip.h>
00046 #include <sys/ioctl.h>
00047
00048 #if defined(__OpenBSD__) || defined(__NetBSD__) || defined(__FreeBSD__)
00049 #include <fcntl.h>
00050 #include <net/route.h>
00051 #endif
00052
00053 #if defined (SOLARIS)
00054 #include <sys/sockio.h>
00055 #endif
00056
00057 #include "asterisk/netsock.h"
00058 #include "asterisk/logger.h"
00059 #include "asterisk/channel.h"
00060 #include "asterisk/options.h"
00061 #include "asterisk/utils.h"
00062 #include "asterisk/lock.h"
00063 #include "asterisk/srv.h"
00064
00065 struct ast_netsock {
00066 ASTOBJ_COMPONENTS(struct ast_netsock);
00067 struct sockaddr_in bindaddr;
00068 int sockfd;
00069 int *ioref;
00070 struct io_context *ioc;
00071 void *data;
00072 };
00073
00074 struct ast_netsock_list {
00075 ASTOBJ_CONTAINER_COMPONENTS(struct ast_netsock);
00076 struct io_context *ioc;
00077 };
00078
00079 static void ast_netsock_destroy(struct ast_netsock *netsock)
00080 {
00081 ast_io_remove(netsock->ioc, netsock->ioref);
00082 close(netsock->sockfd);
00083 free(netsock);
00084 }
00085
00086 struct ast_netsock_list *ast_netsock_list_alloc(void)
00087 {
00088 return ast_calloc(1, sizeof(struct ast_netsock_list));
00089 }
00090
00091 int ast_netsock_init(struct ast_netsock_list *list)
00092 {
00093 memset(list, 0, sizeof(*list));
00094 ASTOBJ_CONTAINER_INIT(list);
00095
00096 return 0;
00097 }
00098
00099 int ast_netsock_release(struct ast_netsock_list *list)
00100 {
00101 ASTOBJ_CONTAINER_DESTROYALL(list, ast_netsock_destroy);
00102 ASTOBJ_CONTAINER_DESTROY(list);
00103
00104 return 0;
00105 }
00106
00107 struct ast_netsock *ast_netsock_find(struct ast_netsock_list *list,
00108 struct sockaddr_in *sa)
00109 {
00110 struct ast_netsock *sock = NULL;
00111
00112 ASTOBJ_CONTAINER_TRAVERSE(list, !sock, {
00113 ASTOBJ_RDLOCK(iterator);
00114 if (!inaddrcmp(&iterator->bindaddr, sa))
00115 sock = iterator;
00116 ASTOBJ_UNLOCK(iterator);
00117 });
00118
00119 return sock;
00120 }
00121
00122 struct ast_netsock *ast_netsock_bindaddr(struct ast_netsock_list *list, struct io_context *ioc, struct sockaddr_in *bindaddr, int tos, ast_io_cb callback, void *data)
00123 {
00124 int netsocket = -1;
00125 int *ioref;
00126
00127 struct ast_netsock *ns;
00128 const int reuseFlag = 1;
00129
00130
00131 netsocket = socket(AF_INET, SOCK_DGRAM, IPPROTO_IP);
00132
00133 if (netsocket < 0) {
00134 ast_log(LOG_ERROR, "Unable to create network socket: %s\n", strerror(errno));
00135 return NULL;
00136 }
00137 if (setsockopt(netsocket, SOL_SOCKET, SO_REUSEADDR, (char *)&reuseFlag, sizeof reuseFlag) < 0) {
00138 ast_log(LOG_WARNING, "Error setting SO_REUSEADDR on sockfd '%d'\n", netsocket);
00139 }
00140 if (bind(netsocket,(struct sockaddr *)bindaddr, sizeof(struct sockaddr_in))) {
00141 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));
00142 close(netsocket);
00143 return NULL;
00144 }
00145 if (option_verbose > 1)
00146 ast_verbose(VERBOSE_PREFIX_2 "Using TOS bits %d\n", tos);
00147
00148 if (setsockopt(netsocket, IPPROTO_IP, IP_TOS, &tos, sizeof(tos)))
00149 ast_log(LOG_WARNING, "Unable to set TOS to %d\n", tos);
00150
00151 ast_enable_packet_fragmentation(netsocket);
00152
00153 if (!(ns = ast_calloc(1, sizeof(struct ast_netsock)))) {
00154 close(netsocket);
00155 return NULL;
00156 }
00157
00158
00159 if (!(ioref = ast_io_add(ioc, netsocket, callback, AST_IO_IN, ns))) {
00160 close(netsocket);
00161 free(ns);
00162 return NULL;
00163 }
00164 ASTOBJ_INIT(ns);
00165 ns->ioref = ioref;
00166 ns->ioc = ioc;
00167 ns->sockfd = netsocket;
00168 ns->data = data;
00169 memcpy(&ns->bindaddr, bindaddr, sizeof(ns->bindaddr));
00170 ASTOBJ_CONTAINER_LINK(list, ns);
00171
00172 return ns;
00173 }
00174
00175 struct ast_netsock *ast_netsock_bind(struct ast_netsock_list *list, struct io_context *ioc, const char *bindinfo, int defaultport, int tos, ast_io_cb callback, void *data)
00176 {
00177 struct sockaddr_in sin;
00178 char *tmp;
00179 char *host;
00180 char *port;
00181 int portno;
00182
00183 memset(&sin, 0, sizeof(sin));
00184 sin.sin_family = AF_INET;
00185 sin.sin_port = htons(defaultport);
00186 tmp = ast_strdupa(bindinfo);
00187
00188 host = strsep(&tmp, ":");
00189 port = tmp;
00190
00191 if (port && ((portno = atoi(port)) > 0))
00192 sin.sin_port = htons(portno);
00193
00194 inet_aton(host, &sin.sin_addr);
00195
00196 return ast_netsock_bindaddr(list, ioc, &sin, tos, callback, data);
00197 }
00198
00199 int ast_netsock_sockfd(const struct ast_netsock *ns)
00200 {
00201 return ns ? ns-> sockfd : -1;
00202 }
00203
00204 const struct sockaddr_in *ast_netsock_boundaddr(const struct ast_netsock *ns)
00205 {
00206 return &(ns->bindaddr);
00207 }
00208
00209 void *ast_netsock_data(const struct ast_netsock *ns)
00210 {
00211 return ns->data;
00212 }
00213
00214 void ast_netsock_unref(struct ast_netsock *ns)
00215 {
00216 ASTOBJ_UNREF(ns, ast_netsock_destroy);
00217 }