test: asio.exe for async reads and writes
Signed-off-by: Casey Bodley <cbodley@citi.umich.edu>
This commit is contained in:
parent
ce200a4c2d
commit
32be705e4d
4 changed files with 481 additions and 0 deletions
138
build.vc10/asio.vcxproj
Normal file
138
build.vc10/asio.vcxproj
Normal file
|
|
@ -0,0 +1,138 @@
|
|||
<?xml version="1.0" encoding="utf-8"?>
|
||||
<Project DefaultTargets="Build" ToolsVersion="4.0" xmlns="http://schemas.microsoft.com/developer/msbuild/2003">
|
||||
<ItemGroup Label="ProjectConfigurations">
|
||||
<ProjectConfiguration Include="Debug|Win32">
|
||||
<Configuration>Debug</Configuration>
|
||||
<Platform>Win32</Platform>
|
||||
</ProjectConfiguration>
|
||||
<ProjectConfiguration Include="Debug|x64">
|
||||
<Configuration>Debug</Configuration>
|
||||
<Platform>x64</Platform>
|
||||
</ProjectConfiguration>
|
||||
<ProjectConfiguration Include="Release|Win32">
|
||||
<Configuration>Release</Configuration>
|
||||
<Platform>Win32</Platform>
|
||||
</ProjectConfiguration>
|
||||
<ProjectConfiguration Include="Release|x64">
|
||||
<Configuration>Release</Configuration>
|
||||
<Platform>x64</Platform>
|
||||
</ProjectConfiguration>
|
||||
</ItemGroup>
|
||||
<PropertyGroup Label="Globals">
|
||||
<ProjectGuid>{F08DE7DF-7B11-4ECF-9E6E-DD8C7DB40B27}</ProjectGuid>
|
||||
<RootNamespace>asio</RootNamespace>
|
||||
</PropertyGroup>
|
||||
<Import Project="$(VCTargetsPath)\Microsoft.Cpp.Default.props" />
|
||||
<PropertyGroup Condition="'$(Configuration)|$(Platform)'=='Debug|Win32'" Label="Configuration">
|
||||
<ConfigurationType>Application</ConfigurationType>
|
||||
<UseDebugLibraries>true</UseDebugLibraries>
|
||||
<CharacterSet>MultiByte</CharacterSet>
|
||||
</PropertyGroup>
|
||||
<PropertyGroup Condition="'$(Configuration)|$(Platform)'=='Debug|x64'" Label="Configuration">
|
||||
<ConfigurationType>Application</ConfigurationType>
|
||||
<UseDebugLibraries>true</UseDebugLibraries>
|
||||
<CharacterSet>MultiByte</CharacterSet>
|
||||
</PropertyGroup>
|
||||
<PropertyGroup Condition="'$(Configuration)|$(Platform)'=='Release|Win32'" Label="Configuration">
|
||||
<ConfigurationType>Application</ConfigurationType>
|
||||
<UseDebugLibraries>false</UseDebugLibraries>
|
||||
<WholeProgramOptimization>true</WholeProgramOptimization>
|
||||
<CharacterSet>MultiByte</CharacterSet>
|
||||
</PropertyGroup>
|
||||
<PropertyGroup Condition="'$(Configuration)|$(Platform)'=='Release|x64'" Label="Configuration">
|
||||
<ConfigurationType>Application</ConfigurationType>
|
||||
<UseDebugLibraries>false</UseDebugLibraries>
|
||||
<WholeProgramOptimization>true</WholeProgramOptimization>
|
||||
<CharacterSet>MultiByte</CharacterSet>
|
||||
</PropertyGroup>
|
||||
<Import Project="$(VCTargetsPath)\Microsoft.Cpp.props" />
|
||||
<ImportGroup Label="ExtensionSettings">
|
||||
</ImportGroup>
|
||||
<ImportGroup Label="PropertySheets" Condition="'$(Configuration)|$(Platform)'=='Debug|Win32'">
|
||||
<Import Project="$(UserRootDir)\Microsoft.Cpp.$(Platform).user.props" Condition="exists('$(UserRootDir)\Microsoft.Cpp.$(Platform).user.props')" Label="LocalAppDataPlatform" />
|
||||
</ImportGroup>
|
||||
<ImportGroup Condition="'$(Configuration)|$(Platform)'=='Debug|x64'" Label="PropertySheets">
|
||||
<Import Project="$(UserRootDir)\Microsoft.Cpp.$(Platform).user.props" Condition="exists('$(UserRootDir)\Microsoft.Cpp.$(Platform).user.props')" Label="LocalAppDataPlatform" />
|
||||
</ImportGroup>
|
||||
<ImportGroup Label="PropertySheets" Condition="'$(Configuration)|$(Platform)'=='Release|Win32'">
|
||||
<Import Project="$(UserRootDir)\Microsoft.Cpp.$(Platform).user.props" Condition="exists('$(UserRootDir)\Microsoft.Cpp.$(Platform).user.props')" Label="LocalAppDataPlatform" />
|
||||
</ImportGroup>
|
||||
<ImportGroup Condition="'$(Configuration)|$(Platform)'=='Release|x64'" Label="PropertySheets">
|
||||
<Import Project="$(UserRootDir)\Microsoft.Cpp.$(Platform).user.props" Condition="exists('$(UserRootDir)\Microsoft.Cpp.$(Platform).user.props')" Label="LocalAppDataPlatform" />
|
||||
</ImportGroup>
|
||||
<PropertyGroup Label="UserMacros" />
|
||||
<PropertyGroup Condition="'$(Configuration)|$(Platform)'=='Debug|Win32'">
|
||||
<OutDir>$(SolutionDir)$(Platform)\$(Configuration)\</OutDir>
|
||||
</PropertyGroup>
|
||||
<PropertyGroup Condition="'$(Configuration)|$(Platform)'=='Debug|Win32'">
|
||||
<IntDir>$(SolutionDir)$(Platform)\$(Configuration)\$(ProjectName)\</IntDir>
|
||||
</PropertyGroup>
|
||||
<PropertyGroup Condition="'$(Configuration)|$(Platform)'=='Release|Win32'">
|
||||
<OutDir>$(SolutionDir)$(Platform)\$(Configuration)\</OutDir>
|
||||
</PropertyGroup>
|
||||
<PropertyGroup Condition="'$(Configuration)|$(Platform)'=='Release|Win32'">
|
||||
<IntDir>$(SolutionDir)$(Platform)\$(Configuration)\$(ProjectName)\</IntDir>
|
||||
</PropertyGroup>
|
||||
<PropertyGroup Condition="'$(Configuration)|$(Platform)'=='Debug|x64'">
|
||||
<OutDir>$(SolutionDir)$(Platform)\$(Configuration)\</OutDir>
|
||||
</PropertyGroup>
|
||||
<PropertyGroup Condition="'$(Configuration)|$(Platform)'=='Debug|x64'">
|
||||
<IntDir>$(SolutionDir)$(Platform)\$(Configuration)\$(ProjectName)\</IntDir>
|
||||
</PropertyGroup>
|
||||
<PropertyGroup Condition="'$(Configuration)|$(Platform)'=='Release|x64'">
|
||||
<OutDir>$(SolutionDir)$(Platform)\$(Configuration)\</OutDir>
|
||||
</PropertyGroup>
|
||||
<PropertyGroup Condition="'$(Configuration)|$(Platform)'=='Release|x64'">
|
||||
<IntDir>$(SolutionDir)$(Platform)\$(Configuration)\$(ProjectName)\</IntDir>
|
||||
</PropertyGroup>
|
||||
<ItemDefinitionGroup Condition="'$(Configuration)|$(Platform)'=='Debug|Win32'">
|
||||
<ClCompile>
|
||||
<WarningLevel>Level3</WarningLevel>
|
||||
<Optimization>Disabled</Optimization>
|
||||
</ClCompile>
|
||||
<Link>
|
||||
<GenerateDebugInformation>true</GenerateDebugInformation>
|
||||
</Link>
|
||||
</ItemDefinitionGroup>
|
||||
<ItemDefinitionGroup Condition="'$(Configuration)|$(Platform)'=='Debug|x64'">
|
||||
<ClCompile>
|
||||
<WarningLevel>Level3</WarningLevel>
|
||||
<Optimization>Disabled</Optimization>
|
||||
</ClCompile>
|
||||
<Link>
|
||||
<GenerateDebugInformation>true</GenerateDebugInformation>
|
||||
</Link>
|
||||
</ItemDefinitionGroup>
|
||||
<ItemDefinitionGroup Condition="'$(Configuration)|$(Platform)'=='Release|Win32'">
|
||||
<ClCompile>
|
||||
<WarningLevel>Level3</WarningLevel>
|
||||
<Optimization>MaxSpeed</Optimization>
|
||||
<FunctionLevelLinking>true</FunctionLevelLinking>
|
||||
<IntrinsicFunctions>true</IntrinsicFunctions>
|
||||
</ClCompile>
|
||||
<Link>
|
||||
<GenerateDebugInformation>true</GenerateDebugInformation>
|
||||
<EnableCOMDATFolding>true</EnableCOMDATFolding>
|
||||
<OptimizeReferences>true</OptimizeReferences>
|
||||
</Link>
|
||||
</ItemDefinitionGroup>
|
||||
<ItemDefinitionGroup Condition="'$(Configuration)|$(Platform)'=='Release|x64'">
|
||||
<ClCompile>
|
||||
<WarningLevel>Level3</WarningLevel>
|
||||
<Optimization>MaxSpeed</Optimization>
|
||||
<FunctionLevelLinking>true</FunctionLevelLinking>
|
||||
<IntrinsicFunctions>true</IntrinsicFunctions>
|
||||
</ClCompile>
|
||||
<Link>
|
||||
<GenerateDebugInformation>true</GenerateDebugInformation>
|
||||
<EnableCOMDATFolding>true</EnableCOMDATFolding>
|
||||
<OptimizeReferences>true</OptimizeReferences>
|
||||
</Link>
|
||||
</ItemDefinitionGroup>
|
||||
<ItemGroup>
|
||||
<ClCompile Include="..\tests\asio\asio.c" />
|
||||
</ItemGroup>
|
||||
<Import Project="$(VCTargetsPath)\Microsoft.Cpp.targets" />
|
||||
<ImportGroup Label="ExtensionTargets">
|
||||
</ImportGroup>
|
||||
</Project>
|
||||
22
build.vc10/asio.vcxproj.filters
Normal file
22
build.vc10/asio.vcxproj.filters
Normal file
|
|
@ -0,0 +1,22 @@
|
|||
<?xml version="1.0" encoding="utf-8"?>
|
||||
<Project ToolsVersion="4.0" xmlns="http://schemas.microsoft.com/developer/msbuild/2003">
|
||||
<ItemGroup>
|
||||
<Filter Include="Source Files">
|
||||
<UniqueIdentifier>{4FC737F1-C7A5-4376-A066-2A32D752A2FF}</UniqueIdentifier>
|
||||
<Extensions>cpp;c;cc;cxx;def;odl;idl;hpj;bat;asm;asmx</Extensions>
|
||||
</Filter>
|
||||
<Filter Include="Header Files">
|
||||
<UniqueIdentifier>{93995380-89BD-4b04-88EB-625FBE52EBFB}</UniqueIdentifier>
|
||||
<Extensions>h;hpp;hxx;hm;inl;inc;xsd</Extensions>
|
||||
</Filter>
|
||||
<Filter Include="Resource Files">
|
||||
<UniqueIdentifier>{67DA6AB6-F800-4c08-8B7A-83BB121AAD01}</UniqueIdentifier>
|
||||
<Extensions>rc;ico;cur;bmp;dlg;rc2;rct;bin;rgs;gif;jpg;jpeg;jpe;resx;tiff;tif;png;wav;mfcribbon-ms</Extensions>
|
||||
</Filter>
|
||||
</ItemGroup>
|
||||
<ItemGroup>
|
||||
<ClCompile Include="..\tests\asio\asio.c">
|
||||
<Filter>Source Files</Filter>
|
||||
</ClCompile>
|
||||
</ItemGroup>
|
||||
</Project>
|
||||
|
|
@ -13,6 +13,8 @@ Project("{8BC9CEB8-8B4A-11D0-8D11-00A0C91BC942}") = "libtirpc", "libtirpc.vcxpro
|
|||
EndProject
|
||||
Project("{8BC9CEB8-8B4A-11D0-8D11-00A0C91BC942}") = "nfs_install", "nfs_install.vcxproj", "{A453DC17-BE6B-4271-A020-66E054AB5908}"
|
||||
EndProject
|
||||
Project("{8BC9CEB8-8B4A-11D0-8D11-00A0C91BC942}") = "asio", "asio.vcxproj", "{F08DE7DF-7B11-4ECF-9E6E-DD8C7DB40B27}"
|
||||
EndProject
|
||||
Global
|
||||
GlobalSection(SolutionConfigurationPlatforms) = preSolution
|
||||
Debug|Win32 = Debug|Win32
|
||||
|
|
@ -69,6 +71,14 @@ Global
|
|||
{A453DC17-BE6B-4271-A020-66E054AB5908}.Release|Win32.Build.0 = Release|Win32
|
||||
{A453DC17-BE6B-4271-A020-66E054AB5908}.Release|x64.ActiveCfg = Release|x64
|
||||
{A453DC17-BE6B-4271-A020-66E054AB5908}.Release|x64.Build.0 = Release|x64
|
||||
{F08DE7DF-7B11-4ECF-9E6E-DD8C7DB40B27}.Debug|Win32.ActiveCfg = Debug|Win32
|
||||
{F08DE7DF-7B11-4ECF-9E6E-DD8C7DB40B27}.Debug|Win32.Build.0 = Debug|Win32
|
||||
{F08DE7DF-7B11-4ECF-9E6E-DD8C7DB40B27}.Debug|x64.ActiveCfg = Debug|x64
|
||||
{F08DE7DF-7B11-4ECF-9E6E-DD8C7DB40B27}.Debug|x64.Build.0 = Debug|x64
|
||||
{F08DE7DF-7B11-4ECF-9E6E-DD8C7DB40B27}.Release|Win32.ActiveCfg = Release|Win32
|
||||
{F08DE7DF-7B11-4ECF-9E6E-DD8C7DB40B27}.Release|Win32.Build.0 = Release|Win32
|
||||
{F08DE7DF-7B11-4ECF-9E6E-DD8C7DB40B27}.Release|x64.ActiveCfg = Release|x64
|
||||
{F08DE7DF-7B11-4ECF-9E6E-DD8C7DB40B27}.Release|x64.Build.0 = Release|x64
|
||||
EndGlobalSection
|
||||
GlobalSection(SolutionProperties) = preSolution
|
||||
HideSolutionNode = FALSE
|
||||
|
|
|
|||
311
tests/asio/asio.c
Normal file
311
tests/asio/asio.c
Normal file
|
|
@ -0,0 +1,311 @@
|
|||
/* Copyright (c) 2010
|
||||
* The Regents of the University of Michigan
|
||||
* All Rights Reserved
|
||||
*
|
||||
* Permission is granted to use, copy and redistribute this software
|
||||
* for noncommercial education and research purposes, so long as no
|
||||
* fee is charged, and so long as the name of the University of Michigan
|
||||
* is not used in any advertising or publicity pertaining to the use
|
||||
* or distribution of this software without specific, written prior
|
||||
* authorization. Permission to modify or otherwise create derivative
|
||||
* works of this software is not granted.
|
||||
*
|
||||
* This software is provided as is, without representation or warranty
|
||||
* of any kind either express or implied, including without limitation
|
||||
* the implied warranties of merchantability, fitness for a particular
|
||||
* purpose, or noninfringement. The Regents of the University of
|
||||
* Michigan shall not be liable for any damages, including special,
|
||||
* indirect, incidental, or consequential damages, with respect to any
|
||||
* claim arising out of or in connection with the use of the software,
|
||||
* even if it has been or is hereafter advised of the possibility of
|
||||
* such damages.
|
||||
*/
|
||||
|
||||
#include <Windows.h>
|
||||
#include <tchar.h>
|
||||
#include <stdio.h>
|
||||
#include <errno.h>
|
||||
|
||||
|
||||
static void PrintErrorMessage(
|
||||
IN DWORD dwError)
|
||||
{
|
||||
LPTSTR lpMsgBuf = NULL;
|
||||
FormatMessage(FORMAT_MESSAGE_ALLOCATE_BUFFER | FORMAT_MESSAGE_FROM_SYSTEM,
|
||||
NULL, dwError, MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT),
|
||||
(LPTSTR)&lpMsgBuf, 0, NULL);
|
||||
_fputts(lpMsgBuf, stderr);
|
||||
LocalFree(lpMsgBuf);
|
||||
}
|
||||
|
||||
static SIZE_T parse_buffer_size(LPCTSTR str)
|
||||
{
|
||||
PTCHAR end;
|
||||
SIZE_T value = _tcstoul(str, &end, 0);
|
||||
|
||||
if (value == ULONG_MAX && errno == ERANGE)
|
||||
return 0;
|
||||
|
||||
if (end) switch (*end) {
|
||||
case 0: break;
|
||||
case TEXT('g'): case TEXT('G'): value <<= 10;
|
||||
case TEXT('m'): case TEXT('M'): value <<= 10;
|
||||
case TEXT('k'): case TEXT('K'): value <<= 10; break;
|
||||
default: value = 0; break;
|
||||
}
|
||||
return value;
|
||||
}
|
||||
|
||||
static DWORD resize_file(
|
||||
HANDLE file,
|
||||
SIZE_T length)
|
||||
{
|
||||
LARGE_INTEGER pointer;
|
||||
DWORD pos, status = NO_ERROR;
|
||||
|
||||
// move to the new eof
|
||||
pointer.QuadPart = length;
|
||||
pos = SetFilePointer(file, pointer.LowPart, &pointer.HighPart, FILE_BEGIN);
|
||||
if (pos == INVALID_SET_FILE_POINTER) {
|
||||
status = GetLastError();
|
||||
if (status) {
|
||||
_ftprintf(stderr, TEXT("SetFilePointer() failed to move to ")
|
||||
TEXT("position %u with %d: "), length, status);
|
||||
goto out;
|
||||
}
|
||||
}
|
||||
|
||||
// set the new file size
|
||||
if (!SetEndOfFile(file)) {
|
||||
status = GetLastError();
|
||||
_ftprintf(stderr, TEXT("SetEndOfFile() failed with %d: "), status);
|
||||
}
|
||||
|
||||
// restore the original file position
|
||||
pos = SetFilePointer(file, 0, 0, FILE_BEGIN);
|
||||
if (pos == INVALID_SET_FILE_POINTER) {
|
||||
status = GetLastError();
|
||||
_ftprintf(stderr, TEXT("SetFilePointer() failed to restore ")
|
||||
TEXT("the original position with %d: "), status);
|
||||
goto out;
|
||||
}
|
||||
out:
|
||||
return status;
|
||||
}
|
||||
|
||||
struct async_job {
|
||||
OVERLAPPED overlapped;
|
||||
LPVOID buffer;
|
||||
DWORD started;
|
||||
DWORD status;
|
||||
DWORD transferred;
|
||||
};
|
||||
|
||||
// make asynchronous calls to WriteFile()
|
||||
static DWORD jobs_write(
|
||||
HANDLE file,
|
||||
struct async_job *jobs,
|
||||
DWORD chunk,
|
||||
DWORD count)
|
||||
{
|
||||
DWORD i, status = NO_ERROR;
|
||||
|
||||
for (i = 0; i < count; i++) {
|
||||
// start the job
|
||||
jobs[i].started = GetTickCount();
|
||||
if (WriteFile(file, jobs[i].buffer, chunk, &jobs[i].transferred, &jobs[i].overlapped)) {
|
||||
jobs[i].status = NO_ERROR;
|
||||
_tprintf(TEXT("job %u wrote %u bytes synchronously in %u ticks\n"),
|
||||
i+1, jobs[i].transferred, GetTickCount() - jobs[i].started);
|
||||
} else {
|
||||
jobs[i].status = GetLastError();
|
||||
if (jobs[i].status != ERROR_IO_PENDING) {
|
||||
status = jobs[i].status;
|
||||
_ftprintf(stderr, TEXT("WriteFile() failed with %d: "), status);
|
||||
break;
|
||||
}
|
||||
}
|
||||
_tprintf(TEXT("started write job %u\n"), i+1);
|
||||
}
|
||||
return status;
|
||||
}
|
||||
|
||||
// make asynchronous calls to ReadFile()
|
||||
static DWORD jobs_read(
|
||||
HANDLE file,
|
||||
struct async_job *jobs,
|
||||
DWORD chunk,
|
||||
DWORD count)
|
||||
{
|
||||
DWORD i, status = NO_ERROR;
|
||||
|
||||
for (i = 0; i < count; i++) {
|
||||
jobs[i].started = GetTickCount();
|
||||
if (ReadFile(file, jobs[i].buffer, chunk, &jobs[i].transferred, &jobs[i].overlapped)) {
|
||||
jobs[i].status = NO_ERROR;
|
||||
_tprintf(TEXT("job %u read %u bytes synchronously in %u ticks\n"),
|
||||
i+1, jobs[i].transferred, GetTickCount() - jobs[i].started);
|
||||
} else {
|
||||
jobs[i].status = GetLastError();
|
||||
if (jobs[i].status != ERROR_IO_PENDING) {
|
||||
status = jobs[i].status;
|
||||
_ftprintf(stderr, TEXT("ReadFile() failed with %d: "), status);
|
||||
break;
|
||||
}
|
||||
}
|
||||
_tprintf(TEXT("started read job %u\n"), i+1);
|
||||
}
|
||||
return status;
|
||||
}
|
||||
|
||||
// use GetOverlappedResult() to wait for all jobs to finish
|
||||
static DWORD jobs_wait(
|
||||
HANDLE file,
|
||||
struct async_job *jobs,
|
||||
DWORD count)
|
||||
{
|
||||
DWORD i, status = NO_ERROR;
|
||||
|
||||
for (i = 0; i < count; i++) {
|
||||
if (jobs[i].status != ERROR_IO_PENDING)
|
||||
continue;
|
||||
if (!GetOverlappedResult(file, &jobs[i].overlapped, &jobs[i].transferred, TRUE)) {
|
||||
status = GetLastError();
|
||||
_ftprintf(stderr, TEXT("GetOverlappedResult() failed with %d: "), status);
|
||||
}
|
||||
_tprintf(TEXT("job %u transferred %u bytes in %u ticks\n"),
|
||||
i+1, jobs[i].transferred, GetTickCount() - jobs[i].started);
|
||||
}
|
||||
return status;
|
||||
}
|
||||
|
||||
static DWORD jobs_run(
|
||||
LPCTSTR filename,
|
||||
PBYTE buffer,
|
||||
SIZE_T length,
|
||||
DWORD count)
|
||||
{
|
||||
HANDLE file;
|
||||
struct async_job jobs[MAXIMUM_WAIT_OBJECTS] = { 0 };
|
||||
const DWORD chunk = (DWORD)(length / count);
|
||||
DWORD i, status = NO_ERROR;
|
||||
|
||||
// open or create the file; specify FILE_FLAG_OVERLAPPED for async io
|
||||
file = CreateFile(filename, GENERIC_READ | GENERIC_WRITE, 0, NULL, CREATE_ALWAYS,
|
||||
FILE_ATTRIBUTE_NORMAL | FILE_FLAG_OVERLAPPED | FILE_FLAG_NO_BUFFERING, NULL);
|
||||
if (file == INVALID_HANDLE_VALUE) {
|
||||
status = GetLastError();
|
||||
_ftprintf(stderr, TEXT("CreateFile('%s') failed with %d: "), filename, status);
|
||||
goto out;
|
||||
}
|
||||
switch (GetLastError()) {
|
||||
case ERROR_ALREADY_EXISTS:
|
||||
_ftprintf(stderr, TEXT("Opened existing file '%s'.\n"), filename);
|
||||
break;
|
||||
default:
|
||||
_ftprintf(stderr, TEXT("Created file '%s'.\n"), filename);
|
||||
break;
|
||||
}
|
||||
|
||||
// set the file length first; writes that extend the file must be synchronous
|
||||
status = resize_file(file, length);
|
||||
if (status)
|
||||
goto out_close;
|
||||
|
||||
// create a separate event for each job, or GetOverlappedResult() will mix them up
|
||||
for (i = 0; i < count; i++) {
|
||||
jobs[i].overlapped.hEvent = CreateEvent(NULL, TRUE, FALSE, NULL);
|
||||
if (jobs[i].overlapped.hEvent == INVALID_HANDLE_VALUE) {
|
||||
status = GetLastError();
|
||||
_ftprintf(stderr, TEXT("CreateEvent() failed with %d: "), status);
|
||||
goto out_events;
|
||||
}
|
||||
jobs[i].buffer = buffer + i * chunk;
|
||||
}
|
||||
|
||||
// start writes and wait for them to finish
|
||||
status = jobs_write(file, jobs, chunk, count);
|
||||
if (status) goto out_events;
|
||||
status = jobs_wait(file, jobs, count);
|
||||
if (status) goto out_events;
|
||||
|
||||
// start reads and wait for them to finish
|
||||
status = jobs_read(file, jobs, chunk, count);
|
||||
if (status) goto out_events;
|
||||
status = jobs_wait(file, jobs, count);
|
||||
if (status) goto out_events;
|
||||
|
||||
out_events:
|
||||
// close the event handles
|
||||
for (i = 0; i < count; i++)
|
||||
CloseHandle(jobs[i].overlapped.hEvent);
|
||||
out_close:
|
||||
CloseHandle(file);
|
||||
out:
|
||||
return status;
|
||||
}
|
||||
|
||||
#define BLOCK_SIZE 512
|
||||
#define PAGE_SIZE 4096
|
||||
|
||||
DWORD __cdecl _tmain(DWORD argc, LPTSTR argv[])
|
||||
{
|
||||
DWORD status = NO_ERROR;
|
||||
LPCTSTR filename;
|
||||
SIZE_T bytes;
|
||||
DWORD count;
|
||||
PBYTE buffer;
|
||||
|
||||
// parse the command line
|
||||
if (argc < 4) {
|
||||
_tprintf(TEXT("Usage: %s <filename> <bytes> <threads>\n"), argv[0]);
|
||||
goto out;
|
||||
}
|
||||
filename = argv[1];
|
||||
|
||||
bytes = parse_buffer_size(argv[2]);
|
||||
if (bytes == 0) {
|
||||
_tprintf(TEXT("Invalid value for bytes: %s\n"), argv[2]);
|
||||
goto out;
|
||||
}
|
||||
// must be a multiple of the page size for VirtualAlloc()
|
||||
if (bytes % PAGE_SIZE) {
|
||||
_tprintf(TEXT("bytes %llu must be a multiple of %u\n"), bytes, PAGE_SIZE);
|
||||
goto out;
|
||||
}
|
||||
|
||||
count = _ttoi(argv[3]);
|
||||
if (count <= 0 || count > MAXIMUM_WAIT_OBJECTS) {
|
||||
_tprintf(TEXT("Invalid value for threads: %s\n"), argv[3]);
|
||||
goto out;
|
||||
}
|
||||
// chunk size must be a multiple of the block size for FILE_FLAG_NO_BUFFERING
|
||||
if ((bytes / count) % BLOCK_SIZE) {
|
||||
_tprintf(TEXT("io size (bytes/threads) %llu must be a multiple of %u\n"), bytes / count, BLOCK_SIZE);
|
||||
goto out;
|
||||
}
|
||||
if (bytes/count > ULONG_MAX) {
|
||||
_tprintf(TEXT("io size (bytes/threads) %llu must fit in 32 bits for WriteFile()\n"), bytes / count);
|
||||
goto out;
|
||||
}
|
||||
|
||||
// allocate the buffer; VirtualAlloc guarantees page alignment for FILE_FLAG_NO_BUFFERING
|
||||
buffer = VirtualAlloc(NULL, bytes, MEM_COMMIT, PAGE_READWRITE);
|
||||
if (buffer == NULL) {
|
||||
status = GetLastError();
|
||||
_ftprintf(stderr, TEXT("VirtualAlloc(%llu) failed with %d: "), bytes, status);
|
||||
goto out;
|
||||
}
|
||||
|
||||
status = jobs_run(filename, buffer, bytes, count);
|
||||
|
||||
// free the buffer
|
||||
if (!VirtualFree(buffer, 0, MEM_RELEASE)) {
|
||||
status = GetLastError();
|
||||
_ftprintf(stderr, TEXT("VirtualFree(%p) failed with %d: "), buffer, status);
|
||||
goto out;
|
||||
}
|
||||
out:
|
||||
if (status) PrintErrorMessage(status);
|
||||
return status;
|
||||
}
|
||||
Loading…
Add table
Add a link
Reference in a new issue