《UNIX网络编程》入门客户端服务器例子

最近在看《UNIX网络编程》（简称unp）和《Linux程序设计》，对于unp中第一个获取服务器时间的例子，实践起来总是有点头痛的，因为作者将声明全部包含在了unp.h里，导致后面编写代码会对这个头文件造成依赖，而学习不到调用了相应功能之后，应该包含哪些确切的头文件。

再者，我下载了unp.h之后，头文件包含再次产生了其他的依赖缺失，因此便参考了《Linux程序设计》中socket一章的入门例子的头文件包含，并且编译中仍然找不到的包含或者是宏定义在unp.h中搜索并粘贴进来，从而不需要再包含类似unp.h之类的自定义头文件。

以下代码在root权限下运行，因为使用的端口号13对于非root用户可能会无权访问。

unp原书中服务器获取机器时间送回客户端，但我在运行中，字符串保存到buffer区的函数snprintf()总是造成了内存错误而中断，所以本着入门的目的，改为了复制一句话到buffer中。

/*client.c*/
#include <sys/types.h>
#include <sys/socket.h>
#include <stdio.h>
#include <sys/un.h>
#include <unistd.h>
#include <stdlib.h>

#include <netinet/in.h>

#define MAXLINE 4096
#define PORT 13
#define    SA    struct sockaddr

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

    if (argc != 2) {
        fprintf(stderr, "usage: client <IP>\n");
        exit(1);
    }

    if ((sockfd = socket(AF_INET, SOCK_STREAM, 0)) < 0) {
        fprintf(stderr, "socket error\n");
        exit(1);
    }

    bzero(&servaddr, sizeof(servaddr));
    servaddr.sin_family = AF_INET;
    servaddr.sin_port = htons(PORT);
    if (inet_pton(AF_INET, argv[1], &servaddr.sin_addr) <= 0) {
        fprintf(stderr, "inet_pton error for %s\n", argv[1]);
        exit(1);
    }

    if (connect(sockfd, (SA *) &servaddr, sizeof(servaddr)) < 0) {
        fprintf(stderr, "connect error\n");
        exit(1);
    }

    while ((n = read(sockfd, recvline, MAXLINE)) > 0) {
        recvline[n] = 0;
        if (fputs(recvline, stdout) == EOF) {
            fprintf(stderr, "fputs error\n");
            exit(1);
        }
    }

    if (n < 0) {
        fprintf(stderr, "read error\n");
        exit(1);
    }

    exit(0);
}

/*server.c*/
#include <sys/types.h>
#include <sys/socket.h>
#include <stdio.h>
#include <sys/un.h>
#include <unistd.h>
#include <stdlib.h>
#include <string.h>
#include <netinet/in.h>

#define MAXLINE 4096
#define    LISTENQ 1024
#define PORT 13
#define SA  struct sockaddr

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

    listenfd = socket(AF_INET, SOCK_STREAM, 0);

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

    bind(listenfd, (SA *) &servaddr, sizeof(servaddr));

    listen(listenfd, LISTENQ);

    while (1) {
        connfd = accept(listenfd, (SA *)NULL, NULL);

        /* write something to buff[]*/
        strcpy(buff, "this is a message\n");

        write(connfd, buff, strlen(buff));
        close(connfd);
    }

    exit(0);
}

编译文件：

1 $ gcc client.c -o client
2 $ gcc server.c -o server

在一个窗口中运行服务器：

1 $ ./server

在另一个窗口运行客户端，从而显示出一句话"this is a message"：

1 $ ./client 127.0.0.1