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: berkley $'

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

 * '$Revision: 653 $'

 */

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

  }

  /**

   * attempts to connect a socket, returns null if it is not successful

   * returns the connected socket if it is successful.

   */

  public static Socket getSocket(String host, int port)

  {

    Socket s = null;

    try

    {

      s = new Socket(host, port);

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

      //System.out.println("socket connnected");

      return s;

    }

    catch(UnknownHostException u)

    {

    }

    catch(IOException i)

    {

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

      //System.out.println("socket not connected");

      return s;

    }

    return s;

  }

  /**

   * 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

      //System.out.println("socket 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("socket available");

      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 HH24: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;

    OutputStreamWriter osw = 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");

      server.setSoTimeout(30000); //set a 30 second timeout

      client = server.accept();

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

      InputStream in = client.getInputStream();      //in from the client

      OutputStream sout = client.getOutputStream();  //out to the client

      osw = new OutputStreamWriter(sout);

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

      //get the length of the file in bytes

      int val2 = -1;

      long fsize = 0;

      while(val2 != 0)

      {

        val2 = in.read();

        if(val2 != 0)

        {

          fsize = val2;

        }

      }

      //System.out.println("file size: " + fsize);

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

      int filenametemp = 1;

      String fn = filename;

      while(nameInUse)

      {

        filename = fn + filenametemp;

        outfile = new File (filedir + filename);

        nameInUse = outfile.exists();

        filenametemp++;

      }

      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;

        int j = 0;

        while (j < fsize) 

        {

          //System.out.println("1 j:" + j);

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

          if (cnt!=-1) 

          {

            out.write(buf, 0, cnt);

          }

          j += cnt;

        }

        out.flush();

        out.close();

        //put the new document info into the DB

        updateDB(accnum, filename, user);

        //System.out.println("sending new docid: " + accnum);

        osw.write("<?xml version=\"1.0\"?><docid>" + accnum + "</docid>");

        osw.flush();

        osw.close();

        //System.out.println("docid sent");

      }

      catch(Exception e)

      {

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

        try

        {

          osw.write("<?xml version=\"1.0\"?><error>" + e.getMessage() + 

                    "</error>");

          osw.flush();

          osw.close();

        }

        catch(Exception ee)

        {

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

        }

        e.printStackTrace(System.out);

      }

    } 

    catch(InterruptedIOException iioe)

    {

      //the accept timeout passed

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

    }

    catch (IOException ex) 

    {

      ex.printStackTrace();

      try

      {

        osw.write("<?xml version=\"1.0\"?><error>" + ex.getMessage() + 

                  "</error>");

        osw.flush();

        osw.close();

      }

      catch(Exception ee)

      {

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

      }

    } 

    finally 

    {

      try

      {

        server.close();

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

        client.close();

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

      } 

      catch (IOException ex) 

      {

        ex.printStackTrace();

      }

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

    }

  }

}