Subversion Repositories mips32r1

/* Implements the XUM bootloader protocol.
 * 
 * Designed by Grant Ayers (ayers@cs.utah.edu) for
 * XUM project of the Gauss group at the University of Utah.
 * 
 * You may reuse this code with proper attribution.
 *
 * Summer 2010
*/
using System;
using System.Collections.Generic;
using System.ComponentModel;
using System.Data;
using System.Drawing;
using System.Linq;
using System.Text;
using System.Windows.Forms;
using System.IO;
using System.Security.Permissions;
using System.IO.Ports;
using System.Diagnostics;


namespace XumBootloader_GUI
{
    public partial class Form1 : Form
    {
        public Form1()
        {
            InitializeComponent();
            
            /* Initialize some form components */
            foreach (string str in Program.getComPortNames())
            {
                try
                {
                    serialPortSelect.Items.Add(str);
                }
                catch (System.ArgumentNullException ex)
                {
                    MessageBox.Show(this, ex.Message + "\nYour system doesn't have any serial ports!");
                    Environment.Exit(0);
                }
            }
            try
            {
                serialPortSelect.SelectedIndex = 0;
                baudRateSelect.SelectedItem = "115200";
                serialPort1.Parity = System.IO.Ports.Parity.None;
                serialPort1.DataBits = 8;
                serialPort1.StopBits = System.IO.Ports.StopBits.One;
                serialPort1.Handshake = System.IO.Ports.Handshake.None;
                //serialPort1.WriteBufferSize = 1048576;  // 1MB
                //serialPort1.ReadBufferSize = 1048576;   // 1MB
                serialPort1.ReadTimeout = 1400; // 1400ms more than enough for 1KB even at 9600 baud.
                serialPort1.WriteTimeout = 1400;
                labelSuccess.Text = "";
                labelSize.Text = "";
                progressBar1.Visible = false;
            }
            catch (Exception ex)
            {
                MessageBox.Show(this, ex.Message);
                Environment.Exit(1);
            }

        }

        private void selectFile_Click(object sender, EventArgs e)
        {
            labelSuccess.Text = "";
            progressBar1.Visible = false;
            if (openFileDialog1.ShowDialog() == DialogResult.OK)
            {
                string filename = openFileDialog1.FileName;
                labelFilename.Text = "Selected: " + filename;
                try
                {
                    System.IO.FileInfo fi = new System.IO.FileInfo(filename);
                    long filesize = fi.Length;
                    if (filesize > 1048576) // currently just 1MB
                    {
                        MessageBox.Show(this, "Maxiumum supported file size is 1MB");
                        labelFilename.Text = "";
                        labelSize.Text = "";
                        openFileDialog1.FileName = "";
                        return;
                    }
                    labelSize.Text = "Size: " + filesize + " bytes.";
                    if ((filesize % 4) != 0)
                        labelSize.Text = labelSize.Text + " (Not aligned to 32 bits!)";
                }
                catch (Exception ex)
                {
                    MessageBox.Show(this, ex.Message);
                    return;
                }
            }
            if (openFileDialog1.FileName == "")
            {
                labelFilename.Text = "No File Selected.";
                labelSize.Text = "";
            }
        }

        private void sendButton_Click(object sender, EventArgs e)
        {
            try
            {
                labelSuccess.Text = "";
                if (openFileDialog1.FileName == "")
                {
                    MessageBox.Show(this, "You must first select a file.");
                    return;
                }
                object portName = serialPortSelect.SelectedItem;
                if (portName == null)
                {
                    MessageBox.Show(this, "You must first select a serial port.");
                    return;
                }
                serialPort1.PortName = portName.ToString();
                object baudRate = baudRateSelect.SelectedItem;
                if (baudRate == null)
                {
                    MessageBox.Show(this, "You must first select the baud rate.");
                    return;
                }
                serialPort1.BaudRate = Convert.ToInt32(baudRate.ToString());
                if (serialPort1.IsOpen) // this doesn't work...
                {
                    MessageBox.Show(this, "Serial port " + portName + " could not be opened and may be in use");
                    return;
                }
                string filename = openFileDialog1.FileName;
                int fileSizeBytes = 0;
                System.IO.FileInfo fi = new System.IO.FileInfo(filename);
                fileSizeBytes = (int)fi.Length; // already verified <= 1MB
                labelSize.Text = "Size: " + fileSizeBytes + " bytes.";
                int offsetWords;
                string offsetStr = setOffset.Text;
                if (offsetStr == "")
                {
                    MessageBox.Show(this, "You must enter an offset between 0 and 262143");
                    return;
                }
                try
                {
                    offsetWords = Convert.ToInt32(offsetStr, 10);
                }
                catch (Exception)
                {
                    MessageBox.Show(this, "\"" + offsetStr + "\" is not a valid number between 0 and 262143");
                    return;
                }
                if ((offsetWords < 0) || (offsetWords > (1048576 - fileSizeBytes)))
                {
                    MessageBox.Show(this, "Offset must be between 0 and (262144 - number of words).");
                    return;
                }
                sendButton.Enabled = false;
                serialPort1.Open();
                /* Send the data */
                fileSizeBytes--;
                UInt32 size_head = (UInt32)((fileSizeBytes / 4));
                UInt32 offs_head = (UInt32)offsetWords;
                byte size1 = (byte)((size_head << 14) >> 30);
                byte size2 = (byte)((size_head << 16) >> 24);
                byte size3 = (byte)((size_head << 24) >> 24);
                byte offs1 = (byte)((offs_head << 14) >> 30);
                byte offs2 = (byte)((offs_head << 16) >> 24);
                byte offs3 = (byte)((offs_head << 24) >> 24);
                byte[] header = { 0x58, 0x55, 0x4d, size1, size2, size3, offs1, offs2, offs3 }; // 'X''U''M' followed by size, offset.
                serialPort1.Write(header, 0, header.Length);
                fileSizeBytes++;

                /* Check that the copy of size3 came back */
                serialPort1.ReadTimeout = 500;  // 500 ms to respond is more than enough.
                byte[] response = new byte[1];
                try
                {
                    serialPort1.Read(response, 0, 1);
                }
                catch (System.TimeoutException)
                {
                    MessageBox.Show(this, "The device did not respond.\n\nIt may be off, disconnected, set at a different baud rate," +
                        "\nor not listening for the XUM boot protocol.");
                    serialPort1.Close();
                    sendButton.Enabled = true;
                    return;
                }
                if (response[0] != size3)
                {
                    MessageBox.Show(this, "An unexpected response was received from the device.\n" +
                        "(Expected " + size3 + ", Received " + response[0] + ")\n\n" +
                        "Make sure it is configured for the XUM boot protocol.");
                    serialPort1.Close();
                    sendButton.Enabled = true;
                    return;
                }

                /* Send and verify the data */
                progressBar1.Visible = true;
                progressBar1.Minimum = 0;
                progressBar1.Maximum = fileSizeBytes;
                progressBar1.Value = 0;
                byte[] fileData = new byte[fileSizeBytes];
                byte[] echoData = new byte[1024];
                System.IO.BinaryReader file = new System.IO.BinaryReader(File.Open(filename, FileMode.Open));
                file.Read(fileData, 0, fileSizeBytes);
                file.Close();
                int sentByteCount = 0;
                int currentSendSize = 0;
                Stopwatch stopwatch = new Stopwatch();
                stopwatch.Start();
                while (sentByteCount < fileSizeBytes)
                {
                    if ((fileSizeBytes - sentByteCount) >= 1024)
                        currentSendSize = 1024;
                    else
                        currentSendSize = (fileSizeBytes - sentByteCount);

                    serialPort1.Write(fileData, sentByteCount, currentSendSize);
                    long waitStart = stopwatch.ElapsedMilliseconds;
                    while (serialPort1.BytesToRead < currentSendSize) {
                        if ((stopwatch.ElapsedMilliseconds - waitStart) > 4000)
                        {
                            MessageBox.Show(this, "The device stopped responding. Giving up." + "\nDevice Ack'd : " + serialPort1.BytesToRead + " of " + currentSendSize);
                            serialPort1.Close();
                            sendButton.Enabled = true;
                            progressBar1.Value = 0;
                            progressBar1.Visible = false;
                            return;
                        }
                    }
                    if (serialPort1.BytesToRead > currentSendSize)
                    {
                        MessageBox.Show(this, "The device sent back an extra " + (serialPort1.BytesToRead - currentSendSize) + " byte(s), which was unexpected.");
                    }
                    serialPort1.Read(echoData, 0, currentSendSize);
                    for (int i = 0; i < currentSendSize; i++)
                    {
                        if (fileData[sentByteCount + i] != echoData[i])
                        {
                            MessageBox.Show(this, "The device sent back non-matching data. There was a transmission error." +
                            "\n\n" + (sentByteCount + i) + " bytes were transmitted successfully.");
                            serialPort1.Close();
                            sendButton.Enabled = true;
                            return;
                        }
                    }
                    sentByteCount += currentSendSize;
                    progressBar1.Value = sentByteCount;
                }
                stopwatch.Stop();
                labelSuccess.Text = "Success (" + (stopwatch.ElapsedMilliseconds / 1000.0) + " seconds).";
                serialPort1.Close();
                sendButton.Enabled = true;
            }
            catch (Exception ex)
            {
                MessageBox.Show(this, ex.Message);
                if (serialPort1.IsOpen)
                    serialPort1.Close();
                sendButton.Enabled = true;
                progressBar1.Value = 0;
                return;
            }

        }


    }
}