一蹄殃、基于TCP的c/s交互流程
C/S
C/S
服務器端：
socket() → bind() → listen() → accept() → recv()/read() ? send()/write() → close()
創(chuàng)建socket → 綁定socket和端口號 → 監(jiān)聽端口號 → 接收來自客戶端的連接請求 → 從socket中讀取字符 → 關閉socket客戶端：
socket() → connect() → send()/write() ? recv()/read() → close()
創(chuàng)建socket → 連接指定服務器的IP/端口號 → 向socket中寫入信息 → 關閉socket
二蔗崎、C++代碼實現(xiàn)
1、server.cpp
按照該實現(xiàn)诀艰，服務器每次只能服務一個客戶端。
#include <iostream>
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <string.h>
#include <unistd.h>
#include <errno.h>
#include <stdio.h>
#include "cs.h"
#include "ErrCode.h"
using namespace std;
#define PORT 9977
#define MSG_SIZE 4096
#define MAX_LINE 20
int server() {
char cMsg[MSG_SIZE], sMsg[MSG_SIZE];
struct sockaddr_in serverAddr; // 一個將來與套接字綁定的結構體
memset(&serverAddr, 0, sizeof(serverAddr));
serverAddr.sin_port = (PORT); // 從服務器的PORT端口接收客戶端信息
serverAddr.sin_family = AF_INET; // 協(xié)議簇，AF_INET表示TCP/IP協(xié)議
serverAddr.sin_addr.s_addr = htonl(INADDR_ANY); // 指定接收的信息來自于某個IP躏将，這里隨意
int fd_skt = socket(AF_INET, SOCK_STREAM, 0); // （1）socket函數(shù)新建套接字fd_skt
if (fd_skt < 0) {
perror("server socket err");
return SERVER_SOCKET_ERR;
}
if (bind(fd_skt, (struct sockaddr*)&serverAddr, sizeof(serverAddr)) < 0 ) { // （2）bind綁定套接字與serverAddr
perror("server bind err");
return SERVER_BIND_ERR;
}
if (listen(fd_skt, MAX_LINE) < 0) { // （3）listen監(jiān)聽套接字fd_skt
perror("server listen err");
return SERVER_LISTEN_ERR;
}
cout << "Listening..." << endl;
while(1) {
memset(cMsg, 0, sizeof(cMsg));
memset(sMsg, 0, sizeof(sMsg));
struct sockaddr_in clientAddr;
socklen_t length = sizeof(clientAddr);
int fd_client = accept(fd_skt, (struct sockaddr*)&clientAddr, &length);
// （4）accept接收連接請求：程序在此阻塞直至產(chǎn)生第一次握手
// 接收到的信息存于第二第三個參數(shù)中
// 返回值是新的文件描述符硬毕，用于后續(xù)read和write調(diào)用
if (fd_client < 0) {
perror("server accept err");
return SERVER_ACCEPT_ERR;
}
cout << "server: got connection from " << clientAddr.sin_addr.s_addr
<< ", port " << PORT
<< ", socket" << fd_client << endl;
while(1) {
int cLen = recv(fd_client, cMsg, sizeof(cMsg), 0); // （5）read將接收到的客戶端消息存在cMsg中
if (cLen <= 0) {
perror("server recv err");
return SERVER_RECV_ERR;
}
cMsg[cLen] = '\0';
cout << "Receive message form <" << clientAddr.sin_addr.s_addr << ">: " << cMsg << endl;
cin.ignore(1024,'\n'); // 去除上一個cin殘留在緩沖區(qū)的\n
cin.getline(sMsg, sizeof(sMsg)); // 不用cin呻引，因為不能含空格
if(strcmp(sMsg, "quit\n") == 0){
break; // 結束對當前客戶端的服務
}
if(send(fd_client, sMsg, strlen(sMsg), 0) <= 0) { // （6）send將服務器的消息發(fā)給客戶端
perror("server send err");
return SERVER_SEND_ERR;
}
// recv和send的返回值：
// >0表示成功，返回實際發(fā)送或接受的字節(jié)數(shù)
// =0表示超時吐咳，對方主動關閉了連接過程
// <0出錯
cout << "Send message to <" << clientAddr.sin_addr.s_addr << ">: " << sMsg << endl;
}
close(fd_client);
cout << "當前客戶端已結束通信逻悠，是否繼續(xù)等待其他客戶端？（1 - 是韭脊， 0 - 否）" << endl;
bool isContinue = false;
cin >> isContinue;
if (!isContinue) {
break;
}
}
close(fd_skt);
return SUCCESS;
}
2童谒、client.cpp
#include <iostream>
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <string.h>
#include <unistd.h>
#include <errno.h>
#include <stdio.h>
#include "cs.h"
#include "ErrCode.h"
using namespace std;
#define PORT 9977
#define MAX_LINE 20
#define MSG_SIZE 4096
#define DEFAULT_SERVER "127.0.0.1"
int client() {
char cMsg[MSG_SIZE], sMsg[MSG_SIZE];
struct sockaddr_in serverAddr; // 一個將來與套接字綁定的結構體
memset(&serverAddr, 0, sizeof(serverAddr));
serverAddr.sin_port = (PORT); // 從客戶端的PORT端口接收服務器信息
serverAddr.sin_family = AF_INET; // 協(xié)議簇，AF_INET表示TCP/IP協(xié)議
cout << "Please input the server address: (輸入1連接默認服務器)" << endl;
char sAddr[20];
cin >> sAddr;
if (strcmp(sAddr, "1") == 0) {
if(inet_pton(AF_INET, DEFAULT_SERVER, (void *)&serverAddr.sin_addr) <= 0) {
perror("client inet_pton err");
return CLIENT_INETPTON_ERR;
}
}
else {
if(inet_pton(AF_INET, DEFAULT_SERVER, (void *)&serverAddr.sin_addr) <= 0) {
perror("client inet_pton err");
return CLIENT_INETPTON_ERR;
}
}
int fd_skt = socket(AF_INET, SOCK_STREAM, 0); // （1）socket函數(shù)新建套接字fd_skt
if (fd_skt < 0) {
perror("client socket err");
return CLIENT_SOCKET_ERR;
}
if (connect(fd_skt, (struct sockaddr*)&serverAddr, sizeof(serverAddr)) < 0) { //（2）connect向服務器發(fā)起連接請求
perror("client connect err");
return CLIENT_CONNECT_ERR;
}
cout << "Connect Success! \nBegin to chat..." << endl;
while(1) {
memset(cMsg, 0, sizeof(cMsg));
memset(sMsg, 0, sizeof(sMsg));
cin.ignore(1024,'\n'); // 去除上一個cin殘留在緩沖區(qū)的\n
cin.getline(cMsg, sizeof(cMsg)); // 不用cin沪羔，因為不能含空格
if(strcmp(cMsg, "quit\n") == 0){
break;
}
if(send(fd_skt, cMsg, strlen(cMsg), 0) < 0) { // （3）send饥伊，客戶端向服務端發(fā)消息
perror("client send err");
return CLIENT_SEND_ERR;
}
cout << "Send message to <" << serverAddr.sin_addr.s_addr << ">: " << cMsg << endl;
int sLen = recv(fd_skt, sMsg, sizeof(sMsg),0); // （4） recv，接收服務器發(fā)來的消息
if(sLen <= 0) {
perror("client recv err");
return CLIENT_RECV_ERR;
}
sMsg[sLen] = '\0';
cout << "Receive message form <" << serverAddr.sin_addr.s_addr << ">: " << sMsg << endl;
}
close(fd_skt);
}
3、main.cpp
#include <iostream>
#include "cs.h"
using namespace std;
int main() {
cout << "————————【chat start】————————" << endl;
while(1) {
cout << "Client or Server?" << endl;
char cs;
cin >> cs;
if (cs == 's') {
if(server() != 0){
break;
}
}
else if (cs == 'c') {
if(client() != 0){
break;
}
}
else {
cout << "err input" << endl;
}
}
cout << "————————【chat end】————————" << endl;
return 0;
}
4琅豆、ErrCode.h
#define SUCCESS 0
#define SERVER_SOCKET_ERR 1000
#define SERVER_BIND_ERR 1001
#define SERVER_LISTEN_ERR 1002
#define SERVER_ACCEPT_ERR 1003
#define SERVER_RECV_ERR 1004
#define SERVER_SEND_ERR 1005
#define CLIENT_INETPTON_ERR 1010
#define CLIENT_SOCKET_ERR 1011
#define CLIENT_CONNECT_ERR 1012
#define CLIENT_SEND_ERR 1013
#define CLIENT_RECV_ERR 1014
5愉豺、cs.h
#include <iostream>
int server();
int client();
3、運行效果
client
server
