#include <windows.h>
#include <iostream>
#include <winsock.h>
using namespace std;
#define NO_FLAGS_SET 0
#define THIS_PORT (u_short) 44967
#define PORT (u_short) 44966
#define DEST_IP_ADDR "127.0.0.1" //Server address
#define FILE_BLOCK_SIZE 409600
#define FILE_BLOCK_NUM 11
#define FILE_NAME_LEN 7
#define DEST_PORT (u_short) 44966
#define CLIENT_NUM 3
char dataBuffer[FILE_BLOCK_SIZE];		// 用于存储接收到的数据
char fileName[100]; //文件名
DWORD ThreadProc(LPVOID clientSock)  
{  
	SOCKET NewConnection = (SOCKET)clientSock;
    char dataBuffer[FILE_BLOCK_SIZE];		// 用于存储接收到的数据
	char fileName[100]; //文件名
	int count;
	FILE *fp = NULL;
	char path[100];//路径 
	while ((count = recv(NewConnection, dataBuffer, sizeof(dataBuffer), 0)) > 0) {	
		if (strlen(dataBuffer) == FILE_NAME_LEN) {		// 发来的是文件名
			strcpy(fileName, dataBuffer);				// 记录文件名
			printf("receving %s\n", fileName);		
		}
		else {
			sprintf(path, "%s%s", "C:\\Users\\大梦一场\\Desktop\\peer4\\",fileName);
			fp = fopen(path, "wb");					// 创建文件
			fwrite(dataBuffer, count, 1, fp);
			printf("receving data %s\n", fileName);
			fclose(fp);
		}
	}
  
    return 0;  
}

int main(void)
{
  
  int numsnt;
  char *getInfo="getInfo";
  WSADATA Data;
  SOCKADDR_IN serverSockAddr;
  SOCKADDR_IN clientSockAddr;
  SOCKET serverSock;
  SOCKET clientSock;
  int addrLen=sizeof(SOCKADDR_IN);
  int status;
  int numrcv;
  char buffer[1024];

  /* initialize the Windows Socket DLL */
  status=WSAStartup(MAKEWORD(1, 1), &Data);
  /*初始化Winsock DLL*/
  if (status != 0)
    cerr << "ERROR: WSAStartup unsuccessful" << endl;

  /* zero the sockaddr_in structure */
  
  memset(&serverSockAddr, 0,sizeof(serverSockAddr));
  
  
  /* specify the port portion of the address */
  serverSockAddr.sin_port=htons(PORT);
  /* specify the address family as Internet */
  serverSockAddr.sin_family=AF_INET;
  /* specify that the address does not matter */
  /*INADDR_ANY 的具体含义是，绑定到0.0.0.0。此时，对所有的地址都将是有效的*/
  serverSockAddr.sin_addr.s_addr=htonl(INADDR_ANY);
  /* create a socket */
  serverSock=socket(AF_INET, SOCK_STREAM, 0);
  if (serverSock == INVALID_SOCKET)
    cerr << "ERROR: socket unsuccessful" << endl;

  /* associate the socket with the address */
  status=bind(serverSock, (LPSOCKADDR) &serverSockAddr,
    sizeof(serverSockAddr));
  if (status == SOCKET_ERROR)
    cerr << "ERROR: bind unsuccessful" << endl;

  /* allow the socket to take connections */
  status=listen(serverSock, 1);
  if (status == SOCKET_ERROR)
    cerr << "ERROR: listen unsuccessful" << endl;	
  
  
  int conn=0;
  while(1)
  {
		int clientAddrLen = sizeof(SOCKADDR);	// 求出地址的长度
		// 接收请求（链接客户端），第二个参数是客户端的地址
		if ((clientSock = accept(serverSock, (LPSOCKADDR)&clientSockAddr, &clientAddrLen)) == INVALID_SOCKET)
		{
			printf("accept failed with error %d\n", WSAGetLastError());
			closesocket(clientSock); // 释放客户端连接	
		}
		// 连接客户端成功，打印客户端的ip地址和端口号
		printf("当前已连接：%s:%d\n", inet_ntoa(clientSockAddr.sin_addr), ntohs(clientSockAddr.sin_port));
		// 开启线程处理客户端连接
		CreateThread(NULL, 0, ThreadProc, (LPVOID)clientSock, 0, NULL);
		conn++;
		
		/* 文件的拼接 */
  		if (conn == CLIENT_NUM) { //  接收完3个进程发送的文件
			Sleep(2000);
			printf("Succeeded in accepting all file blocks. \n");
			int id = 1;			// 文件块编号
			FILE *fpr;			// 文件块读取指针
			FILE *fpw = fopen("C:\\Users\\大梦一场\\Desktop\\DarkSoul.jfif", "wb");
			char fileName[100]; //文件名		
			while (id <= FILE_BLOCK_NUM) {
				sprintf(fileName, "C:\\Users\\大梦一场\\Desktop\\peer4\\part_%02d", id);
				fpr = fopen(fileName, "rb+");
				int count = fread(dataBuffer, 1, FILE_BLOCK_SIZE, fpr);
				fwrite(dataBuffer, count, 1, fpw);
				id++;
				fclose(fpr);
			}
			fclose(fpw);
			Sleep(1000);
			printf("DarkSoul.jfif combining succeeded!\n");
		}
    }
}