Metacat

/**

 *  '$RCSfile$'

 *    Purpose: A test driver for the DataFileServer class

 *  Copyright: 2000 Regents of the University of California and the

 *             National Center for Ecological Analysis and Synthesis

 *    Authors: Chad Berkley

 *    Release: @release@

 *

 *   '$Author: berkley $'

 *     '$Date: 2001-01-10 13:05:31 -0800 (Wed, 10 Jan 2001) $'

 * '$Revision: 653 $'

 */

package edu.ucsb.nceas.metacat;

import java.io.*;

import java.lang.*;

import java.util.*;

import java.sql.*;

import java.net.*;

import edu.ucsb.nceas.metacat.*;

import javax.servlet.http.*;

public class DataStreamTest

{

  DataStreamTest()

  {

  }

  /**

   * sends a file to the data file server socket

   */

  public static String SendFile(String filename, String host, int port, 

                                String cookie) 

  {

    Socket echoSocket = null;

    OutputStream out = null;

    InputStream in = null;

    DataOutputStream dsout = null;

    InputStreamReader isr = null;

		String res = "return";

    System.out.println("host: " + host + " port: " + port + " filename: " +

                       filename + " cookie: " + cookie);

    try 

    {

      echoSocket = DataFileServer.getSocket(host, port);

      while(echoSocket == null) 

      {//loop until the port is there

        echoSocket = DataFileServer.getSocket(host, port);

      }

      //echoSocket = new Socket(host, port);

      out = echoSocket.getOutputStream(); //out to the server

      in = echoSocket.getInputStream();   //in from server

      isr = new InputStreamReader(in);

    } 

    catch (UnknownHostException e) 

    {

      System.err.println("Don't know about host: " + host);

      System.out.println("error: " + e.getMessage());

      e.printStackTrace(System.out);

      System.exit(1);

    } 

    catch (IOException e) 

    {

      System.err.println("Couldn't get I/O for "

                         + "the connection to: "+host);

      System.out.println("error: " + e.getMessage());

      e.printStackTrace(System.out);

      System.exit(1);

    }

    try

    {  

      File file = new File(filename);

      dsout = new DataOutputStream(out);

      FileInputStream fs = new FileInputStream(file);

      // first convert the filename to a byte array

      String fn = file.getName();

      byte[] fname = fn.getBytes();

      // now write the string bytes followed by a '0'

      for (int i=0;i<fname.length;i++) 

      {

        dsout.write(fname[i]);    

      }

      dsout.write(0);  // marks end of name info

      //write the session id to the stream

      byte[] cook = cookie.getBytes();

      for(int i=0; i<cook.length; i++)

      {

        dsout.write(cook[i]);

      }

      dsout.write(0);

      //write the length of the file followed by a 0

      long flength = file.length();

      //System.out.println("file sized (B): " + flength);

      dsout.write((int)flength);

      dsout.write(0);

      dsout.flush();

      // now read response from server

      //InputStreamReader isr = new InputStreamReader(in);

      BufferedReader rin = new BufferedReader(isr);

      // now send the file data

      byte[] buf = new byte[1024];

      int cnt = 0;

      int i = 0;

      while (cnt!=-1) 

      {

        cnt = fs.read(buf);

        //System.out.println("i = "+ i +" Bytes read = " + cnt);

        if (cnt!=-1) 

        {

          dsout.write(buf, 0, cnt);

        }

        i++;

      }

      fs.close();

      dsout.flush();

      //dsout.close();

	  }

	  catch (Exception w) 

    {

      System.out.println("error in DataStreamTest: " + w.getMessage());

    }

    try

    {

      while(!isr.ready()) 

      {

        //System.out.println("not ready");

      }

      int msg = isr.read();

      System.out.print("Message from server: ");

      while(msg != -1)

      {

        System.out.print((char)msg);

        System.out.flush();

        msg = isr.read();

      }

      System.out.println();

      dsout.flush();

      dsout.close();

    }

    catch(Exception e)

    {

      System.out.println("error: " + e.getMessage());

    }

	  return res;

	}

  public static void main(String[] args)

  {

    System.out.println("Starting DataStreamTest");

    try

    {

      String temp = null;

      String servlet = ":8080/berkley/servlet/metacat";

      String protocol = "http://";

      String host = "alpha.nceas.ucsb.edu";

      String server = protocol + host + servlet;

      String filename = null;

      if(args.length == 0)

      {

        filename = "/home/berkley/xmltodb/test.dat";

      }

      else

      {

        filename = args[0];

      }

      //login url

      String u1str = server + "?action=login&username=higgins&password" +

                      "=neetar&qformat=xml";

      System.out.println("url 1: " + u1str);

      URL u1 = new URL(u1str);

      HttpMessage msg = new HttpMessage(u1);

      msg.sendPostMessage();

      String cookie = msg.getCookie();

      //get the port to send the data to

      System.out.println("url 2: " + server + "?action=getdataport");

      URL u2 = new URL(server + "?action=getdataport");

      HttpMessage msg2 = new HttpMessage(u2);

      InputStream in = msg2.sendPostMessage();

      InputStreamReader isr = new InputStreamReader(in);

      char c;

      int i = isr.read();

      String temp2 = null;

      while(i != -1)

      { //get the server response to the getdataport request

        c = (char)i;

        temp2 += c;

        i = isr.read();

      }

      int temp3 = temp2.indexOf("<", 32);

      temp2 = temp2.substring(32, temp3); //parse out the port

      //the port number taken from the xml encoding

      temp2 = temp2.trim();

      int port = (new Integer(temp2)).intValue();

      //while(portIsAvailable(port)) {int x = 1;}

      int index = cookie.indexOf("JSESSIONID=");

      index = cookie.indexOf("=", index);

      int index2 = cookie.indexOf(";", index);

      //get just the session id information from the cookie

      cookie = cookie.substring(index+1, index2);

      SendFile(filename, host, port, cookie);

    }

    catch(Exception e)

    {

      System.out.println("error in main: " + e.getMessage());

      e.printStackTrace(System.out);

    }

  }

}