deluge.java

无线通信的主要编程软件,是无线通信工作人员的必备工具,关天相关教程我会在后续传上.

// $Id: Deluge.java,v 1.14 2005/08/28 00:01:34 jwhui Exp $

/*									tab:2
 *
 *
 * "Copyright (c) 2000-2005 The Regents of the University  of California.  
 * 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 following
 * two paragraphs and the author appear in all copies of this software.
 * 
 * IN NO EVENT SHALL THE UNIVERSITY OF CALIFORNIA BE LIABLE TO ANY PARTY FOR
 * DIRECT, INDIRECT, SPECIAL, INCIDENTAL, OR CONSEQUENTIAL DAMAGES ARISING OUT
 * OF THE USE OF THIS SOFTWARE AND ITS DOCUMENTATION, EVEN IF THE UNIVERSITY OF
 * CALIFORNIA HAS BEEN ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 * 
 * THE UNIVERSITY OF CALIFORNIA SPECIFICALLY DISCLAIMS ANY WARRANTIES,
 * INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY
 * AND FITNESS FOR A PARTICULAR PURPOSE.  THE SOFTWARE PROVIDED HEREUNDER IS
 * ON AN "AS IS" BASIS, AND THE UNIVERSITY OF CALIFORNIA HAS NO OBLIGATION TO
 * PROVIDE MAINTENANCE, SUPPORT, UPDATES, ENHANCEMENTS, OR MODIFICATIONS."
 *
 */

/**
 * @author Jonathan Hui <[e-mail]>
 */

package net.tinyos.tools;

import net.tinyos.deluge.*;
import net.tinyos.message.*;
import net.tinyos.util.*;

import java.io.*; 
import java.util.*;
import java.text.*;

public class Deluge {

  private static final int S_NONE = 0;
  private static final int S_PING = 1;
  private static final int S_INJECT = 2;
  private static final int S_REBOOT = 3;
  private static final int S_ERASE = 4;
  private static final int S_RESET = 5;
  private static final int S_DUMP = 6;

  private int mode = S_NONE;
  private int imageNum = -1;
  private int nodeid = MoteIF.TOS_BCAST_ADDR;
  private boolean verbose = false;
  private String infile = "";
  private String outfile = "";
  private boolean force = false;

  private MoteIF moteif;

  private void usage() {
    throw new IllegalArgumentException( getUsage() );
  }

  public String getUsage() {
    return
      "usage: java net.tinyos.tools.Deluge <action> [options]\n"
    + "  actions are:\n"
    + "   -p,  --ping            : ping status of node\n"
    + "   -i,  --inject          : inject a binary\n"
    + "   -r,  --reboot          : send reboot command\n"
    + "   -e,  --erase           : erase a binary\n"
    + "   -x,  --reset           : reset image\n"
    + "   -d,  --dump            : dumps data injected to node\n"
    + "  options are:\n"
    + "   -id, --nodeid=<num>    : node id\n"
    + "   -ti, --tosimage=<xml>  : tos_image.xml file (program images)\n"
    + "   -in, --imgnum=<num>    : image num\n"
    + "   -o,  --outfile=<xml>   : output file for dump\n"
    + "   -f,  --force           : force the operation, do not ask y/n\n"
    + "   -v,  --verbose         : print all sent/received msgs\n"
    + "   -h,  --help            : print this message\n"
    ;
  }

  private void init( String args[] ) {

    int modeCount = 0;

    if (args.length == 0)
      usage();
    
    for ( int i = 0; i < args.length; i++ ) {
      if (args[i].equals("-p") || args[i].equals("--ping")) {
	mode = S_PING;
	modeCount++;
      }
      else if (args[i].equals("-i") || args[i].equals("--inject")) {
	mode = S_INJECT;
	modeCount++;
      }
      else if (args[i].equals("-r") || args[i].equals("--reboot")) {
	mode = S_REBOOT;
	modeCount++;
      }
      else if (args[i].equals("-e") || args[i].equals("--erase")) {
	mode = S_ERASE;
	modeCount++;
      }
      else if (args[i].equals("-x") || args[i].equals("--reset")) {
	mode = S_RESET;
	modeCount++;
      }
      else if (args[i].equals("-d") || args[i].equals("--dump")) {
	mode = S_DUMP;
	modeCount++;
      }
      else if (args[i].length() > 3 && args[i].substring(0,4).equals("-id=")) {
	nodeid = Integer.parseInt( args[i].substring(4,args[i].length()) );
      }
      else if (args[i].length() > 8 && args[i].substring(0,9).equals("--nodeid=")) {
	nodeid = Integer.parseInt( args[i].substring(9,args[i].length()) );
      }      
      else if (args[i].length() > 3 && args[i].substring(0,4).equals("-ti=")) {
	infile = args[i].substring(4,args[i].length());	
      }
      else if (args[i].length() > 10 && args[i].substring(0,11).equals("--tosimage=")) {
	infile = args[i].substring(11,args[i].length());
      }
      else if (args[i].length() > 3 && args[i].substring(0,4).equals("-in=")) {
	imageNum = Short.parseShort(args[i].substring(4,args[i].length()));
      }
      else if (args[i].length() > 8 && args[i].substring(0,9).equals("--imgnum=")) {
	imageNum = Short.parseShort(args[i].substring(9,args[i].length()));
      }
      else if (args[i].length() > 2 && args[i].substring(0,3).equals("-o=")) {
	outfile = args[i].substring(3,args[i].length());
      }
      else if (args[i].length() > 9 && args[i].substring(0,10).equals("--outfile=")) {
	outfile = args[i].substring(10,args[i].length());
      }
      else if (args[i].equals("-f") || args[i].equals("--force")) {
	force = true;
      }
      else if (args[i].equals("-v") || args[i].equals("--verbose")) {
	verbose = true;
      }
      else if (args[i].equals("-h") || args[i].equals("--help"))
	usage();
      else {
	usage();
      }
    }

    if (modeCount > 1) {
      throw new IllegalArgumentException( "only one action may be specified" );
    }
  }

  public Deluge( String[] args ) {
    init( args );

    try {
      moteif = new MoteIF((Messenger)null);
    } catch (Exception e) {
      e.printStackTrace();
    }

    moteif.start();
  }

  public Deluge( MoteIF moteif, String[] args ) {
    try {
      init( args );
      this.moteif = moteif;
      execute();
    } catch( IllegalArgumentException e ) {
      System.err.println("Error: "+e.getMessage());
    }
  }

  private void execute() {

    Pinger pinger = null;
    
    pinger = new Pinger(moteif, verbose, nodeid);
    pinger.ping();

    System.out.println("--------------------------------------------------");

    switch(mode) {

    case S_PING:

      if (pinger.existsError())
	System.out.println( "Warning: Deluge is not running on one or more nodes." );

      System.out.println("  Currently Executing:");
      System.out.println(pinger.getExecutingIdent());

      for ( int i = 0; i < pinger.getNumImages(); i++ ) {

	if ( imageNum != -1 && imageNum != i )
	  continue;

	if (i == 0)
	  System.out.println("  Stored Image 0 - (Golden Image)");
	else
	  System.out.println("  Stored Image " + i);

	TOSBootImage image = pinger.getImage(i);

	if (pinger.getPingReply(i).get_imgDesc_numPgsComplete() > 0
	    && ( pinger.getPingReply(i).get_imgDesc_vNum() != DelugeConsts.DELUGE_INVALID_VNUM) ) {
	  System.out.println(image);
	  System.out.println("    Num Pages:   " +
			     pinger.getPingReply(i).get_imgDesc_numPgsComplete() +
			     "/" +
			     pinger.getPingReply(i).get_imgDesc_numPgs());
	}
	else if (pinger.getPingReply(i).get_imgDesc_numPgsComplete() == 0
		 && pinger.getPingReply(i).get_imgDesc_numPgs() > 0) {
	  System.out.println("    Prog Name:   <unavailable>");
	  System.out.println("    Compiled On: <unavailable>");
	  System.out.println("    Platform:    <unavailable>");
	  System.out.println("    User ID:     <unavailable>");
	  System.out.println("    Hostname:    <unavailable>");
	  System.out.println("    User Hash:   <unavailable>");
	  System.out.println("    Num Pages:   " +
			     pinger.getPingReply(i).get_imgDesc_numPgsComplete() +
			     "/" +
			     pinger.getPingReply(i).get_imgDesc_numPgs());
	}
	else {
	  System.out.println("    Prog Name:   N/A");
	  System.out.println("    Compiled On: N/A");
	  System.out.println("    Platform:    N/A");
	  System.out.println("    User ID:     N/A");
	  System.out.println("    Hostname:    N/A");
	  System.out.println("    User Hash:   N/A");
	  System.out.println("    Num Pages:   N/A");
	}
      }

      break;

    case S_DUMP:

      if (pinger.existsError())
	System.out.println( "Warning: Deluge is not running on one or more nodes." );

      Downloader downloader = new Downloader(pinger, imageNum, moteif, verbose, outfile);
      
      downloader.extract();

      break;

    case S_INJECT:

      if (pinger.existsError())
	System.out.println( "Warning: Deluge is not running on one or more nodes." );

      TOSBootImage newImage = new TOSBootImage(infile);

      System.out.println("--------------------------------------------------");

      ImageInjector injector = new ImageInjector(pinger, imageNum, newImage,
						 moteif, verbose, force);

      injector.inject();
      
      break;
      
    case S_ERASE:

      if (pinger.existsError())
	System.out.println( "Warning: Deluge is not running on one or more nodes." );

      Eraser eraser = new Eraser(pinger, imageNum, moteif, verbose, force);

      eraser.erase();

      break;

    case S_RESET:

      if (pinger.existsError())
	System.out.println( "Warning: Deluge is not running on one or more nodes." );

      Eraser resetter = new Eraser(pinger, imageNum, moteif, verbose, force);

      resetter.reset();

      break;

    case S_REBOOT:

      if (pinger.existsError())
	System.out.println( "Warning: Deluge is not running on one or more nodes." ); 

      Rebooter rebooter = new Rebooter(pinger, imageNum, moteif, verbose, force);

      rebooter.reboot();

      break;

    default:
      throw new IllegalArgumentException( "unknown mode" );

    }

    System.out.println("--------------------------------------------------");

    System.out.println("DONE!");

  }

  public static void main(String[] args) {
    Deluge deluge = null;
    try {
      deluge = new Deluge(args);
      deluge.execute();
    } catch( IllegalArgumentException e ) {
      if( e.getMessage().startsWith("usage:") ) {
	System.out.println( e.getMessage() );
      } else {
	System.err.println( "Error: "+e.getMessage() );
	System.exit(1);
      }
    }
    System.exit(0);
  }
}