完成端口

#include 
#include 

HANDLE hEvent;
HANDLE hMutex;
char buffer[1024];
int counts;

HANDLE hMutexR;
char b[1024];

typedef struct which
{
    OVERLAPPED overlap;
    int dir;
} which;
#define DIRI    0
#define DIRO    1

which in, out;

int
initialize_serial_port(HANDLE hFile, int speed, char *initialize_string)
{
    DCB dcb;

    if (!GetCommState(hFile, &dcb))
    {
        return -1;
    }
    dcb.fBinary = TRUE;
    dcb.BaudRate = speed;

    // n,8,1
    dcb.fParity = FALSE;
    dcb.Parity = 0;
    dcb.ByteSize = 8;
    dcb.StopBits = 0;
    dcb.fInX  = TRUE;
    dcb.fOutX = TRUE;
    dcb.fOutxDsrFlow = FALSE;
    dcb.fOutxCtsFlow = FALSE;
    dcb.fDtrControl = DTR_CONTROL_ENABLE;
    dcb.fRtsControl = RTS_CONTROL_ENABLE;

    if (!SetCommState(hFile, &dcb))
    {
        return -1;
    }

    if (!SetupComm(hFile, 1024, 1024))
    {
        return -1;
    }

    PurgeComm(hFile, PURGE_RXABORT | PURGE_RXCLEAR);

    return 0;
}

DWORD WINAPI myiocp(LPVOID param)
{
    HANDLE hPort = (HANDLE)param;
    DWORD bytes;
    ULONG_PTR key;
    which *use;
    LPOVERLAPPED overlap;

    while (TRUE)
    {
        GetQueuedCompletionStatus(hPort, &bytes, &key, &overlap, INFINITE);
        use = (which *)overlap;
        if (use->dir == DIRI)
        {
            size_t i, length, found;
            char x[512], *p;

            WaitForSingleObject(hMutexR, INFINITE);
            
            for (;;)
            {
                found = 0; length = strlen(b);
                for (p = x, i = 0; i < length && !found; ++i)
                {
                    switch (b[i])
                    {
                    case '\r':
                        break;
                    case '\n':
                        *p = '\0';
                        found = 1;
                        break;
                    default:
                        *p++ = b[i];
                        break;
                    }
                }
                if (found)
                {
                    if (x[0]) printf("%s\n", x);
                    length -= i;
                    if (length) MoveMemory(b, b + i, length);
                    ZeroMemory(b + length, sizeof(b) - length);
                    if (!strcmp(x, "OK"))
                    {
                        WaitForSingleObject(hMutex, INFINITE);
                        strcpy(buffer + strlen(buffer), "AT\r");
                        if (++counts == 8)
                        {
                            SetEvent(hEvent);
                        }
                        else if (strlen(buffer) == 3)
                        {
                            ZeroMemory(&use->overlap, sizeof(use->overlap));
                            WriteFile((HANDLE)key, buffer, strlen(buffer), &bytes, (LPOVERLAPPED)&out);
                        }
                        ReleaseMutex(hMutex);
                    }
                }
                if (!found) break;
            }
            ZeroMemory(&use->overlap, sizeof(use->overlap));
            ReadFile((HANDLE)key, b + strlen(b), sizeof(b) - strlen(b) - 1, &bytes, &use->overlap);
            ReleaseMutex(hMutexR);
        }
        else
        {
            WaitForSingleObject(hMutex, INFINITE);
            printf("write %d bytes: %.*sw\n", bytes, bytes, buffer);
            MoveMemory(buffer, buffer + bytes, strlen(buffer + bytes) + 1);
            if (strlen(buffer) != 0)
            {
                ZeroMemory(&use->overlap, sizeof(use->overlap));
                WriteFile((HANDLE)key, buffer, strlen(buffer), &bytes, (LPOVERLAPPED)&out);
            }
            ReleaseMutex(hMutex);
        }
    }
}

int
main(int argc, char *argv[])
{
    int i;
    HANDLE hFile, hPort, hThread;
    DWORD bytes;

    hEvent = CreateEvent(NULL, FALSE, FALSE, NULL);
    hMutex = CreateMutex(NULL, FALSE, NULL);
    hMutexR = CreateMutex(NULL, FALSE, NULL);
    hFile = CreateFile("\\COM4", FILE_ALL_ACCESS, 0, NULL, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL | FILE_FLAG_OVERLAPPED, INVALID_HANDLE_VALUE);
    if (hFile == INVALID_HANDLE_VALUE)
    {
        perror("\\COM4");
        return 1;
    }

    if (initialize_serial_port(hFile, 9600, 0) < 0)
    {
        perror("initialize");
        return 1;
    }

    hPort = CreateIoCompletionPort(INVALID_HANDLE_VALUE, NULL, 0, 0);
    if (!hPort)
    {
        printf("error createiocom\n");
        return 1;
    }

    CreateIoCompletionPort(hFile, hPort, (ULONG_PTR)hFile, 0);
    in.dir = DIRI;
    out.dir = DIRO;

    for (i = 0; i < 10; ++i)
    {
        DWORD id;

        hThread = CreateThread(NULL, 0, myiocp, hPort, 0, &id);
        CloseHandle(hThread);
    }

    strcpy(buffer, "AT S7=45 S0=0 L1 V1 X4 &c1 E1 Q0\r");
    bytes = -1;
    WriteFile(hFile, buffer, strlen(buffer), &bytes, &out.overlap);
    ReadFile(hFile, b, sizeof(b), &bytes, &in.overlap);
    printf("bytes = %d\n", bytes);

    WaitForSingleObject(hEvent, INFINITE);

    CloseHandle(hFile);
    printf("Press any key to continue ...\n");
    _getch();

    return 0;

}

文件:	iocp.rar
大小:	1KB
下载:	下载