1. 概述:
該demo主要實現了linux下通過tcp方式的socket通訊,相關接口介紹可以參考<<UNIX環境高級編程>>
2. 場景(服務端(一)<----(一)客戶端):
1) 服務端:
輸入:服務端監聽的port;
接收到客戶端發送的消息,打印到終端,並且回覆客戶端"ok";
當接收消息內容爲"exit"時,選擇退出服務端進程
2) 客戶端:
輸入:客戶端的port,以及服務端ip和port;
輸入消息內容,將消息發送給服務端,阻塞直到接收到服務端的返回信息,並打印到終端(超時未接收到時,報錯);
當輸入消息內容爲"exit"時,選擇退出客戶端進程
3. 測試:
TCP 服務端
客戶端
/*
demo_tcp_server.c
網絡編程demo(tcp服務端)
*/
#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <unistd.h>
#include <signal.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include "lpm.h"
void signal_func() {
printf(RED BLINK"Client Close\n\n"NONE);
}
int main(int argc, char **argv){
int ret;
int server_fd = -1;
int connect_fd = -1;
char server_port[8];
char buf[MAX_BUF];
printf(RED BLINK"TCP 服務端\n"NONE);
printf("請輸入服務端 port\t");
scanf("%s", server_port);
printf("\n");
/*
SIGPIPE : 在reader中止之後寫Pipe的時候發送
SIG_DFL : 默認信號處理程序
SIG_IGN : 忽略信號的處理程序
signal(SIGPIPE,SIG_IGN);
對一個對端已經關閉的socket調用兩次讀/寫, 第二次將會生成SIGPIPE信號, 該信號默認結束進程
*/
if(SIG_ERR == signal(SIGPIPE,signal_func)){
perror("signal:");
return -1;
}
/*
協議域 : AF_INET(IPV4) AF_INET6(IPV6) AF_LOCAL AF_ROUTE 等等
socket類型 : SOCK_STREAM(TCP) SOCK_DGRAM(UDP) 等等
0 : 表示協議域和socket類型選擇默認的協議, 通常是0
*/
server_fd = socket(AF_INET, SOCK_STREAM, 0);
if(server_fd < 0){
/*printf("socket:%s\n", strerror(errno));*/
perror("socket:");
return -1;
}
/*
應用協議 : SOL_SOCKET(套接字) IPPROTO_TCP IPPROTO_IP...
設置項 : SO_REUSEADDR(是否可以重用bind的地址)...
opt : 一些設置項指的是開關...
*/
int opt = 1;
ret = setsockopt(server_fd, SOL_SOCKET, SO_REUSEADDR, (void *)&opt, sizeof(opt));
if(ret < 0){
perror("setsockopt");
close(server_fd);
return -1;
}
struct sockaddr_in server_addr;
memset(&server_addr, 0x0, sizeof(struct sockaddr_in));
server_addr.sin_family = AF_INET;
server_addr.sin_port = htons(atoi(server_port));
server_addr.sin_addr.s_addr = htonl(INADDR_ANY);
ret = bind(server_fd, (struct sockaddr*)&server_addr, sizeof(server_addr));
if(ret < 0){
perror("bind:");
close(server_fd);
return -1;
}
ret = listen(server_fd, MAX_LISTEN);
if(ret < 0){
perror("listen:");
close(server_fd);
return -1;
}
char IP[32];
int addrLen = sizeof(struct sockaddr);
struct sockaddr_in client_addr;
memset(&client_addr, 0x0, sizeof(struct sockaddr_in));
connect_fd = accept(server_fd, (struct sockaddr *)&client_addr, (socklen_t *)&addrLen);
if(connect_fd < 0){
perror("accept:");
close(server_fd);
return -1;
}
memset(IP, 0x0, sizeof(IP));
inet_ntop(AF_INET, (void *)&client_addr.sin_addr.s_addr, IP, sizeof(IP));
printf("客戶端 IP \t%s\n", IP);
printf("客戶端 port\t%d\n", ntohs(client_addr.sin_port));
printf("\n");
ret = send(connect_fd, OK, strlen(OK), 0);
if(ret < 0){
perror("send:");
close(connect_fd);
close(server_fd);
return -1;
}
while(1){
memset(buf, 0x0, sizeof(buf));
ret = recv(connect_fd, buf, sizeof(buf), 0);
if(ret > 0){
printf("消息 \t%s\n", buf);
printf("\n");
}
else if(ret == 0){
/* 無可用數據 或者 對等方已經按序結束 */
printf(RED BLINK"Client Close\n\n"NONE);
break;
}
else if(ret < 0){
perror("recv:");
continue;
}
if(strlen(buf) > 0 && strncmp(buf, EXIT, strlen(buf)) == 0){
break;
}
ret = send(connect_fd, OK, strlen(OK), 0);
if(ret < 0){
perror("send:");
continue;
}
}
/*
SHUT_RD(0) : 關閉sockfd上的讀功能
SHUT_WR(1) : 關閉sockfd的寫功能
SHUT_RDWR(2) : 關閉sockfd的讀寫功能
*/
shutdown(connect_fd, SHUT_RDWR);
shutdown(server_fd, SHUT_RDWR);
close(connect_fd);
close(server_fd);
return 0;
}
/*
demo_tcp_client.c
網絡編程demo(tcp客戶端)
*/
#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <unistd.h>
#include <signal.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include "lpm.h"
void signal_func() {
printf(RED BLINK"Server Close\n\n"NONE);
}
int main(int argc, char **argv){
int ret;
int client_fd = -1;
char client_port[8];
char server_ip[32];
char server_port[8];
char buf[MAX_BUF];
printf(RED BLINK"TCP 客戶端\n"NONE);
printf("請輸入客戶端 port\t");
scanf("%s", client_port);
printf("請輸入服務端 ip \t");
scanf("%s", server_ip);
printf("請輸入服務端 port\t");
scanf("%s", server_port);
printf("\n");
/*
SIGPIPE : 在reader中止之後寫Pipe的時候發送
SIG_DFL : 默認信號處理程序
SIG_IGN : 忽略信號的處理程序
signal(SIGPIPE,SIG_IGN);
對一個對端已經關閉的socket調用兩次讀/寫, 第二次將會生成SIGPIPE信號, 該信號默認結束進程
*/
if(SIG_ERR == signal(SIGPIPE,signal_func)){
perror("signal:");
return -1;
}
/*
協議域 : AF_INET(IPV4) AF_INET6(IPV6) AF_LOCAL AF_ROUTE 等等
socket類型 : SOCK_STREAM(TCP) SOCK_DGRAM(UDP) 等等
0 : 表示協議域和socket類型選擇默認的協議, 通常是0
*/
client_fd = socket(AF_INET, SOCK_STREAM, 0);
if(client_fd < 0){
/*printf("socket:%s\n", strerror(errno));*/
perror("socket:");
return -1;
}
/*
應用協議 : SOL_SOCKET(套接字) IPPROTO_TCP IPPROTO_IP...
設置項 : SO_REUSEADDR(是否可以重用bind的地址)...
opt : 一些設置項指的是開關...
*/
int opt = 1;
ret = setsockopt(client_fd, SOL_SOCKET, SO_REUSEADDR, (void *)&opt, sizeof(opt));
if(ret < 0){
perror("setsockopt");
close(client_fd);
return -1;
}
/*
設置項 : SO_RCVTIMEO(爲recv設置超時時間)...
*/
struct timeval tv;
tv.tv_sec = RECV_TIME_OUT;
tv.tv_usec = 0;
ret = setsockopt(client_fd, SOL_SOCKET, SO_RCVTIMEO, (void *)&tv, sizeof(tv));
if(ret < 0){
perror("setsockopt");
close(client_fd);
return -1;
}
struct sockaddr_in client_addr;
memset(&client_addr, 0x0, sizeof(struct sockaddr_in));
client_addr.sin_family = AF_INET;
client_addr.sin_port = htons(atoi(client_port));
client_addr.sin_addr.s_addr = htonl(INADDR_ANY);
ret = bind(client_fd, (struct sockaddr*)&client_addr, sizeof(client_addr));
if(ret < 0){
perror("bind:");
close(client_fd);
return -1;
}
struct sockaddr_in server_addr;
memset(&server_addr, 0x0, sizeof(struct sockaddr_in));
server_addr.sin_family = AF_INET;
server_addr.sin_port = htons(atoi(server_port));
inet_pton(AF_INET, server_ip, (void *)&(server_addr.sin_addr.s_addr));
ret = connect(client_fd, (struct sockaddr*)&server_addr, sizeof(server_addr));
if(ret < 0){
perror("connect:");
close(client_fd);
return -1;
}
memset(buf, 0x0, sizeof(buf));
ret = recv(client_fd, buf, sizeof(buf), 0);
if(ret > 0){
printf("%s\n", buf);
printf("\n");
}
else if(ret == 0){
/* 無可用數據 或者 對等方已經按序結束 */
printf(RED BLINK"Server Close\n\n"NONE);
close(client_fd);
return -1;
}
else if(ret < 0){
perror("recv:");
close(client_fd);
return -1;
}
while(1){
memset(buf, 0x0, sizeof(buf));
printf("請輸入消息\t");
scanf("%s", buf);
ret = send(client_fd, buf, strlen(buf), 0);
if(ret < 0){
perror("send:");
continue;
}
if(strlen(buf) > 0 && strncmp(buf, EXIT, strlen(buf)) == 0){
break;
}
memset(buf, 0x0, sizeof(buf));
ret = recv(client_fd, buf, sizeof(buf), 0);
if(ret > 0){
printf("%s\n", buf);
printf("\n");
}
else if(ret == 0){
/* 無可用數據 或者 對等方已經按序結束 */
printf(RED BLINK"Server Close\n\n"NONE);
break;
}
else if(ret < 0){
perror("recv:");
continue;
}
}
/*
SHUT_RD(0) : 關閉sockfd上的讀功能
SHUT_WR(1) : 關閉sockfd的寫功能
SHUT_RDWR(2) : 關閉sockfd的讀寫功能
*/
shutdown(client_fd, SHUT_RDWR);
close(client_fd);
return 0;
}
/*
lpm.h
*/
#ifndef _LPM_H_
#define _LPM_H_
#define NONE "\e[0m"
#define BLACK "\e[0;30m"
#define L_BLACK "\e[1;30m"
#define RED "\e[0;31m"
#define L_RED "\e[1;31m"
#define GREEN "\e[0;32m"
#define L_GREEN "\e[1;32m"
#define BROWN "\e[0;33m"
#define YELLOW "\e[1;33m"
#define BLUE "\e[0;34m"
#define L_BLUE "\e[1;34m"
#define PURPLE "\e[0;35m"
#define L_PURPLE "\e[1;35m"
#define CYAN "\e[0;36m"
#define L_CYAN "\e[1;36m"
#define GRAY "\e[0;37m"
#define WHITE "\e[1;37m"
#define BOLD "\e[1m"
#define UNDERLINE "\e[4m"
#define BLINK "\e[5m"
#define REVERSE "\e[7m"
#define HIDE "\e[8m"
#define CLEAR "\e[2J"
#define CLRLINE "\r\e[K" //or "\e[1K\r"
#define MAX_BUF 64
#define RECV_TIME_OUT 3
#define MAX_LISTEN 1
#define MAX_ACCEPT 2
#define OK "ok"
#define EXIT "exit"
#define ERROR "error"
#define LOG() printf("[%s %s] %s: %s: %d\n", __DATE__, __TIME__, __FILE__, __func__, __LINE__)
#endif
#Makefile
CC := gcc
INCLUDE = -I /home/demo/include/
all:
$(CC) demo_tcp_server.c $(INCLUDE) -o demo_tcp_server -Wall -Werror
$(CC) demo_tcp_client.c $(INCLUDE) -o demo_tcp_client -Wall -Werror
clean:
rm demo_tcp_server demo_tcp_client