文件 WIN32 API 上的线程同步

Question

我正在使用 Win32 API 开发一个客户端-服务器应用程序，并且我正在尝试从 login.txt 文件中删除一条记录，该文件存储了我所有登录的用户。问题是我正在使用的文件没有被删除，临时文件没有被重命名（他们做了 2-3 次，但在几次注销后它失败了）。

我的代码是：

EnterCriticalSection(&CriticalSection2);

    TCHAR tmpString[256];
    FILE *tmp = NULL;

    tmp = fopen("C:\\Users\\Liviu\\Desktop\\temporaryFile.txt", "w");

    if (!tmp)
    {
        printf("Temporary file could not open");
        system("pause");
        return;
    }

    TCHAR searchString[20];

    // Removes the possibility of having the ThreadId = password
    sprintf(searchString, "%s%d", ",", GetCurrentThreadId());

    // Paste all the lines execept the specified record to the tmp file
    while (fgets(tmpString, 255, loginFile))
    {
        if (!strstr(tmpString, searchString))
        {
            fputs(tmpString, tmp);
        }
    }

    fclose(tmp);
    fclose(loginFile);

    // I have put "while" here just to make sure this actually works, but it doesn't
    while (remove("C:\\Users\\Liviu\\Desktop\\login.txt") != 0)
        ;

    while (rename("C:\\Users\\Liviu\\Desktop\\temporaryFile.txt", "C:\\Users\\Liviu\\Desktop\\login.txt") != 0)
        ;

    LeaveCriticalSection(&CriticalSection2);

我怎样才能做到这一点？我认为是关于线程同步的，但我不知道如何修复它。

Answer 1

您没有显示您传递给CreateFile() 的实际路径，或者GetLastError() 报告的错误代码，因此没有人可以帮助您了解它失败的原因。

但是，即使CreateFile() 工作正常，您将无效参数传递给ReadFile() 和WriteFile()，因此您的代码仍然无法将数据从一个文件复制到另一个文件。事实上，代码很可能完全崩溃并终止。

试试类似的方法：

int _tmain(int argc, TCHAR *argv[])
{
    int exitCode = 1;

    if (argc < 3)
    {
        _tprintf(_T("not enough arguments!"));
        goto done;
    }

    _tprintf(_T("%s\n%s\n"), argv[1], argv[2]);

    HANDLE hFile1 = CreateFile(argv[1], GENERIC_READ, FILE_SHARE_READ, NULL, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL | FILE_FLAG_SEQUENTIAL_SCAN, NULL);
    if (hFile1 == INVALID_HANDLE_VALUE)
    {
        _tprintf(_T("Cant open File 1, Error: %d\n"), GetLastError());
        goto done;
    }

    HANDLE hFile2 = CreateFile(argv[2], GENERIC_WRITE, FILE_SHARE_READ, NULL, CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL | FILE_FLAG_SEQUENTIAL_SCAN, NULL);
    if (hFile2 == INVALID_HANDLE_VALUE)
    {
        _tprintf(_T("Cant open File 2, Error: %d\n"), GetLastError());
        goto cleanup1;
    }

    BYTE a[100];
    BYTE *ptr;
    DWORD bytesRead;
    DWORD bytesWritten;

    do
    {
        if (!ReadFile(hFile1, a, sizeof(a), &bytesRead, NULL))
        {
            _tprintf(_T("Cant read from File 1, Error: %d\n"), GetLastError());
            goto cleanup;
        }

        if (bytesRead == 0)
            break;

        ptr = a;
        do
        {
            if (!WriteFile(hFile2, ptr, bytesRead, &bytesWritten, NULL))
            {
                _tprintf(_T("Cant write to File 2, Error: %d\n"), GetLastError());
                goto cleanup;
            }
            ptr += bytesWritten;
            bytesRead -= bytesWritten;
        }
        while (bytesRead != 0);
    }
    while (true);

    _tprintf(_T("Finished\n"));
    exitCode = 0;

cleanup:
    CloseHandle(hFile2);
cleanup1:
    CloseHandle(hFile1);

done:
    system("pause");
    return exitCode;
}

话虽如此，您的代码基本上只是在复制 CopyFile() 已经做的事情，所以您可以考虑改用它：

int _tmain(int argc, TCHAR *argv[])
{
    int exitCode = 1;

    if (argc < 3)
    {
        _tprintf(_T("not enough arguments!"));
        goto done;
    }

    _tprintf(_T("%s\n%s\n"), argv[1], argv[2]);

    if (!CopyFile(argv[1], argv[2], FALSE))
    {
        _tprintf(_T("Cant copy File, Error: %d\n"), GetLastError());
        goto done;
    }

    _tprintf(_T("Finished\n"));
    exitCode = 0;

done:
    system("pause");
    return exitCode;
}

Answer 2

如果您的错误是 char * 无法转换为 wchar_t const * 对于 argv 情况，请使用 CreateFileA 而不是 CreateFile，或者切换到使用 int wmain(int, wchar_t * argv[]) 而不是 int main(int, char *[])