网络编程入门

Linux下网络编程

socket函数

#include <sys/socket.h>
int socket(int domain, int type, int protocal);

成功时返回文件描述符，失败时返回-1

bind函数

#include <sys/socket.h>
int bind(int sockfd, struct sockaddr *myaddr, socklen_t addrlen);

成功时返回0，失败时返回-1

listen函数

#include <sys/socket.h>
int listen(int sockfd, int backlog);

成功时返回0，失败时返回-1

accept函数

#include <sys/socket.h>
int accept(int sockfd, struct sockaddr *addr, socklen_t *addrlen);

成功时返回0，失败时返回-1

connect函数

#include <sys/socket.h>
int connect(int sockfd, struct sockaddr *serv_addr, socklen_t addrlen);

成功时返回0，失败时返回-1

open函数

#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>
int open(const char *path, int flag);

成功时返回文件描述符，失败时返回-1

打开模式	含义
O_CREAT	必要时创建文件
O_TRUNC	删除全部现有数据
O_APPEND	维持现有数据，保存到其后面
O_RDONLY	只读打开
O_WRONLY	只写打开
O_RDWR	读写打开

close函数

#include <unistd.h>
int close(fd);

成功时返回0，失败时返回-1

write函数

#include <unistd.h>
ssize_t write(int fd, const void *buf, size_t nbytes);

成功时返回写入的字节数，失败时返回-1
- fd: 显示数据传输对象的文件描述符
- buf: 保存要传输数据的缓冲地址值
- nbytes: 要传输数据的字节数

read函数

#include <unistd.h>
ssize_t read(int fd, void *buf, size_t nbytes);

成功时返回接收的字节数（但遇到文件结尾则返回0），失败时返回-1
- fd: 显示数据传输对象的文件描述符
- buf: 要保存接收数据的缓冲地址值
- nbytes: 要传输数据的字节数

网络编程中接受连接请求的套接字创建过程可整理如下：

• 调用socket函数创建套接字
• 调用bind函数分配IP地址和端口号
• 调用listen函数转为可接收请求状态
• 调用accept函数受理连接请求

server.cpp

#include <cstdio>
#include <string>
#include <cstdlib>
#include <cstring>
#include <iostream>
#include <unistd.h>
#include <arpa/inet.h>
#include <sys/socket.h>
using namespace std;

/*
 * 网络编程中接受连接请求的套接字创建过程可整理为：
 * 1. 调用socket函数创建套接字
 * 2. 调用bind函数分配IP地址和端口号
 * 3. 调用listen函数转为可接收请求状态
 * 4. 调用accept函数受理连接请求 
 * */

void error_handling(string message);

int main(int argc, char ** argv) {
  int serv_sock;
  int clnt_sock;

  struct sockaddr_in serv_addr;
  struct sockaddr_in clnt_addr;
  socklen_t clnt_addr_size;

  char message[] = "hello, world";

  if (argc != 2) {
    printf("Usage: %s <port>\n", argv[0]);
    exit(1);
  }

  serv_sock = socket(PF_INET, SOCK_STREAM, 0);
  if (serv_sock == -1) 
    error_handling("socket() error");

  memset(&serv_addr, 0, sizeof(serv_addr));
  serv_addr.sin_family = AF_INET;
  serv_addr.sin_addr.s_addr = htonl(INADDR_ANY);
  serv_addr.sin_port = htons(atoi(argv[1]));

  if (bind(serv_sock, (struct sockaddr *)&serv_addr, sizeof(serv_addr)) == -1)
    error_handling("bind() error");

  if (listen(serv_sock, 5) == -1)
    error_handling("listen() error");

  clnt_addr_size = sizeof(clnt_addr);
  clnt_sock = accept(serv_sock, (struct sockaddr *)&clnt_addr,  &clnt_addr_size);
  if (clnt_sock == -1)
    error_handling("accept() error");

  write(clnt_sock, message, sizeof(message));
  close(clnt_sock);
  close(serv_sock);
  return 0;
}

void error_handling(string message) {
  cerr << message << endl;
  exit(1);
}

client.cpp

#include <cerrno>
#include <cstdio>
#include <string>
#include <cstdlib>
#include <cstring>
#include <iostream>
#include <unistd.h>
#include <arpa/inet.h>
#include <sys/socket.h>

using namespace std;

void error_handling(string message);

int main(int argc, char ** argv) {
  int sock;
  struct sockaddr_in serv_addr;
  char message[30];
  int str_len;

  if (argc != 3) {
    cout << "Usage: " << argv[0] << " <IP> <port>" << endl;
    exit(1);
  }

  sock = socket(PF_INET, SOCK_STREAM, 0);
  if (sock == -1) 
    error_handling("socket() error");

  memset(&serv_addr, 0, sizeof(serv_addr));
  serv_addr.sin_family = AF_INET;
  serv_addr.sin_addr.s_addr = inet_addr(argv[1]);
  serv_addr.sin_port = htons(atoi(argv[2]));

  if (connect(sock, (struct sockaddr *)&serv_addr, sizeof(serv_addr)) == -1) {
    close(sock);
    cout << "errno = " << errno << endl;
    error_handling("connect() error");
  }

  str_len = read(sock, message, sizeof(message) - 1);
  if (str_len == -1) {
    close(sock);
    error_handling("read() error");
  }

  cout << "Message from server: " << message << endl;
  close(sock);
  return 0;
}
void error_handling(string message) {
  cerr << message << endl;
  exit(1);
}

Windows下网络编程

WSAStartup函数

#include <winsock2.h>
int WSAStartup(WORD wVersionRequested, LPWSADATA lpWSAData);

成功时返回0，失败时返回非0的错误代码值
- wVersionRequested: 程序员要用的Winsock版本
- lpWSAData: WSADATA结构体变量的地址值
- 一般使用4位16进制数表示版本，高8位为副版本号，低8位为主版本号，如0x0201即表示主版本号为1副版本号为2的版本，即1.2版本，通常借助MAKEWORD宏函数MAKEWORD(1,2)表示1.2版本
- lpWSADATA是WSADATA的指针类型

WSAStartup函数调用过程

int main(int argc, char ** argv) {
    WSADATA wsaData;
    ....
    if (WSAStartup(MAKEWORD(2,2), &wsaData) != 0) 
        error_handling("WSAStartup() error");
    ....
    return 0;
}

WSACleanup函数

#include <winsock2.h>
int WSACleanup(void);

成功时返回0，失败时返回SOCKET_ERROR\
调用此函数时，Winsock相关库将归还Windows系统，无法再调用Winsock相关函数

socket函数

#include <winsock2.h>
SOCKET socket(int af, int type, int protocal);

成功时返回套接字句柄，失败时返回INVALID_SOCKET

bind函数

#include <winsock2.h>
int bind(SOCKET s, const struct sockaddr *name, int namelen);

成功时返回0，失败时返回SOCKET_ERROR

listen函数

#include <winsock2.h>
int listen(SOCKET s, int backlog);

成功时返回0，失败时返回SOCKET_ERROR

accept函数

#include <winsock2.h>
SOCKET accept(SOCKET s, struct sockaddr * addr, int *addrlen);

成功时返回套接字句柄，失败时返回INVALID_SOCKET

connect函数

#include <winsock2.h>
int connect(SOCKET s, const struct sockaddr * name, int namelen);

成功时返回0，失败时返回SOCKET_ERROR

closesocket函数

#include <winsock2.h>
int closesocket(SOCKET s);

成功时返回0，失败时返回SOCKET_ERROR

send函数

#include <winsock2.h>
int send(SOCKET s, const char *buf, int len, int flags);

成功时返回传输字节数，失败时返回SOCKET_ERROR
- s: 表示数据传输对象连接的套接字句柄
- buf: 保存带传输数据的缓冲地址值
- len: 要传输的字节数
- flags: 传输数据时用到的多种选项信息

recv函数

#include <winsock2.h>
int recv(SOCKET s, const char *buf, int len, int flags);

成功时返回接收的字节数(收到EOF时为0)，失败时返回SOCKET_ERROR
- s: 表示数据传输对象连接的套接字句柄
- buf: 保存接收数据的缓冲地址值
- len: 能够接收的最大字节数
- flags: 接收数据时用到的多种选项信息

server.cpp

#include <cstdio>
#include <string>
#include <cstring>
#include <cstdlib>
#include <iostream>
#include <WinSock2.h>
using namespace std;

void ErrorHandling(string message);

int main(int argc, char ** argv) {
    WSADATA wsaData;
    SOCKET hServSock, hClntSock;
    SOCKADDR_IN servAddr, clntSock;

    char message[] = "hello, world";
    if (argc != 2) {
        cout << "Usage: " << argv[0] << " <port>" << endl;
        exit(1);
    }

    if (WSAStartup(MAKEWORD(2, 2), &wsaData) == INVALID_SOCKET)
        ErrorHandling("WSAStartup() error");

    hServSock = socket(PF_INET, SOCK_STREAM, 0);
    if (hServSock == SOCKET_ERROR)
        ErrorHandling("socket() error");

    memset(&servAddr, 0, sizeof(servAddr));
    servAddr.sin_family = AF_INET;
    servAddr.sin_addr.s_addr = htonl(INADDR_ANY);
    servAddr.sin_port = htons(atoi(argv[1]));

    if (bind(hServSock, (SOCKADDR *)&servAddr, sizeof(servAddr)) == SOCKET_ERROR) {
        closesocket(hServSock);
        ErrorHandling("bind() error");
    }

    if (listen(hServSock, 5) == SOCKET_ERROR) {
        closesocket(hServSock);
        ErrorHandling("listen() error");
    }

    int szClntSock = sizeof(clntSock);
    hClntSock = accept(hServSock, (SOCKADDR *)&clntSock, &szClntSock);
    if (hClntSock == INVALID_SOCKET) {
        closesocket(hServSock);
        ErrorHandling("accept() error");
    }

    send(hClntSock, message, sizeof(message), 0);

    closesocket(hServSock);
    closesocket(hClntSock);
    WSACleanup();
    return 0;
}

void ErrorHandling(string message) {
    WSACleanup();
    cerr << message << endl;
    exit(1);
}

client.cpp

#include <cstdio>
#include <string>
#include <cstring>
#include <cstdlib>
#include <iostream>
#include <WinSock2.h>
using namespace std;

void ErrorHandling(string message);

int main(int argc, char ** argv) {
    WSADATA wsaData;
    SOCKET hSocket;
    SOCKADDR_IN servAddr;

    char message[30];
    int strLen;
    if (argc != 3) {
        cout << "Usage: " << argv[0] << "<IP> <port>" << endl;
        exit(1);
    }

    if (WSAStartup(MAKEWORD(2, 2), &wsaData) != 0)
        ErrorHandling("WSAStartup() error");

    hSocket = socket(PF_INET, SOCK_STREAM, 0);
    if (hSocket == INVALID_SOCKET)
        ErrorHandling("socket() error");

    memset(&servAddr, 0, sizeof(servAddr));
    servAddr.sin_family = AF_INET;
    servAddr.sin_addr.s_addr = inet_addr(argv[1]);
    servAddr.sin_port = htons(atoi(argv[2]));

    if (connect(hSocket, (SOCKADDR *)&servAddr, sizeof(servAddr)) == SOCKET_ERROR) {
        closesocket(hSocket);
        ErrorHandling("connect() error");
    }

    strLen = recv(hSocket, message, sizeof(message) - 1, 0);
    if (strLen == -1) {
        closesocket(hSocket);
        ErrorHandling("recv() error");
    }

    cout << "Message from server: " << message << endl;

    closesocket(hSocket);
    WSACleanup();
    return 0;
}

void ErrorHandling(string message) {
    WSACleanup();
    cerr << message << endl;
    exit(1);
}

运行结果