C語言Socket編程（計算機網絡作業）

最近我計算機網絡課程要做作業了，沒辦法跟着老師一步一步的寫C語言的代碼，使用的計算就是Socket通信發送消息；代碼實現的功能很簡單，客戶端向服務器端發送消息，服務器端接收客戶端發來的消息，並且輸出顯示到屏幕上；代碼中的每個socket方法的返回值都應該做判斷出錯情況，例如：創建SOCKET的創建的時候s句柄（或對象），需要判斷返回值是否為INVALID_SOCKET，以及socket的所有操作的返回值是否為SOCKET_ERROR，以保證程序的穩定性，這里只是測試代碼。

老師是錄制的視頻講解的，其他其中使用的到的方法要求我們自己去查相關的MSDN資料庫，如果有不懂的，就自己上網查。

服務器端接收代碼：

#include <Winsock2.h>
#pragma comment(lib,"Ws2_32.lib")
#include <stdio.h>
#include <memory.h>

void main()
{
WSAData wsd;
WSAStartup(MAKEWORD(2,0),&wsd);

SOCKET s =NULL;
s=socket(AF_INET,SOCK_STREAM,IPPROTO_TCP);
struct sockaddr_in ch;
memset(&ch,0,sizeof(ch));
ch.sin_family=AF_INET;
ch.sin_addr.s_addr=INADDR_ANY;
ch.sin_port=htons(1041);
int b=bind(s,(struct sockaddr *) &ch,sizeof(ch));
#define QUEUE_SIZE 5
int l=listen(s,QUEUE_SIZE);
printf("正在監聽本機的1041端口！\n");
SOCKET sc=accept(s,0,0);
printf("客戶端已經連接到本機的1041端口！\n");
#define BUF_SIZE 4096
int receByt=0;
while(1)
{
char buf[BUF_SIZE];
receByt=recv(sc,buf,BUF_SIZE,0);
buf[receByt]='\0';
if(receByt>0)
{
printf("接收的消息是：%s\n",buf);
}
else
{
printf("接收消息結束！");
break;
}

}
int ic=closesocket(sc);
int is=closesocket(s);

}

======================================================
客戶端發送的代碼：

#include <Winsock2.h>
#pragma comment(lib,"Ws2_32.lib")
#include <stdio.h>
#include <memory.h>
#include <string.h>

void main()
{
WSAData wsd;
WSAStartup(MAKEWORD(2,0),&wsd);

SOCKET s =NULL;
s=socket(AF_INET,SOCK_STREAM,IPPROTO_TCP);
struct sockaddr_in ch;
memset(&ch,0,sizeof(ch));
ch.sin_family=AF_INET;
ch.sin_addr.s_addr=inet_addr("127.0.0.1");
ch.sin_port=htons(1041);

int c=connect(s,(struct sockaddr *) &ch,sizeof(ch));
printf("已經連接到服務器的1041端口！現在可以向服務器發送消息了！\n");
#define BUF_SIZE 4096
char info[1024],buf[BUF_SIZE];

while(1)
{
gets(info);
if(info[0]=='\0')
break;
strcpy(buf,info);
int nsend=send(s,buf,strlen(buf),0);

}
int ic=closesocket(s);
}

========================================================================

程序代碼經過了優化，並且整合多線程，把接收和發送放到同一個文件中，使用參數模式調用發送和接收模塊。增加了創建SOCKET的創建的時候s句柄（或對象）判斷返回值是否為INVALID_SOCKET，以及socket的bind操作的返回值是否為SOCKET_ERROR，其他socket的操作應該也判斷SOCKET_ERROR，以保證程序的穩定性，這里只是測試代碼就不去寫這么多了，剩下的就由你個人發揮。

#include <Winsock2.h>
#pragma comment(lib,"Ws2_32.lib")
#include <stdio.h>
#include <memory.h>
#include <string.h>
#include <pthread.h>


void Receive();
void Send();
void creatThread();

SOCKET s =NULL;
pthread_t t[1000];
int threadCount=0;

void main(int argc,char* argv[])
{
    printf("本程序制作人學號：713901040041\n");
    printf("程序說明：服務器端和客戶端為同一個程序，請使用不同的參數運行.\n"); printf("接收程序請使用 r參數；發送程序請使用 s參數。\n"); //printf("len : %d\n", argc); //printf("count %d\n",argc); //printf("value: %s\n",argv[1]); //printf("%d",argv[1][0]=='r'); if(argc<=1) { printf("please input program arguments ...\n"); exit(0); } if(argc>1 && argv[1][0]=='r') { printf("run receive ...\n"); Receive(); } if(argc>1 && argv[1][0]=='s') { printf("run send ...\n"); Send(); } } void* receiveWork(void * args) { SOCKET sc=accept(s,0,0); if(sc==INVALID_SOCKET) { printf("sc Error"); } creatThread(); printf("----------客戶端已經連接到本機的%d線程連接！\n",threadCount-2); #define BUF_SIZE 4096 int receByt=0; while(1) { char buf[BUF_SIZE]; receByt=recv(sc,buf,BUF_SIZE,0); buf[receByt]='\0'; if(receByt>0) { printf("線程接收的消息是：%s\n",buf); } else { printf("客戶端已退出，"); break; } } int ic=closesocket(sc); printf("服務器結束連接！\n"); return NULL; } void creatThread() { pthread_create(&t[threadCount++],NULL,receiveWork,NULL); } void Receive() { WSAData wsd; WSAStartup(MAKEWORD(2,0),&wsd); s=socket(AF_INET,SOCK_STREAM,IPPROTO_TCP); if(s==INVALID_SOCKET) { printf("socket created Error"); } struct sockaddr_in ch; memset(&ch,0,sizeof(ch)); ch.sin_family=AF_INET; ch.sin_addr.s_addr=INADDR_ANY; ch.sin_port=htons(1041); int b=bind(s,(struct sockaddr *) &ch,sizeof(ch)); if(b==SOCKET_ERROR) { printf("bind 失敗，出錯代碼是：%d\n",WSAGetLastError()); exit(0); } #define QUEUE_SIZE 5 int l=listen(s,QUEUE_SIZE); printf("正在監聽本機的1041端口！\n"); creatThread(); for(int i=0;i<1000;i++) { pthread_join(t[i],NULL); } int is=closesocket(s); } void Send() { WSAData wsd; WSAStartup(MAKEWORD(2,0),&wsd); SOCKET s =NULL; s=socket(AF_INET,SOCK_STREAM,IPPROTO_TCP); if(s==INVALID_SOCKET) { printf("socket created Error"); } struct sockaddr_in ch; memset(&ch,0,sizeof(ch)); ch.sin_family=AF_INET; ch.sin_addr.s_addr=inet_addr("127.0.0.1"); ch.sin_port=htons(1041); int c=connect(s,(struct sockaddr *) &ch,sizeof(ch)); printf("已經連接到服務器的1041端口！現在可以向服務器發送消息了！\n"); #define BUF_SIZE 4096 char info[1024],buf[BUF_SIZE]; while(1) { gets(info); if(info[0]=='\0') break; strcpy(buf,info); int nsend=send(s,buf,strlen(buf),0); } int ic=closesocket(s); }