Subversion Repositories HomeAutomation

#include "DownloadState.h"

static HexFile * hf;

//static CSerial * serial;

static SerialConnection * serial;

static HANDLE  hRunMutex;                   // "Keep Running" mutex 

void downloadState(void *pMyID);

typedef enum { SEND_START, WAIT_ACK, SEND_PGM_DATA, WAIT_OWN_PACKET, WAIT_PGM_ACK, RESEND_ADDR, WAIT_DONE, SEND_DONE, DONE, WAIT_ADDR_ACK,DEBUG_STATE1,DEBUG_STATE2 } dState;

void downloadState(void *pMyID)

{

	long lLastError = 0;

	dState pgs = SEND_START;

	DWORD t = 0;

	CanProtoMessage * outCm = (CanProtoMessage *)malloc(sizeof(CanProtoMessage));

	CanProtoMessage * cm = (CanProtoMessage *)malloc(sizeof(CanProtoMessage));

	unsigned char data[8];

	unsigned char byteSent = 0;

	unsigned long currentAddress = 0;

	unsigned char bytesToWrite[PACKET_LENGTH];

	while ( WaitForSingleObject( hRunMutex, 75L ) == WAIT_TIMEOUT )

	{

		bool hasMessage = serial->getMessage(cm);

		if (hasMessage) 

		{

			unsigned char *dtmp = cm->getData();

			for(int i=0;i<8;i++) data[i]=dtmp[i];

			printf("[%ld] Has message\n",GetTickCount());

		}

		switch(pgs)

		{

			 case SEND_START:

				// Send boot start packet to target.

				// and wait for ack.

				currentAddress = hf->getAddrLower();

				data[RID_INDEX] = TARGET_ID;

				data[ADDRU_INDEX] = (unsigned char)((currentAddress & 0xFF0000) >> 16);

				data[ADDRH_INDEX] = (unsigned char)((currentAddress & 0xFF00) >> 8);

				data[ADDRL_INDEX] = (unsigned char)(currentAddress & 0xFF);

				outCm = &CanProtoMessage::CanProtoMessage(FUNCT_BOOTLOADER, FUNCC_BOOT_INIT, MY_NID, MY_ID, 8, data);

				outCm->getBytes(bytesToWrite);

				serial->write(bytesToWrite,PACKET_LENGTH);

				t = GetTickCount();

				pgs = WAIT_ACK;

			break;

			case WAIT_ACK:

				// Check for timeout, resend start packet in that case..

				if ((GetTickCount() - t) > TIMEOUT_MS)

				{

					pgs = SEND_START;

				}

				// If received ack.

				if (hasMessage && cm->getFunct() == FUNCT_BOOTLOADER && cm->getFuncc() == FUNCC_BOOT_ACK && cm->getNid() == MY_NID && cm->getSid() == TARGET_ID)

				{

					// If no error

					if (data[ERR_INDEX] == ERR_NO_ERROR)

					{

						// Start sending program data..

						byteSent = 0;

						pgs = SEND_PGM_DATA;

					}

					else 

					{ 

						// else resend start..

						pgs = SEND_START;

					}

				}

			break;

			case SEND_PGM_DATA:

				// Send program data.

				for (int i = 0; i < 8; i++)

				{ 

					// Populate data.

					data[i] = hf->getByte(currentAddress + i + byteSent);

				}

				outCm = &CanProtoMessage::CanProtoMessage(FUNCT_BOOTLOADER, (unsigned int)((unsigned int)((unsigned int)(floor(((double)byteSent/8.0)))<<2) + (FUNCC_BOOT_PGM)), MY_NID, MY_ID, 8, data);

				outCm->getBytes(bytesToWrite);

				serial->write(bytesToWrite,PACKET_LENGTH);

				t = GetTickCount();

				pgs = WAIT_OWN_PACKET;

			break;

			case WAIT_OWN_PACKET:

				// Wait reciving own packet, if timeout, resend last.

				if ((GetTickCount() - t) > TIMEOUT_MS)

				{

					//pgs = dState.SEND_PGM_DATA;

					pgs = RESEND_ADDR;

				}

				// If received own packet..

				if (hasMessage && outCm->equals(cm))

				{

					byteSent += 8;

					// Check if 64 bytes sent or not..

					if (byteSent == 64)

					{

						// Yes, wait for ack.

						t = GetTickCount();

						pgs = WAIT_PGM_ACK;

					}

					else

					{ 

						// No, send more.

						pgs = SEND_PGM_DATA;

					}

				}

			break;

			case WAIT_PGM_ACK:

				// Wait for pgm ack.

				if ((GetTickCount() - t) > 2*TIMEOUT_MS)

				{

					// Woops, error. Resend address and start over.

					pgs = RESEND_ADDR;

				}

				// If received ack.

				if (hasMessage && cm->getFunct() == FUNCT_BOOTLOADER && cm->getFuncc() == FUNCC_BOOT_ACK && cm->getNid() == MY_NID && cm->getSid() == TARGET_ID)

				{

					// If no error

					if (data[ERR_INDEX] == ERR_NO_ERROR)

					{

						currentAddress += 64;

						// Check if end.

						if (currentAddress > hf->getAddrUpper())

						{

							// Yes, we are done, send done.

							pgs = SEND_DONE;

						}

						else

						{ 

							// More to write.

							byteSent = 0;

							pgs = SEND_PGM_DATA;

						}

					}

					else

					{

						// else resend start..

						pgs = RESEND_ADDR;

					}

				}

			break;

			case SEND_DONE:

				// Send done

				outCm = &CanProtoMessage::CanProtoMessage(FUNCT_BOOTLOADER, FUNCC_BOOT_DONE, MY_NID, MY_ID, 8, data);

				outCm->getBytes(bytesToWrite);

				serial->write(bytesToWrite,PACKET_LENGTH);

				t = GetTickCount();

				pgs = WAIT_DONE;

			break;

			case RESEND_ADDR:

				// Resend addr.

				data[RID_INDEX] = TARGET_ID;

				data[ADDRU_INDEX] = (byte)((currentAddress & 0xFF0000) >> 16);

				data[ADDRH_INDEX] = (byte)((currentAddress & 0xFF00) >> 8);

				data[ADDRL_INDEX] = (byte)(currentAddress & 0xFF);

				outCm = &CanProtoMessage::CanProtoMessage(FUNCT_BOOTLOADER, FUNCC_BOOT_ADDR, MY_NID, MY_ID, 8, data);

				outCm->getBytes(bytesToWrite);

				serial->write(bytesToWrite,PACKET_LENGTH);

				t = GetTickCount();

				pgs = WAIT_ADDR_ACK;

			break;

			case WAIT_ADDR_ACK:

				// Check for timeout, resend addr packet in that case..

				if ((GetTickCount() - t) > TIMEOUT_MS)

				{

					pgs = RESEND_ADDR;

				}

				// If received ack.

				if (hasMessage && cm->getFunct() == FUNCT_BOOTLOADER && cm->getFuncc() == FUNCC_BOOT_ACK && cm->getNid() == MY_NID && cm->getSid() == TARGET_ID)

				{

					// If no error

					if (data[ERR_INDEX] == ERR_NO_ERROR)

					{

						// Start sending program data..

						byteSent = 0;

						pgs = SEND_PGM_DATA;

					}

					else

					{

						// else resend addr..

						pgs = RESEND_ADDR;

					}

				}

			break;

			case WAIT_DONE:

				// Check for timeout, resend done packet in that case..

				if ((GetTickCount() - t) > TIMEOUT_MS)

				{

					pgs = SEND_DONE;

				}

				// If received ack.

				if (hasMessage && cm->getFunct() == FUNCT_BOOTLOADER && cm->getFuncc() == FUNCC_BOOT_ACK && cm->getNid() == MY_NID && cm->getSid() == TARGET_ID)

				{

					// If no error

					if (data[ERR_INDEX] == ERR_NO_ERROR)

					{

						// Start sending program data..

						byteSent = 0;

						pgs = DONE;

					}

					else

					{

						// else resend addr..

						pgs = RESEND_ADDR;

					}

				}

			break;

			case DONE:

				downloadStop();

				printf("DONE!\n");

			break;

		}

	}

	return;

}

void downloadStart(SerialConnection * ser,HexFile *hex)

{

	hf = hex;

	serial = ser;

	hRunMutex = CreateMutex( NULL, TRUE, NULL );      // Set 

	_beginthread( downloadState, 0, 0 );

}

void downloadStop( void )

{

	ReleaseMutex( hRunMutex );

	CloseHandle( hRunMutex );

}