testprogs/win32: add an option to trigger small reads in npecho_client2.c

[samba.git] / testprogs / win32 / npecho / npecho_client2.c

diff --git a/testprogs/win32/npecho/npecho_client2.c b/testprogs/win32/npecho/npecho_client2.c
index 892574beb0534112707e561fcd16703db8049763..ebf4f9ad173750cc2871a48206b163de81e47aa7 100755 (executable)
--- a/testprogs/win32/npecho/npecho_client2.c
+++ b/testprogs/win32/npecho/npecho_client2.c
@@ -15,11 +15,20 @@ int main(int argc, char *argv[])
        HANDLE h;
        DWORD numread = 0;
        char *outbuffer = malloc(sizeof(ECHODATA)*2);
+       BOOL small_reads = FALSE;
+       DWORD state = 0;
+       DWORD flags = 0;
 
        if (argc == 1) {
-               printf("Usage: %s pipename\n", argv[0]);
-               printf("  Where pipename is something like \\\\servername\\NPECHO\n");
-               return -1;
+               goto usage;
+       } else if (argc >= 3) {
+               if (strcmp(argv[2], "large") == 0) {
+                       small_reads = FALSE;
+               } else if (strcmp(argv[2], "small") == 0) {
+                       small_reads = TRUE;
+               } else {
+                       goto usage;
+               }
        }
 
        h = CreateFile(argv[1], GENERIC_READ|GENERIC_WRITE, 0, NULL, OPEN_EXISTING, 0, NULL);
@@ -28,14 +37,69 @@ int main(int argc, char *argv[])
                return -1;
        }
 
-       Sleep(5000);
+       GetNamedPipeHandleState(h, &state, NULL, NULL, NULL, NULL, 0);
 
-       if (!ReadFile(h, outbuffer, sizeof(ECHODATA)*2, &numread, NULL)) {
-               printf("Error reading: %d\n", GetLastError());
-               return -1;
+       if (state & PIPE_READMODE_MESSAGE) {
+               printf("message read mode\n");
+       } else {
+               printf("byte read mode\n");
        }
 
+       Sleep(5000);
+
+       if (small_reads) {
+               DWORD ofs, size, nread;
+               const char *more = "";
+               printf("small reads\n");
+               numread = 0;
+               ofs = 0;
+               size = sizeof(ECHODATA)/2;
+               if (ReadFile(h, outbuffer+ofs, size, &nread, NULL)) {
+                       if (state & PIPE_READMODE_MESSAGE) {
+                               printf("Error message mode small read succeeded\n");
+                               return -1;
+                       }
+               } else if (GetLastError() == ERROR_MORE_DATA) {
+                       if (!(state & PIPE_READMODE_MESSAGE)) {
+                               printf("Error byte mode small read returned ERROR_MORE_DATA\n");
+                               return -1;
+                       }
+                       more = " more data";
+               } else {
+                       printf("Error reading: %d\n", GetLastError());
+                       return -1;
+               }
+               printf("Small Read: %d%s\n", nread, more);
+               numread += nread;
+               ofs = size;
+               size = sizeof(ECHODATA) - ofs;
+               if (!ReadFile(h, outbuffer+ofs, size, &nread, NULL)) {
+                       printf("Error reading: %d\n", GetLastError());
+                       return -1;
+               }
+               printf("Small Read: %d\n", nread);
+               numread += nread;
+       } else {
+               printf("large read\n");
+               if (!ReadFile(h, outbuffer, sizeof(ECHODATA)*2, &numread, NULL)) {
+                       printf("Error reading: %d\n", GetLastError());
+                       return -1;
+               }
+       }
        printf("Read: %s %d\n", outbuffer, numread);
+       if (state & PIPE_READMODE_MESSAGE) {
+               if (numread != sizeof(ECHODATA)) {
+                       printf("message mode returned %d bytes should be %s\n",
+                              numread, sizeof(ECHODATA));
+                       return -1;
+               }
+       } else {
+               if (numread != (sizeof(ECHODATA)*2)) {
+                       printf("message mode returned %d bytes should be %s\n",
+                              numread, (sizeof(ECHODATA)*2));
+                       return -1;
+               }
+       }
 
        if (!ReadFile(h, outbuffer, sizeof(ECHODATA)*2, &numread, NULL)) {
                printf("Error reading: %d\n", GetLastError());
@@ -45,4 +109,9 @@ int main(int argc, char *argv[])
        printf("Read: %s %d\n", outbuffer, numread);
 
        return 0;
+usage:
+       printf("Usage: %s pipename [read]\n", argv[0]);
+       printf("  Where pipename is something like \\\\servername\\NPECHO\n");
+       printf("  Where read is something 'large' or 'small'\n");
+       return -1;
 }