套接字编程常用函数

基本TCP套接字编程

socket函数

这是一个进程在执行网络I/O之前必须调用的函数，用于指定期望的通信协议类型

#include <sys/socket.h>

int socket(int family, int type, int protocol);   // 调用成功返回非负的套接字描述符，出错返回-1

connect函数

TCP客户端用connect函数来建立与TCP服务器的连接

#include <sys/socket.h>

int connect(int sockfd, const struct sockaddr *servaddr, socklen_t addrlen);   // 调用成功返回0，出错返回-1

bind函数

bind函数用于将一个本地协议地址赋予一个套接字

#include <sys/socket.h>

int bind(int sockfd, const struct sockaddr *myaddr, socklen_t addrlen);   // 调用成功返回0，出错返回-1

listen函数

listen函数仅由TCP服务器调用，listen函数将一个未连接的套接字转换成一个被动套接字，指
示内核应接受指向该套接字的连接请求。除此之外， listen函数规定了内核应该为相应套接字
排队的最大连接个数(backlog = 未完成连接队列 + 已完成连接队列， 内核允许在这个套接字
上排队的最大客户连接数)。

#include <sys/socket.h>

int listen(int sockfd, int backlog);   // 调用成功返回0，出错返回-1

accept函数

accept函数由TCP服务器调用，用于从已完成连接队列头返回下一个已完成连接

#include <sys/socket.h>

// cliaddr、addrlen用于返回已连接对端进程(客户)的协议地址
// 调用成功返回非负描述符，出错返回-1
int accept(int sockfd, struct sockaddr *cliaddr, socklen_t *addrlen);  

close函数

close函数可用于关闭套接字，并终止TCP连接

#include <unistd.h>

int close(int sockfd);     // 调用成功返回0，出错返回-1

getsockname函数和getpeername函数

这两个函数用于返回与某个套接字关联的本地协议地址(getsockname)，或者返回与某个套接字
关联的外地协议地址(getpeername)

#include <sys/socket.h>

int getsockname(int sockfd, struct sockaddr *localaddr, socklen_t *addrlen);   // 调用成功返回0，出错返回-1

int getpeername(int sockfd, struct sockaddr *peeraddr, socklen_t *addrlen);    // 调用成功返回0，出错返回-1

用于获取个设置套接字选项

#include <sys/socket.h>

// 调用成功返回0，出错返回-1
int getsockopt(int sockfd, int level, int optname, void *optval, socklen_t *optlen);       

// 调用成功返回0，出错返回-1
int setsockopt(int sockfd, int level, int optname, const void *optval, socklen_t optlen);   

示例：简单的时间获取程序————客户端和服务端实现

#include "net.h"

#define SA struct sockaddr

int main(int argc, char **argv)
{
    int sockfd, n;
    char recvline[MAXLINE + 1];
    struct sockaddr_in servaddr;

    if (argc != 2)
    {
        printf("Error: Options!
");
          return -1;
    }

    if ( (sockfd = socket(AF_INET, SOCK_STREAM, 0)) < 0 )
    {
        printf("Error: Socket!
");
          return -1;
    }

    bzero(&servaddr, sizeof(servaddr));
    servaddr.sin_family = AF_INET;
    servaddr.sin_port = htons(13);

    if (inet_pton(AF_INET, argv[1], &servaddr.sin_addr) <= 0)
    {
        printf("Error: inet_pton for %s
", argv[1]);
                return -1;
    }

    if (connect(sockfd, (SA *) &servaddr, sizeof(servaddr)) < 0)
    {
        printf("Error: Connect!
");
    }

    while ( (n = read(sockfd, recvline, MAXLINE)) > 0 )
    {
        recvline[n] = '';   

              if (fputs(recvline, stdout) == EOF)
              {
                        printf("Error: Fputs!
");
              }
    }

    if (n < 0)
    {
        printf("Erroe: Read!
");
    }

    return 0;
}

#include "net.h"

#define SA struct sockaddr

int main(int argc, char **argv)
{
    int listenfd, connfd;
    struct sockaddr_in servaddr;
    char buff[MAXLINE];
    time_t ticks;

    if ( (listenfd = socket(AF_INET, SOCK_STREAM, 0)) < 0 )
    {
        printf("Error: Socket!
");
                return -1;
    }

    bzero(&servaddr, sizeof(servaddr));
    servaddr.sin_family = AF_INET;
    servaddr.sin_addr.s_addr = htonl(INADDR_ANY);
    servaddr.sin_port = htons(13);

    if (bind(listenfd, (SA *) &servaddr, sizeof(servaddr)) < 0)
    {
        printf("Error: Bind!
");
                return -1;
    }

    if (listen(listenfd, LISTENQ) < 0)
    {
        printf("Error: Listen!
");
                return -1;
    }

    for ( ; ; )
    {
        if ( (connfd = accept(listenfd, (SA *) NULL, NULL)) < 0 )
        {
            printf("Error: Accept!
");
            return -1;
        }

                ticks = time(NULL);
                snprintf(buff, sizeof(buff), "%.24s
", ctime(&ticks));

                if (write(connfd, buff, strlen(buff)) < 0)
                {
                    printf("Error: Write!
");
                    return -1;
                }

                if (close(connfd) < 0)
                {
                    printf("Error: Close!
");
                    return -1;
                }
    }

    return 0;
}

 基本UDP套接字编程

recvfrom和sendto函数

类似于标准的read和write函数

#include <sys/socket.h>

// 从套接字sockfd的接受缓冲区中读取nbytes字节的数据写入到指针buff所指向的位置
// 将发送此数据的对端的协议地址用from返回
ssize_t recvfrom(int sockfd, void *buff, size_t nbytes, int flags, struct sockaddr *from, socklen_t *addrlen);

// 将指针buff指向的位置存放的nbytes字节的数据发送到协议地址为to的对端
ssize_t sendto(int sockfd, void *buff, size_t nbytes, int flags, struct sockaddr *to, socklen_t *addrlen);

// 上述两个函数调用成功时返回读或写的字节数，出错返回-1