arduino_ci 0.3.0 → 1.3.0

data/cpp/arduino/HardwareSerial.h

@@ -44,19 +44,19 @@ class HardwareSerial : public StreamTape
     operator bool() { return true; }
 };
 
-#if defined(UBRRH) || defined(UBRR0H)
+#if NUM_SERIAL_PORTS >= 1
   extern HardwareSerial Serial;
   #define HAVE_HWSERIAL0
 #endif
-#if defined(UBRR1H)
+#if NUM_SERIAL_PORTS >= 2
   extern HardwareSerial Serial1;
   #define HAVE_HWSERIAL1
 #endif
-#if defined(UBRR2H)
+#if NUM_SERIAL_PORTS >= 3
   extern HardwareSerial Serial2;
   #define HAVE_HWSERIAL2
 #endif
-#if defined(UBRR3H)
+#if NUM_SERIAL_PORTS >= 4
   extern HardwareSerial Serial3;
   #define HAVE_HWSERIAL3
 #endif

data/cpp/arduino/IPAddress.h

@@ -0,0 +1,59 @@
+#pragma once
+
+#include <stdint.h>
+
+class IPAddress {
+private:
+  union {
+    uint8_t bytes[4];
+    uint32_t dword;
+    operator uint8_t *() const { return (uint8_t *)bytes; }
+  } _address;
+
+public:
+  // Constructors
+  IPAddress() : IPAddress(0, 0, 0, 0) {}
+  IPAddress(uint8_t octet1, uint8_t octet2, uint8_t octet3, uint8_t octet4) {
+    _address.bytes[0] = octet1;
+    _address.bytes[1] = octet2;
+    _address.bytes[2] = octet3;
+    _address.bytes[3] = octet4;
+  }
+  IPAddress(uint32_t dword) { _address.dword = dword; }
+  IPAddress(const uint8_t bytes[]) {
+    _address.bytes[0] = bytes[0];
+    _address.bytes[1] = bytes[1];
+    _address.bytes[2] = bytes[2];
+    _address.bytes[3] = bytes[3];
+  }
+  IPAddress(unsigned long dword) { _address.dword = (uint32_t)dword; }
+
+  // Accessors
+  uint32_t asWord() const { return _address.dword; }
+  uint8_t *raw_address() { return _address.bytes; }
+
+  // Comparisons
+  bool operator==(const IPAddress &rhs) const {
+    return _address.dword == rhs.asWord();
+  }
+
+  bool operator!=(const IPAddress &rhs) const {
+    return _address.dword != rhs.asWord();
+  }
+
+  // Indexing
+  uint8_t operator[](int index) const { return _address.bytes[index]; }
+  uint8_t &operator[](int index) { return _address.bytes[index]; }
+
+  // Conversions
+  operator uint32_t() const { return _address.dword; };
+
+  friend class EthernetClass;
+  friend class UDP;
+  friend class Client;
+  friend class Server;
+  friend class DhcpClass;
+  friend class DNSClient;
+};
+
+const IPAddress INADDR_NONE(0, 0, 0, 0);

data/cpp/arduino/Print.h

@@ -2,6 +2,8 @@
 
 #include <stdio.h>
 #include <avr/pgmspace.h>
+
+#include "Printable.h"
 #include "WString.h"
 
 #define DEC 10
@@ -12,22 +14,17 @@
 #endif
 #define BIN 2
 
-class Print;
-
-class Printable
-{
-  public:
-    virtual size_t printTo(Print& p) const = 0;
-};
-
 class Print
 {
+  private:
+    int write_error;
+  protected:
+    void setWriteError(int err = 1) { write_error = err; }
   public:
-    Print() {}
+    Print() : write_error(0) {}
 
-    // Arduino's version of this is richer but until I see an actual error case I'm not sure how to mock
-    int getWriteError() { return 0; }
-    void clearWriteError() { }
+    int getWriteError() { return write_error; }
+    void clearWriteError() { setWriteError(0); }
     virtual int availableForWrite() { return 0; }
 
     virtual size_t write(uint8_t) = 0;

data/cpp/arduino/Printable.h

@@ -0,0 +1,8 @@
+#pragma once
+
+class Print;
+
+class Printable {
+public:
+  virtual size_t printTo(Print &p) const = 0;
+};

data/cpp/arduino/SPI.h

@@ -40,7 +40,11 @@
 
 class SPISettings {
 public:
-  SPISettings(uint32_t clock, uint8_t bitOrder, uint8_t dataMode){};
+  uint8_t bitOrder;
+
+  SPISettings(uint32_t clock, uint8_t bitOrder = MSBFIRST, uint8_t dataMode = SPI_MODE0) {
+    this->bitOrder = bitOrder;
+  };
   SPISettings(){};
 };
 
@@ -68,6 +72,7 @@ public:
   // and configure the correct settings.
   void beginTransaction(SPISettings settings)
   {
+    this->bitOrder = settings.bitOrder;
     #ifdef SPI_TRANSACTION_MISMATCH_LED
     if (inTransactionFlag) {
       pinMode(SPI_TRANSACTION_MISMATCH_LED, OUTPUT);
@@ -94,10 +99,12 @@ public:
   uint16_t transfer16(uint16_t data) {
     union { uint16_t val; struct { uint8_t lsb; uint8_t msb; }; } in, out;
     in.val = data;
-    if (!(SPCR & (1 << DORD))) {
+    if (bitOrder == MSBFIRST) {
       out.msb = transfer(in.msb);
       out.lsb =  transfer(in.lsb);
-    } else {
+    }
+    else
+    {
       out.lsb =  transfer(in.lsb);
       out.msb = transfer(in.msb);
     }
@@ -143,6 +150,7 @@ private:
   #endif
 
   bool isStarted = false;
+  uint8_t bitOrder;
   String* dataIn;
   String* dataOut;
 };

data/cpp/arduino/Server.h

@@ -0,0 +1,5 @@
+#pragma once
+
+#include <Stream.h>
+
+class Server : public Print {};

data/cpp/arduino/Udp.h

@@ -0,0 +1,27 @@
+#pragma once
+
+#include <IPAddress.h>
+#include <Stream.h>
+
+class UDP : public Stream {
+protected:
+  uint8_t *rawIPAddress(IPAddress &addr) { return addr.raw_address(); };
+
+public:
+  UDP() {
+    // The Stream mock defines a String buffer but never puts anyting in it!
+    if (!mGodmodeDataIn) {
+      mGodmodeDataIn = new String;
+    }
+  }
+  ~UDP() {
+    if (mGodmodeDataIn) {
+      delete mGodmodeDataIn;
+      mGodmodeDataIn = nullptr;
+    }
+  }
+  virtual size_t write(uint8_t value) {
+    mGodmodeDataIn->concat(value);
+    return 1;
+  }
+};

data/cpp/arduino/Wire.h

@@ -1,138 +1,258 @@
+/*
+ * The Wire Library (https://www.arduino.cc/en/Reference/Wire)
+ * allows you to communicate with I2C/TWI devices. The general
+ * TWI protocol supports one "master" device and many "slave"
+ * devices that share the same two wires (SDA and SCL for data
+ * and clock respectively).
+ *
+ * You initialize the library by calling begin() as a master or
+ * begin(myAddress) as a slave (with an int from 8-127). In the
+ * initial mock implementation we support only the master role.
+ *
+ * To send bytes from a master to a slave, start with
+ * beginTransmission(slaveAddress), then use write(byte) to
+ * enqueue data, and finish with endTransmission().
+ *
+ * When a master wants to read, it starts with a call to
+ * requestFrom(slaveAddress, quantity) which blocks until the
+ * request finishes. The return value is either 0 (if the slave
+ * does not respond) or the number of bytes requested (which
+ * might be more than the number sent since reading is simply
+ * looking at a pin value at each clock tick).
+ *
+ * A master can write to or read from two or more slaves in
+ * quick succession (say, during one loop() function), so our
+ * mock needs to support preloading data to be read from multiple
+ * slaves and archive data sent to multiple slaves.
+ *
+ * In the mock, this is handled by having an array of wireData_t
+ * structures, each of which contains a deque for input and a
+ * deque for output. You can preload data to be read and you can
+ * look at a log of data that has been written.
+ */
 
 #pragma once
 
 #include <inttypes.h>
 #include "Stream.h"
+#include <cassert>
+#include <deque>
+using std::deque;
+
+const size_t SLAVE_COUNT = 128;
+const size_t BUFFER_LENGTH = 32;
+
+struct wireData_t {
+  uint8_t misoSize;          // bytes remaining for this read
+  uint8_t mosiSize;          // bytes included in this write
+  deque<uint8_t> misoBuffer; // master in, slave out
+  deque<uint8_t> mosiBuffer; // master out, slave in
+};
+
+// Some inspiration taken from
+// https://github.com/arduino/ArduinoCore-megaavr/blob/d2a81093ba66d22dbda14c30d146c231c5910734/libraries/Wire/src/Wire.cpp
+class TwoWire : public ObservableDataStream {
+private:
+  bool _didBegin = false;
+  wireData_t* in = nullptr;  // pointer to current slave for writing
+  wireData_t* out = nullptr; // pointer to current slave for reading
+  wireData_t slaves[SLAVE_COUNT];
 
-class TwoWire : public ObservableDataStream
-{
 public:
-  TwoWire() {
+
+  //////////////////////////////////////////////////////////////////////////////////////////////
+  // testing methods
+  //////////////////////////////////////////////////////////////////////////////////////////////
+
+  // initialize all the mocks
+  void resetMocks() {
+    _didBegin = false;
+    in = nullptr;  // pointer to current slave for writing
+    out = nullptr; // pointer to current slave for reading
+    for (int i = 0; i < SLAVE_COUNT; ++i) {
+      slaves[i].misoSize = 0;
+      slaves[i].mosiSize = 0;
+      slaves[i].misoBuffer.clear();
+      slaves[i].mosiBuffer.clear();
+    }
   }
 
-  // https://www.arduino.cc/en/Reference/WireBegin
-  // Initiate the Wire library and join the I2C bus as a master or slave. This should normally be called only once.
-  void begin() {
-    isMaster = true;
+  // to verify that Wire.begin() was called at some point
+  bool didBegin() { return _didBegin; }
+
+  // to access the MISO buffer, which allows you to mock what the master will read in a request
+  deque<uint8_t>* getMiso(uint8_t address) {
+    return &slaves[address].misoBuffer;
   }
-  void begin(int address) {
-    i2cAddress = address;
-    isMaster = false;
+
+  // to access the MOSI buffer, which records what the master sends during a write
+  deque<uint8_t>* getMosi(uint8_t address) {
+    return &slaves[address].mosiBuffer;
   }
-  void begin(uint8_t address) {
-    begin((int)address);
+
+
+  //////////////////////////////////////////////////////////////////////////////////////////////
+  // mock implementation
+  //////////////////////////////////////////////////////////////////////////////////////////////
+
+  // constructor initializes internal data
+  TwoWire() {
+    resetMocks();
   }
-  void end() {
-    // TODO: implement
+
+  // https://www.arduino.cc/en/Reference/WireBegin
+  // Initiate the Wire library and join the I2C bus as a master or slave. This
+  // should normally be called only once.
+  void begin() { begin(0); }
+  void begin(uint8_t address) {
+    assert(address == 0);
+    _didBegin = true;
   }
+  void begin(int address) { begin((uint8_t)address); }
+  // NOTE: end() is not part of the published API so we ignore it
+  void end() {}
 
   // https://www.arduino.cc/en/Reference/WireSetClock
-  // This function modifies the clock frequency for I2C communication. I2C slave devices have no minimum working
-  // clock frequency, however 100KHz is usually the baseline.
-  void setClock(uint32_t) {
-    // TODO: implement?
-  }
+  // This function modifies the clock frequency for I2C communication. I2C slave
+  // devices have no minimum working clock frequency, however 100KHz is usually
+  // the baseline.
+  // Since the mock does not actually write pins we ignore this.
+  void setClock(uint32_t clock) {}
 
   // https://www.arduino.cc/en/Reference/WireBeginTransmission
-  // Begin a transmission to the I2C slave device with the given address. Subsequently, queue bytes for
-  // transmission with the write() function and transmit them by calling endTransmission().
-  void beginTransmission(int address) {
-    // TODO: implement
-  }
+  // Begin a transmission to the I2C slave device with the given address.
+  // Subsequently, queue bytes for transmission with the write() function and
+  // transmit them by calling endTransmission().
+  // For the mock we update our output to the proper destination.
   void beginTransmission(uint8_t address) {
-    beginTransmission((int)address);
+    assert(_didBegin);
+    assert(address > 0 && address < SLAVE_COUNT);
+    assert(out == nullptr);
+    out = &slaves[address];
+    out->mosiSize = 0;
   }
+  void beginTransmission(int address) { beginTransmission((uint8_t)address); }
 
   // https://www.arduino.cc/en/Reference/WireEndTransmission
-  // Ends a transmission to a slave device that was begun by beginTransmission() and transmits the bytes that were
-  // queued by write().
-  uint8_t endTransmission(uint8_t sendStop) {
-    // TODO: implement
+  // Ends a transmission to a slave device that was begun by beginTransmission()
+  // and transmits the bytes that were queued by write().
+  // In the mock we just leave the bytes there in the buffer
+  // to be read by the testing API and we ignore the sendStop.
+  uint8_t endTransmission(bool sendStop) {
+    assert(_didBegin);
+    assert(out);
+    out = nullptr;
     return 0; // success
   }
-  uint8_t endTransmission(void) {
-    return endTransmission((uint8_t)true);
-  }
+  uint8_t endTransmission(void) { return endTransmission(true); }
 
   // https://www.arduino.cc/en/Reference/WireRequestFrom
-  // Used by the master to request bytes from a slave device. The bytes may then be retrieved with the
-  // available() and read() functions.
-  uint8_t requestFrom(int address, int quantity, int stop) {
-    // TODO: implement
-    return 0; // number of bytes returned from the slave device
+  // Used by the master to request bytes from a slave device. The bytes may then
+  // be retrieved with the available() and read() functions.
+  uint8_t requestFrom(uint8_t address, uint8_t quantity, uint32_t _iaddress, uint8_t _isize, uint8_t stop) {
+    assert(_didBegin);
+    assert(address > 0 && address < SLAVE_COUNT);
+    assert(quantity <= BUFFER_LENGTH);
+    in = &slaves[address];
+    // do we have enough data in the input buffer
+    if (quantity <= (in->misoBuffer).size()) { // enough data
+      in->misoSize = quantity;
+      return quantity;
+    } else { // not enough data
+      in->misoSize = 0;
+      in = nullptr;
+      return 0;
+    }
   }
-  uint8_t requestFrom(int address, int quantity) {
-    int stop = true;
-    return requestFrom(address, quantity, stop);
+
+  uint8_t requestFrom(uint8_t address, uint8_t quantity, uint8_t stop) {
+    return requestFrom((uint8_t)address, (uint8_t)quantity, (uint32_t)0, (uint8_t)0, (uint8_t)stop);
   }
+
   uint8_t requestFrom(uint8_t address, uint8_t quantity) {
-    return requestFrom((int)address, (int)quantity);
+    return requestFrom((uint8_t)address, (uint8_t)quantity, (uint8_t)true);
   }
-  uint8_t requestFrom(uint8_t address, uint8_t quantity, uint8_t stop) {
-    return requestFrom((int)address, (int)quantity, (int)stop);
+
+  uint8_t requestFrom(int address, int quantity) {
+    return requestFrom((uint8_t)address, (uint8_t)quantity, (uint8_t)true);
   }
-  uint8_t requestFrom(uint8_t, uint8_t, uint32_t, uint8_t, uint8_t) {
-    // TODO: implement
-    return 0;
+  uint8_t requestFrom(int address, int quantity, int stop) {
+    return requestFrom((uint8_t)address, (uint8_t)quantity, (uint8_t)stop);
   }
 
   // https://www.arduino.cc/en/Reference/WireWrite
-  // Writes data from a slave device in response to a request from a master, or queues bytes for transmission from a
-  // master to slave device (in-between calls to beginTransmission() and endTransmission()).
+  // Writes data from a slave device in response to a request from a master, or
+  // queues bytes for transmission from a master to slave device (in-between
+  // calls to beginTransmission() and endTransmission()).
   size_t write(uint8_t value) {
-    // TODO: implement
-    return 0; // number of bytes written
+    assert(out);
+    assert(++(out->mosiSize) <= BUFFER_LENGTH);
+    (out->mosiBuffer).push_back(value);
+    return 1; // number of bytes written
+  }
+  size_t write(const char *str) {
+    return str == NULL ? 0 : write((const uint8_t *)str, String(str).length());
   }
-  size_t write(const char *str) { return str == NULL ? 0 : write((const uint8_t *)str, String(str).length()); }
   size_t write(const uint8_t *buffer, size_t size) {
     size_t n;
-    for (n = 0; size && write(*buffer++) && ++n; --size);
+    for (n = 0; size && write(*buffer++) && ++n; --size)
+      ;
     return n;
   }
-  size_t write(const char *buffer, size_t size) { return write((const uint8_t *)buffer, size); }
+  size_t write(const char *buffer, size_t size) {
+    return write((const uint8_t *)buffer, size);
+  }
   size_t write(unsigned long n) { return write((uint8_t)n); }
   size_t write(long n) { return write((uint8_t)n); }
   size_t write(unsigned int n) { return write((uint8_t)n); }
   size_t write(int n) { return write((uint8_t)n); }
 
   // https://www.arduino.cc/en/Reference/WireAvailable
-  // Returns the number of bytes available for retrieval with read(). This should be called on a master device after a
-  // call to requestFrom() or on a slave inside the onReceive() handler.
+  // Returns the number of bytes available for retrieval with read(). This
+  // should be called on a master device after a call to requestFrom() or on a
+  // slave inside the onReceive() handler.
   int available(void) {
-    // TODO: implement
-    return 0; // number of bytes available for reading
+    assert(in);
+    return in->misoSize;
   }
 
   // https://www.arduino.cc/en/Reference/WireRead
-  // Reads a byte that was transmitted from a slave device to a master after a call to requestFrom() or was transmitted
-  // from a master to a slave. read() inherits from the Stream utility class.
-  int read(void) {
-    // TODO: implement
-    return '\0'; // The next byte received
+  // Reads a byte that was transmitted from a slave device to a master after a
+  // call to requestFrom() or was transmitted from a master to a slave. read()
+  // inherits from the Stream utility class.
+  // In the mock we simply return the next byte from the input buffer.
+  uint8_t read(void) {
+    uint8_t value = peek();
+    --in->misoSize;
+    in->misoBuffer.pop_front();
+    return value; // The next byte received
   }
-  int peek(void) {
-    // TODO: implement
-    return 0;
+
+  // part of the Stream API
+  uint8_t peek(void) {
+    assert(in);
+    assert(0 < in->misoSize);
+    return in->misoBuffer.front(); // The next byte received
   }
+
+  // part of the Stream API
   void flush(void) {
-    // TODO: implement
+    // NOTE: commented out in the megaavr repository
+    // data already at the (mock) destination
   }
 
   // https://www.arduino.cc/en/Reference/WireOnReceive
-  // Registers a function to be called when a slave device receives a transmission from a master.
-  void onReceive( void (*callback)(int) ) {
-    // TODO: implement
-  }
+  // Registers a function to be called when a slave device receives a
+  // transmission from a master.
+  // We don't (yet) support the slave role in the mock
+  void onReceive(void (*callback)(int)) { assert(false); }
 
   // https://www.arduino.cc/en/Reference/WireOnRequest
-  // Register a function to be called when a master requests data from this slave device.
-  void onRequest( void (*callback)(void) ) {
-    // TODO: implement
-  }
+  // Register a function to be called when a master requests data from this
+  // slave device.
+  // We don't (yet) support the slave role in the mock
+  void onRequest(void (*callback)(void)) { assert(false); }
 
-private:
-  int i2cAddress;
-  bool isMaster = false;
 };
 
 extern TwoWire Wire;