0

我使用的是 Linux 系统,而不是 Windows 系统。我在下面发布了一些代码。请记住,此代码从未打算成为“生产质量”。

#include <unistd.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <stdarg.h>
#include <netdb.h>
#include <errno.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>

#define PORT 9909

void die ( const char *fmt, ... )
{
    va_list vargs;
    va_start( vargs, fmt );
    vfprintf( stderr, fmt, vargs );
    va_end( vargs );
    exit( 1 );
}

int main ( int argc, char **argv )
{
    /* *** */
    int listener = socket( PF_INET, SOCK_STREAM, 0 );
    if( listener < 0 ) die( "socket(listener)" );

    int flag = 1;
    if( setsockopt( listener, SOL_SOCKET, SO_REUSEADDR, (char*)&flag, sizeof(int) ) < 0 )
        die( "setsockopt()" );

    struct sockaddr_in svr_addr;
    memset( &svr_addr, 0, sizeof(struct sockaddr) );
        svr_addr.sin_family = PF_INET;
        svr_addr.sin_port = htons( PORT );
        svr_addr.sin_addr.s_addr = INADDR_ANY;
    if( bind( listener, (struct sockaddr*)&svr_addr, (socklen_t)sizeof(struct sockaddr) ) < 0 )
        die( "bind()" );

    if( listen( listener, 10 ) < 0 )
        die( "listen()" );

    /* *** */

    fd_set fd_master;
    fd_set fd_select;
    int fd_max = listener; 

    FD_ZERO( &fd_master );
    FD_ZERO( &fd_select );

    FD_SET( listener, &fd_master );

    while( 1 )
    {
        fd_select = fd_master;
        if( select( fd_max + 1, &fd_select, NULL, NULL, NULL ) < 0 )
            die( "select()" );

        for( int ifd = 0; ifd <= fd_max; ++ifd )
        {
            if( ! FD_ISSET( ifd, &fd_select ) ) continue;

            struct sockaddr_in cli_addr; memset( &cli_addr, 0, sizeof(cli_addr) );
            socklen_t cli_alen = sizeof(cli_addr);

            if( ifd == listener )
            {
                int cli = accept( listener, (struct sockaddr*)&cli_addr, &cli_alen );
                if( cli < 0 ) die( "accept()" );

                FD_SET( cli, &fd_master );
                if( cli > fd_max ) fd_max = cli;
                printf( "new connection> %s:%u\n", inet_ntoa( cli_addr.sin_addr ), ntohs( cli_addr.sin_port ) );
                fflush( stdout );
            }
            else
            {
                char buf[256];
                cli_alen = sizeof(cli_addr);
                ssize_t nbytes = recvfrom( ifd, buf, sizeof(buf), 0, (struct sockaddr*)&cli_addr, &cli_alen );
                if( nbytes <= 0 )
                {
                    close( ifd );
                    FD_CLR( ifd, &fd_master );
                    if( nbytes == 0 )
                        printf( "connection hung up> %u\n", ifd );
                    else
                        printf( "recvfrom() : %s\n", strerror( errno ) );
                    fflush( stdout );
                }
                else
                {
                    // build a "from identifier" for each of the recipients
                    char msg[sizeof(buf) * 2];
                    sprintf( msg, "%s:%u> ", inet_ntoa( cli_addr.sin_addr ), ntohs( cli_addr.sin_port ) );
                    memcpy( msg + strlen( msg ), buf, nbytes );
                    nbytes += strlen( msg );

                    // send incoming data to all clients (excluding the originator)
                    for( int ofd = 0; ofd <= fd_max; ++ofd )
                    {
                        if( FD_ISSET( ofd, &fd_master ) )
                            if( ofd != listener && ofd != ifd )
                                if( send( ofd, msg, nbytes, 0 ) < 0 )
                                    { printf( "send() %s\n", strerror( errno ) ); fflush( stdout ); }
                    }
                }
            }
        }
    }

    return 0;
}

当代码运行并且您从两个或多个客户端(通过 telnet)连接时,每条消息都将发送者显示为“0.0.0.0”,端口为 0。

recvfrom() 的 Windows 文档指出“[t]he fromfromlen参数对于面向连接的套接字将被忽略。” Linux 和 POSIX 文档没有做出这样的声明,甚至说 recvfrom() “......可以用于在套接字上接收数据,无论它是否面向连接。” 没有在哪里说src_addraddrlen将被忽略......所以我希望这些被填写。

4

2 回答 2

4

在连接的套接字上,您必须调用getpeername然后继续使用您的inet_ntoa(考虑inet_ntop改用它,因为它支持多个地址系列)。根据手册页:

int getpeername(int socket, struct sockaddr *restrict address, socklen_t *restrict address_len);
于 2017-08-19T17:15:25.333 回答
1

它没有说src_addraddrlen将被忽略。

这是不真实的。它说

如果 src_addr 不为 NULL,且底层协议提供源地址,则填写此源地址。【重点补充】

您可以争论是否可以说 TCP 提供了源地址,但您不能声称“它在哪里说……”。

于 2017-08-19T18:47:37.303 回答