簡單的linux下socket程式設計，分別基於TCP和UDP協議實現的簡單程式

linux下socket程式設計可以概括為以下幾個函式的運用：

• socket()
• bind()
• listen()
• connect()
• accept()
• read()
• write()
• close()函式

基於TCP實現

流程

server程式碼

#include <stdio.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>


int main(int argc, char *argv[])
{
	int server_sockfd;//伺服器端套接字
	int client_sockfd;//客戶端套接字
	int len;
	struct sockaddr_in my_addr;   //伺服器網路地址結構體
	struct sockaddr_in remote_addr; //客戶端網路地址結構體
	int sin_size;
	char buf[BUFSIZ];  //資料傳送的緩衝區
	memset(&my_addr,0,sizeof(my_addr)); //資料初始化--清零
	my_addr.sin_family=AF_INET; //設定為IP通訊
	my_addr.sin_addr.s_addr=INADDR_ANY;//伺服器IP地址--允許連線到所有本地地址上
	my_addr.sin_port=htons(8000); //伺服器埠號
	
	/*建立伺服器端套接字--IPv4協議，面向連線通訊，TCP協議*/
	if((server_sockfd=socket(PF_INET,SOCK_STREAM,0))<0)
	{  
		perror("socket error");
		return 1;
	}


	/*將套接字繫結到伺服器的網路地址上*/
	if(bind(server_sockfd,(struct sockaddr *)&my_addr,sizeof(struct sockaddr))<0)
	{
		perror("bind error");
		return 1;
	}
	
	/*監聽連線請求--監聽佇列長度為5*/
	if(listen(server_sockfd,5)<0)
	{
		perror("listen error");
		return 1;
	};
	
	sin_size=sizeof(struct sockaddr_in);
	
	/*等待客戶端連線請求到達*/
	if((client_sockfd=accept(server_sockfd,(struct sockaddr *)&remote_addr,&sin_size))<0)
	{
		perror("accept error");
		return 1;
	}
	printf("accept client %s/n",inet_ntoa(remote_addr.sin_addr));
	len=send(client_sockfd,"Welcome to my server/n",21,0);//傳送歡迎資訊
	
	/*接收客戶端的資料並將其傳送給客戶端--recv返回接收到的位元組數，send返回傳送的位元組數*/
	while((len=recv(client_sockfd,buf,BUFSIZ,0))>0))
	{
		buf[len]='/0';
		printf("%s/n",buf);
		if(send(client_sockfd,buf,len,0)<0)
		{
			perror("write error");
			return 1;
		}
	}


	/*關閉套接字*/
	close(client_sockfd);
	close(server_sockfd);
    
	return 0;
}
 

#include <stdio.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>

int main(int argc, char *argv[])
{
	int client_sockfd;
	int len;
	struct sockaddr_in remote_addr; //伺服器端網路地址結構體
	char buf[BUFSIZ];  //資料傳送的緩衝區
	memset(&remote_addr,0,sizeof(remote_addr)); //資料初始化--清零
	remote_addr.sin_family=AF_INET; //設定為IP通訊
	remote_addr.sin_addr.s_addr=inet_addr("127.0.0.1");//伺服器IP地址
	remote_addr.sin_port=htons(8000); //伺服器埠號
	
	/*建立客戶端套接字--IPv4協議，面向連線通訊，TCP協議*/
	if((client_sockfd=socket(PF_INET,SOCK_STREAM,0))<0)
	{
		perror("socket error");
		return 1;
	}
	
	/*將套接字繫結到伺服器的網路地址上*/
	if(connect(client_sockfd,(struct sockaddr *)&remote_addr,sizeof(struct sockaddr))<0)
	{
		perror("connect error");
		return 1;
	}
	printf("connected to server/n");
	len=recv(client_sockfd,buf,BUFSIZ,0);//接收伺服器端資訊
        buf[len]='/0';
	printf("%s",buf); //列印伺服器端資訊
	
	/*迴圈的傳送接收資訊並列印接收資訊（可以按需傳送）--recv返回接收到的位元組數，send返回傳送的位元組數*/
	while(1)
	{
		printf("Enter string to send:");
		scanf("%s",buf);
		if(!strcmp(buf,"quit")
			break;
		len=send(client_sockfd,buf,strlen(buf),0);
		len=recv(client_sockfd,buf,BUFSIZ,0);
		buf[len]='/0';
		printf("received:%s/n",buf);
	}
	
	/*關閉套接字*/
	close(client_sockfd);
    
	return 0;
}
 

基於UDP實現

server程式碼

#include <stdio.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>

int main(int argc, char *argv[])
{
	int server_sockfd;
	int len;
	struct sockaddr_in my_addr;   //伺服器網路地址結構體
        struct sockaddr_in remote_addr; //客戶端網路地址結構體
	int sin_size;
	char buf[BUFSIZ];  //資料傳送的緩衝區
	memset(&my_addr,0,sizeof(my_addr)); //資料初始化--清零
	my_addr.sin_family=AF_INET; //設定為IP通訊
	my_addr.sin_addr.s_addr=INADDR_ANY;//伺服器IP地址--允許連線到所有本地地址上
	my_addr.sin_port=htons(8000); //伺服器埠號
	
	/*建立伺服器端套接字--IPv4協議，面向無連線通訊，UDP協議*/
	if((server_sockfd=socket(PF_INET,SOCK_DGRAM,0))<0)
	{  
		perror("socket error");
		return 1;
	}
	
	/*將套接字繫結到伺服器的網路地址上*/
	if (bind(server_sockfd,(struct sockaddr *)&my_addr,sizeof(struct sockaddr))<0)
	{
		perror("bind error");
		return 1;
	}
	sin_size=sizeof(struct sockaddr_in);
	printf("waiting for a packet.../n");
	
	/*接收客戶端的資料並將其傳送給客戶端--recvfrom是無連線的*/
	if((len=recvfrom(server_sockfd,buf,BUFSIZ,0,(struct sockaddr *)&remote_addr,&sin_size))<0)
	{
		perror("recvfrom error"); 
		return 1;
	}
	printf("received packet from %s:/n",inet_ntoa(remote_addr.sin_addr));
	buf[len]='/0';
	printf("contents: %s/n",buf);

	/*關閉套接字*/
	close(server_sockfd);

        return 0;
}
 

#include <stdio.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>

int main(int argc, char *argv[])
{
	int client_sockfd;
	int len;
	struct sockaddr_in remote_addr; //伺服器端網路地址結構體
	int sin_size;
	char buf[BUFSIZ];  //資料傳送的緩衝區
	memset(&remote_addr,0,sizeof(remote_addr)); //資料初始化--清零
	remote_addr.sin_family=AF_INET; //設定為IP通訊
	remote_addr.sin_addr.s_addr=inet_addr("127.0.0.1");//伺服器IP地址
	remote_addr.sin_port=htons(8000); //伺服器埠號

        /*建立客戶端套接字--IPv4協議，面向無連線通訊，UDP協議*/
	if((client_sockfd=socket(PF_INET,SOCK_DGRAM,0))<0)
	{  
		perror("socket error");
		return 1;
	}
	strcpy(buf,"This is a test message"); // 傳送的內容
	printf("sending: '%s'/n",buf); 
	sin_size=sizeof(struct sockaddr_in);
	
	/*向伺服器傳送資料包*/
	if((len=sendto(client_sockfd,buf,strlen(buf),0,(struct sockaddr *)&remote_addr,sizeof(struct sockaddr)))<0)
	{
		perror("recvfrom"); 
		return 1;
	}

	/*關閉套接字*/
	close(client_sockfd);

	return 0;
}