Telescope/第二周/tcpSocket/fileOperation.c

#include"fileOperation.h"
long g_fileSize;  //�����ļ���С��ȫ�ֱ�����ʼ��Ϊ��
//long g_fileSize = 11152;
char *g_fileBuf;  //�����ļ�����
char g_recvBuf[1024];  //�����ļ���С
//int main()


bool readFile(struct MsgHeader* pmsg)
{
	FILE* read = fopen(pmsg->fileInfo.fileName, "rb"); //�ļ�ָ�룬rbΪ�����Ʒ�ʽ��
	if (!read)  //�����ȡΪ��
	{
		perror("�ļ���ȡʧ��\n");
		printf("�Ҳ���[%s]�ļ�...\n", pmsg->fileInfo.fileName);
		struct MsgHeader msg = { .msgID = MSG_OPENFILE_FAILD };
		return false;
	}

	//��ȡ�ļ���С����λ�ֽ�
	fseek(read, 0, SEEK_END);  //���ļ�λ��ָ���ƶ������
	g_fileSize = ftell(read);  //ftell��ȡ��ǰ�ļ�λ��ָ��
	fseek(read, 0, SEEK_SET);  //�ƶ�����ͷ
	printf("filesize:%d\n", g_fileSize);

	//�����ڴ�
	g_fileBuf = calloc(g_fileSize, sizeof(char));  //calloc���ڴ�ȫ����ʼ��Ϊ0
	if (!g_fileBuf)
	{
		return false;
	}
	//���ļ������ڴ�����
	fread(g_fileBuf, sizeof(char), g_fileSize, read);

	fclose(read);
	return true;
}

bool sendFile(SOCKET s, struct MsgHeader* pmsg)
{
	readFile(pmsg);  //��ȥ��server�����ٶ��ļ�����
	//���ļ����ƺʹ�С�����ͻ���
	struct MsgHeader msg = { .msgID = MSG_FILESIZE,.fileInfo.fileSize = g_fileSize };
	//fileName = C:\Users\Katherine\Desktop\TyporaHotKey.ahk
	
	char tfname[200] = { 0 }, text[100];  //tfname�ļ���TyporaHotKey��text��׺��.ahk
	_splitpath(pmsg->fileInfo.fileName, NULL, NULL, tfname, text);  //�ָ�·��
	strcat(tfname, text);  //ƴ��ΪTyporaHotKey.ahk������tfname
	strcpy(msg.fileInfo.fileName, tfname);  //������msg.fileInfo.fileName

	//int ret1 = send(s,  //������
	int ret2 = send(s, (char*)&msg, sizeof(struct MsgHeader), 0);  //����С
	
	//�����ļ�����g_fileBuf��Ҫ����Ӧ�ò���
	int ret = send(s, g_fileBuf, msg.fileInfo.fileSize,0);  //ʵ�ʷ��͵��ļ���Сret
	if (ret == SOCKET_ERROR)
	{
		err("sendFile");
		return false;
	}
	printf("���ͳɹ�(%d)Byte\n", ret);

	return true;
}

void downloadFileName(SOCKET serfd)
{
	char fileName[1024] = "��ã����Ǹ��ڵ�̩ɽ~";
	gets_s(fileName, 1023);		//��ȡҪ���ص��ļ�

	struct MsgHeader file = { .msgID = MSG_FILENAME };
	strcpy(file.fileInfo.fileName, fileName);

	send(serfd, (char*)&file, sizeof(struct MsgHeader), 0);

}

bool recvFile(SOCKET s)
{
	//�����ļ���С
	recv(s, g_recvBuf, 1023, 0);
	struct MsgHeader* msg = (struct MsgHeader*)g_recvBuf;
	printf("���յ��ļ���СΪ��(%d)Byte\n", msg->fileInfo.fileSize);
	//strcpy(msg->fileInfo.fileName, pmsg->fileInfo.fileName);
	//�����ļ�ʱ��Ҫ�����ڴ棡��
	if (g_fileBuf == NULL)
	{
		g_fileBuf = calloc(msg->fileInfo.fileSize, sizeof(char));
		if (!g_fileBuf)
		{
			return false;
		}
	}
	int ret = recv(s, g_fileBuf, msg->fileInfo.fileSize, 0);
	//int ret = recv(s, g_fileBuf, 1023, 0);
	if (ret == 0)
	{
		printf("��������������...\n");  
	}
	else if (ret < 0)
	{
		err("recv");
	}
	saveFile(msg);
	return true;
}

bool saveFile(struct MsgHeader* pmsg)
{
	FILE* write = fopen(&(pmsg->fileInfo.fileName), "wb"); //�ļ�ָ�룬wbֻ����д����
	if (!write)  //�����ȡΪ��
	{
		perror("�ļ���ȡʧ��\n");
		return false;
	}

	fwrite(g_fileBuf, sizeof(char), pmsg->fileInfo.fileSize, write);

	fclose(write);
	return true;
}