UDP的傳輸功能

最近看了幾個聊天室和發送文件的代碼，覺得比較有意思。於是我就想自己也試着弄一個這樣的程序出來，下面開始介紹我亂搞的這個程序。界面是模仿飛鴿的界面，但功能比那個菜多了，比山寨還山寨。只有傳消息和傳文件的功能。

首先定義兩個結構體：

//這個是用戶信息結構體，有主機名和IP地址。呆會要添加到列表狂裏

typedef struct userInfo{
 char myhost[256];
 char ip[256];
}USERINFO,*PUSERINFO;

 

//這個是包含窗口句柄和套接字句柄的結構體。在向進程傳遞指針時用到這個

struct RECVPARAM{
 HWND hWnd;
 SOCKET sock;
};

 

BOOL CMsgDlg::OnInitDialog() { WSADATA wsadata; WSAStartup(MAKEWORD(2,2),&wsadata); /*下面部分是控件的初始化工作*/ //初始化列表控件 m_list.SetExtendedStyle(LVS_EX_FULLROWSELECT|LVS_EX_GRIDLINES|LVS_EX_HEADERDRAGDROP); m_list.InsertColumn(0,"IP地址",LVCFMT_LEFT,100); m_list.InsertColumn(1,"羣組",LVCFMT_LEFT,100); m_list.InsertColumn(2,"主機名",LVCFMT_LEFT,100); //用戶登陸後執行的操作 //獲得主機名 PUSERINFO puser=(PUSERINFO)::GlobalAlloc(GPTR,sizeof(userInfo));//申請一個指針對象 gethostname(puser->myhost,256); //獲得IP地址 hostent*phost=gethostbyname(puser->myhost); char *p=phost->h_addr_list[0]; in_addr sin; memcpy(&sin.S_un.S_addr,p,phost->h_length); strcpy(puser->ip,inet_ntoa(sin)); AddMyInfo(puser);//先添加自己的信息 PostAndRecvInfo(puser);//再發送給所有其他用戶自己的信息 GlobalFree(p);//記得要釋放，不然結束程序時會提示出錯 //顯示在線人數 CString struser; struser.Format("在線%d人",user); SetDlgItemText(IDC_EDIT1,struser); /*發送文件部分的操作,專門建立一個套接字在5000端口上收發文件*/ m_socket=socket(AF_INET,SOCK_DGRAM,0); if(INVALID_SOCKET==m_socket) { MessageBox("套接字創建失敗！"); return FALSE; } SOCKADDR_IN addrSock; addrSock.sin_family=AF_INET; addrSock.sin_port=htons(5000); addrSock.sin_addr.S_un.S_addr=htonl(INADDR_ANY); //綁定套接字 BOOL resue; setsockopt(m_socket,SOL_SOCKET,SO_REUSEADDR,(char*)&resue,sizeof(BOOL)); int retval; retval=bind(m_socket, (SOCKADDR*)&addrSock, sizeof(SOCKADDR)); if(SOCKET_ERROR==retval) { closesocket(m_socket); MessageBox("綁定失敗了!"); return FALSE; } //套接字設置成阻塞的 u_long ul=0; ioctlsocket(m_socket,FIONBIO,(u_long*)&ul); //產生一個用於接收數據的線程 struct RECVPARAM *pRecvParam=new RECVPARAM; pRecvParam->sock=m_socket; pRecvParam->hWnd=m_hWnd; HANDLE hThread=CreateThread(NULL, 0, RecvProc, (LPVOID)pRecvParam, 0, NULL); CloseHandle(hThread); return TRUE; // return TRUE unless you set the focus to a control } void CMsgDlg::AddMyInfo(userInfo * p) { //現在自己的界面上添加自己的信息 user = m_list.InsertItem(0, "1");//插入一行 m_list.SetItemText(user,0,p->ip); m_list.SetItemText(user,2,p->myhost);//對應行的某列加數據 user++;//在線人數 } void CMsgDlg::PostAndRecvInfo(userInfo *p) { //先發送廣播給所有用戶 SOCKET broadsocket=socket(AF_INET,SOCK_DGRAM,0); BOOL bBroad=TRUE; setsockopt(broadsocket,SOL_SOCKET,SO_BROADCAST,(char*)&bBroad,sizeof(BOOL)); //設置廣播地址 SOCKADDR_IN m_cast; m_cast.sin_addr.S_un.S_addr=INADDR_BROADCAST; m_cast.sin_family=AF_INET; m_cast.sin_port=htons(6000); //發送廣播 char *sendbuf=new char[256]; CString use; int lenhost=strlen(p->myhost); use+=char(lenhost);//第一位爲主機名長度 use+=p->myhost;//後面幾位是主機名 int lenip=strlen(p->ip); use+=char(lenip);//下一位是ip地址的長度 use+=p->ip;//最後是ip地址 int len=use.GetLength(); sendbuf=use.GetBuffer(len); sendbuf[len]='/0'; if(sendto(broadsocket,sendbuf,strlen(sendbuf),0,(SOCKADDR*)&m_cast,sizeof(m_cast))==SOCKET_ERROR) { CString str; str.Format("%d",WSAGetLastError()); MessageBox(str); MessageBox("發送廣播數據失敗!"); return; } //廣播發送完，開始準備讀取 s=socket(AF_INET,SOCK_DGRAM,0); SOCKADDR_IN m_addr; m_addr.sin_addr.S_un.S_addr=INADDR_ANY; m_addr.sin_family=AF_INET; m_addr.sin_port=htons(6000); BOOL reuse=TRUE; setsockopt(s,SOL_SOCKET,SO_REUSEADDR,(char*)&reuse,sizeof(reuse)); if(bind(s,(SOCKADDR*)&m_addr,sizeof(SOCKADDR))==SOCKET_ERROR) { //CString str; //str.Format("%d",WSAGetLastError()); //MessageBox(str); MessageBox("綁定失敗!"); return ; } if(WSAAsyncSelect(s,m_hWnd,WM_SOCK,FD_READ)==SOCKET_ERROR) { MessageBox("註冊網絡讀取事件失敗!"); return ; } } void CMsgDlg::OnSock(WPARAM wparam,LPARAM lparam) { switch(LOWORD(lparam)) { case FD_READ: static int flag=0; WSABUF wsabuf; wsabuf.buf=new char[200]; wsabuf.len=200; DWORD dwRead; DWORD dwFlag=0; SOCKADDR_IN addrFrom; int len=sizeof(SOCKADDR); CString strTemp; CString str; if(SOCKET_ERROR==WSARecvFrom(s,&wsabuf,1,&dwRead,&dwFlag, (SOCKADDR*)&addrFrom,&len,NULL,NULL)) { //MessageBox("接收數據失敗!"); return ; } //發送過來的是廣播用戶信息 if(wsabuf.buf[0]>0&&wsabuf.buf[0]<65) { flag++; if(flag%2==0) { char *host; int hostlen=(int)wsabuf.buf[0]; host=(char*)malloc(hostlen+1); for(int i=0;i<hostlen;i++) host[i]=wsabuf.buf[i+1]; host[hostlen]='/0'; char *ip; int iplen=(int)wsabuf.buf[hostlen+1]; ip=(char*)malloc(iplen+1); for(i=0;i<iplen;i++) ip[i]=wsabuf.buf[hostlen+2+i]; ip[iplen]='/0'; int row=m_list.InsertItem(user,"12"); m_list.SetItemText(row,0,ip); m_list.SetItemText(row,2,host); user++; //SetTimer(1,1000,NULL); } } //發送來的是聊天消息 if(wsabuf.buf[0]=='M') { CString strrecv; int len=(int)wsabuf.buf[1]; char *recv=new char[len+1]; for(int i=0;i<len;i++) recv[i]=wsabuf.buf[i+2]; recv[len]='/0'; strrecv.Format("用戶%s發來消息:%s",inet_ntoa(addrFrom.sin_addr),recv); SetDlgItemText(IDC_SEND,strrecv); } break; } } //定時器，每隔一段時間刷新一下在線人數 void CMsgDlg::OnTimer(UINT nIDEvent) { // TODO: Add your message handler code here and/or call default CString struser; struser.Format("在線%d人",user); SetDlgItemText(IDC_EDIT1,struser); CDialog::OnTimer(nIDEvent); } //下面是發送消息的函數 void CMsgDlg::OnBtnsend() { // TODO: Add your control notification handler code here //獲得選中行的某一列內容(這裏爲第一列) CString strsend; int nItem = m_list.GetItemCount(); for (int i=0; i<nItem; i++) { if (m_list.GetItemState(i, LVIS_SELECTED) == LVIS_SELECTED) // 該行選中的話 { sItem1 = m_list.GetItemText(i, 0);// i爲該行索引, 0表示第1列 break; } } GetDlgItemText(IDC_SEND,strsend); char *sendbuf=new char[256]; sendbuf[0]='M'; int len=strsend.GetLength(); sendbuf[1]=char(len); strcpy(&sendbuf[2],strsend.GetBuffer(len)); SOCKADDR_IN m_addrto; m_addrto.sin_addr.S_un.S_addr=inet_addr(sItem1); m_addrto.sin_family=AF_INET; m_addrto.sin_port=htons(6000); //套接字s在前面已經創建並綁定過了，這裏直接發送就可以了 int ret=sendto(s,sendbuf,strlen(sendbuf),0,(SOCKADDR*)&m_addrto,sizeof(SOCKADDR)); if(ret==SOCKET_ERROR) { MessageBox("發送數據失敗!"); return; } SetDlgItemText(IDC_SEND,""); } /*下面開始文件傳輸部分的操作*/ //這個是嚮導裏添加的一個右鍵的命令相應 void CMsgDlg::OnRclickList1(NMHDR* pNMHDR, LRESULT* pResult) { // TODO: Add your control notification handler code here NM_LISTVIEW* pNMListView = (NM_LISTVIEW*)pNMHDR; //爲-1時就表示右擊了列表框上的某一行 if(pNMListView->iItem != -1) { DWORD dwPos = GetMessagePos(); CPoint point( LOWORD(dwPos), HIWORD(dwPos) ); CMenu menu; VERIFY( menu.LoadMenu( IDR_MENU1 ) ); CMenu* popup = menu.GetSubMenu(0); ASSERT( popup != NULL ); popup->TrackPopupMenu(TPM_LEFTALIGN | TPM_RIGHTBUTTON, point.x, point.y, this ); } *pResult = 0; } //發送文件的函數，這裏發送文件和接收文件用的都是網上現成的代碼，不是自己的成果 void CMsgDlg::OnSendfile() { // TODO: Add your command handler code here CFileDialog dlg(TRUE); if(IDOK==dlg.DoModal()) { m_filePath=dlg.GetPathName(); if (m_posting) //bool isPosting 表示程序是否正在發送文件 { MessageBox("數據發送中，請稍候再試。"); return; } WIN32_FIND_DATA FindFileData; if (INVALID_HANDLE_VALUE == FindFirstFile(m_filePath, &FindFileData)) { MessageBox("文件路徑錯誤或文件不存在!/n請重新指定文件路徑。"); return; } SOCKADDR_IN addrTo; addrTo.sin_family=AF_INET; addrTo.sin_port=htons(5000); addrTo.sin_addr.S_un.S_addr=inet_addr(sItem1); //構建文件信息數據塊 char sendBuf[274]; int i; //消息頭 sendBuf[0] = 'H'; //文件名 for (i = 1; i <= 256 && FindFileData.cFileName[i-1] != '/0'; i++) sendBuf[i] = FindFileData.cFileName[i-1]; sendBuf[i] = '/0'; //文件大小 _itoa(FindFileData.nFileSizeLow, &sendBuf[257], 10); sendBuf[273] = '/0'; //發送數據塊 sendto(s, sendBuf, 274, 0, (SOCKADDR*)&addrTo, sizeof(SOCKADDR)); //打開文件，等待讀取 if (!(m_file = fopen(m_filePath, "rb"))) { MessageBox("讀取文件失敗！"); } m_nSend=0;//文件塊數 m_nFileSize_s = FindFileData.nFileSizeLow; //文件大小 // m_progress.SetRange(0, m_nFileSize_s/256+1);//設置發送進度條 m_posting = true; //標明發送正進行 MessageBox("發送文件成功"); } } DWORD WINAPI CMsgDlg::RecvProc(LPVOID lpParameter) { SOCKET sock=((RECVPARAM*)lpParameter)->sock; HWND hWnd=((RECVPARAM*)lpParameter)->hWnd; delete lpParameter; SOCKADDR_IN addrFrom; addrFrom.sin_port=htons(5000); addrFrom.sin_family=AF_INET; addrFrom.sin_addr.S_un.S_addr=INADDR_ANY; int len=sizeof(SOCKADDR); char recvBuf[274]; //256+17字節的接受緩衝數組 char fileName[256]; //256字節的文件名存儲區 int retval, i; FILE* file = NULL; while(1) { //接收UDP數據 retval=recvfrom(sock,recvBuf,274,0, (SOCKADDR*)&addrFrom,&len); if(SOCKET_ERROR==retval) break; //收到消息頭爲'R',即對方同意讓你繼續發送數據 if (recvBuf[0] == 'R') { char wParam = 'R'; ::PostMessage(hWnd, WM_READY_TO_RECEIVE, (WPARAM)&wParam, 0); } //收到消息頭爲'D',即對方拒絕讓你繼續發送數據 else if (recvBuf[0] == 'D') { char wParam = 'D'; ::PostMessage(hWnd, WM_READY_TO_RECEIVE, (WPARAM)&wParam, 0); } //收到消息頭爲'H',即對方申請給你發送信息,並送來文件的信息 else if (recvBuf[0] == 'H') { //從收到的數據中提取文件名信息 for (i = 1; i <= 256 && recvBuf[i] != '/0'; i++) fileName[i-1] = recvBuf[i]; //recvBuf[1]到recvBuf[256]爲文件名 fileName[i-1] = '/0'; //從收到的數據中提取文件大小信息 CString recvMsg; nFileSize = atoi(&recvBuf[257]); //recvBuf[257]開始是文件大小信息,把字符串變整數 recvMsg.Format("收到來自於(%s)的文件：%s/n文件大小：%i字節/n是否接收？", inet_ntoa(addrFrom.sin_addr), fileName, nFileSize); //用消息框提示用戶有人要發送文件 if (IDOK == AfxMessageBox(recvMsg, MB_OKCANCEL)) { //若用戶同意接收，提供一個文件保存對話框用於設定保存的路徑 CFileDialog saveDlg(false, NULL, fileName); if (IDOK == saveDlg.DoModal()) { //創建一個文件用於複製接收的文件數據 if (!(file = fopen(saveDlg.GetPathName(), "wb"))) { AfxMessageBox("創建本地文件失敗！"); continue; } char wParam = 'H'; ::PostMessage(hWnd, WM_READY_TO_RECEIVE, (WPARAM)&wParam, (LPARAM)&addrFrom); } else { char wParam = 'C'; ::PostMessage(hWnd, WM_READY_TO_RECEIVE, (WPARAM)&wParam, (LPARAM)&addrFrom); } } else //用戶拒絕接收 { char wParam = 'C'; ::PostMessage(hWnd, WM_READY_TO_RECEIVE, (WPARAM)&wParam, (LPARAM)&addrFrom); } } //收到的消息頭爲'F',即對方發來的是文件數據 else if (recvBuf[0] == 'F') { //將文件數據寫入本地文件中 fwrite(&recvBuf[18], 1, 256, file); //recvBuf[18]開始是文件的數據塊了 char wParam = 'F'; ::PostMessage(hWnd, WM_READY_TO_RECEIVE, (WPARAM)&wParam, (LPARAM)&addrFrom); } //收到的消息頭爲'E',即對方發來最後一個數據塊 else if (recvBuf[0] == 'E') { //獲取數據塊的大小 int bufSize = atoi(&recvBuf[1]); //將數據塊寫入本地文件，並關閉文件 fwrite(&recvBuf[0x12], 1, bufSize, file); fclose(file); char wParam = 'E'; ::PostMessage(hWnd, WM_READY_TO_RECEIVE, (WPARAM)&wParam, (LPARAM)&addrFrom); } else AfxMessageBox("傳送數據過程中出現錯誤！"); } return 0; } void CMsgDlg::OnReadyToRecv(WPARAM wParam,LPARAM lParam) { char sendBuf[0x112]; SOCKADDR_IN addrTo; addrTo.sin_family=AF_INET; addrTo.sin_port=htons(5000); addrTo.sin_addr.S_un.S_addr=inet_addr(sItem1); int nRead; CString str; switch (*(char*)wParam) { //對方拒絕接收文件，關閉已打開的文件 case 'D': MessageBox("對方拒絕接受你發送的文件！"); fclose(m_file); m_posting = false; break; //對方同意接收文件 case 'R': nRead = fread(&sendBuf[18], 1, 256, m_file); //讀取的文件小於256字節，則讀到文件尾 if (nRead < 0x100) { sendBuf[0] = 'E'; _itoa(nRead, &sendBuf[1], 10); sendto(m_socket, sendBuf, nRead+0x12, 0, (SOCKADDR*)&addrTo, sizeof(SOCKADDR)); fclose(m_file); m_posting = false; MessageBox("發送完畢！"); } //讀到文件等於256字節，則文件還未讀完 else { sendBuf[0] = 'F'; sendto(m_socket, sendBuf, 0x112, 0, (SOCKADDR*)&addrTo, sizeof(SOCKADDR)); m_nSend++; m_send.Format("發送進度：(%.1f%%)", (float)m_nSend/(m_nFileSize_s/0x100+1)*100); } break; //同意接收對方文件 case 'H': MessageBox("同意接收了"); m_nRecv = 0; case 'F': sendto(m_socket, "R", 2, 0, (SOCKADDR*)&addrTo, sizeof(SOCKADDR)); m_nRecv++; //str.Format("%d",m_nRecv); //MessageBox(str); m_recv.Format("接收進度：(%.1f%%)", (float)m_nRecv/(nFileSize/0x100+1)*100); break; //接受完畢，提示用戶 case 'E': MessageBox("接收完畢！"); break; //拒絕接收，通知對方 case 'C': sendto(m_socket, "D", 2, 0, (SOCKADDR*)&addrTo, sizeof(SOCKADDR)); break; } }

代碼很麻煩也很亂，但基本功能是實現了。還剩下用戶正常退出或異常退出時處理沒有寫，這個目前還沒想到該怎麼弄纔好，先留着以後學了別的知識或許就能輕易解決了。

不怕自己笨，就怕自己不努力。