| 1 | // Licensed to the .NET Foundation under one or more agreements. |
| 2 | // The .NET Foundation licenses this file to you under the MIT license. |
| 3 | // See the LICENSE file in the project root for more information. |
| 4 | |
| 5 | /*===================================================================== |
| 6 | ** |
| 7 | ** Source: WriteFile.c (test 4) |
| 8 | ** |
| 9 | ** Purpose: Tests the PAL implementation of the WriteFile function. |
| 10 | ** Performs multiple writes to a file at different locations |
| 11 | ** then verifies the results with GetFileSize. |
| 12 | ** |
| 13 | ** dependency: |
| 14 | ** CreateFile. |
| 15 | ** GetFileSize. |
| 16 | ** FlushFileBuffers |
| 17 | ** SetFilePointer. |
| 18 | ** CloseHandle. |
| 19 | ** DeleteFile. |
| 20 | ** |
| 21 | ** |
| 22 | ** |
| 23 | **===================================================================*/ |
| 24 | |
| 25 | |
| 26 | #include <palsuite.h> |
| 27 | |
| 28 | BOOL CleanUp(HANDLE hFile, const char * fileName) |
| 29 | { |
| 30 | BOOL bRc = TRUE; |
| 31 | if (CloseHandle(hFile) != TRUE) |
| 32 | { |
| 33 | bRc = FALSE; |
| 34 | Trace("WriteFile: ERROR -> Unable to close file \"%s\"," |
| 35 | " error: %ld.\n" , fileName, GetLastError()); |
| 36 | } |
| 37 | if (!DeleteFileA(fileName)) |
| 38 | { |
| 39 | bRc = FALSE; |
| 40 | Trace("WriteFile: ERROR -> Unable to delete file \"%s\"," |
| 41 | " error: %ld.\n" , fileName, GetLastError()); |
| 42 | } |
| 43 | return bRc; |
| 44 | } |
| 45 | |
| 46 | int __cdecl main(int argc, char *argv[]) |
| 47 | { |
| 48 | const char* szStringTest = "1234567890" ; |
| 49 | const char* szWritableFile = "writeable.txt" ; |
| 50 | HANDLE hFile = NULL; |
| 51 | DWORD dwBytesWritten; |
| 52 | |
| 53 | if (0 != PAL_Initialize(argc,argv)) |
| 54 | { |
| 55 | return FAIL; |
| 56 | } |
| 57 | |
| 58 | /* create the test file */ |
| 59 | hFile = CreateFile(szWritableFile, |
| 60 | GENERIC_WRITE, |
| 61 | FILE_SHARE_WRITE, |
| 62 | NULL, |
| 63 | CREATE_ALWAYS, |
| 64 | FILE_ATTRIBUTE_NORMAL, |
| 65 | NULL); |
| 66 | |
| 67 | if(hFile == INVALID_HANDLE_VALUE) |
| 68 | { |
| 69 | Fail("WriteFile: ERROR -> Unable to create file \"%s\".\n" , |
| 70 | szWritableFile); |
| 71 | } |
| 72 | |
| 73 | |
| 74 | /* test wtriting to the file */ |
| 75 | if( WriteFile(hFile, /* HANDLE handle to file */ |
| 76 | szStringTest, /* data buffer */ |
| 77 | strlen(szStringTest), /* number of bytes to write */ |
| 78 | &dwBytesWritten, /* number of bytes written */ |
| 79 | NULL) /* overlapped buffer */ |
| 80 | ==0) |
| 81 | { |
| 82 | Trace("WriteFile: ERROR -> Unable to write to file error: %ld \n" , |
| 83 | GetLastError()); |
| 84 | CleanUp(hFile,szWritableFile); |
| 85 | Fail("" ); |
| 86 | } |
| 87 | |
| 88 | if(!FlushFileBuffers(hFile)) |
| 89 | { Trace("WriteFile: ERROR -> Call to FlushFile Buffers failed " |
| 90 | "error %ld \n" ,GetLastError()); |
| 91 | CleanUp(hFile,szWritableFile); |
| 92 | Fail("" ); |
| 93 | } |
| 94 | |
| 95 | /* check the file size */ |
| 96 | if(GetFileSize(hFile, NULL)!=strlen(szStringTest)) |
| 97 | { |
| 98 | Trace("WriteFile: ERROR -> writing %u chars to empty file " |
| 99 | "caused its size to become %u\n" ,strlen(szStringTest), |
| 100 | GetFileSize(hFile, NULL)); |
| 101 | CleanUp(hFile,szWritableFile); |
| 102 | Fail("" ); |
| 103 | } |
| 104 | |
| 105 | /* test writing to the file at position 5. */ |
| 106 | SetFilePointer( |
| 107 | hFile, /* handle to file */ |
| 108 | 0x5, /* bytes to move pointer */ |
| 109 | NULL, /* bytes to move pointer */ |
| 110 | FILE_BEGIN /* starting point */ |
| 111 | ); |
| 112 | |
| 113 | |
| 114 | if( WriteFile(hFile, /* HANDLE handle to file */ |
| 115 | szStringTest, /* data buffer */ |
| 116 | strlen(szStringTest), /* number of bytes to write */ |
| 117 | &dwBytesWritten, /* number of bytes written */ |
| 118 | NULL) /* overlapped buffer */ |
| 119 | ==0) |
| 120 | { |
| 121 | Trace("WriteFile: ERROR -> Unable to write to file after " |
| 122 | " moiving the file poiner to 5 error: %ld \n" , |
| 123 | GetLastError()); |
| 124 | CleanUp(hFile,szWritableFile); |
| 125 | Fail("" ); |
| 126 | } |
| 127 | |
| 128 | |
| 129 | if(!FlushFileBuffers(hFile)) |
| 130 | { |
| 131 | Trace("WriteFile: ERROR -> Call to FlushFile Buffers failed " |
| 132 | "error %ld \n" ,GetLastError()); |
| 133 | CleanUp(hFile,szWritableFile); |
| 134 | Fail("" ); |
| 135 | } |
| 136 | |
| 137 | /* Check the file size */ |
| 138 | if(GetFileSize(hFile, NULL)!=(strlen(szStringTest)+5)) |
| 139 | { |
| 140 | Trace("WriteFile: ERROR -> writing %u chars to the file after " |
| 141 | "sitting the file pointer to 5 resulted in wrong file size; " |
| 142 | "Expected %u resulted %u." ,strlen(szStringTest), |
| 143 | (strlen(szStringTest)+5),GetFileSize(hFile, NULL)); |
| 144 | CleanUp(hFile,szWritableFile); |
| 145 | Fail("" ); |
| 146 | } |
| 147 | |
| 148 | if (!CleanUp(hFile,szWritableFile)) |
| 149 | { |
| 150 | Fail("" ); |
| 151 | } |
| 152 | |
| 153 | PAL_Terminate(); |
| 154 | return PASS; |
| 155 | } |
| 156 | |