Author: sebb
Date: Thu Feb 9 23:19:19 2017
New Revision: 1782415
URL: http://svn.apache.org/viewvc?rev=1782415&view=rev
Log:
CLI app for running TFTPServer
Added:
commons/proper/net/trunk/src/test/java/org/apache/commons/net/tftp/TFTPServerMain.java
(with props)
Added:
commons/proper/net/trunk/src/test/java/org/apache/commons/net/tftp/TFTPServerMain.java
URL:
http://svn.apache.org/viewvc/commons/proper/net/trunk/src/test/java/org/apache/commons/net/tftp/TFTPServerMain.java?rev=1782415&view=auto
==============================================================================
---
commons/proper/net/trunk/src/test/java/org/apache/commons/net/tftp/TFTPServerMain.java
(added)
+++
commons/proper/net/trunk/src/test/java/org/apache/commons/net/tftp/TFTPServerMain.java
Thu Feb 9 23:19:19 2017
@@ -0,0 +1,150 @@
+/*
+ * Licensed to the Apache Software Foundation (ASF) under one or more
+ * contributor license agreements. See the NOTICE file distributed with
+ * this work for additional information regarding copyright ownership.
+ * The ASF licenses this file to You under the Apache License, Version 2.0
+ * (the "License"); you may not use this file except in compliance with
+ * the License. You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package org.apache.commons.net.tftp;
+
+import java.io.File;
+import java.io.IOException;
+import java.util.HashMap;
+import java.util.Map;
+import java.util.Random;
+
+/**
+ * Main class for TFTPServer.
+ * This allows CLI use of the server.
+ */
+public class TFTPServerMain {
+
+ private static final String USAGE =
+ "Usage: TFTPServerMain [options] [port]\n\n" +
+ "port - the port to use (default 6901)\n" +
+ "\t-p path to server directory (default java.io.tempdir)\n" +
+ "\t-r randomly introduce errors\n" +
+ "\t-v verbose (trace packets)\n"
+ ;
+
+ public static void main(String [] args) throws Exception {
+ int port = 6901;
+ int argc;
+ Map<String,String> opts = new HashMap<String,String>();
+ opts.put("-p", System.getProperty("java.io.tmpdir"));
+ // Parse options
+ for (argc = 0; argc < args.length; argc++)
+ {
+ String arg = args[argc];
+ if (arg.startsWith("-"))
+ {
+ if (arg.equals("-v")) {
+ opts.put(arg, arg);
+ } else if (arg.equals("-r")) {
+ opts.put(arg, arg);
+ } else if (arg.equals("-p")) {
+ opts.put(arg, args[++argc]);
+ } else {
+ System.err.println("Error: unrecognized option.");
+ System.err.print(USAGE);
+ System.exit(1);
+ }
+ } else {
+ break;
+ }
+ }
+
+ if (argc < args.length) {
+ port = Integer.parseInt(args[argc]);
+ argc++;
+ }
+ final boolean verbose = opts.containsKey("-v");
+ final boolean randomErrors = opts.containsKey("-r");
+ final Random rand = randomErrors ? new Random() : null;
+
+ final File serverDirectory = new File(opts.get("-p"));
+ System.out.println("Server directory: " + serverDirectory);
+ final TFTPServer tftpS = new TFTPServer(serverDirectory,
serverDirectory, port,
+ TFTPServer.ServerMode.GET_AND_PUT, null, null){
+ @Override
+ void sendData(TFTP tftp, TFTPPacket packet) throws IOException {
+ if (rand == null) {
+ super.sendData(tftp, packet);
+ return;
+ }
+ int rint = rand.nextInt(10);
+ switch(rint) {
+ case 0:
+ System.out.println("Bump port " + packet);
+ int port = packet.getPort();
+ packet.setPort(port+5);
+ super.sendData(tftp, packet);
+ packet.setPort(port);
+ break;
+ case 1:
+ if (packet instanceof TFTPDataPacket) {
+ TFTPDataPacket data = (TFTPDataPacket) packet;
+ System.out.println("Change data block num");
+ data._blockNumber--;
+ super.sendData(tftp, packet);
+ data._blockNumber++;
+ }
+ if (packet instanceof TFTPAckPacket) {
+ TFTPAckPacket ack = (TFTPAckPacket) packet;
+ System.out.println("Change ack block num");
+ ack._blockNumber--;
+ super.sendData(tftp, packet);
+ ack._blockNumber++;
+ }
+ break;
+ case 2:
+ System.out.println("Drop packet: " + packet);
+ break;
+ case 3:
+ System.out.println("Dupe packet: " + packet);
+ super.sendData(tftp, packet);
+ super.sendData(tftp, packet);
+ break;
+ default:
+ super.sendData(tftp, packet);
+ break;
+ }
+ }
+
+ TFTP newTFTP(){
+ if (verbose) {
+ return new TFTP() {
+ @Override
+ protected void trace(String direction, TFTPPacket
packet) {
+ System.out.println(direction + " " +
packet.toString());
+ }
+ };
+ } else {
+ return new TFTP();
+ }
+ }
+ };
+
+ Runtime.getRuntime().addShutdownHook(new Thread() {
+ public void run() {
+ System.out.println("Server shutting down");
+ tftpS.shutdown();
+ System.out.println("Server exit");
+ }
+ }
+ );
+ System.out.println("Started the server on " + port);
+ Thread.sleep(99999999L);
+ }
+
+}
Propchange:
commons/proper/net/trunk/src/test/java/org/apache/commons/net/tftp/TFTPServerMain.java
------------------------------------------------------------------------------
svn:eol-style = native
