Metacat

/**

 *  '$RCSfile$'

 *    Purpose: A Class that implements a socket server for data files

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

 *             National Center for Ecological Analysis and Synthesis

 *    Authors: Chad Berkley

 *    Release: @release@

 *

 *   '$Author$'

 *     '$Date$'

 * '$Revision$'

 */

package edu.ucsb.nceas.metacat;

import java.net.*;

import java.io.*;

import java.util.Properties;

import java.util.Date;

import java.text.SimpleDateFormat;

import java.sql.*;

public class DataFileServer extends Thread 

{

  MetaCatUtil util = new MetaCatUtil();

  static String filedir = "";

  String user = null;

  String sess_id = null;

  int port;

  static String httpstring = "http://dev.nceas.ucsb.edu:8090/metacat/upload/";

  protected Socket s;

  DataFileServer (int port, String user, String sess_id) 

  {

    this.sess_id = sess_id;

    this.user = user;

    this.port = port;

    filedir = util.getOption("datafilepath");

  }

  /**

   * returns true if the port specified is not in use.  false otherwise

   */

  public static boolean portIsAvailable(int port)

  {

    Socket s;

    String host = "localhost";

    try

    {

      s = new Socket(host, port);

      s.close();

      //we could create a socket on this port so the port is not available

      return false;

    }

    catch(UnknownHostException u)

    {

      //it better know localhost!

    }

    catch(IOException i)

    {

      //an ioexception is thrown if the port is not in use

      //System.out.println("port not in use");

      return true;

    }

    return true;

  }

  /**

   * Updates xml_documents with the new data document information

   * @param accnum the accession number of the new data file

   * @param filename the filename of the new data file

   * @param userOwner the document's owner's username

   */

  private void updateDB(String accnum, String filename, String userOwner)

  {

    SimpleDateFormat formatter = new SimpleDateFormat ("yy-MM-dd HH:mm:ss");

    java.util.Date localtime = new java.util.Date();

    String dateString = formatter.format(localtime);

    String sqlDateString = "to_date('" + dateString + "', 'YY-MM-DD HH:MI:SS')";

    StringBuffer sql = new StringBuffer();

    sql.append("insert into xml_documents (docid, docname, doctype, ");

    sql.append("user_owner, user_updated, server_location, rev, date_created");

    sql.append(", date_updated, public_access) values ('");

    sql.append(accnum).append("','").append(filename).append("','BIN','");

    sql.append(userOwner).append("','").append(userOwner).append("','1','");

    sql.append("1',").append(sqlDateString).append(",");

    sql.append(sqlDateString).append(",'0')");

    //System.out.println("sql: " + sql.toString());

    try

    {

      Connection conn = util.openDBConnection();

      PreparedStatement pstmt = conn.prepareStatement(sql.toString());

      pstmt.execute();

      conn.close();

    }

    catch(Exception e)

    {

      System.out.println("error with db connection in DataFileServer.updateDB" +

                         ": " + e.getMessage());

    }

  }

  /**

   * This method is invoked when this class is broken off into a new thread

   */

  public void run () 

  {

    ServerSocket server = null;

    Socket client = null;

    try

    {

      //System.out.println("Starting on port " + port);

      //System.out.println("Waiting for sess_id: " + sess_id);

      server = new ServerSocket((new Integer(port)).intValue());

      //System.out.println("Waiting");

      client = server.accept();

      //System.out.println("Accepted from " + client.getInetAddress());

      InputStream in = client.getInputStream();

      OutputStreamWriter osw = new OutputStreamWriter(client.getOutputStream());

      //System.out.println("output stream received");

      // first read to get the file name

      byte[] str = new byte[1024];

      int val = -1;

      int i = 0;

      //get the filename

      while (val!=0) 

      {

        //System.out.println("i: " + i);

        val = in.read();

        str[i] = (byte)val;

        i++;

      }

      String filename = (new String(str,0, i-1)).trim();

      //System.out.println("filename: " + filename);

      val = -1;

      i = 0;

      str = new byte[1024];

      //get the session id

      while (val!=0) 

      {

        val = in.read();

        str[i] = (byte)val;

        i++;

      }

      String session_id = (new String(str,0, i-1)).trim();

      //System.out.println("session_id: " + session_id);

      if(!sess_id.equals(session_id))

      {//this is an unauthorized connection to this port

        System.out.println("unauthorized request");

        return;

      }

      else

      {

        System.out.println("User authenticated on port " + port);

      }

      String restext=null;

      File outfile = new File(filedir + filename);

      //System.out.println("outfile: " + filedir + filename);

      boolean nameInUse = outfile.exists();

      if(!nameInUse) 

      {

        try

        {

          AccessionNumber anum = new AccessionNumber();

          String accnum = anum.generate(null, "INSERT");

          //osw.write(restext,0,restext.length());

          //osw.flush();

          FileOutputStream out = new FileOutputStream(outfile);

          byte[] buf = new byte[1024];

          int cnt = 0;

          while (cnt!=-1) 

          {

            cnt = in.read(buf, 0, 1024);

            if (cnt!=-1) 

            {

              out.write(buf, 0, cnt);

            }

          }

          out.flush();

          out.close();

          //put the new document info into the DB

          updateDB(accnum, filename, user);

        }

        catch(Exception e)

        {

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

          e.printStackTrace(System.out);

        }

      }

      else

      {//error the file name is already in use on the server

        //send an error message here!

        System.out.println("error: name in use.");

      }

    } 

    catch (IOException ex) 

    {

      ex.printStackTrace();

    } 

    finally 

    {

      try

      {

        client.close();

        System.out.println("client socket closed");

        server.close();

        System.out.println("server socket closed");

      } 

      catch (IOException ex) 

      {

        ex.printStackTrace();

      }

      System.out.println("done");

    }

  }

}

/**

 *        Name: HttpMessage.java

 *     Purpose: Used for Java applet/application communication

 *              with servlet. Based on code given in the book

 *              "Java Servlet Programming" by Hunter & crawford

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

 *              National Center for Ecological Analysis and Synthesis

 *     Authors: Dan Higgins modified for metacat by Chad Berkley

 *

 *     Version: '$Id$'

 *

 *     '$Author$'

 *     '$Date$'

 *     '$Revision$'

 */

package edu.ucsb.nceas.metacat;

import java.io.*;

import java.net.*;

import java.util.*;

public class HttpMessage {

    public String contype;

    URL servlet = null;

    String argString = null;

    static String cookie = null;

    public HttpMessage(URL servlet) {

        this.servlet = servlet;

    }

    /**

     * return the cookie that this message object contains

     */

    public String getCookie()

    {

      return cookie;

    }

    /**

     * Performs a GET request to the previously given servlet

     * with no query string

    */

    public InputStream sendGetMessage() throws IOException {

        return sendGetMessage(null);

    }

    /**

     * Performs a GET request to the previously given servlet

     *Builds a query string from the supplied Properties list.

     */

    public InputStream sendGetMessage(Properties args) throws IOException {

        argString = ""; //default

        if (args != null) {

            argString = "?" + toEncodedString(args);

        }

        URL url = new URL(servlet.toExternalForm() + argString);

        // turn off caching

        URLConnection con = url.openConnection();

        con.setUseCaches(false);

        contype = con.getContentType();

        return con.getInputStream();

    }

    /**

     * Performs a POST request to the previously given servlet

     * with no query string

     */

    public InputStream sendPostMessage() throws IOException {

        return sendPostMessage(null);

    }

    /**

     * Builds post data from the supplied properties list

     */

    public InputStream sendPostMessage(Properties args) throws IOException {

        argString = ""; //default

        if (args != null) {

            argString = toEncodedString(args); 

        }

        URLConnection con = servlet.openConnection();

        if (cookie!=null) {

            int k = cookie.indexOf(";");

            if (k>0) {

                cookie = cookie.substring(0, k);

            }

            //System.out.println("Cookie = " + cookie);

            con.setRequestProperty("Cookie", cookie); 

            con.setRequestProperty("User-Agent", "MORPHO");  // add 10/26/00 by DFH so Metacat can determine where request come from

        }

        //prepare for both input and output

        con.setDoInput(true);

        con.setDoOutput(true);

        //turn off caching

        con.setUseCaches(false);

        //work around a Netscape bug

        //con.setRequestProperty("Content-Type", "application/x-www-form-urlencoded");

        //Write the arguments as post data

        DataOutputStream out = new DataOutputStream(con.getOutputStream());

        out.writeBytes(argString);

        out.flush();

        contype = con.getContentType();

        String temp = con.getHeaderField("Set-Cookie");

        if (temp!=null) {

            cookie = temp;

        }

        //System.out.println(cookie);

        out.close();

        return con.getInputStream();

    }

    /**

     * Returns the urls argument strings

     */

    public String getArgString() {

        String argString1 = argString;

        if (!argString1.startsWith("?")) {

            argString1 = "?"+argString1;}

        return argString1;

    }

    /**

     * Converts a Properties list to a URL-encoded query string

     */

    private String toEncodedString(Properties args) {

        StringBuffer buf = new StringBuffer();

        Enumeration names = args.propertyNames();

        while (names.hasMoreElements()) {

            String name = (String)names.nextElement();

            String value = args.getProperty(name);

            buf.append(URLEncoder.encode(name) + "=" + URLEncoder.encode(value));

            if (names.hasMoreElements()) buf.append("&");

        }

        return buf.toString();

    }

}

/**

 *  '$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$'

 *     '$Date$'

 * '$Revision$'

 */

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;

		String res = "return";

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

                       filename + " cookie: " + cookie);

    try 

    {

      //while(DataFileServer.portIsAvailable(port)) 

      //{//loop until the port is there

      //}

      echoSocket = new Socket(host, port);

      out = echoSocket.getOutputStream();

      in = echoSocket.getInputStream();

    } 

    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);

      DataOutputStream 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);

      // 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());

    }

	  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 = "/home/berkley/xmltodb/test.dat";

      //login url

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

                      "=neetar&qformat=xml";

      System.out.println("u1: " + 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("u2: " + 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();

      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);

      //this for loop is a hack to give the servlet enough time to start

      //the socket thread.  Sometimes it doesn't work

      //for(int j=0; j<100000000; j++) { int x = 1; }

      SendFile(filename, host, port, cookie);

    }

    catch(Exception e)

    {

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

      e.printStackTrace(System.out);

    }

  }

}