Annotation of embedaddon/quagga/babeld/net.c, revision 1.1
1.1 ! misho 1: /*
! 2: * This file is free software: you may copy, redistribute and/or modify it
! 3: * under the terms of the GNU General Public License as published by the
! 4: * Free Software Foundation, either version 2 of the License, or (at your
! 5: * option) any later version.
! 6: *
! 7: * This file is distributed in the hope that it will be useful, but
! 8: * WITHOUT ANY WARRANTY; without even the implied warranty of
! 9: * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
! 10: * 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, see <http://www.gnu.org/licenses/>.
! 14: *
! 15: * This file incorporates work covered by the following copyright and
! 16: * permission notice:
! 17: *
! 18: Copyright (c) 2007, 2008 by Juliusz Chroboczek
! 19:
! 20: Permission is hereby granted, free of charge, to any person obtaining a copy
! 21: of this software and associated documentation files (the "Software"), to deal
! 22: in the Software without restriction, including without limitation the rights
! 23: to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
! 24: copies of the Software, and to permit persons to whom the Software is
! 25: furnished to do so, subject to the following conditions:
! 26:
! 27: The above copyright notice and this permission notice shall be included in
! 28: all copies or substantial portions of the Software.
! 29:
! 30: THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
! 31: IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
! 32: FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
! 33: AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
! 34: LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
! 35: OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
! 36: THE SOFTWARE.
! 37: */
! 38:
! 39: #include <unistd.h>
! 40: #include <fcntl.h>
! 41: #include <string.h>
! 42: #include <sys/ioctl.h>
! 43: #include <sys/types.h>
! 44: #include <sys/uio.h>
! 45: #include <sys/socket.h>
! 46: #include <netinet/in.h>
! 47: #include <arpa/inet.h>
! 48: #include <errno.h>
! 49:
! 50: #include "babeld.h"
! 51: #include "util.h"
! 52: #include "net.h"
! 53:
! 54: int
! 55: babel_socket(int port)
! 56: {
! 57: struct sockaddr_in6 sin6;
! 58: int s, rc;
! 59: int saved_errno;
! 60: int one = 1, zero = 0;
! 61: const int ds = 0xc0; /* CS6 - Network Control */
! 62:
! 63: s = socket(PF_INET6, SOCK_DGRAM, 0);
! 64: if(s < 0)
! 65: return -1;
! 66:
! 67: rc = setsockopt(s, IPPROTO_IPV6, IPV6_V6ONLY, &one, sizeof(one));
! 68: if(rc < 0)
! 69: goto fail;
! 70:
! 71: rc = setsockopt(s, SOL_SOCKET, SO_REUSEADDR, &one, sizeof(one));
! 72: if(rc < 0)
! 73: goto fail;
! 74:
! 75: rc = setsockopt(s, IPPROTO_IPV6, IPV6_MULTICAST_LOOP,
! 76: &zero, sizeof(zero));
! 77: if(rc < 0)
! 78: goto fail;
! 79:
! 80: rc = setsockopt(s, IPPROTO_IPV6, IPV6_UNICAST_HOPS,
! 81: &one, sizeof(one));
! 82: if(rc < 0)
! 83: goto fail;
! 84:
! 85: rc = setsockopt(s, IPPROTO_IPV6, IPV6_MULTICAST_HOPS,
! 86: &one, sizeof(one));
! 87: if(rc < 0)
! 88: goto fail;
! 89:
! 90: #ifdef IPV6_TCLASS
! 91: rc = setsockopt(s, IPPROTO_IPV6, IPV6_TCLASS, &ds, sizeof(ds));
! 92: #else
! 93: rc = -1;
! 94: errno = ENOSYS;
! 95: #endif
! 96: if(rc < 0)
! 97: perror("Couldn't set traffic class");
! 98:
! 99: rc = fcntl(s, F_GETFL, 0);
! 100: if(rc < 0)
! 101: goto fail;
! 102:
! 103: rc = fcntl(s, F_SETFL, (rc | O_NONBLOCK));
! 104: if(rc < 0)
! 105: goto fail;
! 106:
! 107: rc = fcntl(s, F_GETFD, 0);
! 108: if(rc < 0)
! 109: goto fail;
! 110:
! 111: rc = fcntl(s, F_SETFD, rc | FD_CLOEXEC);
! 112: if(rc < 0)
! 113: goto fail;
! 114:
! 115: memset(&sin6, 0, sizeof(sin6));
! 116: sin6.sin6_family = AF_INET6;
! 117: sin6.sin6_port = htons(port);
! 118: rc = bind(s, (struct sockaddr*)&sin6, sizeof(sin6));
! 119: if(rc < 0)
! 120: goto fail;
! 121:
! 122: return s;
! 123:
! 124: fail:
! 125: saved_errno = errno;
! 126: close(s);
! 127: errno = saved_errno;
! 128: return -1;
! 129: }
! 130:
! 131: int
! 132: babel_recv(int s, void *buf, int buflen, struct sockaddr *sin, int slen)
! 133: {
! 134: struct iovec iovec;
! 135: struct msghdr msg;
! 136: int rc;
! 137:
! 138: memset(&msg, 0, sizeof(msg));
! 139: iovec.iov_base = buf;
! 140: iovec.iov_len = buflen;
! 141: msg.msg_name = sin;
! 142: msg.msg_namelen = slen;
! 143: msg.msg_iov = &iovec;
! 144: msg.msg_iovlen = 1;
! 145:
! 146: rc = recvmsg(s, &msg, 0);
! 147: return rc;
! 148: }
! 149:
! 150: int
! 151: babel_send(int s,
! 152: void *buf1, int buflen1, void *buf2, int buflen2,
! 153: struct sockaddr *sin, int slen)
! 154: {
! 155: struct iovec iovec[2];
! 156: struct msghdr msg;
! 157: int rc;
! 158:
! 159: iovec[0].iov_base = buf1;
! 160: iovec[0].iov_len = buflen1;
! 161: iovec[1].iov_base = buf2;
! 162: iovec[1].iov_len = buflen2;
! 163: memset(&msg, 0, sizeof(msg));
! 164: msg.msg_name = (struct sockaddr*)sin;
! 165: msg.msg_namelen = slen;
! 166: msg.msg_iov = iovec;
! 167: msg.msg_iovlen = 2;
! 168:
! 169: again:
! 170: rc = sendmsg(s, &msg, 0);
! 171: if(rc < 0) {
! 172: if(errno == EINTR)
! 173: goto again;
! 174: else if(errno == EAGAIN) {
! 175: int rc2;
! 176: rc2 = wait_for_fd(1, s, 5);
! 177: if(rc2 > 0)
! 178: goto again;
! 179: errno = EAGAIN;
! 180: }
! 181: }
! 182: return rc;
! 183: }
! 184:
! 185: int
! 186: tcp_server_socket(int port, int local)
! 187: {
! 188: struct sockaddr_in6 sin6;
! 189: int s, rc, saved_errno;
! 190: int one = 1;
! 191:
! 192: s = socket(PF_INET6, SOCK_STREAM, 0);
! 193: if(s < 0)
! 194: return -1;
! 195:
! 196: rc = setsockopt(s, SOL_SOCKET, SO_REUSEADDR, &one, sizeof(one));
! 197: if(rc < 0)
! 198: goto fail;
! 199:
! 200: rc = fcntl(s, F_GETFL, 0);
! 201: if(rc < 0)
! 202: goto fail;
! 203:
! 204: rc = fcntl(s, F_SETFL, (rc | O_NONBLOCK));
! 205: if(rc < 0)
! 206: goto fail;
! 207:
! 208: rc = fcntl(s, F_GETFD, 0);
! 209: if(rc < 0)
! 210: goto fail;
! 211:
! 212: rc = fcntl(s, F_SETFD, rc | FD_CLOEXEC);
! 213: if(rc < 0)
! 214: goto fail;
! 215:
! 216: memset(&sin6, 0, sizeof(sin6));
! 217: sin6.sin6_family = AF_INET6;
! 218: sin6.sin6_port = htons(port);
! 219: if(local) {
! 220: rc = inet_pton(AF_INET6, "::1", &sin6.sin6_addr);
! 221: if(rc < 0)
! 222: goto fail;
! 223: }
! 224: rc = bind(s, (struct sockaddr*)&sin6, sizeof(sin6));
! 225: if(rc < 0)
! 226: goto fail;
! 227:
! 228: rc = listen(s, 2);
! 229: if(rc < 0)
! 230: goto fail;
! 231:
! 232: return s;
! 233:
! 234: fail:
! 235: saved_errno = errno;
! 236: close(s);
! 237: errno = saved_errno;
! 238: return -1;
! 239: }
FreeBSD-CVSweb <freebsd-cvsweb@FreeBSD.org>