Return to receivedata.c CVS log

Up to [ELWIX - Embedded LightWeight unIX -] / embedaddon / miniupnpc / src

Annotation of embedaddon/miniupnpc/src/receivedata.c, revision 1.1

1.1     ! misho       1: /* $Id: receivedata.c,v 1.10 2021/03/02 23:33:07 nanard Exp $ */
        !             2: /* Project : miniupnp
        !             3:  * Website : http://miniupnp.free.fr/
        !             4:  * Author : Thomas Bernard
        !             5:  * Copyright (c) 2011-2021 Thomas Bernard
        !             6:  * This software is subject to the conditions detailed in the
        !             7:  * LICENCE file provided in this distribution. */
        !             8: 
        !             9: #include <stdio.h>
        !            10: #include <string.h>
        !            11: #ifdef _WIN32
        !            12: #include <winsock2.h>
        !            13: #include <ws2tcpip.h>
        !            14: #else /* _WIN32 */
        !            15: #include <unistd.h>
        !            16: #if defined(__amigaos__) && !defined(__amigaos4__)
        !            17: #define socklen_t int
        !            18: #else /* #if defined(__amigaos__) && !defined(__amigaos4__) */
        !            19: #include <sys/select.h>
        !            20: #endif /* #else defined(__amigaos__) && !defined(__amigaos4__) */
        !            21: #include <sys/socket.h>
        !            22: #include <netinet/in.h>
        !            23: #if !defined(__amigaos__) && !defined(__amigaos4__)
        !            24: #include <poll.h>
        !            25: #endif /* !defined(__amigaos__) && !defined(__amigaos4__) */
        !            26: #include <errno.h>
        !            27: #define MINIUPNPC_IGNORE_EINTR
        !            28: #endif /* _WIN32 */
        !            29: 
        !            30: #include "receivedata.h"
        !            31: 
        !            32: int
        !            33: receivedata(SOCKET socket,
        !            34:             char * data, int length,
        !            35:             int timeout, unsigned int * scope_id)
        !            36: {
        !            37: #ifdef MINIUPNPC_GET_SRC_ADDR
        !            38:        struct sockaddr_storage src_addr;
        !            39:        socklen_t src_addr_len = sizeof(src_addr);
        !            40: #endif /* MINIUPNPC_GET_SRC_ADDR */
        !            41:     int n;
        !            42: #if !defined(_WIN32) && !defined(__amigaos__) && !defined(__amigaos4__)
        !            43:        /* using poll */
        !            44:     struct pollfd fds[1]; /* for the poll */
        !            45: #ifdef MINIUPNPC_IGNORE_EINTR
        !            46:     do {
        !            47: #endif /* MINIUPNPC_IGNORE_EINTR */
        !            48:         fds[0].fd = socket;
        !            49:         fds[0].events = POLLIN;
        !            50:         n = poll(fds, 1, timeout);
        !            51: #ifdef MINIUPNPC_IGNORE_EINTR
        !            52:     } while(n < 0 && errno == EINTR);
        !            53: #endif /* MINIUPNPC_IGNORE_EINTR */
        !            54:     if(n < 0) {
        !            55:         PRINT_SOCKET_ERROR("poll");
        !            56:         return -1;
        !            57:     } else if(n == 0) {
        !            58:                /* timeout */
        !            59:         return 0;
        !            60:     }
        !            61: #else  /* !defined(_WIN32) && !defined(__amigaos__) && !defined(__amigaos4__) */
        !            62:        /* using select under _WIN32 and amigaos */
        !            63:     fd_set socketSet;
        !            64:     TIMEVAL timeval;
        !            65:     FD_ZERO(&socketSet);
        !            66:     FD_SET(socket, &socketSet);
        !            67:     timeval.tv_sec = timeout / 1000;
        !            68:     timeval.tv_usec = (timeout % 1000) * 1000;
        !            69:     n = select(FD_SETSIZE, &socketSet, NULL, NULL, &timeval);
        !            70:     if(n < 0) {
        !            71:         PRINT_SOCKET_ERROR("select");
        !            72:         return -1;
        !            73:     } else if(n == 0) {
        !            74:         return 0;
        !            75:     }
        !            76: #endif /* !defined(_WIN32) && !defined(__amigaos__) && !defined(__amigaos4__) */
        !            77: #ifdef MINIUPNPC_GET_SRC_ADDR
        !            78:        memset(&src_addr, 0, sizeof(src_addr));
        !            79:        n = recvfrom(socket, data, length, 0,
        !            80:                     (struct sockaddr *)&src_addr, &src_addr_len);
        !            81: #else  /* MINIUPNPC_GET_SRC_ADDR */
        !            82:        n = recv(socket, data, length, 0);
        !            83: #endif /* MINIUPNPC_GET_SRC_ADDR */
        !            84:        if(n<0) {
        !            85:                PRINT_SOCKET_ERROR("recv");
        !            86:        }
        !            87: #ifdef MINIUPNPC_GET_SRC_ADDR
        !            88:        if (src_addr.ss_family == AF_INET6) {
        !            89:                const struct sockaddr_in6 * src_addr6 = (struct sockaddr_in6 *)&src_addr;
        !            90: #ifdef DEBUG
        !            91:                printf("scope_id=%u\n", src_addr6->sin6_scope_id);
        !            92: #endif /* DEBUG */
        !            93:                if(scope_id)
        !            94:                        *scope_id = src_addr6->sin6_scope_id;
        !            95:        } else {
        !            96:                if(scope_id)
        !            97:                        *scope_id = 0;
        !            98:        }
        !            99: #else  /* MINIUPNPC_GET_SRC_ADDR */
        !           100:        if(scope_id)
        !           101:                *scope_id = 0;
        !           102: #endif /* MINIUPNPC_GET_SRC_ADDR */
        !           103:        return n;
        !           104: }
        !           105: