messenger.cpp

局域网聊天工具。VC开发的一个局域网聊天工具。程序有详细注释。供网络编程的人参考和学习

/*************************************************************************
Module: Hello.c
Notices: Copyright (c) hahaLiu 2008
*************************************************************************/

#include <windows.h>
#include <commctrl.h>
#include <stdio.h>
#include "CmnHdr.h"
#include "Resource.h"

#pragma comment(lib, "Ws2_32.lib")
#pragma comment(lib, "Comctl32.lib")

#define WM_TRAYICON WM_USER+1
#define MAINTRAYICON 1
#define ID_LISTVIEW 100

HINSTANCE hInst = NULL;
HWND hWnd = NULL;
HWND hWndList = NULL;
HANDLE hFileFriends = NULL;
SOCKET s;

int Port = 5150;

struct FRIEND {
	TCHAR szName[20];
	TCHAR szIPAddress[20];
};

extern TCHAR szNewIP[20];
int WINAPI AddFriendDlgProc(HWND hwnd, UINT uMsg, WPARAM wParam, LPARAM lParam);
DWORD WINAPI ReceiveThread(PVOID pvParam);
int CALLBACK ConversationDlgProc(HWND hwndDlg, UINT uMsg, WPARAM wParam, LPARAM lParam);
void SendToAllFriends(TCHAR* szBuf, int len);

//////////////////////////////////////////////////////////////////////////////////

DWORD WINAPI AssistThread(PVOID pvParam) { 

	FRIEND frnd;
	frnd = *(FRIEND*) pvParam;
	DialogBoxParam(hInst, MAKEINTRESOURCE(IDD_CONVERSATION), NULL, ConversationDlgProc, (LPARAM) &frnd);
	return 0;
}

/////////////////////////////////////////////////////////////////////////////////

DWORD WINAPI ReceiveThread(PVOID pvParam) { 
	
	TCHAR szBuf[1024];
	SOCKADDR_IN from;
	int ret;
	static onlinemsgcount;
	int fromlen = sizeof(from);
	while(TRUE) {
		ret = recvfrom(s, szBuf, sizeof(szBuf), 0, (SOCKADDR*)&from, &fromlen);
		if(strcmp(szBuf, "beginconversation msg") == 0) {
			// received beginconversation msg, start a conversation dialog,
			//and make it jump in the tray area ..... later
			FRIEND frnd;
			strcpy(frnd.szIPAddress, inet_ntoa(from.sin_addr));
			// get that guy's name
			int nCount = ListView_GetItemCount(hWndList);
			TCHAR sz[20];
			for(int i = 1; i < nCount; i++) {
				ListView_GetItemText(hWndList, i, 2, sz, sizeof(sz));
				if(strcmp(frnd.szIPAddress, sz) == 0) {
					ListView_GetItemText(hWndList, i, 0, sz, sizeof(sz));
					strcpy(frnd.szName, sz);
					i = nCount - 1;
				}
			}
			TCHAR szTemp[40];
			wsprintf(szTemp, "Conversation with %s", frnd.szName);
			HWND hConversationDlg = FindWindow(NULL, szTemp);
			if(hConversationDlg == NULL) {
				//DialogBoxParam(hInst, MAKEINTRESOURCE(IDD_CONVERSATION), NULL, ConversationDlgProc, (LPARAM) &frnd);
				DWORD dwThreadID;
				CreateThread(NULL, 0, AssistThread, &frnd, 0, &dwThreadID);	
			}
		}
		else if(strcmp(szBuf, "online msg") == 0) {
			//if received online msg, put that guy into online state(update itsicon etc.);
			onlinemsgcount++;
			int nCount = ListView_GetItemCount(hWndList);
			TCHAR sz[20];
			for(int i = 1; i < nCount; i++) {
				ListView_GetItemText(hWndList, i, 2, sz, sizeof(sz));
				if(strcmp(inet_ntoa(from.sin_addr), sz) == 0) {
					nCount = i;
				}
			}
			LVITEM li;
			li.mask = LVIF_STATE | LVIF_IMAGE; 
			li.state = 0; 
			li.stateMask = 0; 
			li.iSubItem = 0;
			li.iItem = nCount;
			li.iImage = 0;
			ListView_SetItem(hWndList, &li);
			strcpy(sz, "online msg");
			ListView_SetItemText(hWndList, nCount, 1, "On");
			if(onlinemsgcount == 1)
				SendToAllFriends(sz, sizeof(sz));
		}
		else if(strcmp(szBuf, "offline msg") == 0) {
			//if received offline msg, put that guy into offline state(update itsicon etc.);
			int nCount = ListView_GetItemCount(hWndList);
			TCHAR sz[20];
			for(int i = 1; i < nCount; i++) {
				ListView_GetItemText(hWndList, i, 2, sz, sizeof(sz));
				if(strcmp(inet_ntoa(from.sin_addr), sz) == 0) {
					nCount = i;
				}
			}
			LVITEM li;
			li.mask = LVIF_STATE | LVIF_IMAGE; 
			li.state = 0; 
			li.stateMask = 0; 
			li.iSubItem = 0;
			li.iItem = nCount;
			li.iImage = 1;
			ListView_SetItem(hWndList, &li);
			ListView_SetItemText(hWndList, nCount, 1, "Off");
		}
		else if(strcmp(szBuf, "transmit file msg") == 0) {
			//create a save as dialog, and save the file.
			// will be completed later ......
			MessageBox(NULL, "Not completed yet ......", "Messenger", 0);
		}
		else {
			//if received conversation content, put the content in the corresponding 
			//conversation dialog(if it's already closed by the user, start one;
			// get that guy's name
			int nCount = ListView_GetItemCount(hWndList);
			TCHAR sz[20];
			FRIEND frnd;
			for(int i = 1; i < nCount; i++) {
				ListView_GetItemText(hWndList, i, 2, sz, sizeof(sz));
				if(strcmp(inet_ntoa(from.sin_addr), sz) == 0) {
					ListView_GetItemText(hWndList, i, 0, sz, sizeof(sz));
					i = nCount - 1;
				}
			}
			strcpy(frnd.szName, sz);
			strcpy(frnd.szIPAddress, inet_ntoa(from.sin_addr));
			TCHAR szTemp[40];
			wsprintf(szTemp, "Conversation with %s", sz);
			HWND hConversationDlg = FindWindow(NULL, szTemp);
			if(hConversationDlg == NULL) {
				//DialogBoxParam(hInst, MAKEINTRESOURCE(IDD_CONVERSATION), NULL, ConversationDlgProc, (LPARAM) &frnd);
				DWORD dwThreadID;
				CreateThread(NULL, 0, AssistThread, &frnd, 0, &dwThreadID);	
			}
			hConversationDlg = FindWindow(NULL, szTemp);
			HWND hwndReceive = GetDlgItem(hConversationDlg, IDC_EDIT_DIALOG);
			TCHAR szContent[4 * 1024], szContent2[4 * 1024];
			Edit_GetText(hwndReceive, szContent, sizeof(szContent));
			wsprintf(szContent2, "%s\r\n\r\n%s:\r\n%s", szContent, frnd.szName, szBuf);
			Edit_SetText(hwndReceive, szContent2);
		}
	}
	return 0;
}

////////////////////////////////////////////////////////////////////////////////

void Conversation_OnCommand(HWND hwnd, int id, HWND hwndCtl, UINT codeNotify, FRIEND frnd)
{
	switch(id)
	{
	case IDCANCEL:
		DestroyWindow(hwnd);
		break;
	case IDC_SEND: {
		// get the address to send to 
		SOCKADDR_IN sa;
		TCHAR szBuf[1024];
		int ret;
		sa.sin_family = AF_INET;
		sa.sin_port = htons(Port);
		sa.sin_addr.s_addr = inet_addr(frnd.szIPAddress);
		HWND hwndSend = GetDlgItem(hwnd, IDC_TOBE_SENT);
		Edit_GetText(hwndSend, szBuf, sizeof(szBuf));
		HWND hwndReceive = GetDlgItem(hwnd, IDC_EDIT_DIALOG);
		TCHAR szContent[4 * 1024], szContent2[4 * 1024];
		Edit_GetText(hwndReceive, szContent, sizeof(szContent));
		wsprintf(szContent2, "%s\r\n\r\nMyself:\r\n%s", szContent, szBuf);
		ret = sendto(s, szBuf, sizeof(szBuf), 0, (SOCKADDR*)&sa, sizeof(sa));
		if(ret == SOCKET_ERROR) {
			MessageBox(NULL, "sendto failed!", "Messenger", 0);
		}
		Edit_SetText(hwndReceive, szContent2);

		break; }
	case IDC_SEND_FILE:
		MessageBox(NULL, "Not completed yet...", "Messenger", 0);
		break;
	}

}

///////////////////////////////////////////////////////////////////////////////////

BOOL CALLBACK ConversationDlgProc(HWND hwndDlg, UINT uMsg, WPARAM wParam, LPARAM lParam) {

	static FRIEND frnd;
	switch(uMsg) {
	case WM_INITDIALOG: {
		frnd = *(FRIEND*) lParam;
		TCHAR szCaption[40];
		wsprintf(szCaption, "Conversation with %s", frnd.szName);
		SetWindowText(hwndDlg, szCaption);
		return TRUE; }
	case WM_COMMAND:
		Conversation_OnCommand(hwndDlg, wParam, (HWND)lParam, NULL, frnd);
		return TRUE;
	}
	return FALSE;
}

////////////////////////////////////////////////////////////////////////////////////

void SaveFriendList() {
	
	HANDLE hFile = CreateFile("FriendList", GENERIC_WRITE, 0, NULL, TRUNCATE_EXISTING, 
		FILE_ATTRIBUTE_NORMAL, NULL);
	TCHAR szBuf[40];
	FRIEND frnd;
	DWORD dwBytesWritten;

	if(hFile == INVALID_HANDLE_VALUE)
		MessageBox(NULL, "kjflkdsjflj", "kdjf", 0);
	// get the count of the list, and write it to the first part of the file
	int nCount = ListView_GetItemCount(hWndList);
	wsprintf(szBuf, "%d", nCount);
	WriteFile(hFile, szBuf, sizeof(szBuf), &dwBytesWritten, NULL);

	// iterate through the list, retrieve the info, and write them to the file
	for(int i = 1; i < nCount; i++) {
		ListView_GetItemText(hWndList, i, 0, frnd.szName, sizeof(frnd.szName));
		ListView_GetItemText(hWndList, i, 2, frnd.szIPAddress, sizeof(frnd.szIPAddress));
		WriteFile(hFile, (LPCVOID)& frnd, sizeof(FRIEND), &dwBytesWritten, NULL);
	}
	CloseHandle(hFile);

	return;
}

/////////////////////////////////////////////////////////////////////////////////////

void LoadFriendList() {

	HANDLE hFile;
	hFile = CreateFile("FriendList", GENERIC_READ, 0, NULL, CREATE_NEW, 
		FILE_ATTRIBUTE_NORMAL, NULL);
	if(hFile == INVALID_HANDLE_VALUE) { // critical error
		// the file already exists, open it, and load the data
		hFile = CreateFile("FriendList", GENERIC_ALL, 0, NULL, OPEN_ALWAYS, 
			FILE_ATTRIBUTE_NORMAL, NULL);
		TCHAR szBuf[40];
		int nCount;
		DWORD dwBytesRead;
		FRIEND* pfrnd;
		ReadFile(hFile, szBuf, sizeof(szBuf), &dwBytesRead, NULL);
		nCount = atoi(szBuf);
		for(int i = 1; i < nCount; i++) {
			ReadFile(hFile, szBuf, sizeof(szBuf), &dwBytesRead, NULL);
			pfrnd = (FRIEND*) szBuf;
			// add this friend info to the list
			LVITEM li;
			li.mask = LVIF_STATE | LVIF_IMAGE; 
			li.state = 0; 
			li.stateMask = 0; 
			li.iSubItem = 0;
			li.iItem = i;
			li.iImage = 1;
			ListView_InsertItem(hWndList, &li);
			ListView_SetItemText(hWndList, i, 0, pfrnd->szName);
			ListView_SetItemText(hWndList, i, 1, "Off");
			ListView_SetItemText(hWndList, i, 2, pfrnd->szIPAddress);	
		}
	} else {
		// do nothing if there's no friend list right now
	}
	CloseHandle(hFile);

	return;
}

///////////////////////////////////////////////////////////////////////////////////////

HWND CreateListView (HWND hwndParent) { 
	
	HWND hWndListView;
	RECT rcl; 
	INITCOMMONCONTROLSEX icex;