开通博客赚积分
发布资源赚积分
分类

源码开发语言/平台
当前位置：首页 > 源码/资料 > Internet/网络编程 > P2P编程 > 查看源码
P2P文件传输.cpp
资源名称：p2p.rar　[点击查看]
上传用户：lj0533
上传日期：2022-08-08
资源大小：1392k
文件大小：12k
源码类别：

P2P编程

开发平台：

Visual C++

P2P文件传输.cpp：源码内容
  1. // p2p.cpp : 定义应用程序的入口点。
  2. //
  4. #include "stdafx.h"
  5. #include "p2p.h"
  6. #include "winsock2.h"
  7. #pragma comment(lib,"ws2_32.lib")
  8. #include <Commdlg.h> 
  9. #define MAX_LOADSTRING 100
  10. #include"stdio.h"
  11. #include"windows.h"
  12. // 全局变量:
  14. HINSTANCE hInst; // 当前实例
  15. TCHAR szTitle[MAX_LOADSTRING]; // 标题栏文本
  16. TCHAR szWindowClass[MAX_LOADSTRING]; // 主窗口类名
  18. //// 此代码模块中包含的函数的前向声明:
  19. ATOM MyRegisterClass(HINSTANCE hInstance);
  20. BOOL InitInstance(HINSTANCE, int);
  21. LRESULT CALLBACK WndProc(HWND, UINT, WPARAM, LPARAM);
  22. INT_PTR CALLBACK About(HWND, UINT, WPARAM, LPARAM);
  24. int APIENTRY _tWinMain(HINSTANCE hInstance,
  25.                        HINSTANCE hPrevInstance,
  26.                        LPTSTR    lpCmdLine,
  27.                        int       nCmdShow)
  28. {
  29.     //UNREFERENCED_PARAMETER(hPrevInstance);
  30.     //UNREFERENCED_PARAMETER(lpCmdLine);
  32.     // // TODO: 在此放置代码。
  33.     //MSG msg;
  34.     //HACCEL hAccelTable;
  36.     //// 初始化全局字符串
  37.     //LoadString(hInstance, IDS_APP_TITLE, szTitle, MAX_LOADSTRING);
  38.     //LoadString(hInstance, IDC_P2P, szWindowClass, MAX_LOADSTRING);
  39.     //MyRegisterClass(hInstance);
  41.     //// 执行应用程序初始化:
  42.     //if (!InitInstance (hInstance, nCmdShow))
  43.     //{
  44.     // return FALSE;
  45.     //}
  47.     //hAccelTable = LoadAccelerators(hInstance, MAKEINTRESOURCE(IDC_P2P));
  49.     //// 主消息循环:
  50.     //while (GetMessage(&msg, NULL, 0, 0))
  51.     //{
  52.     // if (!TranslateAccelerator(msg.hwnd, hAccelTable, &msg))
  53.     // {
  54.     // TranslateMessage(&msg);
  55.     // DispatchMessage(&msg);
  56.     // }
  57.     //}
  58.     DialogBox(hInstance,MAKEINTRESOURCE(IDD_DIALOG1),NULL,About);
  60.     //return (int) msg.wParam;
  61. }
  65. ////
  66. ////  函数: MyRegisterClass()
  67. ////
  68. ////  目的: 注册窗口类。
  69. ////
  70. ////  注释:
  71. ////
  72. ////    仅当希望
  73. ////    此代码与添加到 Windows 95 中的“RegisterClassEx”
  74. ////    函数之前的 Win32 系统兼容时，才需要此函数及其用法。调用此函数十分重要，
  75. ////    这样应用程序就可以获得关联的
  76. ////    “格式正确的”小图标。
  77. ////
  78. //ATOM MyRegisterClass(HINSTANCE hInstance)
  79. //{
  80. // WNDCLASSEX wcex;
  81. //
  82. // wcex.cbSize = sizeof(WNDCLASSEX);
  83. //
  84. // wcex.style = CS_HREDRAW | CS_VREDRAW;
  85. // wcex.lpfnWndProc = WndProc;
  86. // wcex.cbClsExtra = 0;
  87. // wcex.cbWndExtra = 0;
  88. // wcex.hInstance = hInstance;
  89. // wcex.hIcon = LoadIcon(hInstance, MAKEINTRESOURCE(IDI_P2P));
  90. // wcex.hCursor = LoadCursor(NULL, IDC_ARROW);
  91. // wcex.hbrBackground = (HBRUSH)(COLOR_WINDOW+1);
  92. // wcex.lpszMenuName = MAKEINTRESOURCE(IDC_P2P);
  93. // wcex.lpszClassName = szWindowClass;
  94. // wcex.hIconSm = LoadIcon(wcex.hInstance, MAKEINTRESOURCE(IDI_SMALL));
  95. //
  96. // return RegisterClassEx(&wcex);
  97. //}
  98. //
  99. ////
  100. ////   函数: InitInstance(HINSTANCE, int)
  101. ////
  102. ////   目的: 保存实例句柄并创建主窗口
  103. ////
  104. ////   注释:
  105. ////
  106. ////        在此函数中，我们在全局变量中保存实例句柄并
  107. ////        创建和显示主程序窗口。
  108. ////
  109. //BOOL InitInstance(HINSTANCE hInstance, int nCmdShow)
  110. //{
  111. //   HWND hWnd;
  112. //
  113. //   hInst = hInstance; // 将实例句柄存储在全局变量中
  114. //
  115. //   hWnd = CreateWindow(szWindowClass, szTitle, WS_OVERLAPPEDWINDOW,
  116. //      CW_USEDEFAULT, 0, CW_USEDEFAULT, 0, NULL, NULL, hInstance, NULL);
  117. //
  118. //   if (!hWnd)
  119. //   {
  120. //      return FALSE;
  121. //   }
  122. //
  123. //   ShowWindow(hWnd, nCmdShow);
  124. //   UpdateWindow(hWnd);
  125. //
  126. //   return TRUE;
  127. //}
  128. //
  129. ////
  130. ////  函数: WndProc(HWND, UINT, WPARAM, LPARAM)
  131. ////
  132. ////  目的: 处理主窗口的消息。
  133. ////
  134. ////  WM_COMMAND - 处理应用程序菜单
  135. ////  WM_PAINT - 绘制主窗口
  136. ////  WM_DESTROY - 发送退出消息并返回
  137. ////
  138. ////
  139. //LRESULT CALLBACK WndProc(HWND hWnd, UINT message, WPARAM wParam, LPARAM lParam)
  140. //{
  141. // int wmId, wmEvent;
  142. // PAINTSTRUCT ps;
  143. // HDC hdc;
  144. //
  145. // switch (message)
  146. // {
  147. // case WM_COMMAND:
  148. // wmId    = LOWORD(wParam);
  149. // wmEvent = HIWORD(wParam);
  150. // // 分析菜单选择:
  151. // switch (wmId)
  152. // {
  153. // case IDM_ABOUT:
  154. // DialogBox(hInst, MAKEINTRESOURCE(IDD_ABOUTBOX), hWnd, About);
  155. // break;
  156. // case IDM_EXIT:
  157. // DestroyWindow(hWnd);
  158. // break;
  159. // default:
  160. // return DefWindowProc(hWnd, message, wParam, lParam);
  161. // }
  162. // break;
  163. // case WM_PAINT:
  164. // hdc = BeginPaint(hWnd, &ps);
  165. // // TODO: 在此添加任意绘图代码...
  166. // EndPaint(hWnd, &ps);
  167. // break;
  168. // case WM_DESTROY:
  169. // PostQuitMessage(0);
  170. // break;
  171. // default:
  172. // return DefWindowProc(hWnd, message, wParam, lParam);
  173. // }
  174. // return 0;
  175. //}
  177. unsigned short nPort;
  178. char szIP[15];
  179. WSADATA wsaData;
  180. char szBufRecv[1000];
  181. TCHAR szFile[1024]; // 保存获取文件名称的缓冲区。 ;
  182. sockaddr_in saSrv;
  183. SOCKET sRecv1;
  184. SOCKET sRecv2;
  185. SOCKET sSend1;
  186. SOCKET sSend2;
  187. char szBuf[1024];
  188. char filename[1024];
  190. // “关于”框的消息处理程序。
  191. INT_PTR CALLBACK About(HWND hDlg, UINT message, WPARAM wParam, LPARAM lParam)
  192. {
  193.     UNREFERENCED_PARAMETER(lParam);
  194.     switch (message)
  195.     {
  196.     case WM_INITDIALOG:
  197.         SetDlgItemText(hDlg,IDC_EDIT_IP,"127.0.0.1");
  198.         //SetDlgItemText(hDlg,IDC_EDIT_PORT,"8888");
  199.         return (INT_PTR)TRUE;
  201.     case WM_COMMAND:
  202.         if(LOWORD(wParam)==IDC_RADIO1)
  203.             SendMessage(GetDlgItem(hDlg,IDC_RADIO1),BM_SETCHECK,BST_CHECKED,0);
  205.         if(LOWORD(wParam)==IDC_RADIO2)
  206.             SendMessage(GetDlgItem(hDlg,IDC_RADIO2),BM_SETCHECK,BST_CHECKED,0);
  208.         if(BST_CHECKED==SendMessage(GetDlgItem(hDlg,IDC_RADIO1),BM_GETCHECK,0,0))
  209.         {
  210.             if (LOWORD(wParam)==IDC_RADIO1)
  211.             {
  213.                 //初始化
  214.                 WSAStartup(MAKEWORD(2,2),&wsaData); //ws2_32.dll 
  215.                 //创建套接字
  216.                 sRecv1=socket(AF_INET,SOCK_DGRAM,IPPROTO_UDP);
  217.                 //bind
  218.                 saSrv.sin_family =AF_INET;
  219.                 saSrv.sin_addr.S_un.S_addr =INADDR_ANY; //inet_addr("127.0.0.1");
  220.                 saSrv.sin_port =htons(7777);
  221.                 bind(sRecv1,(sockaddr*)&saSrv,sizeof(saSrv));
  222.     
  223.                 WSAStartup(MAKEWORD(2,2),&wsaData);
  224.                 //创建（接收）套接字
  225.                 sRecv2=socket(AF_INET,SOCK_DGRAM,IPPROTO_UDP);
  226.                 sockaddr_in saSrv;
  227.                 saSrv.sin_family=AF_INET;
  228.                 saSrv.sin_addr.S_un.S_addr=INADDR_ANY;
  229.                 saSrv.sin_port=htons(8888);
  230.                 bind(sRecv2,(sockaddr*)&saSrv,sizeof(sockaddr_in));
  233.             }
  234.             if (LOWORD(wParam) == IDOK2)
  235.             {
  236.                 memset(filename,0,1024);
  237.                 recv(sRecv1,filename,1024,0);
  238.                 BOOL GetSaveFileName();
  239.                 {
  240.                     OPENFILENAME ofn;      // 公共对话框结构
  241.                     // 初始化选择文件对话框。
  242.                     ZeroMemory(&ofn, sizeof(ofn));
  243.                     ofn.lStructSize = sizeof(ofn);
  244.                     ofn.hwndOwner = NULL;
  245.                     ofn.lpstrFile = filename;
  246.                     ofn.nMaxFile = sizeof(filename);
  247.                     ofn.lpstrFilter = _T("All*.*Text*.TXT");
  248.                     ofn.nFilterIndex = 1;
  249.                     ofn.lpstrFileTitle =filename;
  250.                     ofn.nMaxFileTitle = 0;
  251.                     ofn.lpstrInitialDir = NULL;
  252.                     ofn.Flags = OFN_SHOWHELP | OFN_OVERWRITEPROMPT;
  253.                     // 显示打开选择文件对话框。
  254.                     if ( GetSaveFileName(&ofn) )
  255.                     {
  256.                         //显示选择的文件。
  257.                         SetDlgItemText(hDlg,IDC_EDIT1,filename);
  258.                         return 0;
  259.                     }
  260.                 }
  261.                 shutdown(sRecv1,2);
  262.                 closesocket(sRecv1);
  263.                 WSACleanup();
  264.             }
  265.             if (LOWORD(wParam) == IDOK3)
  266.             {
  267.                 recv(sRecv2,filename,1024,0);
  268.                 FILE *fp; //定义文件类型的指针
  269.                 fp=fopen(filename,"wb");
  270.                 char buf[1024]; 
  271.                 memset(buf,0,1024);
  272.                 int nNumRead =1;
  273.                 while(nNumRead)
  274.                 {
  275.                     nNumRead = recv(sRecv2,buf,1024, 0 );
  276.                     fwrite(buf, 1, nNumRead, fp );
  278.                 }
  279.                 MessageBox(hDlg,"接收完成","",MB_OK);
  280.                 return FALSE;  
  281.                 fclose(fp);
  282.                 shutdown( sRecv2,2);
  283.                 closesocket(sRecv2);
  284.                 WSACleanup();
  285.             }      
  286.         }
  287.         if(BST_CHECKED==SendMessage(GetDlgItem(hDlg,IDC_RADIO2),BM_GETCHECK,0,0))
  288.         {
  289.             if(LOWORD(wParam)==IDOK2)
  290.             { 
  292.                 //初始化
  293.     WSAStartup(MAKEWORD(2,2),&wsaData);
  294.                 //创建套接字
  295.                 sSend1=socket(AF_INET,SOCK_DGRAM,IPPROTO_UDP);
  296.                 if(sSend1==INVALID_SOCKET)
  297.                 {
  298.                     MessageBox(hDlg,"创建套接字失败","",MB_OK);
  299.                     break;
  300.                 }
  301.                 sockaddr_in saDst;
  302.                 saDst.sin_family=AF_INET;
  303.                 WORD nPort;
  304.                 GetDlgItemText(hDlg,IDC_EDIT_IP,szIP,15);
  305.                 //GetDlgItemText(hDlg,IDC_EDIT2,szPort,5);
  306.                 saDst.sin_addr.S_un.S_addr=inet_addr(szIP);
  307.                 saDst.sin_port=htons(7777);  
  308.                 BOOL GetOpenFileName();
  309.                 {
  310.                     OPENFILENAME ofn;      // 公共对话框结构。         
  311.                     // 初始化选择文件对话框。
  312.                     ZeroMemory(&ofn, sizeof(ofn));
  313.                     ofn.lStructSize = sizeof(ofn);
  314.                     ofn.hwndOwner =NULL;
  315.                     ofn.lpstrFile = szFile;
  316.                     ofn.lpstrFile[0] = _T('');
  317.                     ofn.nMaxFile = sizeof(szFile);
  318.                     ofn.lpstrFilter = _T("All*.*Text*.TXT");
  319.                     ofn.nFilterIndex = 1;
  320.                     ofn.lpstrFileTitle = NULL;
  321.                     ofn.nMaxFileTitle = 0;
  322.                     ofn.lpstrInitialDir = NULL;
  323.                     ofn.Flags = OFN_PATHMUSTEXIST | OFN_FILEMUSTEXIST;
  325.                     // 显示打开选择文件对话框。
  326.                     if ( GetOpenFileName(&ofn) )
  327.                     {
  328.                         //显示选择的文件。 
  329.                         SetDlgItemText(hDlg,IDC_EDIT1,szFile);
  330.                         GetDlgItemText(hDlg,IDC_EDIT1,szFile,1024);
  331.                         sendto(sSend1,szFile,strlen(szFile),0,(sockaddr*)&saDst,sizeof(saDst));
  332.                         //shutdown(sSend1, 2);
  333.                         //closesocket(sSend1);
  334.                         //WSACleanup();
  335.                     }
  336.                 }
  337.             }
  338.             if (LOWORD(wParam) == IDOK)
  339.             {
  340.                 WSAStartup(MAKEWORD(2,2),&wsaData);
  341.                 //创建套接字
  342.                 sSend2=socket(AF_INET,SOCK_DGRAM,IPPROTO_UDP);
  343.                 if(sSend2==INVALID_SOCKET)
  344.                 {
  345.                     MessageBox(hDlg,"创建套接字失败","",MB_OK);
  346.                     break;
  347.                 }
  348.                 sockaddr_in saDst;
  349.                 saDst.sin_family=AF_INET;
  350.                 WORD nPort;
  351.                 GetDlgItemText(hDlg,IDC_EDIT_IP,szIP,15);
  352.                 saDst.sin_addr.S_un.S_addr=inet_addr(szIP);
  353.                 saDst.sin_port=htons(8888);  
  354.                 GetDlgItemText(hDlg,IDC_EDIT1,szFile,1024);
  355.                 FILE *fp; //定义文件类型的指针
  356.                 char filename[1024];
  357.                 strcpy(filename,szFile);
  358.                 fp=fopen(filename,"rb");
  359.                 if(fp==NULL)
  360.                 {
  361.                     MessageBox(hDlg,"文件打开失败","",MB_OK);
  362.                     return TRUE;
  363.                 }
  364.                 char buf[1024]; 
  365.                 memset(buf,0,1024);
  366.                 int nNumRead = 0;
  367.                 while( !feof( fp) )
  368.                 {
  369.                     nNumRead = fread( buf, 1, 1024, fp );
  370.                     int nCount=sendto(sSend2,buf,strlen(buf),0,(sockaddr*)&saDst,sizeof(saDst));
  371.                     if(nCount==SOCKET_ERROR)
  372.                     {
  373.                         MessageBox(hDlg,"发送数据失败","",MB_OK);
  374.                         return FALSE;
  375.                     }
  376.                 }
  377.                 fclose(fp);
  378.                 shutdown(sSend2, 2 );
  379.                 closesocket(sSend2);
  380.                 WSACleanup();
  381.                 MessageBox(hDlg,"发送完成","",MB_OK);
  382.                 return FALSE;
  383.             }
  384.         }
  385.         if (LOWORD(wParam) == IDCANCEL)
  386.         {
  388.             EndDialog(hDlg, LOWORD(wParam));
  389.             return (INT_PTR)TRUE;
  390.         }
  391.         break;
  392.     }
  393.     return (INT_PTR)FALSE;
  394. }