comparison: src/protocols/jabber/socket.c
src/protocols/jabber/socket.c
- changeset 2086
- 007508451e2c
- child 3127
- 4213ad5b231c
equal
deleted
inserted
replaced
| 2085:4659ae7bc69c | 2086:007508451e2c |
|---|---|
| 1 /* | |
| 2 * This program is free software; you can redistribute it and/or modify | |
| 3 * it under the terms of the GNU General Public License as published by | |
| 4 * the Free Software Foundation; either version 2 of the License, or | |
| 5 * (at your option) any later version. | |
| 6 * | |
| 7 * This program is distributed in the hope that it will be useful, | |
| 8 * but WITHOUT ANY WARRANTY; without even the implied warranty of | |
| 9 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the | |
| 10 * GNU General Public License for more details. | |
| 11 * | |
| 12 * You should have received a copy of the GNU General Public License | |
| 13 * along with this program; if not, write to the Free Software | |
| 14 * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA. | |
| 15 * | |
| 16 * Jabber | |
| 17 * Copyright (C) 1998-1999 The Jabber Team http://jabber.org/ | |
| 18 */ | |
| 19 | |
| 20 #include "libxode.h" | |
| 21 | |
| 22 /* socket.c | |
| 23 * | |
| 24 * Simple wrapper to make socket creation easy. | |
| 25 * type = NETSOCKET_SERVER is local listening socket | |
| 26 * type = NETSOCKET_CLIENT is connection socket | |
| 27 * type = NETSOCKET_UDP | |
| 28 */ | |
| 29 | |
| 30 int make_netsocket(u_short port, char *host, int type) | |
| 31 { | |
| 32 int s, flag = 1; | |
| 33 struct sockaddr_in sa; | |
| 34 struct in_addr *saddr; | |
| 35 int socket_type; | |
| 36 | |
| 37 /* is this a UDP socket or a TCP socket? */ | |
| 38 socket_type = (type == NETSOCKET_UDP)?SOCK_DGRAM:SOCK_STREAM; | |
| 39 | |
| 40 bzero((void *)&sa,sizeof(struct sockaddr_in)); | |
| 41 | |
| 42 if((s = socket(AF_INET,socket_type,0)) < 0) | |
| 43 return(-1); | |
| 44 if(setsockopt(s, SOL_SOCKET, SO_REUSEADDR, (char*)&flag, sizeof(flag)) < 0) | |
| 45 return(-1); | |
| 46 | |
| 47 saddr = make_addr(host); | |
| 48 if(saddr == NULL) | |
| 49 return(-1); | |
| 50 sa.sin_family = AF_INET; | |
| 51 sa.sin_port = htons(port); | |
| 52 | |
| 53 if(type == NETSOCKET_SERVER) | |
| 54 { | |
| 55 /* bind to specific address if specified */ | |
| 56 if(host != NULL) | |
| 57 sa.sin_addr.s_addr = saddr->s_addr; | |
| 58 | |
| 59 if(bind(s,(struct sockaddr*)&sa,sizeof sa) < 0) | |
| 60 { | |
| 61 close(s); | |
| 62 return(-1); | |
| 63 } | |
| 64 } | |
| 65 if(type == NETSOCKET_CLIENT) | |
| 66 { | |
| 67 sa.sin_addr.s_addr = saddr->s_addr; | |
| 68 if(connect(s,(struct sockaddr*)&sa,sizeof sa) < 0) | |
| 69 { | |
| 70 close(s); | |
| 71 return(-1); | |
| 72 } | |
| 73 } | |
| 74 if(type == NETSOCKET_UDP) | |
| 75 { | |
| 76 /* bind to all addresses for now */ | |
| 77 if(bind(s,(struct sockaddr*)&sa,sizeof sa) < 0) | |
| 78 { | |
| 79 close(s); | |
| 80 return(-1); | |
| 81 } | |
| 82 | |
| 83 /* specify default recipient for read/write */ | |
| 84 sa.sin_addr.s_addr = saddr->s_addr; | |
| 85 if(connect(s,(struct sockaddr*)&sa,sizeof sa) < 0) | |
| 86 { | |
| 87 close(s); | |
| 88 return(-1); | |
| 89 } | |
| 90 } | |
| 91 | |
| 92 | |
| 93 return(s); | |
| 94 } | |
| 95 | |
| 96 | |
| 97 struct in_addr *make_addr(char *host) | |
| 98 { | |
| 99 struct hostent *hp; | |
| 100 static struct in_addr addr; | |
| 101 char myname[MAXHOSTNAMELEN + 1]; | |
| 102 | |
| 103 if(host == NULL || strlen(host) == 0) | |
| 104 { | |
| 105 gethostname(myname,MAXHOSTNAMELEN); | |
| 106 hp = gethostbyname(myname); | |
| 107 if(hp != NULL) | |
| 108 { | |
| 109 return (struct in_addr *) *hp->h_addr_list; | |
| 110 } | |
| 111 }else{ | |
| 112 addr.s_addr = inet_addr(host); | |
| 113 if(addr.s_addr != -1) | |
| 114 { | |
| 115 return &addr; | |
| 116 } | |
| 117 hp = gethostbyname(host); | |
| 118 if(hp != NULL) | |
| 119 { | |
| 120 return (struct in_addr *) *hp->h_addr_list; | |
| 121 } | |
| 122 } | |
| 123 return NULL; | |
| 124 } | |
| 125 | |
| 126 /* Sets a file descriptor to close on exec. "flag" is 1 to close on exec, 0 to | |
| 127 * leave open across exec. | |
| 128 * -- EJB 7/31/2000 | |
| 129 */ | |
| 130 int set_fd_close_on_exec(int fd, int flag) | |
| 131 { | |
| 132 int oldflags = fcntl(fd,F_GETFL); | |
| 133 int newflags; | |
| 134 | |
| 135 if(flag) | |
| 136 newflags = oldflags | FD_CLOEXEC; | |
| 137 else | |
| 138 newflags = oldflags & (~FD_CLOEXEC); | |
| 139 | |
| 140 if(newflags==oldflags) | |
| 141 return 0; | |
| 142 return fcntl(fd,F_SETFL,(long)newflags); | |
| 143 } | |
| 144 |
