【文章內(nèi)容簡介】 設(shè)計 10 return SUCCEED。 } } pthread_mutex_unlock(amp。user_table_mutex)。 return FAIL。 } /************************************************************ *函數(shù)名稱 : init_server_socket *功能說明 : 初始化服務(wù)器用于監(jiān)聽的的 socket *正常返回值 : 已經(jīng)初始化的服務(wù)器用于監(jiān)聽的的 socket *函數(shù)執(zhí)行失敗 : 輸出錯誤信息 ,退出程序 *************************************************************/ int init_server_socket() { struct sockaddr_in server_addr。 bzero(amp。server_addr,sizeof(server_addr))。 = AF_INET。 = htons(INADDR_ANY)。 = htons(CHAT_SERVER_PORT)。 int server_socket = socket(AF_INET,SOCK_STREAM,0)。 if( server_socket 0) { printf(Create Socket Failed!)。 exit(1)。 } if( bind(server_socket,(struct sockaddr*)amp。server_addr,sizeof(server_addr))) { printf(Server Bind Port : %d Failed!, CHAT_SERVER_PORT)。 exit(1)。 } if ( listen(server_socket, LENGTH_OF_LISTEN_QUEUE) ) { printf(Server Listen Failed!)。 exit(1)。 《網(wǎng)絡(luò)編程》課程設(shè)計 11 } return server_socket。 } /************************************************************ *函數(shù)名稱 : process_request *功能說明 : 根據(jù)接收到的報文的內(nèi)容 ,進(jìn)行響應(yīng)的服務(wù) . * 服務(wù)類型包括 :注冊 ,登錄 ,獲取已登錄用戶列表 ,向用戶發(fā)送信息 ,退出 ,修改注冊信息 *正常返回值 : 服務(wù)器對用戶發(fā)回的響應(yīng)類型 *函數(shù)執(zhí)行失敗 : 沒有檢測和處理 *參數(shù)說明 : client_socket 是服務(wù)器與用戶通信的 socket * receive_buffer 為服務(wù)器收到的報文的內(nèi)容 *依賴自定義數(shù)據(jù)結(jié)構(gòu) : chat_package *************************************************************/ int process_request(int client_socket, char * receive_buffer) { chat_package send_buffer。 bzero((char*)amp。send_buffer,BUFFER_SIZE)。 char * user_ID = ((chat_package *)receive_buffer)from。 char * password = ((chat_package *)receive_buffer)password。 char * receiver = ((chat_package *)receive_buffer)to。 printf(Request %d from client\n,((chat_package *)receive_buffer)type)。 switch(((chat_package *)receive_buffer)type) { case REGISTER: = user_register(user_ID, password, client_socket)。 break。 case LOGIN: = login(user_ID, password, client_socket)。 break。 case GET_USER_LIST: memcpy(, USER_LIST, OPTION_SIZE)。 = get_active_user_list()。 break。 case TALK_TO: = SUCCEED。 send(client_socket, (chat_package *)amp。send_buffer,BUFFER_SIZE,0)。 《網(wǎng)絡(luò)編程》課程設(shè)計 12 client_socket = look_up_socket(receiver)。 = TRANSFER。 memcpy(, ((chat_package *)receive_buffer)from, MESSAGE_SIZE)。 memcpy(, ((chat_package *)receive_buffer)message, MESSAGE_SIZE)。 break。 case EXIT: deactive_user(client_socket)。 return 。 break。 case CHANGE: = user_change_register(user_ID, password, client_socket)。 } printf(Answer %d to client\n,)。 send(client_socket, (chat_package *)amp。send_buffer,BUFFER_SIZE,0)。 return 。 } /************************************************************ *函數(shù)名稱 : talk_to_client *功能說明 : 對單獨(dú)的一個用戶的各種請求進(jìn)行服務(wù) ,當(dāng)用戶的請求 為 EXIT時 ,結(jié)束本線程 *函數(shù)執(zhí)行失敗 : 通信失敗時 ,顯示錯誤信息 ,結(jié)束本線程 *依賴自定義數(shù)據(jù)結(jié)構(gòu) : chat_package *************************************************************/ void * talk_to_client(void * new_server_socket_to_client) { int new_server_socket = (int)new_server_socket_to_client。 int request = NO_COMMAND。 while(request!=EXIT) { chat_package buffer。 bzero((char*)amp。buffer,BUFFER_SIZE)。 int length = recv(new_server_socket,(char*)amp。buffer,BUFFER_SIZE,0)。 if (length 0) 《網(wǎng)絡(luò)編程》課程設(shè)計 13 { printf(Server Recieve Data Failed!\n)。 close(new_server_socket)。 pthread_exit(NULL)。 } if (length==0) { close(new_server_socket)。 pthread_exit(NULL)。 } request = process_request(new_server_socket, (char*)amp。buffer)。 } close(new_server_socket)。 pthread_exit(NULL)。 } int main(int argc, char **argv) { init_user_table()。 pthread_mutex_init(amp。user_table_mutex, NULL)。 int server_socket = init_server_socket()。 pthread_t child_thread。 pthread_attr_t child_thread_attr。 pthread_attr_init(amp。child_thread_attr)。 pthread_attr_setdetachstate(amp。child_thread_attr,PTHREAD_CREATE_DETACHED)。 while (1) { struct sockaddr_in client_addr。 socklen_t length = sizeof(client_addr)。 int new_server_socket = accept(server_socket,(struct sockaddr*)amp。client_addr,amp。length)。 if ( new_server_socket 0) { printf(Server Accept Failed!\n)。 《網(wǎng)絡(luò)編程》課程設(shè)計 14 break。 } if( pthread_create(amp。child_thread,amp。child_thread_attr,talk_to_client, (void *)new_server_socket) 0 ) printf(pthread_create Failed : %s\n,strerror(errno))。 } close(server_socket)。 pthread_attr_destroy(amp。child_thread_attr)。 pthread_mutex_destroy(amp。user_table_mutex)。 pthread_exit (NULL)。 return 0。 } 客戶端 /************************************************************ * 基于 socket 的聊天客戶機(jī) *設(shè)計思路 : * 客戶機(jī)提出各種請求 ,服務(wù)器根據(jù)不同請求 ,發(fā)送不同的響應(yīng) . *關(guān)鍵問題 : * 客戶機(jī)和 服務(wù)器之間協(xié)議制訂 *************************************************************/ include include // for fork include sys/ // for signal include sys/ // for wait include static char user_ID[BUFFER_SIZE]。 //一個很大的緩沖區(qū) ,實際 ID_SIZE內(nèi)有效 static char password[BUFFER_SIZE]。 // 一個很大的緩沖區(qū) , 實際PASSWORD_SIZEE 內(nèi)有效 static int client_socket。 /************************************************************ *函數(shù)名稱 : connect_to_server *正常返回值 :