socket编程入门（二）

在动手编写socket通信程序之前，还需要补充一些可能用到的函数。本文列出这些函数。

编程使用到的函数

• socket
  int socket(int domain,int type,int protocol)
  用于创建一个套接字用于通信，成功返回非负整数，失败返回-1
• bind
  int bind(int sockfd,const struct sockaddr* addr,socklen_t addrlen)
  用于将addr指向地址和sock描述符绑定，成功返回0，失败返回-1
• listen
  int listen(int sockfd,int backlog)
  backlog：规定内核为此套接字排队的最大连接个数
  将套接字用于监听进入的连接，失败返回-1
  socket函数返回的默认是主动套接字，在调用listen函数后变为被动套接字。
  主动套接字用于调用connect来主动发起连接。
  被动套接字被动地调用accept等待连接到来。
  对于给定的监听套接口，内核要维护两个队列：
  
  • 已由客户发出并到达服务器，服务器正在等待完成相应的TCP三路握手过程的队列a
  • 已完成连接的队列b
  • a+b=backlog
• accept
  int accept(int sockfd,struct sockaddr* addr,socklen_t* addrlen)
  从已完成连接队列返回第一个连接，如果已完成连接队列为空，则阻塞，失败返回-1
• setsockopt(int sockfd,int level,int optname,const void* optval,socklen_t optlen);
  用于设定套接字选项。sockfd：套接字描述符；level：指定系统中解释选项的代码或为通用套接字代码，或者为某个特定于协议的代码；（UNP第三版卷一P151）
  optval：待设置的新值；optlen：新值长度。

了解以上函数后，就可以编写socket程序了，这里选择一个比较简单的示例：回射客户服务器。下面给出它的编程模型。

最简单回射客户服务器模型

代码

makefile

.PHONY:clean all
CC=gcc -std=c99
CFLAGS=-Wall -g
#OBJS=cli.o srv.o
BIN=cli srv
all:$(BIN)
cli:cli.o
    $(CC) $(CFLAGS) $^ -o $@
srv:srv.o
    $(CC) $(CFLAGS) $^ -o $@
.c.o:
    $(CC) $(CFLAGS) -c $< -o $@
clean:
    rm -rf $(BIN) *.o

cli:

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

#define ERR_EXIT(m)\
    do\
    {\
        perror(m);\
        exit(EXIT_FAILURE);\
    }while(0)

int main(int argc,char* argv[])
{

    int sock;
    if((sock = socket(PF_INET,SOCK_STREAM,IPPROTO_TCP)) < 0)
        ERR_EXIT("socket");
    struct sockaddr_in servaddr;
    memset(&servaddr,0,sizeof(servaddr));
    servaddr.sin_family = AF_INET;
    servaddr.sin_port = htons(5188);
    //servaddr.sin_addr.s_addr = htonl("127.0.0.1");
    servaddr.sin_addr.s_addr = inet_addr("127.0.0.1");
    //inet_aton("127.0.0.1",&servaddr.sin_addr);

    if(connect(sock,(struct sockaddr*)&servaddr,sizeof(servaddr)) < 0)
        ERR_EXIT("connect");

    char sendbuf[1024] = {0};

    char recvbuf[1024] = {0};
    while(fgets(sendbuf,sizeof(sendbuf),stdin)!=NULL)
    {
        write(sock,sendbuf,strlen(sendbuf));
        read(sock,recvbuf,sizeof(recvbuf));

        fputs(recvbuf,stdout);
        memset(sendbuf,0,sizeof(sendbuf));
        memset(recvbuf,0,sizeof(recvbuf));
    }

    return 0;
}

srv:

#include <stdio.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <stdlib.h>
#include <string.h>
#include <errno.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <unistd.h>
#define ERR_EXIT(m)\
    do\
    {\
        perror(m);\
        exit(EXIT_FAILURE);\
    }while(0)

int main(int argc,char* argv[])
{

    int listenfd;
    if((listenfd = socket(PF_INET,SOCK_STREAM,IPPROTO_TCP)) < 0)
        ERR_EXIT("socket");
    struct sockaddr_in servaddr;
    memset(&servaddr,0,sizeof(servaddr));
    servaddr.sin_family = AF_INET;
    servaddr.sin_port = htons(5188);
    servaddr.sin_addr.s_addr = htonl(INADDR_ANY);
    //servaddr.sin_addr.s_addr = inet_addr("127.0.0.1");
    //inet_aton("127.0.0.1",&servaddr.sin_addr);
    if(bind(listenfd,(struct sockaddr*)&servaddr,sizeof(servaddr)) < 0)
        ERR_EXIT("bind");
    if(listen(listenfd,SOMAXCONN) < 0)
        ERR_EXIT("listen");
    struct sockaddr_in peeraddr;
    socklen_t peerlen = sizeof(peeraddr);
    int conn;
    if((conn = accept(listenfd,(struct sockaddr*)&peeraddr,&peerlen)) < 0)
        ERR_EXIT("accept");
    printf("ip=%s portr=%d\n",inet_ntoa(peeraddr.sin_addr),ntohs(peeraddr.sin_port));   
    char recvbuf[1024];
    while(1)
    {
        memset(recvbuf,0,sizeof(recvbuf));

        int ret = read(conn,recvbuf,sizeof(recvbuf));
        fputs(recvbuf,stdout);
        write(conn,recvbuf,ret);
    }
    return 0;
}