#include <windows.h>
#include "debug.h"
#include "stringutils.h"
#include "vectorutils.h"
#include "args.h"

// send wav_message:
// prun tstdevio -d WAV1: -c 0x1d000c

bool dev_ioctl(std::string& devname, DWORD code, const ByteVector& request, ByteVector& reply)
{
    HANDLE hDev= CreateFile(ToTString(devname).c_str(), 0xc0000000, 3, NULL, 3, 4, NULL);
    if (hDev==NULL || hDev==INVALID_HANDLE_VALUE) {
        error("opendev");
        return false;
    }
    DWORD nReturned=0;
    if (!DeviceIoControl(hDev, code, 
                request.empty()?NULL:(void*)vectorptr(request), request.size(),
                reply.empty()?NULL:(void*)vectorptr(reply), reply.size(),
                &nReturned, NULL)) {
        error("DeviceIoControl(%08lx, %d->%d), ret=%d", code, request.size(), reply.size(), nReturned);
        CloseHandle(hDev);
        return false;
    }
    debug("nret=%d : %s\n", nReturned, hexdump(reply).c_str());
    if (nReturned<reply.size())
        reply.resize(nReturned);
    CloseHandle(hDev);
    return true;
}

int WINAPI WinMain( HINSTANCE hInstance,
                   HINSTANCE hPrevInstance,
                   LPTSTR    lpCmdLine,
                   int       nCmdShow)
{
    DebugSetLogfile("tstdevio.log");

    debug("devio\n");
    StringList args;
    if (!SplitString(ToString(lpCmdLine), args, false))
    {
        error("Error in commandline");
        return false;
    }
    ByteVector request;
    ByteVector reply;
    DWORD replysize=0;
    DWORD code=0;
    std::string devname;

    for (StringList::iterator i= args.begin() ; i!=args.end() ; ++i)
    {
        std::string& arg= *i;
        if (arg[0]=='-') switch(arg[1])
        {
            case 'd': HANDLESTLSTROPTION(devname); break;
            case 'c': HANDLESTLULOPTION(code, DWORD); break;
            case 'r': HANDLESTLULOPTION(replysize, DWORD); break;
        }
        else {
            DWORD val= strtoul(arg.c_str(), 0, 16);
            if (arg.size()>4)
                BV_AppendDword(request, val);
            else if (arg.size()>2)
                BV_AppendWord(request, val);
            else 
                BV_AppendByte(request, val);
        }
    }
    reply.resize(replysize);

    debug("sending %s %08lx %s\n", devname.c_str(), code, hexdump(request).c_str());
    debug("expecting %d bytes reply\n", replysize);

    if (!dev_ioctl(devname, code, request, reply))
    {
        debug("error in ioctl\n");
    }
    debug("got %d bytes reply\n", reply.size());
    debug("request: %s\n", hexdump(request).c_str());
    debug("reply  : %s\n", hexdump(reply).c_str());
    debug("\n");
    return 0;
}