sin/samba-mirror
1
0
Fork 0
mirror of https://github.com/samba-team/samba.git synced 2025-01-21 18:04:06 +03:00
Code

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

Browse Source 
metze
This commit is contained in:
Stefan Metzmacher 2009-02-09 13:31:03 +01:00
parent d78fd19547
commit 3ce895181b
1 changed files with 76 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

83
testprogs/win32/npecho/npecho_client2.c
View File

@ -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;
}