/*-----------------------------------------------------------------------------

    This is a part of the Microsoft Source Code Samples. 
    Copyright (C) 1995 Microsoft Corporation.
    All rights reserved. 
    This source code is only intended as a supplement to 
    Microsoft Development Tools and/or WinHelp documentation.
    See these sources for detailed information regarding the 
    Microsoft samples programs.

    MODULE: Init.c

    PURPOSE: 初始化全局数据和通信端口连接。 关闭通信端口并清除全局数据。

    FUNCTIONS:
        GlobalInitialize  - 初始化全局变量和系统对象
        GlobalCleanup     - 清除全局变量和系统对象
        ClearTTYContents  - 清除tty缓冲区
        InitNewFont       - 为TTY子窗口创建一个新字体
        CreateTTYInfo     - 创建动态tty数据，控制tty的行为
        DestroyTTYInfo    - 销毁tty信息数据
        StartThreads      - 打开端口时启动工作线程
        SetupCommPort     - 第一次打开港口
        WaitForThreads    - 设置线程退出事件并等待工作线程退出
        BreakDownCommPort - 关闭与comm端口的连接
        DisconnectOK      - 询问用户是否可以断开连接

-----------------------------------------------------------------------------*/

#include <windows.h>
#include <commctrl.h>
#include "mttty.h"

/*
    Prototypes for functions called only within this file
*/
void StartThreads( void );
DWORD WaitForThreads( DWORD );

/*
    TimeoutsDefault
		我们需要ReadIntervalTimeout来执行读取操作 我们这样做是为了超时并变得重叠。 在这里指定1会使ReadFile很快返回
		这样我们的读取线程将继续执行。
		参考定义
*/
COMMTIMEOUTS gTimeoutsDefault = { 0x01, 0, 0, 0, 0 };


/*-----------------------------------------------------------------------------

FUNCTION: GlobalInitialize

PURPOSE: 在创建任何窗口之前初始化全局变量

COMMENTS: 相应函数 GlobalCleanup

HISTORY:   Date:      Author:     Comment:
           10/27/95   AllenD      Wrote it

-----------------------------------------------------------------------------*/
void GlobalInitialize()
{
    int cyMenuHeight, cyCaptionHeight, cyFrameHeight;

    //
    // critical sections in status reporting & node management
    // 初始化互斥对象
    InitializeCriticalSection(&gStatusCritical);
    InitializeCriticalSection(&gcsWriterHeap);
    InitializeCriticalSection(&gcsDataHeap);

    //
    // 状态消息事件
    //
    ghStatusMessageEvent = CreateEvent(NULL, FALSE, FALSE, NULL);
    if (ghStatusMessageEvent == NULL)
        ErrorReporter("CreateEvent (Status message event)");

    //
    //线程退出事件
    //
    ghThreadExitEvent = CreateEvent(NULL, TRUE, FALSE, NULL);
    if (ghThreadExitEvent == NULL)
        ErrorReporter("CreateEvent (Thread exit event)");        

    //
    //用于文件传输状态栏
    //
    InitCommonControls();

    //
    // 字体控制
    //
    ghFontStatus = CreateStatusEditFont();

    //
    //下面的内容用于调整tty窗口和对话框窗口的大小
    //
    gwBaseY = HIWORD(GetDialogBaseUnits());
    cyMenuHeight = GetSystemMetrics(SM_CYMENU);
    cyCaptionHeight = GetSystemMetrics(SM_CYCAPTION);
    cyFrameHeight = GetSystemMetrics(SM_CYFRAME);
    gcyMinimumWindowHeight = cyMenuHeight + \
                            4 * cyCaptionHeight + \
                            2 * cyFrameHeight +
                            (SETTINGSFACTOR + STATUSFACTOR) * gwBaseY ;
    return ;
}


/*-----------------------------------------------------------------------------

FUNCTION: GlobalCleanup

PURPOSE: 清除所有全局变量

COMMENTS: Partner to GlobalInitialize

HISTORY:   Date:      Author:     Comment:
           10/27/95   AllenD      Wrote it

-----------------------------------------------------------------------------*/
void GlobalCleanup()
{
    DeleteCriticalSection(&gStatusCritical);
    DeleteCriticalSection(&gcsWriterHeap);
    DeleteCriticalSection(&gcsDataHeap);
    DeleteObject(ghFontStatus);
    CloseHandle(ghStatusMessageEvent);
    CloseHandle(ghThreadExitEvent);
    HeapDestroy(ghStatusMessageHeap);
    return;
}


/*-----------------------------------------------------------------------------

FUNCTION: ClearTTYContents

PURPOSE: 清除tty缓冲区

RETURN: always TRUE

HISTORY:   Date:      Author:     Comment:
           10/27/95   AllenD      Wrote it

-----------------------------------------------------------------------------*/
BOOL ClearTTYContents()
{
    FillMemory(TTYInfo.Screen, MAXCOLS*MAXROWS, ' ');
    return TRUE;
}


/*-----------------------------------------------------------------------------

FUNCTION: InitNewFont(LOGFONT, COLORREF)

PURPOSE: 初始化在TTY窗口上使用的字体

PARAMETERS:
    LogFont  - 逻辑字体
    rgbColor - 字体颜色

COMMENTS: 当用户建立新连接或更改TTY字体时调用。

HISTORY:   Date:      Author:     Comment:
           10/27/95   AllenD      Wrote it

-----------------------------------------------------------------------------*/
void InitNewFont(LOGFONT LogFont, COLORREF rgbColor)
{
    TEXTMETRIC tm;
    HDC        hDC;

    //
    //如果旧字体存在，那么我应该删除它
    //
    if (TTYInfo.hTTYFont)
        DeleteObject(TTYInfo.hTTYFont);

    TTYInfo.lfTTYFont = LogFont;
    TTYInfo.hTTYFont = CreateFontIndirect(&(TTYInfo.lfTTYFont));
    TTYInfo.rgbFGColor = rgbColor;

    hDC = GetDC( ghwndMain ) ;
    SelectObject( hDC, TTYInfo.hTTYFont ) ;
    GetTextMetrics( hDC, &tm ) ;
    ReleaseDC( ghwndMain, hDC ) ;

    //
	//字符的宽度和高度
    //
    TTYInfo.xChar = tm.tmAveCharWidth  ;
    TTYInfo.yChar = tm.tmHeight + tm.tmExternalLeading ;

    TTYInfo.xOffset = 0 ;
    TTYInfo.yOffset = TTYInfo.yChar * TTYInfo.nRow;
}

/*-----------------------------------------------------------------------------

FUNCTION: InitTTYInfo

PURPOSE: 初始化TTY数据

COMMENTS: 这个结构是这个程序所有部分使用的TTY属性的集合

HISTORY:   Date:      Author:     Comment:
           10/27/95   AllenD      Wrote it
            2/14/96   AllenD      Removed npTTYInfo

-----------------------------------------------------------------------------*/
BOOL InitTTYInfo()
{
    //
    // initialize generial TTY info
    //
	TTYInfo.hCommPort      = NULL ;
    TTYInfo.fConnected     = FALSE ;
    TTYInfo.fLocalEcho     = FALSE ;
    TTYInfo.wCursorState  = CS_HIDE ;
    TTYInfo.bPort          = 1 ;
    TTYInfo.dwBaudRate      = 921600 ;
    TTYInfo.bByteSize      = 8 ;
    TTYInfo.bParity        = NOPARITY ;
    TTYInfo.bStopBits      = ONESTOPBIT ;
    TTYInfo.fAutowrap      = TRUE;
    TTYInfo.fNewLine       = FALSE;
    TTYInfo.xSize         = 0 ;
    TTYInfo.ySize         = 0 ;
    TTYInfo.xScroll       = 0 ;
    TTYInfo.yScroll       = 0 ;
    TTYInfo.nColumn        = 0 ;
    TTYInfo.nRow           = MAXROWS - 1 ;
    TTYInfo.fDisplayErrors = TRUE ;

    //
    // timeouts
    //
    TTYInfo.timeoutsnew   = gTimeoutsDefault;

    //
    // read state and status events
    //
    gdwReceiveState            = RECEIVE_TTY;
    TTYInfo.dwEventFlags    = EVENTFLAGS_DEFAULT;
    TTYInfo.chFlag      = FLAGCHAR_DEFAULT;

    //
    // Flow Control Settings
    //
    TTYInfo.fDtrControl    = DTR_CONTROL_ENABLE;
    TTYInfo.fRtsControl    = RTS_CONTROL_ENABLE;
    TTYInfo.chXON       = ASCII_XON;
    TTYInfo.chXOFF      = ASCII_XOFF;
    TTYInfo.wXONLimit      = 0;
    TTYInfo.wXOFFLimit     = 0;
    TTYInfo.fCTSOutFlow    = FALSE;
    TTYInfo.fDSROutFlow    = FALSE;
    TTYInfo.fDSRInFlow     = FALSE;
    TTYInfo.fXonXoffOutFlow  = FALSE;
    TTYInfo.fXonXoffInFlow   = FALSE;
    TTYInfo.fTXafterXoffSent = FALSE;

    TTYInfo.fNoReading       = FALSE;
    TTYInfo.fNoWriting       = FALSE;
    TTYInfo.fNoEvents        = FALSE;
    TTYInfo.fNoStatus        = FALSE;
    TTYInfo.fDisplayTimeouts    = FALSE;

    //
    // setup default font information
    // 
    TTYInfo.lfTTYFont.lfHeight =         20 ;
    TTYInfo.lfTTYFont.lfWidth =          0 ;
    TTYInfo.lfTTYFont.lfEscapement =     0 ;
    TTYInfo.lfTTYFont.lfOrientation =    0 ;
    TTYInfo.lfTTYFont.lfWeight =         0 ;
    TTYInfo.lfTTYFont.lfItalic =         0 ;
    TTYInfo.lfTTYFont.lfUnderline =      0 ;
    TTYInfo.lfTTYFont.lfStrikeOut =      0 ;
    TTYInfo.lfTTYFont.lfCharSet =        OEM_CHARSET ;
    TTYInfo.lfTTYFont.lfOutPrecision =   OUT_DEFAULT_PRECIS ;
    TTYInfo.lfTTYFont.lfClipPrecision =  CLIP_DEFAULT_PRECIS ;
    TTYInfo.lfTTYFont.lfQuality =        DEFAULT_QUALITY ;
    TTYInfo.lfTTYFont.lfPitchAndFamily = FIXED_PITCH | FF_MODERN ;
    strcpy( TTYInfo.lfTTYFont.lfFaceName, "宋体" ) ;
	TTYInfo.DownloadReday = FALSE;

    InitNewFont( TTYInfo.lfTTYFont, RGB(0,0,0));

    ClearTTYContents();

    return ( TRUE ) ;
}

/*-----------------------------------------------------------------------------

FUNCTION: DestroyTTYInfo

PURPOSE: 释放与TTYInfo结构体相关的对象

HISTORY:   Date:      Author:     Comment:
           10/27/95   AllenD      Wrote it
            2/14/96   AllenD      Removed npTTYInfo

-----------------------------------------------------------------------------*/
void DestroyTTYInfo()
{
    DeleteObject(TTYInfo.hTTYFont);
}

/*-----------------------------------------------------------------------------

FUNCTION: StartThreads

PURPOSE:创建读取/状态和写入线程

HISTORY:   Date:      Author:     Comment:
           10/27/95   AllenD      Wrote it

-----------------------------------------------------------------------------*/
void StartThreads(void)
{
	DWORD			dwReadStatId;
	DWORD			dwWriterId;
	TTYInfo.hReaderStatus = CreateThread(NULL, 0, (LPTHREAD_START_ROUTINE) ReaderAndStatusProc, (LPVOID) ghWndTTY, 0, &dwReadStatId);
	if (TTYInfo.hReaderStatus == NULL)
		ErrorInComm("CreateThread(Reader/Status)");
	TTYInfo.hWriter = CreateThread(NULL, 0, (LPTHREAD_START_ROUTINE) WriterProc, (LPVOID) NULL, 0, &dwWriterId);
	if (TTYInfo.hWriter == NULL)
		ErrorInComm("CreateThread (Writer)");
	return;
}



/*-----------------------------------------------------------------------------

FUNCTION: SetupCommPort( void )

PURPOSE: 配置串口

RETURN: 
	如果成功返回串口句柄, 否则返回NULL

HISTORY:   Date:      Author:     Comment:
           10/27/95   AllenD      Wrote it

-----------------------------------------------------------------------------*/
HANDLE SetupCommPort()
{
	int myintPort;
    //
	//从设置对话框获取tty设置
    //
    UpdateTTYInfo();
	myintPort = atoi(gszPort+3);

    //
	//打开通信端口句柄
    //
	sprintf(gszPort, "\\\\.\\COM%d", myintPort);///可以显示COM10以上端口//add by itas109 2014-01-09
    TTYInfo.hCommPort = CreateFile( gszPort,  
                                      GENERIC_READ | GENERIC_WRITE, 
                                      0, 
                                      0, 
                                      OPEN_EXISTING,
                                      FILE_ATTRIBUTE_NORMAL | FILE_FLAG_OVERLAPPED,
                                      0);

    if (TTYInfo.hCommPort == INVALID_HANDLE_VALUE) {   
        ErrorReporter("CreateFile");
        return NULL;
    }

	//设置新的超时参数
    if (!GetCommTimeouts( TTYInfo.hCommPort, &(TTYInfo.timeoutsorig)))
        ErrorReporter("GetCommTimeouts");

    //
    // Set port state
    //
    UpdateConnection();

    //
    // set comm buffer sizes
    //
    SetupComm(TTYInfo.hCommPort, MAX_READ_BUFFER, MAX_WRITE_BUFFER);

    //
    // raise DTR
    //
    if (!EscapeCommFunction(TTYInfo.hCommPort, SETDTR))
        ErrorReporter("EscapeCommFunction (SETDTR)");

    //
    // start threads and set initial thread state to not done
    //
    StartThreads();

    //
    // set overall connect flag
    //
    TTYInfo.fConnected = TRUE ;

    return TTYInfo.hCommPort;
}

/*-----------------------------------------------------------------------------

FUNCTION: WaitForThreads(DWORD)

PURPOSE: Waits a specified time for the worker threads to exit

PARAMETERS:
    dwTimeout - milliseconds to wait until timeout

RETURN:
    WAIT_OBJECT_0 - successful wait, threads are not running
    WAIT_TIMEOUT  - at least one thread is still running
    WAIT_FAILED   - failure in WaitForMultipleObjects

HISTORY:   Date:      Author:     Comment:
           10/27/95   AllenD      Wrote it

----------------------------------------------------------------------------*/
DWORD WaitForThreads(DWORD dwTimeout)
{	
    HANDLE hThreads[2];
    DWORD  dwRes;

    hThreads[0] = TTYInfo.hReaderStatus;
    hThreads[1] = TTYInfo.hWriter;

    //
    // set thread exit event here
    //
    SetEvent(ghThreadExitEvent);

    dwRes = WaitForMultipleObjects(2, hThreads, TRUE, dwTimeout);
    switch(dwRes)
    {
        case WAIT_OBJECT_0:
        case WAIT_OBJECT_0 + 1: 
            dwRes = WAIT_OBJECT_0;
            break;

        case WAIT_TIMEOUT:
            
            if (WaitForSingleObject(TTYInfo.hReaderStatus, 0) == WAIT_TIMEOUT)
                OutputDebugString("Reader/Status Thread didn't exit.\n\r");

            if (WaitForSingleObject(TTYInfo.hWriter, 0) == WAIT_TIMEOUT)
                OutputDebugString("Writer Thread didn't exit.\n\r");

            break;

        default:
            ErrorReporter("WaitForMultipleObjects");
            break;
    }

    //
    // reset thread exit event here
    //
    ResetEvent(ghThreadExitEvent);

    return dwRes;
}

/*-----------------------------------------------------------------------------

FUNCTION: BreakDownCommPort

PURPOSE: Closes a connection to a comm port

RETURN:
    TRUE  - successful breakdown of port
    FALSE - port isn't connected

COMMENTS: Waits for threads to exit,
          clears DTR, restores comm port timeouts, purges any i/o
          and closes all pertinent handles

HISTORY:   Date:      Author:     Comment:
           10/27/95   AllenD      Wrote it

-----------------------------------------------------------------------------*/
BOOL BreakDownCommPort()
{
    if (!TTYInfo.fConnected)
        return FALSE;

    TTYInfo.fConnected = FALSE;

    //
    // wait for the threads for a small period
    //
    if (WaitForThreads(20000) != WAIT_OBJECT_0)
        /*
            if threads haven't exited, then they will
            interfere with a new connection.  I must abort
            the entire program.
        */
        ErrorHandler("Error closing port.");

    //
    // lower DTR
    //
    if (!EscapeCommFunction(TTYInfo.hCommPort, CLRDTR))
        ErrorReporter("EscapeCommFunction(CLRDTR)");

    //
    // restore original comm timeouts
    //
    if (!SetCommTimeouts(TTYInfo.hCommPort,  &(TTYInfo.timeoutsorig)))
        ErrorReporter("SetCommTimeouts (Restoration to original)");

    //
    // Purge reads/writes, input buffer and output buffer
    //
    if (!PurgeComm(TTYInfo.hCommPort, PURGE_FLAGS))
        ErrorReporter("PurgeComm");

    CloseHandle(TTYInfo.hCommPort);
    CloseHandle(TTYInfo.hReaderStatus);
    CloseHandle(TTYInfo.hWriter);

    return TRUE;
}
 
/*-----------------------------------------------------------------------------

FUNCTION: DisconnectOK

PURPOSE: Asks user if it is OK to disconnect

RETURN:
    TRUE  - OK to disconnect
    FALSE - Disconnect not OK

HISTORY:   Date:      Author:     Comment:
           10/27/95   AllenD      Wrote it

-----------------------------------------------------------------------------*/
BOOL DisconnectOK()
{
    if (!TTYInfo.fConnected)
        return TRUE;
    
    return ((MessageBox(ghwndMain, "OK to Disconnect?", gszPort, MB_YESNO)) == IDYES);
}



BOOL QuerySingleSerialPortEx(int nPort)
{
	char strComm[50];
	BOOL bValid = TRUE;
	HANDLE hComm;
	int nError;


	sprintf(strComm,"COM%d",nPort);
	if (nPort >= 10)
	{//微软的技术手册规定打开端口号大于9的串口时应该使用"\\.\COM10"格式的文件名，更多的信息请参考msdn。
		sprintf(strComm,"\\\\.\\COM%d",nPort);
	}

	hComm = CreateFile(strComm,GENERIC_READ | GENERIC_WRITE,0,NULL,\
	OPEN_EXISTING,FILE_ATTRIBUTE_NORMAL| FILE_FLAG_OVERLAPPED, NULL); 
	nError = GetLastError();

	if(INVALID_HANDLE_VALUE == hComm)     
	{//***********如果没有该设备，或者被其他应用程序占用********
		if(2 == nError)
			bValid = -1;
		else if(5 == nError)
			bValid = FALSE;
		else
			bValid = FALSE;
	}
	CloseHandle(hComm); //重要 关闭文件句柄 = 关闭串口
	return bValid;
}

//利用API查询指定范围内的串口状态信息，并分别返回指定范围内系统存在的串口，
//当前未被占用的可用串口，已被占用的串口和系统不存在的串口
void QuerySerialPortStatusEx(int  *AryNoneOpenedPort,int nStartPort,int nLastPort)
{
	char strTmp[50],strComm[50],strFree[50];
	int n1 = nStartPort,n2=nLastPort,i,loop = 0;
	HANDLE hComm;
	int nLastError;
	//对输入的参数进行检查
	if(nStartPort < 1) nStartPort = 1;
	if(nStartPort > 65535) nStartPort = 65535;
	if(nLastPort < 1) nLastPort = 1;
	if(nLastPort > 65535) nLastPort = 65535;
	nStartPort = min(n1,n2);
	nLastPort  = max(n1,n2);
	memset(AryNoneOpenedPort,0,255);


	for (i = nStartPort;i<=nLastPort;i++)
	{
		sprintf(strComm,"COM%d",i);
		if (i >= 10)
		{//微软的技术手册规定打开端口号大于9的串口时应该使用"\\.\COM10"格式的文件名，更多的信息请参考msdn。
			sprintf(strComm,"\\\\.\\COM%d",i);
		}
		// 这里的CreateFile函数起了很大的作用，可以用来创建系统设备文件，如果该设备不存在或者被占用，则会返回一个错误，即下面的 INVALID_HANDLE_VALUE ，据此可以判断可使用性。详细参见MSDN中的介绍。
		hComm = CreateFile(strComm,GENERIC_READ | GENERIC_WRITE,0,NULL,\
			OPEN_EXISTING,FILE_ATTRIBUTE_NORMAL| FILE_FLAG_OVERLAPPED, NULL); 
		nLastError = GetLastError();
		//strComm.TrimLeft("\\\\.\\");

		if(hComm != INVALID_HANDLE_VALUE)     
		{//串口存在且未被占用
			AryNoneOpenedPort[loop] = i;
			loop++;
		}
		CloseHandle(hComm); //关闭文件句柄 - 必要
	}

}