AutoHandle get_current_process_token() noexcept
{
HANDLE token = nullptr;
OpenProcessToken(GetCurrentProcess(), TOKEN_QUERY, &token);
return AutoHandle(token);
}
void run()
{
if (_arguments.count() == 1) {
console.write("Usage: run [-c] <name of file to send>\n");
return;
}
int fileNameArgument = 1;
bool comFile = false;
if (_arguments[1] == String("-c")) {
comFile = true;
fileNameArgument = 2;
}
String fileName = _arguments[fileNameArgument];
String data = File(fileName).contents();
int l = data.length();
_com = AutoHandle(CreateFile(
L"COM3",
GENERIC_READ | GENERIC_WRITE,
0, // must be opened with exclusive-access
NULL, // default security attributes
OPEN_EXISTING, // must use OPEN_EXISTING
0, // not overlapped I/O
NULL), // hTemplate must be NULL for comm devices
String("COM port"));
//IF_ZERO_THROW(SetupComm(_com, 1024, 1024));
DCB deviceControlBlock;
SecureZeroMemory(&deviceControlBlock, sizeof(DCB));
IF_ZERO_THROW(GetCommState(_com, &deviceControlBlock));
deviceControlBlock.DCBlength = sizeof(DCB);
deviceControlBlock.BaudRate = 19200;
deviceControlBlock.fBinary = TRUE;
deviceControlBlock.fParity = FALSE;
deviceControlBlock.fOutxCtsFlow = FALSE;
deviceControlBlock.fOutxDsrFlow = FALSE;
// DTR_CONTROL_ENABLE causes Arduino to reset on connect
//deviceControlBlock.fDtrControl = DTR_CONTROL_ENABLE;
deviceControlBlock.fDtrControl = DTR_CONTROL_DISABLE;
deviceControlBlock.fDsrSensitivity = FALSE;
deviceControlBlock.fTXContinueOnXoff = TRUE;
deviceControlBlock.fOutX = TRUE;
deviceControlBlock.fInX = TRUE;
deviceControlBlock.fErrorChar = FALSE;
deviceControlBlock.fNull = FALSE;
deviceControlBlock.fRtsControl = RTS_CONTROL_DISABLE;
deviceControlBlock.fAbortOnError = TRUE;
deviceControlBlock.wReserved = 0;
deviceControlBlock.ByteSize = 8;
deviceControlBlock.Parity = NOPARITY;
deviceControlBlock.StopBits = ONESTOPBIT;
deviceControlBlock.XonChar = 17;
deviceControlBlock.XoffChar = 19;
IF_ZERO_THROW(SetCommState(_com, &deviceControlBlock));
//COMMTIMEOUTS timeOuts;
//SecureZeroMemory(&timeOuts, sizeof(COMMTIMEOUTS));
//timeOuts.ReadIntervalTimeout = 0;
//timeOuts.ReadTotalTimeoutMultiplier = 0;
//timeOuts.ReadTotalTimeoutConstant = 0;
//IF_ZERO_THROW(SetCommTimeouts(_com, &timeOuts));
IF_ZERO_THROW(SetCommMask(_com, EV_RXCHAR));
//IF_ZERO_THROW(ClearCommBreak(_com));
//IF_ZERO_THROW(PurgeComm(_com, PURGE_RXCLEAR | PURGE_TXCLEAR));
//IF_ZERO_THROW(FlushFileBuffers(_com));
//DWORD error;
//IF_ZERO_THROW(ClearCommError(_com, &error, NULL));
//_com.set(CreateFile(L"run.output", GENERIC_WRITE, 0, NULL,
// CREATE_ALWAYS, 0, NULL));
Sleep(2000);
ReaderThread thread(this);
//thread.start();
sendByte(0x7f); // Put Arduino in raw mode
sendByte(0x76); // Clear keyboard buffer
sendByte(0x72); // Put Arduino in tester mode
//sendByte(0x78); // Put Arduino in tester raw mode
// The buffer in the Arduino only holds 255 bytes, so we have to send
// it in chunks. We do this by buffering the data on the host PC side,
// and sending the buffer when it's full or when we're done.
_bufferCount = 0;
// When running a .com file, we need the instruction pointer to start
// at 0x100. We do this by prepending 0x100 NOP bytes at the beginning.
// In DOS this area would contain the Program Segment Prefix structure.
console.write(hex(l) + "\n");
Byte checkSum = 0;
if (comFile) {
addLength(l + 0x100);
for (int i = 0; i < 0x100; ++i) {
addByte(0x90);
checkSum += 0x90;
}
}
else
addLength(l);
for (int i = 0; i < l; ++i) {
addByte(data[i]); // Send data byte
checkSum += data[i];
if ((i & 0xff) == 0)
console.write(".");
}
addByte(checkSum);
flush();
IF_ZERO_THROW(FlushFileBuffers(_com));
console.write("Upload complete.\n");
// Dump bytes from COM port to stdout until we receive ^Z
//thread.join();
}
static AutoHandle current() { return AutoHandle( GetCurrentThread() ); }