www.pudn.com > 多功能聊天程序.rar > SpeakDlg.cpp


// SpeakDlg.cpp : implementation file 
// 
 
#include "stdafx.h" 
#include "Speak.h" 
#include "SpeakDlg.h" 
 
#ifdef _DEBUG 
#define new DEBUG_NEW 
#undef THIS_FILE 
static char THIS_FILE[] = __FILE__; 
#endif 
 
 
///////////////////////////////////////////////////////////////////////////// 
// CAboutDlg dialog used for App About 
 
class CAboutDlg : public CDialog 
{ 
public: 
	CAboutDlg(); 
 
// Dialog Data 
	//{{AFX_DATA(CAboutDlg) 
	enum { IDD = IDD_ABOUTBOX }; 
	//}}AFX_DATA 
 
	// ClassWizard generated virtual function overrides 
	//{{AFX_VIRTUAL(CAboutDlg) 
	protected: 
	virtual void DoDataExchange(CDataExchange* pDX);    // DDX/DDV support 
	//}}AFX_VIRTUAL 
 
// Implementation 
protected: 
	//{{AFX_MSG(CAboutDlg) 
	//}}AFX_MSG 
	DECLARE_MESSAGE_MAP() 
}; 
 
CAboutDlg::CAboutDlg() : CDialog(CAboutDlg::IDD) 
{ 
	//{{AFX_DATA_INIT(CAboutDlg) 
	//}}AFX_DATA_INIT 
} 
 
void CAboutDlg::DoDataExchange(CDataExchange* pDX) 
{ 
	CDialog::DoDataExchange(pDX); 
	//{{AFX_DATA_MAP(CAboutDlg) 
	//}}AFX_DATA_MAP 
} 
 
BEGIN_MESSAGE_MAP(CAboutDlg, CDialog) 
	//{{AFX_MSG_MAP(CAboutDlg) 
		// No message handlers 
	//}}AFX_MSG_MAP 
END_MESSAGE_MAP() 
 
///////////////////////////////////////////////////////////////////////////// 
// CSpeakDlg dialog 
 
CSpeakDlg::CSpeakDlg(CWnd* pParent /*=NULL*/) 
	: CDialog(CSpeakDlg::IDD, pParent) 
{ 
	//{{AFX_DATA_INIT(CSpeakDlg) 
	m_Name = _T(""); 
	//}}AFX_DATA_INIT 
	// Note that LoadIcon does not require a subsequent DestroyIcon in Win32 
	m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME); 
} 
 
void CSpeakDlg::DoDataExchange(CDataExchange* pDX) 
{ 
	CDialog::DoDataExchange(pDX); 
	//{{AFX_DATA_MAP(CSpeakDlg) 
	DDX_Control(pDX, IDC_IPADDRESS, m_IPADDRESS); 
	DDX_Text(pDX, IDC_Name, m_Name); 
	//}}AFX_DATA_MAP 
} 
 
BEGIN_MESSAGE_MAP(CSpeakDlg, CDialog) 
	//{{AFX_MSG_MAP(CSpeakDlg) 
	ON_WM_SYSCOMMAND() 
	ON_WM_PAINT() 
	ON_WM_QUERYDRAGICON() 
	ON_BN_CLICKED(IDC_ConnectBTN, OnConnectBTN) 
	ON_BN_CLICKED(IDC_CUTUPBTN, OnCutupbtn) 
	ON_WM_TIMER() 
	ON_BN_CLICKED(IDC_TestBTN, OnTestBTN) 
	//}}AFX_MSG_MAP 
	ON_MESSAGE(WM_READREADY,OnReadReady) 
END_MESSAGE_MAP() 
 
///////////////////////////////////////////////////////////////////////////// 
// CSpeakDlg message handlers 
 
BOOL CSpeakDlg::OnInitDialog() 
{ 
	CDialog::OnInitDialog(); 
 
	// Add "About..." menu item to system menu. 
 
	// IDM_ABOUTBOX must be in the system command range. 
	ASSERT((IDM_ABOUTBOX & 0xFFF0) == IDM_ABOUTBOX); 
	ASSERT(IDM_ABOUTBOX < 0xF000); 
 
	CMenu* pSysMenu = GetSystemMenu(FALSE); 
	if (pSysMenu != NULL) 
	{ 
		CString strAboutMenu; 
		strAboutMenu.LoadString(IDS_ABOUTBOX); 
		if (!strAboutMenu.IsEmpty()) 
		{ 
			pSysMenu->AppendMenu(MF_SEPARATOR); 
			pSysMenu->AppendMenu(MF_STRING, IDM_ABOUTBOX, strAboutMenu); 
		} 
	} 
 
	// Set the icon for this dialog.  The framework does this automatically 
	//  when the application's main window is not a dialog 
	SetIcon(m_hIcon, TRUE);			// Set big icon 
	SetIcon(m_hIcon, FALSE);		// Set small icon 
	CFile MyFile; 
	MyFile.Open("voice.txt",CFile::modeRead); 
    int filelong=0; 
	filelong=MyFile.GetLength(); 
	 
	char pbuf[500]; 
	MyFile.Read( pbuf, sizeof( pbuf ) );  
	pbuf[filelong]='\0'; 
 
	MyFile.Close();  
	strIP=""; 
	m_Name=""; 
	int i=2; 
	char sign=pbuf[0]; 
	while(pbuf[i]!='#') 
	{ 
		strIP+=pbuf[i]; 
		i++; 
	} 
	i++; 
	while(pbuf[i]!='#') 
	{ 
		m_Name+=pbuf[i]; 
		i++; 
	}	 
//	MessageBox(strIP); 
	//else exit(0); 
	//strIP="192.168.1.98"; 
	this->UpdateData(false); 
	m_IPADDRESS.SetWindowText(strIP); 
	// TODO: Add extra initialization here 
	if (WSAStartup(WINSOCK_VERSION,&wsaData)) 
	{ 
		MessageBeep(MB_ICONSTOP); 
		MessageBox("Winsock could not be initialized!", AfxGetAppName(), MB_OK|MB_ICONSTOP); 
	    WSACleanup(); 
	    return(FALSE); 
	} 
	m_bWaiting=TRUE; 
	m_bPassive=TRUE; 
	m_bClose=TRUE; 
	m_RemoteIP=0; 
	m_nBlockSize=BLOCK; 
	nID=0; 
	 
	chSndOut=(unsigned char*)malloc(BLOCK); 
 
	Openserver(); 
	sndIn.SetBits(8); 
	sndIn.SetBlockProp(m_nBlockSize,MAX); 
	sndIn.SetSampleRate(11025); 
	sndOut.SetBits(8); 
	sndOut.SetBlockProp(m_nBlockSize,MAX); 
	sndOut.SetSampleRate(11025); 
	sndIn.SetTestProp(TRUE); 
	sndIn.SetSoundOut(&sndOut); 
	sndIn.SetCompressionObject(&cmpSend); 
 
	cmpSend.SelectMethod(3); 
	cmpReceive.SelectMethod(3); 
	cmpReceive.SetWaveFormat(1,11025,8); 
	cmpReceive.Initialize(); 
	cmpReceive.SetDstSamples(BLOCK/2,(unsigned char*)chRecv); 
	cmpReceive.PrepareSpace(FALSE); 
	if(sign=='3') 
	{ 
		(CEdit *)GetDlgItem(IDC_ConnectBTN)->EnableWindow(FALSE); 
		this->OnConnectBTN(); 
	} 
	return TRUE;  // return TRUE  unless you set the focus to a control 
} 
 
void CSpeakDlg::OnSysCommand(UINT nID, LPARAM lParam) 
{ 
	if ((nID & 0xFFF0) == IDM_ABOUTBOX) 
	{ 
		CAboutDlg dlgAbout; 
		dlgAbout.DoModal(); 
	} 
	else 
	{ 
		CDialog::OnSysCommand(nID, lParam); 
	} 
} 
 
// If you add a minimize button to your dialog, you will need the code below 
//  to draw the icon.  For MFC applications using the document/view model, 
//  this is automatically done for you by the framework. 
 
void CSpeakDlg::OnPaint()  
{ 
	if (IsIconic()) 
	{ 
		CPaintDC dc(this); // device context for painting 
 
		SendMessage(WM_ICONERASEBKGND, (WPARAM) dc.GetSafeHdc(), 0); 
 
		// Center icon in client rectangle 
		int cxIcon = GetSystemMetrics(SM_CXICON); 
		int cyIcon = GetSystemMetrics(SM_CYICON); 
		CRect rect; 
		GetClientRect(&rect); 
		int x = (rect.Width() - cxIcon + 1) / 2; 
		int y = (rect.Height() - cyIcon + 1) / 2; 
 
		// Draw the icon 
		dc.DrawIcon(x, y, m_hIcon); 
	} 
	else 
	{ 
		CDialog::OnPaint(); 
	} 
} 
 
// The system calls this to obtain the cursor to display while the user drags 
//  the minimized window. 
HCURSOR CSpeakDlg::OnQueryDragIcon() 
{ 
	return (HCURSOR) m_hIcon; 
} 
 
void CSpeakDlg::OnConnectBTN()  
{ 
	// TODO: Add your control notification handler code here 
	m_bPassive=FALSE; 
	Connect(TRUE); 
} 
 
void CSpeakDlg::OnCutupbtn()  
{ 
	// TODO: Add your control notification handler code here 
	char *chClose=(char*)malloc(12); 
	wsprintf(chClose,"CLOSE"); 
	m_bClose=TRUE; 
	sendto(m_sSend,chClose,sizeof(chClose),0,(LPSOCKADDR)&m_RemoteAddr,sizeof(m_RemoteAddr)); 
	sndIn.CloseMic(); 
	sndOut.CloseSpk(); 
	m_RemoteIP=0; 
    m_bWaiting=TRUE; 
	m_bPassive=TRUE; 
	free(chClose); 
} 
 
void CSpeakDlg::Openserver() 
{ 
	m_sSend=socket(AF_INET,SOCK_DGRAM,0);  
	m_sRecv=socket(AF_INET,SOCK_DGRAM,0);  
	m_sAccept=socket(AF_INET,SOCK_STREAM,0); 
	char chName[64]; 
 
	gethostname(chName,sizeof(chName)); 
 
	addrSend.sin_family=AF_INET; 
	addrSend.sin_addr.S_un.S_addr=INADDR_ANY;//inet_addr(m_chLocalIp); 
	addrSend.sin_port=htons(SENDPORT); 
	if (bind(m_sSend,(LPSOCKADDR)&addrSend,sizeof(addrSend))== SOCKET_ERROR) 
	{ 
		ReportWinsockErr("Could not bind server socket."); 
	} 
	 
	addrRecv.sin_family=AF_INET; 
	addrRecv.sin_addr.S_un.S_addr=INADDR_ANY;//inet_addr(m_chLocalIp); 
	addrRecv.sin_port=htons(RECEIVEPORT); 
	if (bind(m_sRecv,(LPSOCKADDR)&addrRecv,sizeof(addrRecv))==SOCKET_ERROR) 
	{ 
		ReportWinsockErr("Could not bind server socket."); 
	} 
 
	int iError=WSAAsyncSelect(m_sRecv,m_hWnd,WM_READREADY,FD_READ|FD_CLOSE); 
	if(iError==SOCKET_ERROR) 
	{ 
		ReportWinsockErr("Could not Select server socket."); 
	} 
} 
 
BOOL CSpeakDlg::Connect(BOOL b) 
{ 
	unsigned long ip; 
	char* str=new char[64]; 
	char chQuery[64]; 
	SOCKADDR_IN to; 
	 
	m_bClose=FALSE; 
	if(b){ 
		//CIPAddress dlg; 
		//if(dlg.DoModal()==IDCANCEL) 
		//	return FALSE; 
		//else{ 
			m_RemoteIP=inet_addr(strIP); 
			nID=SetTimer(1,10000,NULL); 
		//} 
	} 
	int i=GetLocalIP(&ip,1); 
	in_addr t; 
	t.S_un.S_addr=ip; 
	if(i!=0) 
		str=inet_ntoa(t); 
	wsprintf(chQuery,"PHONE %s",str); 
 
	to.sin_family=AF_INET; 
	to.sin_port=htons(RECEIVEPORT); 
	to.sin_addr.S_un.S_addr=m_RemoteIP;//inet_addr("210.32.151.247");  
 
	int iError=sendto(m_sSend,chQuery,sizeof(chQuery),0,(LPSOCKADDR)&to,sizeof(to)); 
	if(iError==SOCKET_ERROR) 
	{ 
		ReportWinsockErr("无法建立连接."); 
		return FALSE; 
	} 
	return TRUE; 
} 
 
int CSpeakDlg::GetLocalIP(unsigned long *ip, int len) 
{ 
	char chName[128]; 
 
	gethostname(chName,sizeof(chName)); 
	struct hostent* pHost; 
	pHost = gethostbyname(chName);  
	CString str; 
	int i=0; 
	for(i=0; pHost!=NULL && pHost->h_addr_list[i]!=NULL; i++ ){ 
		 int j;   
		 for(j = 0;jh_length;j++ ){ 
			 CString addr; 
			 if( j > 0 ) 
				 str += ".";  
 			addr.Format("%u", (unsigned int)((unsigned char*)pHost->h_addr_list[i])[j]); 
			str += addr; 
		 } 
		 if(i>len) 
			 return 0; 
		ip[i]=inet_addr(str); 
		str.Empty(); 
 	} 
	return i; 
} 
 
void CSpeakDlg::ReportWinsockErr(LPSTR lpszErrorMsg) 
{ 
	char chMsgBuffer[100]; 
	wsprintf(chMsgBuffer, "\nWinsock error %d: %s\n\n", WSAGetLastError(), lpszErrorMsg); 
	MessageBox(chMsgBuffer,  AfxGetAppName(), MB_OK|MB_ICONSTOP); 
	return;    
} 
 
void CSpeakDlg::OnTimer(UINT nIDEvent)  
{ 
	// TODO: Add your message handler code here and/or call default 
	switch(nIDEvent){ 
	case 1: 
		if(m_bPassive==FALSE){ 
			AfxMessageBox("对方不同意或者不在同一个协议下!"); 
			m_bPassive=TRUE; 
		} 
	case 2:{ 
		   } 
	} 
	CDialog::OnTimer(nIDEvent); 
} 
 
void CSpeakDlg::OnTestBTN()  
{ 
	// TODO: Add your control notification handler code here 
	m_nBlockSize=BLOCK; 
	 
	sndIn.SetBits(8); 
	sndIn.SetBlockProp(m_nBlockSize,MAX); 
	sndIn.SetSampleRate(11025); 
	 
	sndOut.SetBits(8); 
	sndOut.SetBlockProp(m_nBlockSize,MAX); 
	sndOut.SetSampleRate(11025); 
	sndIn.SetTestProp(TRUE); 
	sndIn.SetSoundOut(&sndOut); 
	sndIn.OpenMic(); 
	sndOut.OpenSpk(); 
} 
 
LRESULT CSpeakDlg::OnReadReady(WPARAM wParam, LPARAM lParam) 
{ 
	int iRecievedNum; 
	char* ip=(char*)malloc(64); 
	 
	if(m_bWaiting){ 
		iRecievedNum=recv(m_sRecv,chRecv,sizeof(chRecv),0); 
		if(iRecievedNum==SOCKET_ERROR) 
		{ 
			ReportWinsockErr("Could not Receive the Data."); 
			return 0L; 
		} 
		if(chRecv[0]=='P' && chRecv[1]=='H' ){ 
			m_bWaiting=FALSE; 
			sscanf(chRecv,"PHONE %s",ip); 
			m_RemoteIP=inet_addr(ip); 
			m_RemoteAddr.sin_addr.S_un.S_addr=m_RemoteIP; 
			m_RemoteAddr.sin_family=AF_INET;  
			m_RemoteAddr.sin_port=htons(RECEIVEPORT); 
			if(m_bPassive){ 
				char *call=(char*)malloc(128); 
				wsprintf(call,"请求语音聊天%s\n, 接受么?",ip);  
				if(1==AfxMessageBox(call,MB_OKCANCEL)) 
					Connect(FALSE); 
				else 
					return 0L; 
				free(call); 
			} 
			m_bPassive=TRUE; 
			sndIn.SetTestProp(FALSE); 
			sndIn.SetSocketParam(m_sSend,m_RemoteAddr); 
			sndIn.OpenMic(); 
			sndOut.OpenSpk(); 
		} 
		else{ 
		//	MessageBox("Error On the Buffer"); 
			return 0L; 
		} 
	} 
	else{ 
		int tmp=sizeof(m_RemoteAddr); 
		iRecievedNum=recvfrom(m_sRecv,chRecv,BLOCK,0,(LPSOCKADDR)&m_RemoteAddr,&tmp); 
		if((chRecv[0]=='P' && chRecv[1]=='H')) 
			return 0L; 
		if(chRecv[0]=='C' && chRecv[1]=='O'){ 
			if(!m_bClose) 
				OnCutupbtn(); 
			return 0L; 
		} 
		if(iRecievedNum==SOCKET_ERROR) 
		{ 
			ReportWinsockErr("无法接收数据"); 
			return 0L; 
		} 
		cmpReceive.SetDstSamples(iRecievedNum,(unsigned char*)chRecv); 
		long temp=cmpReceive.UnConvert(TRUE,NULL,0); 
		if(temp