cvoicechatdlg.cpp

CVoiceChat.rar.局域网音频聊天软件

// CVoiceChatDlg.cpp : implementation file
//

#include "stdafx.h"
#include "CVoiceChat.h"
#include "CVoiceChatDlg.h"
#include "ClientSock.h"
#include "ServerSock.h"
#include "Mmsystem.h"
#include "QueueItem.h"

#ifdef _DEBUG
#define new DEBUG_NEW
#undef THIS_FILE
static char THIS_FILE[] = __FILE__;
#endif

/////////////////////////////////////////////////////////////////////////////
// CAboutDlg dialog used for App About
#define INP_BUFFER_SIZE 4096
static HWAVEIN      hWaveIn ;        // 录音设备句柄
static HWAVEOUT     hWaveOut ;       // 播放设备句柄
static PBYTE        pBufferIn[2];    // 用于接收和播放的两块缓冲区
static PBYTE        pBufferOut[2];   // 用于发送和录音的两块缓冲区
static PWAVEHDR     pWaveHdrIn[2];   // 用于录音的PWAVEHDR结构数组
static PWAVEHDR     pWaveHdrOut[2];  // 用于播放的PWAVEHDR结构数组
static WAVEFORMATEX waveform ;       // 用于打开音频设备的WAVEFORMATEX结构
BOOL                bDisconnectState=TRUE; //是否处于未连接状态
bool                bwavedeviceInit=false;

int                 nIn=0;       // pBufferIn[2]中，当前播放缓冲区号
int                 nOut=0;      // pBufferOut[2]中，当前录音缓冲区号
sockaddr_in server,client;
CString m_serverIP,m_clientIP;
CString LocalIP;
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()
};
CCVoiceChatDlg  *pDlg;
extern CCVoiceChatApp theApp;
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()

/////////////////////////////////////////////////////////////////////////////
// CCVoiceChatDlg dialog
HANDLE sendthread,recvthread;
DWORD WINAPI ThreadSendPro(LPVOID lp);
DWORD WINAPI ThreadRecvPro(LPVOID lp);
CCVoiceChatDlg::CCVoiceChatDlg(CWnd* pParent /*=NULL*/)
	: CDialog(CCVoiceChatDlg::IDD, pParent)
{
	//{{AFX_DATA_INIT(CCVoiceChatDlg)
	m_info = _T("");
	m_IP = _T("");
	//}}AFX_DATA_INIT
	// Note that LoadIcon does not require a subsequent DestroyIcon in Win32
	m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME);
	//m_firstsock=true;
	Contrlsock= new CServerSock;
	//m_clientsock=true;
	Clientsock=new CClientSock;
	m_clientstate=false;
	m_serverstate=false;
	sendthread=::CreateThread(NULL,0,ThreadSendPro,NULL,CREATE_SUSPENDED,NULL);
    recvthread=::CreateThread(NULL,0,ThreadRecvPro,NULL,CREATE_SUSPENDED,NULL);
}

void CCVoiceChatDlg::DoDataExchange(CDataExchange* pDX)
{
	CDialog::DoDataExchange(pDX);
	//{{AFX_DATA_MAP(CCVoiceChatDlg)
	DDX_Control(pDX, IDC_BTN_Link, m_link);
	DDX_Control(pDX, IDC_BTN_End, m_end);
	DDX_Control(pDX, IDC_BTN_Agreen, m_agreen);
	DDX_Control(pDX, IDC_BTN_Abjust, m_abjust);
	DDX_Text(pDX, IDC_EDIT_Info, m_info);
	DDX_Text(pDX, IDC_EDIT_IP, m_IP);
	//}}AFX_DATA_MAP
}

BEGIN_MESSAGE_MAP(CCVoiceChatDlg, CDialog)
	//{{AFX_MSG_MAP(CCVoiceChatDlg)
	ON_WM_SYSCOMMAND()
	ON_WM_PAINT()
	ON_WM_QUERYDRAGICON()
	ON_BN_CLICKED(IDC_BTN_Link, OnBTNLink)
	ON_BN_CLICKED(IDC_BTN_Agreen, OnBTNAgreen)
	ON_BN_CLICKED(IDC_BTN_Abjust, OnBTNAbjust)
	ON_BN_CLICKED(IDC_BTN_End, OnBTNEnd)
	ON_MESSAGE(MM_WIM_OPEN,ON_MM_WIM_OPEN)
	ON_MESSAGE(MM_WIM_DATA,ON_MM_WIM_DATA)
	ON_MESSAGE(MM_WIM_CLOSE,ON_MM_WIM_CLOSE)
	ON_MESSAGE(MM_WOM_OPEN,ON_MM_WOM_OPEN)
	ON_MESSAGE(UM_SOCK,ONDATASOCKETPRC)
	//}}AFX_MSG_MAP
END_MESSAGE_MAP()

/////////////////////////////////////////////////////////////////////////////
// CCVoiceChatDlg message handlers

BOOL CCVoiceChatDlg::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

	pDlg=(CCVoiceChatDlg *)theApp.m_pMainWnd;

	m_abjust.EnableWindow(false);
	m_agreen.EnableWindow(false);
	m_end.EnableWindow(false);
 
    WORD wVersionRequested;
    WSADATA wsaData;
	//CString LocalIP;
    char name[255];
    PHOSTENT hostinfo;
    wVersionRequested = MAKEWORD( 2, 0 );
    if(WSAStartup( wVersionRequested, &wsaData ) == 0 )
    {
       if( gethostname ( name, sizeof(name)) == 0)
       {
           if((hostinfo = gethostbyname(name)) != NULL)
           {
               LocalIP = inet_ntoa (*(struct in_addr *)*hostinfo->h_addr_list);
	       }
       }            
 //      WSACleanup( );
    } 
	//AfxSocketInit();
	if(Listensock.Create(5000)==0)
	{
		AfxMessageBox("服务器套接字创建失败!!!");
		return 0;
	}

	// 开始侦听
	if(Listensock.Listen(5)==0)
	{
		AfxMessageBox("套接字监听失败!!!");
		return 0;
	}
	
	
	return TRUE;  // return TRUE  unless you set the focus to a control
}

void CCVoiceChatDlg::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 CCVoiceChatDlg::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 CCVoiceChatDlg::OnQueryDragIcon()
{
	return (HCURSOR) m_hIcon;
}
//按下呼叫按扭响应(客户端)
void CCVoiceChatDlg::OnBTNLink() 
{

    if(Clientsock->Create(SOCK_STREAM)==0)
	{
		AfxMessageBox("客户端套接字创建失败!!!");
		return ;
	}
	UpdateData(true);
	m_serverIP=m_IP;
	if(Clientsock->Connect(m_IP,5000)==0)
	{
		AfxMessageBox("呼叫失败!!!");
        Clientsock->Close();
		return ;
	}
	else
	{
		//添加播放等待接听的提示音乐
        PlaySound("./PhoneIn.wav",NULL,SND_SYNC|SND_LOOP| SND_ASYNC);
	}
	
	UpdateData(false);
	m_link.EnableWindow(false);//设置呼叫按扭为按下
	m_abjust.EnableWindow(false);//
	m_agreen.EnableWindow(false);
	m_end.EnableWindow(true);
	
	
}

void CCVoiceChatDlg::OnBTNAgreen() 
{
	//设置按纽状态
	m_link.EnableWindow(false);
	m_abjust.EnableWindow(false);
	m_agreen.EnableWindow(false);
	m_end.EnableWindow(true);
	//关掉提示语音
	PlaySound(NULL,NULL,NULL);
	UpdateData(true);
	m_info="正在通话中.....";
	UpdateData(false);
	char *buf;
	buf="电话已接听!!";
	Contrlsock->Send(buf,strlen(buf)+1,0);
	m_serverstate=true;//设置服务端正在通话中,可以考虑多用户连接情况
    bDisconnectState=false;
	//启动数据报套接字用于语音传输
	SetDataSock();
	//添加初始化音频函数
	if( bwavedeviceInit==false)InitAudioDevice();
	{
		AfxMessageBox("初始化音频设备失败!");
		return ;
	}
		
	//开始录音
	RecordBegin();
	//启动发送与放音线程
	::ResumeThread(sendthread);
	::ResumeThread(recvthread);

}

void CCVoiceChatDlg::OnBTNAbjust() 
{
	//设置按纽状态
	m_abjust.EnableWindow(false);
	m_link.EnableWindow(true);
	m_agreen.EnableWindow(false);
	m_end.EnableWindow(false);
	//关掉提示语音
	PlaySound(NULL,NULL,NULL);
	char *buf;
	buf="对方拒绝了你的请求!!!";
	UpdateData(true);
	m_info="你拒绝了对方请求!!";
    UpdateData(false);
	Contrlsock->Send(buf,strlen(buf)+1,0);
	
	Sleep(10);
	Contrlsock->Close();
	Contrlsock=new CServerSock;
}

void CCVoiceChatDlg::OnBTNEnd() 
{
	// TODO: Add your control notification handler code here
    //设置挂断按纽状态
	m_link.EnableWindow(true);
	m_abjust.EnableWindow(false);
	m_agreen.EnableWindow(false);
	m_end.EnableWindow(false);
	if(m_clientstate==true)
	{
		
		char *buf;
		buf="对方终止了聊天..........";
		Clientsock->Send(buf,strlen(buf)+1,0);
		UpdateData(true);
		m_info="通话结束!!!";
		UpdateData(false);
		//将队列清空
		sendlist.deleteall();
		recvlist.deleteall();
		Sleep(10);
		::SuspendThread(sendthread);
		::SuspendThread(recvthread);
		m_clientstate=false;
		bDisconnectState=true;
		Clientsock->Close();
        Clientsock=new CClientSock;
		::closesocket(m_datasock);
		return ;
	}
	if(m_serverstate==true)
	{
		char *buf;
		buf="对方终止了聊天..........";
		Contrlsock->Send(buf,strlen(buf)+1,0);
		UpdateData(true);
		m_info="通话结束!!!";
		UpdateData(false);
		m_serverstate=false;
		bDisconnectState=true;
		Contrlsock->Close();
		Contrlsock=new CServerSock;
		::closesocket(m_datasock);
		//清空队列
		sendlist.deleteall();
		recvlist.deleteall();
		Sleep(10);
		::SuspendThread(sendthread);
		::SuspendThread(recvthread);