libsrc/wolioclinerecvw.cpp at master · LTRData/libsrc · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
#define WIN32_LEAN_AND_MEAN
#include <wio.h>

#ifdef _MSC_VER
#pragma comment(lib, "kernel32")
#endif

//-- wio.h definitions
//--------------- SafeSend(), LineRecv(), BufRecv() and BufSend() functions are
//--------------- used for I/O on overlapped streams such as sockets, comm
//--------------- devices or named pipes.

/* This one is like LineRecv() but receives a line with Unicode characters.
 */
EXTERN_C DWORD
LineRecvW(HANDLE h, LPWSTR pBuf, DWORD dwBufSize, DWORD dwTimeout)
{
  WOverlappedIOC ol;
  if (!ol)
    return 0;

  return ol.LineRecvW(h, pBuf, dwBufSize, dwTimeout);
}

DWORD
WOverlappedIOC::LineRecvW(HANDLE h, LPWSTR pBuf, DWORD dwBufChrs,
			  DWORD dwTimeout)
{
  if (dwBufChrs < 2)
    return 0;

  bool bGood = true;
  DWORD dwLen = 0;
  for (LPWSTR ptr = pBuf; dwLen < dwBufChrs - 2; )
    {
      DWORD dwReadLen;
      if (!ReadFile(h, ptr, sizeof WCHAR, &dwReadLen, NULL))
	{
	  bGood = false;
	  break;
	}

      if (dwReadLen != sizeof WCHAR)
	{
	  bGood = false;
	  SetLastError(ERROR_HANDLE_EOF);
	  break;
	}

      if (ptr[0] == L'\r')
	continue;

      if (ptr[0] == L'\n')
	break;

      dwLen ++;
      ptr ++;
    }

  if (bGood && (dwLen == 0))
    SetLastError(ERROR_SUCCESS);

  if (bGood)
    {
      pBuf[dwLen] = 0;
      return dwLen;
    }

  if (GetLastError() != ERROR_INVALID_PARAMETER)
    return 0;

  dwLen = 0;
  bGood = true;
  for (LPWSTR ptr = pBuf; dwLen < dwBufChrs - 2; )
    {
      if (!Read(h, ptr, sizeof(WCHAR)))
	{
	  bGood = false;
	  break;
	}

      if (!Wait(dwTimeout))
	{
	  bGood = false;
	  break;
	}

      DWORD dwReadLen;
      if (!GetResult(&dwReadLen))
	{
	  bGood = false;
	  break;
	}

      if (dwReadLen != sizeof(WCHAR))
	{
	  bGood = false;
	  SetLastError(ERROR_HANDLE_EOF);
	  break;
	}

      if (ptr[0] == L'\r')
	continue;

      if (ptr[0] == L'\n')
	break;

      dwLen ++;
      ptr ++;
    }

  if (bGood & (dwLen == 0))
    SetLastError(ERROR_SUCCESS);

  pBuf[dwLen] = 0;
  return dwLen;
}