Super Duper update to Deluge T2. The manual and the testing scripts are temporary...

[tinyos-2.x.git] / tos / lib / net / Deluge / DelugeManagerP.nc

diff --git a/tos/lib/net/Deluge/DelugeManagerP.nc b/tos/lib/net/Deluge/DelugeManagerP.nc
new file mode 100644 (file)
index 0000000..d7146af
--- /dev/null
+++ b/tos/lib/net/Deluge/DelugeManagerP.nc
@@ -0,0 +1,136 @@
+/* Copyright (c) 2007 Johns Hopkins University.
+*  All rights reserved.
+*
+*  Permission to use, copy, modify, and distribute this software and its
+*  documentation for any purpose, without fee, and without written
+*  agreement is hereby granted, provided that the above copyright
+*  notice, the (updated) modification history and the author appear in
+*  all copies of this source code.
+*
+*  THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS `AS IS'
+*  AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
+*  IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
+*  ARE DISCLAIMED.  IN NO EVENT SHALL THE COPYRIGHT HOLDERS OR CONTRIBUTORS
+*  BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
+*  CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, LOSS OF USE, DATA,
+*  OR PROFITS) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
+*  CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
+*  ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
+*  THE POSSIBILITY OF SUCH DAMAGE.
+*/
+
+/**
+ * @author Razvan Musaloiu-E. <razvanm@cs.jhu.edu>
+ * @author Chieh-Jan Mike Liang <cliang4@cs.jhu.edu>
+ */
+
+generic module DelugeManagerP()
+{
+  uses {
+    interface DisseminationUpdate<DelugeCmd>;
+    interface AMSend as SerialAMSender;
+    interface Receive as SerialAMReceiver;
+    interface Timer<TMilli> as DelayTimer;
+    interface NetProg;
+    interface Leds;
+    interface StorageMap[uint8_t volumeId];
+    interface DelugeMetadata;
+    interface ObjectTransfer;
+    interface DelugeVolumeManager;
+    interface Resource;
+    command void stop();
+  }
+}
+
+implementation
+{
+  typedef nx_struct SerialReqPacket {
+    nx_uint8_t cmd;
+    nx_uint8_t imgNum;
+  } SerialReqPacket;
+  
+  typedef nx_struct SerialReplyPacket {
+    nx_uint8_t error;
+  } SerialReplyPacket;
+
+  message_t serialMsg;
+  DelugeCmd delugeCmd;
+
+  void sendReply(error_t error)
+  {
+    uint8_t len = sizeof(SerialReplyPacket);
+    SerialReplyPacket *reply = (SerialReplyPacket *)call SerialAMSender.getPayload(&serialMsg, len);
+    if (reply == NULL) {
+      return;
+    }
+    reply->error = error;
+    call SerialAMSender.send(AM_BROADCAST_ADDR, &serialMsg, len);
+  }
+
+  event message_t* SerialAMReceiver.receive(message_t* msg, void* payload, uint8_t len)
+  {
+    SerialReqPacket *request = (SerialReqPacket *)payload;
+    memset(&delugeCmd, 0, sizeof(DelugeCmd));
+
+    call stop();
+    delugeCmd.type = request->cmd;
+
+    switch (request->cmd) {
+    case DELUGE_CMD_STOP:
+      call DisseminationUpdate.change(&delugeCmd);
+    case DELUGE_CMD_LOCAL_STOP:
+      sendReply(SUCCESS);
+      call Resource.release();
+      break;
+    case DELUGE_CMD_ONLY_DISSEMINATE:
+    case DELUGE_CMD_DISSEMINATE_AND_REPROGRAM:
+      if (call Resource.immediateRequest() == SUCCESS) {
+       call DelugeMetadata.read(request->imgNum);
+      } else {
+       sendReply(FAIL);
+      }
+      break;
+    case DELUGE_CMD_REPROGRAM:
+    case DELUGE_CMD_REBOOT:
+      call DelayTimer.startOneShot(1024);
+      sendReply(SUCCESS);
+      break;
+    }
+    return msg;
+  }
+
+  event void DelayTimer.fired()
+  {
+    switch (delugeCmd.type) {
+    case DELUGE_CMD_REPROGRAM:
+      call NetProg.programImageAndReboot(call StorageMap.getPhysicalAddress[delugeCmd.imgNum](0));
+      break;
+    case DELUGE_CMD_REBOOT:
+      call NetProg.reboot();
+      break;
+    }
+  }
+
+  event void DelugeMetadata.readDone(uint8_t imgNum, DelugeIdent* ident, error_t error)
+  {
+    delugeCmd.imgNum = imgNum;
+    sendReply(error);
+    if (error != SUCCESS) {
+      return;
+    }
+    switch (delugeCmd.type) {
+    case DELUGE_CMD_ONLY_DISSEMINATE:
+    case DELUGE_CMD_DISSEMINATE_AND_REPROGRAM:
+      delugeCmd.uidhash = ident->uidhash;
+      delugeCmd.size = ident->size;
+      call DisseminationUpdate.change(&delugeCmd);
+      call ObjectTransfer.publish(delugeCmd.uidhash, delugeCmd.size, delugeCmd.imgNum);
+      break;
+    }    
+  }
+
+  event void Resource.granted() {}
+  event void ObjectTransfer.receiveDone(error_t error) {}
+  event void SerialAMSender.sendDone(message_t* msg, error_t error) {}
+  event void DelugeVolumeManager.eraseDone(uint8_t imgNum) {}
+}