Create the AHCI branch for Aman's work
[reactos.git] / sdk / tools / pipetools / piperead.cpp
1 //
2 // piperead.cpp
3 //
4 // Martin Fuchs, 30.11.2003
5 //
6 // Jan Roeloffzen, 26.1.2010
7 // Pipe client, based on msdn example
8
9
10 #define WIN32_LEAN_AND_MEAN
11 #include <errno.h>
12 #include <windows.h>
13 #include <stdio.h>
14
15 #define PIPEREAD_VERSION "0.3"
16 #define PIPEREAD_NOPIPE (-101)
17
18 // This definition currently missing in MinGW.
19 #ifndef FILE_FLAG_FIRST_PIPE_INSTANCE
20 #define FILE_FLAG_FIRST_PIPE_INSTANCE 0x00080000
21 #endif
22
23
24 #define BUFSIZE 1024
25
26 static void print_error(DWORD win32_error)
27 {
28 fprintf(stderr, "WIN32 error %lu\n", win32_error);
29 }
30
31 static int pipeServer(char *path)
32 {
33 HANDLE hPipe = CreateNamedPipe(path, PIPE_ACCESS_DUPLEX|FILE_FLAG_FIRST_PIPE_INSTANCE, PIPE_WAIT|PIPE_TYPE_BYTE, 1, 4096, 4096, 30000, NULL);
34
35 if (hPipe == INVALID_HANDLE_VALUE) {
36 print_error(GetLastError());
37 return 1;
38 }
39
40 for(;;) {
41 DWORD read;
42 BYTE buffer[BUFSIZE];
43
44 if (!ReadFile(hPipe, buffer, sizeof(buffer), &read, NULL)) {
45 DWORD error = GetLastError();
46
47 if (error == ERROR_PIPE_LISTENING) {
48 Sleep(1000);
49 } else if (error == ERROR_BROKEN_PIPE) {
50 CloseHandle(hPipe);
51
52 hPipe = CreateNamedPipe(path, PIPE_ACCESS_DUPLEX|FILE_FLAG_FIRST_PIPE_INSTANCE, PIPE_WAIT|PIPE_TYPE_BYTE, 1, 4096, 4096, 30000, NULL);
53
54 if (hPipe == INVALID_HANDLE_VALUE) {
55 fprintf(stderr,"INVALID_HANDLE_VALUE\n");
56 print_error(GetLastError());
57 return 1;
58 }
59 } else {
60 fprintf(stderr,"error %lu\n",error);
61 print_error(error);
62 break;
63 }
64 }
65
66 if (read)
67 fwrite(buffer, read, 1, stdout);
68 }
69
70 if (!CloseHandle(hPipe))
71 print_error(GetLastError());
72
73 return 0;
74 }
75
76
77 static int pipeClient(char *path)
78 {
79 HANDLE hPipe=INVALID_HANDLE_VALUE;
80 TCHAR chBuf[BUFSIZE];
81 BOOL fSuccess = FALSE;
82 DWORD cbRead;
83 DWORD Err;
84 int res = 0;
85
86 setvbuf(stdout, NULL, _IONBF, 0);
87 while (1) {
88 hPipe = CreateFile(path, // pipe name
89 GENERIC_READ,
90 0, // no sharing
91 NULL, // default security attributes
92 OPEN_EXISTING, // opens existing pipe
93 0, // default attributes
94 NULL); // no template file
95
96 // Break if the pipe handle is valid.
97 if (hPipe != INVALID_HANDLE_VALUE)
98 break;
99
100 // Exit if an error other than ERROR_PIPE_BUSY occurs.
101 Err = GetLastError();
102 if (Err != ERROR_PIPE_BUSY) {
103 if (ERROR_FILE_NOT_FOUND == Err)
104 {
105 res = PIPEREAD_NOPIPE;
106 return res;
107 }
108 else
109 {
110 fprintf(stderr,"Could not open pipe %s. Error=%lu\n", path, Err );
111 res = -1;
112 }
113 break;
114 }
115
116 // All pipe instances are busy, so wait for 20 seconds.
117 if ( ! WaitNamedPipe(path, 20000)) {
118 fprintf(stderr,"Could not open pipe: 20 second wait timed out.");
119 res = -2;
120 break;
121 }
122 }
123
124 if (!res) do {
125 fSuccess = ReadFile(hPipe, // pipe handle
126 chBuf, // buffer to receive reply
127 BUFSIZE, // size of buffer
128 &cbRead, // number of bytes read
129 NULL); // not overlapped
130
131 if ( ! fSuccess ) {
132 Err = GetLastError();
133 if ( Err == ERROR_MORE_DATA ) {
134 fSuccess = TRUE;
135 } else {
136 fprintf(stderr, "ReadFile: Error %lu \n", Err );
137 res = -9;
138 break;
139 }
140 }
141
142 fwrite(chBuf,1,cbRead,stdout);
143 } while ( fSuccess);
144
145 if ( ! fSuccess) {
146 fprintf(stderr, "ReadFile from pipe failed. Error=%lu\n", GetLastError() );
147 }
148
149 if (hPipe != INVALID_HANDLE_VALUE)
150 CloseHandle(hPipe);
151
152 return res;
153
154 }
155
156 static int fileClient(const char *path)
157 {
158 int res = 0;
159 FILE *fin;
160 int c;
161
162 setvbuf(stdout, NULL, _IONBF, 0);
163 if (!(fin = fopen(path, "r"))) {
164 fprintf(stderr,"Could not fopen %s (%s)\n", path, strerror(errno) );
165 return -1;
166 }
167
168 while ((c = fgetc(fin)) != EOF) {
169 fputc(c, stdout);
170 }
171
172 fclose(fin);
173 return res;
174 }
175
176 void usage(void)
177 {
178 fprintf(stderr, "piperead " PIPEREAD_VERSION "\n\n");
179 fprintf(stderr, "Usage: piperead [-c] <named pipe>\n");
180 fprintf(stderr, "-c means Client mode\n");
181 fprintf(stderr, "Example: piperead -c \\\\.\\pipe\\kdbg | log2lines -c\n\n");
182 }
183
184 int main(int argc, char** argv)
185 {
186 char path[MAX_PATH];
187 const char* pipe_name;
188 const char* clientMode;
189 int res = 0;
190
191 pipe_name = "com_1";
192 clientMode = NULL;
193 switch (argc) {
194 case 3:
195 clientMode = *++argv;
196 if (strcmp(clientMode,"-c") != 0) {
197 fprintf(stderr,"Invalid option: %s\n", clientMode);
198 clientMode = NULL;
199 res = -6;
200 }
201 //fall through
202 case 2:
203 pipe_name = *++argv;
204 if (strcmp(pipe_name,"-h") == 0) {
205 res = -7;
206 }
207 break;
208 default:
209 res = -8;
210 break;
211 }
212 if (res) {
213 usage();
214 return res;
215 }
216
217 if ( pipe_name[0] == '\\' ) {
218 //assume caller specified full path
219 sprintf(path, "%s", pipe_name);
220 } else {
221 sprintf(path, "\\\\.\\pipe\\%s", pipe_name);
222 }
223
224 if ( clientMode ) {
225 res = pipeClient(path);
226 if (res == PIPEREAD_NOPIPE) {
227 res = fileClient(pipe_name);
228 }
229 } else {
230 res = pipeServer(path);
231 }
232
233 return res;
234 }