Project

General

Profile

1
/**
2
 *  '$RCSfile$'
3
 *    Purpose: A Class that implements a metadata catalog as a java Servlet
4
 *  Copyright: 2000 Regents of the University of California and the
5
 *             National Center for Ecological Analysis and Synthesis
6
 *    Authors: Matt Jones, Dan Higgins, Jivka Bojilova, Chad Berkley
7
 *    Release: @release@
8
 *
9
 *   '$Author: tao $'
10
 *     '$Date: 2003-04-30 15:04:32 -0700 (Wed, 30 Apr 2003) $'
11
 * '$Revision: 1629 $'
12
 *
13
 * This program is free software; you can redistribute it and/or modify
14
 * it under the terms of the GNU General Public License as published by
15
 * the Free Software Foundation; either version 2 of the License, or
16
 * (at your option) any later version.
17
 *
18
 * This program is distributed in the hope that it will be useful,
19
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
20
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
21
 * GNU General Public License for more details.
22
 *
23
 * You should have received a copy of the GNU General Public License
24
 * along with this program; if not, write to the Free Software
25
 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
26
 */
27

    
28
package edu.ucsb.nceas.metacat;
29

    
30
import com.oreilly.servlet.multipart.FilePart;
31
import com.oreilly.servlet.multipart.MultipartParser;
32
import com.oreilly.servlet.multipart.ParamPart;
33
import com.oreilly.servlet.multipart.Part;
34

    
35
import java.io.File;
36
import java.io.PrintWriter;
37
import java.io.IOException;
38
import java.io.StringReader;
39
import java.io.FileInputStream;
40
import java.io.BufferedInputStream;
41
import java.util.Enumeration;
42
import java.util.Hashtable;
43
import java.util.ResourceBundle;
44
import java.util.Random;
45
import java.util.PropertyResourceBundle;
46
import java.util.Vector;
47
import java.net.URL;
48
import java.net.MalformedURLException;
49
import java.sql.PreparedStatement;
50
import java.sql.ResultSet;
51
import java.sql.Connection;
52
import java.sql.SQLException;
53
import java.lang.reflect.*;
54
import java.net.*;
55
import java.util.zip.*;
56

    
57
import javax.servlet.ServletConfig;
58
import javax.servlet.ServletContext;
59
import javax.servlet.ServletException;
60
import javax.servlet.ServletInputStream;
61
import javax.servlet.http.HttpServlet;
62
import javax.servlet.http.HttpServletRequest;
63
import javax.servlet.http.HttpServletResponse;
64
import javax.servlet.http.HttpSession;
65
import javax.servlet.http.HttpUtils;
66
import javax.servlet.ServletOutputStream;
67

    
68
import org.ecoinformatics.eml.EMLParser;
69

    
70
import org.xml.sax.SAXException;
71

    
72
/**
73
 * A metadata catalog server implemented as a Java Servlet
74
 *
75
 * <p>Valid parameters are:<br>
76
 * action=query -- query the values of all elements and attributes
77
 *                     and return a result set of nodes<br>
78
 * action=squery -- structured query (see pathquery.dtd)<br>
79
 * action= -- export a zip format for data packadge<br>
80
 * action=read -- read any metadata/data file from Metacat and from Internet<br>
81
 * action=insert -- insert an XML document into the database store<br>
82
 * action=update -- update an XML document that is in the database store<br>
83
 * action=delete --  delete an XML document from the database store<br>
84
 * action=validate -- vallidate the xml contained in valtext<br>
85
 * doctype -- document type list returned by the query (publicID)<br>
86
 * qformat=xml -- display resultset from query in XML<br>
87
 * qformat=html -- display resultset from query in HTML<br>
88
 * qformat=zip -- zip resultset from query<br>
89
 * docid=34 -- display the document with the document ID number 34<br>
90
 * doctext -- XML text of the document to load into the database<br>
91
 * acltext -- XML access text for a document to load into the database<br>
92
 * dtdtext -- XML DTD text for a new DTD to load into Metacat XML Catalog<br>
93
 * query -- actual query text (to go with 'action=query' or 'action=squery')<br>
94
 * valtext -- XML text to be validated<br>
95
 * abstractpath -- XPath in metadata document to read from<br>
96
 * action=getaccesscontrol -- retrieve acl info for Metacat document<br>
97
 * action=getdoctypes -- retrieve all doctypes (publicID)<br>
98
 * action=getdtdschema -- retrieve a DTD or Schema file<br>
99
 * action=getdataguide -- retrieve a Data Guide<br>
100
 * action=getprincipals -- retrieve a list of principals in XML<br>
101
 * datadoc -- data document name (id)<br>
102
 * <p>
103
 * The particular combination of parameters that are valid for each
104
 * particular action value is quite specific.  This documentation
105
 * will be reorganized to reflect this information.
106
 */
107
public class MetaCatServlet extends HttpServlet {
108

    
109
  private ServletConfig config = null;
110
  private ServletContext context = null;
111
  private String resultStyleURL = null;
112
  private String xmlcatalogfile = null;
113
  private String saxparser = null;
114
  private String datafilepath = null;
115
  private File dataDirectory = null;
116
  private String servletpath = null;
117
  private String htmlpath = null;
118
  private PropertyResourceBundle options = null;
119
  private MetaCatUtil util = null;
120
  private DBConnectionPool connPool = null;
121
  private static final String PROLOG = "<?xml version=\"1.0\"?>";
122
  private static final String SUCCESS = "<success>";
123
  private static final String SUCCESSCLOSE = "</success>";
124
  private static final String ERROR = "<error>";
125
  private static final String ERRORCLOSE = "</error>";
126
  public static final String SCHEMALOCATIONKEYWORD = ":schemaLocation";
127
  public static final String NONAMESPACELOCATION = ":noNamespaceSchemaLocation";
128
  public static final String EML2KEYWORD =":eml";
129

    
130
  /**
131
   * Initialize the servlet by creating appropriate database connections
132
   */
133
  public void init( ServletConfig config ) throws ServletException {
134
    try {
135
      super.init( config );
136
      this.config = config;
137
      this.context = config.getServletContext();
138
      System.out.println("MetaCatServlet Initialize");
139

    
140
      util = new MetaCatUtil();
141

    
142
      //initial DBConnection pool
143
      connPool = DBConnectionPool.getInstance();
144

    
145
      // Get the configuration file information
146
      resultStyleURL = util.getOption("resultStyleURL");
147
      xmlcatalogfile = util.getOption("xmlcatalogfile");
148
      saxparser = util.getOption("saxparser");
149
      datafilepath = util.getOption("datafilepath");
150
      dataDirectory = new File(datafilepath);
151
      servletpath = util.getOption("servletpath");
152
      htmlpath = util.getOption("htmlpath");
153

    
154

    
155
    } catch ( ServletException ex ) {
156
      throw ex;
157
    } catch (SQLException e) {
158
      MetaCatUtil.debugMessage("Error in MetacatServlet.init: "
159
                                          +e.getMessage(), 20);
160
    }
161
  }
162

    
163
  /**
164
   * Close all db connections from the pool
165
   */
166
  public void destroy() {
167
      // Close all db connection
168
      System.out.println("Destroying MetacatServlet");
169
      connPool.release();
170
  }
171

    
172
  /** Handle "GET" method requests from HTTP clients */
173
  public void doGet (HttpServletRequest request, HttpServletResponse response)
174
    throws ServletException, IOException {
175

    
176
    // Process the data and send back the response
177
    handleGetOrPost(request, response);
178
  }
179

    
180
  /** Handle "POST" method requests from HTTP clients */
181
  public void doPost( HttpServletRequest request, HttpServletResponse response)
182
    throws ServletException, IOException {
183

    
184
    // Process the data and send back the response
185
    handleGetOrPost(request, response);
186
  }
187

    
188
  /**
189
   * Control servlet response depending on the action parameter specified
190
   */
191
  private void handleGetOrPost(HttpServletRequest request,
192
                               HttpServletResponse response)
193
                               throws ServletException, IOException
194
  {
195

    
196
    if ( util == null ) {
197
        util = new MetaCatUtil();
198
    }
199
    /*MetaCatUtil.debugMessage("Connection pool size: "
200
                                     +connPool.getSizeOfDBConnectionPool(),10);
201
    MetaCatUtil.debugMessage("Free DBConnection number: "
202
                                  +connPool.getFreeDBConnectionNumber(), 10);*/
203
    //If all DBConnection in the pool are free and DBConnection pool
204
    //size is greater than initial value, shrink the connection pool
205
    //size to initial value
206
    DBConnectionPool.shrinkDBConnectionPoolSize();
207

    
208
    //Debug message to print out the method which have a busy DBConnection
209
    connPool.printMethodNameHavingBusyDBConnection();
210

    
211
    String ctype = request.getContentType();
212
    if (ctype != null && ctype.startsWith("multipart/form-data")) {
213
      handleMultipartForm(request, response);
214
    } else {
215

    
216

    
217
      String name = null;
218
      String[] value = null;
219
      String[] docid = new String[3];
220
      Hashtable params = new Hashtable();
221
      Enumeration paramlist = request.getParameterNames();
222

    
223

    
224
      while (paramlist.hasMoreElements()) {
225

    
226
        name = (String)paramlist.nextElement();
227
        value = request.getParameterValues(name);
228

    
229
        // Decode the docid and mouse click information
230
        if (name.endsWith(".y")) {
231
          docid[0] = name.substring(0,name.length()-2);
232
          params.put("docid", docid);
233
          name = "ypos";
234
        }
235
        if (name.endsWith(".x")) {
236
          name = "xpos";
237
        }
238

    
239
        params.put(name,value);
240
      }
241

    
242

    
243
      //handle param is emptpy
244
      if (params.isEmpty() || params == null)
245
      {
246
        return;
247
      }
248
      //if the user clicked on the input images, decode which image
249
      //was clicked then set the action.
250
      String action = ((String[])params.get("action"))[0];
251
      util.debugMessage("Line 230: Action is: " + action, 1);
252

    
253
      // This block handles session management for the servlet
254
      // by looking up the current session information for all actions
255
      // other than "login" and "logout"
256
      String username = null;
257
      String password = null;
258
      String[] groupnames = null;
259
      String sess_id = null;
260

    
261
      // handle login action
262
      if (action.equals("login")) {
263
        PrintWriter out = response.getWriter();
264
        handleLoginAction(out, params, request, response);
265
        out.close();
266

    
267
      // handle logout action
268
      } else if (action.equals("logout")) {
269
        PrintWriter out = response.getWriter();
270
        handleLogoutAction(out, params, request, response);
271
        out.close();
272

    
273
      // handle shrink DBConnection request
274
      } else if (action.equals("shrink")) {
275
        PrintWriter out = response.getWriter();
276
        boolean success = false;
277
        //If all DBConnection in the pool are free and DBConnection pool
278
        //size is greater than initial value, shrink the connection pool
279
        //size to initial value
280
        success = DBConnectionPool.shrinkConnectionPoolSize();
281
        if (success)
282
        {
283
          //if successfully shrink the pool size to initial value
284
          out.println("DBConnection Pool shrink successfully");
285
        }//if
286
        else
287
        {
288
          out.println("DBConnection pool couldn't shrink successfully");
289
        }
290
       //close out put
291
        out.close();
292

    
293
      // aware of session expiration on every request
294
      } else {
295

    
296
        HttpSession sess = request.getSession(true);
297
        if (sess.isNew()) {
298
          // session expired or has not been stored b/w user requests
299
          username = "public";
300
          sess.setAttribute("username", username);
301
        } else {
302
          username = (String)sess.getAttribute("username");
303
          password = (String)sess.getAttribute("password");
304
          groupnames = (String[])sess.getAttribute("groupnames");
305
          try {
306
            sess_id = (String)sess.getId();
307
          } catch(IllegalStateException ise) {
308
            System.out.println("error in handleGetOrPost: this shouldn't " +
309
                               "happen: the session should be valid: " +
310
                               ise.getMessage());
311
          }
312
        }
313
      }
314

    
315
       // Now that we know the session is valid, we can delegate the request
316
      // to a particular action handler
317
      if(action.equals("query")) {
318
        PrintWriter out = response.getWriter();
319
        handleQuery(out,params,response,username,groupnames);
320
        out.close();
321
      } else if(action.equals("squery")) {
322
        PrintWriter out = response.getWriter();
323
        if(params.containsKey("query")) {
324
         handleSQuery(out, params,response,username,groupnames);
325
         out.close();
326
        } else {
327
          out.println("Illegal action squery without \"query\" parameter");
328
          out.close();
329
        }
330
      } else if (action.equals("export")) {
331

    
332
        handleExportAction(params, response, username, groupnames, password);
333
      } else if (action.equals("read")) {
334
        handleReadAction(params, response, username,password, groupnames);
335
      } else if (action.equals("readinlinedata")) {
336
        handleReadInlineDataAction(params, response, username, 
337
                                   password, groupnames);
338
      } else if (action.equals("insert") || action.equals("update")) {
339
        PrintWriter out = response.getWriter();
340
        if ( (username != null) &&  !username.equals("public") ) {
341
          handleInsertOrUpdateAction(out,params,username,groupnames);
342
        } else {
343
          out.println("Permission denied for user "+username +" " + action);
344
       }
345
        out.close();
346
      } else if (action.equals("delete")) {
347
        PrintWriter out = response.getWriter();
348
        if ( (username != null) &&  !username.equals("public") ) {
349
          handleDeleteAction(out, params, response, username, groupnames);
350
        } else {
351
          out.println("Permission denied for " + action);
352
        }
353
        out.close();
354
      } else if (action.equals("validate")) {
355
        PrintWriter out = response.getWriter();
356
        handleValidateAction(out, params);
357
        out.close();
358
      } else if (action.equals("setaccess")) {
359
         PrintWriter out = response.getWriter();
360
         handleSetAccessAction(out, params, username);
361
        out.close();
362
      } else if (action.equals("getaccesscontrol")) {
363
        PrintWriter out = response.getWriter();
364
        handleGetAccessControlAction(out,params,response,username,groupnames);
365
        out.close();
366
      } else if (action.equals("getprincipals")) {
367
        PrintWriter out = response.getWriter();
368
        handleGetPrincipalsAction(out, username, password);
369
        out.close();
370
      } else if (action.equals("getdoctypes")) {
371
        PrintWriter out = response.getWriter();
372
        handleGetDoctypesAction(out, params, response);
373
        out.close();
374
      } else if (action.equals("getdtdschema")) {
375
        PrintWriter out = response.getWriter();
376
        handleGetDTDSchemaAction(out, params, response);
377
        out.close();
378
      } else if (action.equals("getdataguide")) {
379
        PrintWriter out = response.getWriter();
380
        handleGetDataGuideAction(out, params, response);
381
        out.close();
382
      } else if (action.equals("getlastdocid")) {
383
        PrintWriter out = response.getWriter();
384
        handleGetMaxDocidAction(out, params, response);
385
        out.close();
386
      } else if (action.equals("getrevisionanddoctype")) {
387
        PrintWriter out = response.getWriter();
388
        handleGetRevisionAndDocTypeAction(out, params);
389
        out.close();
390
      } else if (action.equals("login") || action.equals("logout")) {
391
      } else if (action.equals("protocoltest")) {
392
        String testURL = "metacat://dev.nceas.ucsb.edu/NCEAS.897766.9";
393
        try {
394
          testURL = ((String[])params.get("url"))[0];
395
        } catch (Throwable t) {
396
        }
397
        String phandler = System.getProperty("java.protocol.handler.pkgs");
398
        response.setContentType("text/html");
399
        PrintWriter out = response.getWriter();
400
        out.println("<body bgcolor=\"white\">");
401
        out.println("<p>Handler property: <code>" + phandler + "</code></p>");
402
        out.println("<p>Starting test for:<br>");
403
        out.println("    " + testURL + "</p>");
404
        try {
405
          URL u = new URL(testURL);
406
          out.println("<pre>");
407
          out.println("Protocol: " + u.getProtocol());
408
          out.println("    Host: " + u.getHost());
409
          out.println("    Port: " + u.getPort());
410
          out.println("    Path: " + u.getPath());
411
          out.println("     Ref: " + u.getRef());
412
          String pquery = u.getQuery();
413
          out.println("   Query: " + pquery);
414
          out.println("  Params: ");
415
          if (pquery != null) {
416
            Hashtable qparams = util.parseQuery(u.getQuery());
417
            for (Enumeration en = qparams.keys(); en.hasMoreElements(); ) {
418
              String pname = (String)en.nextElement();
419
              String pvalue = (String)qparams.get(pname);
420
              out.println("    " + pname + ": " + pvalue);
421
            }
422
          }
423
          out.println("</pre>");
424
          out.println("</body>");
425
          out.close();
426
        } catch (MalformedURLException mue) {
427
          System.out.println("bad url from MetacatServlet.handleGetOrPost");
428
          out.println(mue.getMessage());
429
          mue.printStackTrace(out);
430
          out.close();
431
        }
432
      } else {
433
        PrintWriter out = response.getWriter();
434
        out.println("<?xml version=\"1.0\"?>");
435
        out.println("<error>");
436
        out.println("Error: action not registered.  Please report this error.");
437
        out.println("</error>");
438
        out.close();
439
      }
440

    
441
      //util.closeConnections();
442
      // Close the stream to the client
443
      //out.close();
444
    }
445
  }
446

    
447
  // LOGIN & LOGOUT SECTION
448
  /**
449
   * Handle the login request. Create a new session object.
450
   * Do user authentication through the session.
451
   */
452
  private void handleLoginAction(PrintWriter out, Hashtable params,
453
               HttpServletRequest request, HttpServletResponse response) {
454

    
455
    AuthSession sess = null;
456
    String un = ((String[])params.get("username"))[0];
457
    String pw = ((String[])params.get("password"))[0];
458
    String action = ((String[])params.get("action"))[0];
459
    String qformat = ((String[])params.get("qformat"))[0];
460

    
461
    try {
462
      sess = new AuthSession();
463
    } catch (Exception e) {
464
      System.out.println("error in MetacatServlet.handleLoginAction: " +
465
                          e.getMessage());
466
      out.println(e.getMessage());
467
      return;
468
    }
469
    boolean isValid = sess.authenticate(request, un, pw);
470
    // format and transform the output
471
    if (qformat.equals("xml")) {
472
      response.setContentType("text/xml");
473
      out.println(sess.getMessage());
474
    } else {
475

    
476
      try {
477

    
478
        DBTransform trans = new DBTransform();
479
        response.setContentType("text/html");
480
        trans.transformXMLDocument(sess.getMessage(), "-//NCEAS//login//EN",
481
                                   "-//W3C//HTML//EN", qformat, out);
482

    
483
      } catch(Exception e) {
484

    
485
        MetaCatUtil.debugMessage("Error in MetaCatServlet.handleLoginAction: "
486
                                +e.getMessage(), 30);
487
      }
488

    
489
    // any output is returned
490
    }
491
  }
492

    
493
  /**
494
   * Handle the logout request. Close the connection.
495
   */
496
  private void handleLogoutAction(PrintWriter out, Hashtable params,
497
               HttpServletRequest request, HttpServletResponse response) {
498

    
499
    String qformat = ((String[])params.get("qformat"))[0];
500

    
501
    // close the connection
502
    HttpSession sess = request.getSession(false);
503
    if (sess != null) { sess.invalidate();  }
504

    
505
    // produce output
506
    StringBuffer output = new StringBuffer();
507
    output.append("<?xml version=\"1.0\"?>");
508
    output.append("<logout>");
509
    output.append("User logged out");
510
    output.append("</logout>");
511

    
512
    //format and transform the output
513
    if (qformat.equals("xml")) {
514
      response.setContentType("text/xml");
515
      out.println(output.toString());
516
    } else {
517

    
518
      try {
519

    
520
        DBTransform trans = new DBTransform();
521
        response.setContentType("text/html");
522
        trans.transformXMLDocument(output.toString(), "-//NCEAS//login//EN",
523
                                   "-//W3C//HTML//EN", qformat, out);
524

    
525
      } catch(Exception e) {
526

    
527
        MetaCatUtil.debugMessage("Error in MetaCatServlet.handleLogoutAction"
528
                                  +e.getMessage(), 30);
529
      }
530
    }
531
  }
532
  // END OF LOGIN & LOGOUT SECTION
533

    
534
  // SQUERY & QUERY SECTION
535
  /**
536
   * Retreive the squery xml, execute it and display it
537
   *
538
   * @param out the output stream to the client
539
   * @param params the Hashtable of parameters that should be included
540
   * in the squery.
541
   * @param response the response object linked to the client
542
   * @param conn the database connection
543
   */
544
  protected void handleSQuery(PrintWriter out, Hashtable params,
545
                 HttpServletResponse response, String user, String[] groups)
546
  {
547
    String xmlquery = ((String[])params.get("query"))[0];
548
    String qformat = ((String[])params.get("qformat"))[0];
549
    String resultdoc = null;
550
    MetaCatUtil.debugMessage("xmlquery: "+xmlquery, 30);
551
    double startTime = System.currentTimeMillis()/1000;
552
    Hashtable doclist = runQuery(xmlquery, user, groups);
553
    double docListTime = System.currentTimeMillis()/1000;
554
    MetaCatUtil.debugMessage("Time for getting doc list: "
555
                                            +(docListTime-startTime), 30);
556

    
557
    resultdoc = createResultDocument(doclist, transformQuery(xmlquery));
558
    double toStringTime = System.currentTimeMillis()/1000;
559
    MetaCatUtil.debugMessage("Time to create xml string: "
560
                              +(toStringTime-docListTime), 30);
561
    //format and transform the results
562
    double outPutTime = 0;
563
    if(qformat.equals("xml")) {
564
      response.setContentType("text/xml");
565
      out.println(resultdoc);
566
      outPutTime = System.currentTimeMillis()/1000;
567
      MetaCatUtil.debugMessage("Output time: "+(outPutTime-toStringTime), 30);
568
    } else {
569
      transformResultset(resultdoc, response, out, qformat);
570
      outPutTime = System.currentTimeMillis()/1000;
571
      MetaCatUtil.debugMessage("Output time: "+(outPutTime-toStringTime), 30);
572
    }
573
  }
574

    
575
   /**
576
    * Create the xml query, execute it and display the results.
577
    *
578
    * @param out the output stream to the client
579
    * @param params the Hashtable of parameters that should be included
580
    * in the squery.
581
    * @param response the response object linked to the client
582
    */
583
  protected void handleQuery(PrintWriter out, Hashtable params,
584
                 HttpServletResponse response, String user, String[] groups)
585
  {
586
    //create the query and run it
587
    String xmlquery = DBQuery.createSQuery(params);
588
    Hashtable doclist = runQuery(xmlquery, user, groups);
589
    String qformat = ((String[])params.get("qformat"))[0];
590
    String resultdoc = null;
591

    
592
    resultdoc = createResultDocument(doclist, transformQuery(params));
593

    
594
    //format and transform the results
595
    if(qformat.equals("xml")) {
596
      response.setContentType("text/xml");
597
      out.println(resultdoc);
598
    } else {
599
      transformResultset(resultdoc, response, out, qformat);
600
    }
601
  }
602

    
603
  /**
604
   * Removes the <?xml version="x"?> tag from the beginning of xmlquery
605
   * so it can properly be placed in the <query> tag of the resultset.
606
   * This method is overwritable so that other applications can customize
607
   * the structure of what is in the <query> tag.
608
   *
609
   * @param xmlquery is the query to remove the <?xml version="x"?> tag from.
610
   */
611
  protected String transformQuery(Hashtable params)
612
  {
613
    //DBQuery.createSQuery is a re-calling of a previously called
614
    //function but it is necessary
615
    //so that overriding methods have access to the params hashtable
616
    String xmlquery = DBQuery.createSQuery(params);
617
    //the <?xml version="1.0"?> tag is the first 22 characters of the
618
    xmlquery = xmlquery.trim();
619
    int index = xmlquery.indexOf("?>");
620
    return xmlquery.substring(index + 2, xmlquery.length());
621
  }
622

    
623
  /**
624
   * removes the <?xml version="1.0"?> tag from the beginning.  This takes a
625
   * string as a param instead of a hashtable.
626
   *
627
   * @param xmlquery a string representing a query.
628
   */
629
  protected String transformQuery(String xmlquery)
630
  {
631
    xmlquery = xmlquery.trim();
632
    int index = xmlquery.indexOf("?>");
633
    return xmlquery.substring(index + 2, xmlquery.length());
634
  }
635

    
636
  /**
637
   * Run the query and return a hashtable of results.
638
   *
639
   * @param xmlquery the query to run
640
   */
641
  private Hashtable runQuery(String xmlquery, String user, String[] groups)
642
  {
643
    Hashtable doclist=null;
644

    
645
    try
646
    {
647

    
648
      DBQuery queryobj = new DBQuery(saxparser);
649
      doclist = queryobj.findDocuments(new StringReader(xmlquery),user,groups);
650

    
651
      return doclist;
652
    }
653
    catch (Exception e)
654
    {
655

    
656
      MetaCatUtil.debugMessage("Error in MetacatServlet.runQuery: "
657
                                                      + e.getMessage(), 30);
658
      doclist = null;
659
      return doclist;
660
    }
661
  }
662

    
663
  /**
664
   * Transorms an xml resultset document to html and sends it to the browser
665
   *
666
   * @param resultdoc the string representation of the document that needs
667
   * to be transformed.
668
   * @param response the HttpServletResponse object bound to the client.
669
   * @param out the output stream to the client
670
   * @param qformat the name of the style-set to use for transformations
671
   */
672
  protected void transformResultset(String resultdoc,
673
                                    HttpServletResponse response,
674
                                    PrintWriter out, String qformat)
675
  {
676

    
677
    try {
678

    
679
      DBTransform trans = new DBTransform();
680
      response.setContentType("text/html");
681
      trans.transformXMLDocument(resultdoc, "-//NCEAS//resultset//EN",
682
                                 "-//W3C//HTML//EN", qformat, out);
683

    
684
    }
685
    catch(Exception e)
686
    {
687

    
688
      MetaCatUtil.debugMessage("Error in MetaCatServlet.transformResultset:"
689
                                +e.getMessage(), 30);
690
    }
691
  }
692

    
693
  /**
694
   * Transforms a hashtable of documents to an xml or html result.
695
   *
696
   * @param doclist- the hashtable to transform
697
   * @param xmlquery- the query that returned the doclist result
698
   */
699
  protected String createResultDocument(Hashtable doclist, String xmlquery)
700
  {
701
    // Create a buffer to hold the xml result
702
    StringBuffer resultset = new StringBuffer();
703

    
704
    // Print the resulting root nodes
705
    String docid = null;
706
    String document = null;
707
    resultset.append("<?xml version=\"1.0\"?>\n");
708
    resultset.append("<resultset>\n");
709

    
710
    resultset.append("  <query>" + xmlquery + "</query>");
711

    
712
    if(doclist != null)
713
    {
714
      Enumeration doclistkeys = doclist.keys();
715
      while (doclistkeys.hasMoreElements())
716
      {
717
        docid = (String)doclistkeys.nextElement();
718
        document = (String)doclist.get(docid);
719
        resultset.append("  <document>" + document + "</document>");
720
      }
721
    }
722

    
723
    resultset.append("</resultset>");
724
    return resultset.toString();
725
  }
726
  // END OF SQUERY & QUERY SECTION
727

    
728
 //Exoport section
729
 /**
730
   * Handle the "export" request of data package from Metacat in zip format
731
   * @param params the Hashtable of HTTP request parameters
732
   * @param response the HTTP response object linked to the client
733
   * @param user the username sent the request
734
   * @param groups the user's groupnames
735
   */
736
  private void handleExportAction(Hashtable params,
737
    HttpServletResponse response, String user, String[] groups, String passWord)
738
  {
739
    // Output stream
740
    ServletOutputStream out = null;
741
    // Zip output stream
742
    ZipOutputStream zOut = null;
743
    DocumentImpl docImpls=null;
744
    DBQuery queryObj=null;
745

    
746
    String[] docs = new String[10];
747
    String docId = "";
748

    
749
    try
750
    {
751
      // read the params
752
      if (params.containsKey("docid"))
753
      {
754
        docs = (String[])params.get("docid");
755
      }//if
756
      // Create a DBuery to handle export
757
      queryObj = new DBQuery(saxparser);
758
      // Get the docid
759
      docId=docs[0];
760
      // Make sure the client specify docid
761
      if (docId == null || docId.equals(""))
762
      {
763
        response.setContentType("text/xml"); //MIME type
764
        // Get a printwriter
765
        PrintWriter pw = response.getWriter();
766
        // Send back message
767
        pw.println("<?xml version=\"1.0\"?>");
768
        pw.println("<error>");
769
        pw.println("You didn't specify requested docid");
770
        pw.println("</error>");
771
        // Close printwriter
772
        pw.close();
773
        return;
774
      }//if
775
      // Get output stream
776
      out = response.getOutputStream();
777
      response.setContentType("application/zip"); //MIME type
778
      zOut = new ZipOutputStream(out);
779
      zOut =queryObj.getZippedPackage(docId, out, user, groups, passWord);
780
      zOut.finish(); //terminate the zip file
781
      zOut.close();  //close the zip stream
782

    
783
    }//try
784
    catch (Exception e)
785
    {
786
      try
787
      {
788
        response.setContentType("text/xml"); //MIME type
789
        // Send error message back
790
        if (out != null)
791
        {
792
            PrintWriter pw = new PrintWriter(out);
793
            pw.println("<?xml version=\"1.0\"?>");
794
            pw.println("<error>");
795
            pw.println(e.getMessage());
796
            pw.println("</error>");
797
            // Close printwriter
798
            pw.close();
799
            // Close output stream
800
            out.close();
801
        }//if
802
        // Close zip output stream
803
        if ( zOut != null )
804
        {
805
          zOut.close();
806
        }//if
807
      }//try
808
      catch (IOException ioe)
809
      {
810
        MetaCatUtil.debugMessage("Problem with the servlet output " +
811
                           "in MetacatServlet.handleExportAction: " +
812
                           ioe.getMessage(), 30);
813
      }//catch
814

    
815
      MetaCatUtil.debugMessage("Error in MetacatServlet.handleExportAction: " +
816
                         e.getMessage(), 30);
817
      e.printStackTrace(System.out);
818

    
819
    }//catch
820

    
821
  }//handleExportAction
822

    
823
  
824
   //read inline data section
825
 /**
826
   * In eml2 document, the xml can have inline data and data was stripped off 
827
   * and store in file system. This action can be used to read inline data only
828
   * @param params the Hashtable of HTTP request parameters
829
   * @param response the HTTP response object linked to the client
830
   * @param user the username sent the request
831
   * @param groups the user's groupnames
832
   */
833
  private void handleReadInlineDataAction(Hashtable params,
834
                                          HttpServletResponse response,
835
                                          String user, String passWord,
836
                                          String[] groups)
837
  {
838
    String[] docs = new String[10];
839
    String inlineDataId = null;
840
    String docId = "";
841
    ServletOutputStream out = null;
842

    
843
    try
844
    {
845
      // read the params
846
      if (params.containsKey("inlinedataid"))
847
      {
848
        docs = (String[])params.get("inlinedataid");
849
      }//if
850
      // Get the docid
851
      inlineDataId=docs[0];
852
      // Make sure the client specify docid
853
      if (inlineDataId == null || inlineDataId.equals(""))
854
      {
855
        throw new Exception("You didn't specify requested inlinedataid");
856
      }//if
857
      
858
      // check for permission
859
      docId = MetaCatUtil.getDocIdWithoutRevFromInlineDataID(inlineDataId);
860
      PermissionController controller = new PermissionController(docId);
861
      // check top level read permission
862
      if (!controller.hasPermission(user, groups, 
863
                                    AccessControlInterface.READSTRING))
864
      {
865
          throw new Exception("User "+ user + " doesn't have permission "+
866
                              " to read document " + docId);
867
      }//if
868
      // if the document has subtree control, we need to check subtree control
869
      else if(controller.hasSubTreeAccessControl())
870
      {
871
        // get node id for inlinedata
872
        long nodeId=getInlineDataNodeId(inlineDataId, docId);
873
        if (!controller.hasPermissionForSubTreeNode(user, groups,
874
                                     AccessControlInterface.READSTRING, nodeId))
875
        {
876
           throw new Exception("User "+ user + " doesn't have permission "+
877
                              " to read inlinedata " + inlineDataId);
878
        }//if
879
        
880
      }//else
881
      
882
      // Get output stream
883
      out = response.getOutputStream();
884
      // read the inline data from the file
885
      String inlinePath = MetaCatUtil.getOption("inlinedatafilepath");
886
      File lineData = new File(inlinePath, inlineDataId);
887
      FileInputStream input = new FileInputStream(lineData);
888
      byte [] buffer = new byte[4*1024];
889
      int bytes = input.read(buffer);
890
      while (bytes != -1)
891
      {
892
        out.write(buffer, 0, bytes);
893
        bytes = input.read(buffer);
894
      }
895
      out.close();
896

    
897
    }//try
898
    catch (Exception e)
899
    {
900
      try
901
      { 
902
        PrintWriter pw = null;
903
        // Send error message back
904
        if (out != null)
905
        {
906
            pw = new PrintWriter(out);
907
        }//if
908
        else
909
        {
910
          pw = response.getWriter();
911
        } 
912
         pw.println("<?xml version=\"1.0\"?>");
913
         pw.println("<error>");
914
         pw.println(e.getMessage());
915
         pw.println("</error>");
916
         // Close printwriter
917
         pw.close();
918
         // Close output stream if out is not null
919
         if (out != null)
920
         {
921
           out.close();
922
         }
923
     }//try
924
     catch (IOException ioe)
925
     {
926
        MetaCatUtil.debugMessage("Problem with the servlet output " +
927
                           "in MetacatServlet.handleExportAction: " +
928
                           ioe.getMessage(), 30);
929
     }//catch
930

    
931
      MetaCatUtil.debugMessage("Error in MetacatServlet.handleReadInlineDataAction: " 
932
                                + e.getMessage(), 30);
933
   
934
    }//catch
935

    
936
  }//handleReadInlineDataAction
937
  
938
  /*
939
   * Get the nodeid from xml_nodes for the inlinedataid
940
   */
941
  private long getInlineDataNodeId(String inLineDataId, String docId) 
942
                                   throws SQLException
943
  {
944
    long nodeId = 0;
945
    String INLINE = "inline";
946
    boolean hasRow;
947
    PreparedStatement pStmt = null;
948
    DBConnection conn = null;
949
    int serialNumber = -1;
950
    String sql ="SELECT nodeid FROM xml_nodes WHERE docid=? AND nodedata=? " +
951
                "AND nodetype='TEXT' AND parentnodeid IN " +
952
                "(SELECT nodeid FROM xml_nodes WHERE docid=? AND " + 
953
                "nodetype='ELEMENT' AND nodename='" + INLINE + "')";
954
  
955
    try
956
    {
957
      //check out DBConnection
958
      conn=DBConnectionPool.getDBConnection("AccessControlList.isAllowFirst");
959
      serialNumber=conn.getCheckOutSerialNumber();
960
    
961
      pStmt = conn.prepareStatement(sql);
962
      //bind value
963
      pStmt.setString(1, docId);//docid
964
      pStmt.setString(2, inLineDataId);//inlinedataid
965
      pStmt.setString(3, docId);
966
      // excute query 
967
      pStmt.execute();
968
      ResultSet rs = pStmt.getResultSet();
969
      hasRow=rs.next();
970
      // get result
971
      if (hasRow)
972
      {
973
        nodeId = rs.getLong(1);  
974
      }//if
975
     
976
    }//try
977
    catch (SQLException e)
978
    {
979
      throw e;
980
    }
981
    finally
982
    {
983
      try
984
      {
985
        pStmt.close();
986
      }
987
      finally
988
      {
989
        DBConnectionPool.returnDBConnection(conn, serialNumber);
990
      }
991
    }
992
    MetaCatUtil.debugMessage("The nodeid for inlinedataid " + inLineDataId +
993
                             " is: "+nodeId, 35);
994
    return nodeId;
995
  }
996
  
997
  
998
  
999
  // READ SECTION
1000
  /**
1001
   * Handle the "read" request of metadata/data files from Metacat
1002
   * or any files from Internet;
1003
   * transformed metadata XML document into HTML presentation if requested;
1004
   * zip files when more than one were requested.
1005
   *
1006
   * @param params the Hashtable of HTTP request parameters
1007
   * @param response the HTTP response object linked to the client
1008
   * @param user the username sent the request
1009
   * @param groups the user's groupnames
1010
   */
1011
  private void handleReadAction(Hashtable params, HttpServletResponse response,
1012
                                String user, String passWord, String[] groups)
1013
  {
1014
    ServletOutputStream out = null;
1015
    ZipOutputStream zout = null;
1016
    PrintWriter pw = null;
1017
    boolean zip = false;
1018
    boolean withInlineData = true;
1019

    
1020
    try {
1021
      String[] docs = new String[0];
1022
      String docid = "";
1023
      String qformat = "";
1024
      String abstrpath = null;
1025

    
1026
      // read the params
1027
      if (params.containsKey("docid")) {
1028
        docs = (String[])params.get("docid");
1029
      }
1030
      if (params.containsKey("qformat")) {
1031
        qformat = ((String[])params.get("qformat"))[0];
1032
      }
1033
      // the param for only metadata (eml)
1034
      if (params.containsKey("inlinedata"))
1035
      {
1036
        
1037
        String inlineData = ((String[])params.get("inlinedata"))[0];
1038
        if (inlineData.equalsIgnoreCase("false"))
1039
        {
1040
          withInlineData = false;
1041
        }
1042
      } 
1043
      if (params.containsKey("abstractpath")) {
1044
        abstrpath = ((String[])params.get("abstractpath"))[0];
1045
        if ( !abstrpath.equals("") && (abstrpath != null) ) {
1046
          viewAbstract(response, abstrpath, docs[0]);
1047
          return;
1048
        }
1049
      }
1050
      if ( (docs.length > 1) || qformat.equals("zip") ) {
1051
        zip = true;
1052
        out = response.getOutputStream();
1053
        response.setContentType("application/zip"); //MIME type
1054
        zout = new ZipOutputStream(out);
1055
      }
1056
      // go through the list of docs to read
1057
      for (int i=0; i < docs.length; i++ ) {
1058
        try {
1059

    
1060
          URL murl = new URL(docs[i]);
1061
          Hashtable murlQueryStr = util.parseQuery(murl.getQuery());
1062
          // case docid="http://.../?docid=aaa"
1063
          // or docid="metacat://.../?docid=bbb"
1064
          if (murlQueryStr.containsKey("docid")) {
1065
            // get only docid, eliminate the rest
1066
            docid = (String)murlQueryStr.get("docid");
1067
            if ( zip ) {
1068
              addDocToZip(docid, zout, user, groups);
1069
            } else {
1070
              readFromMetacat(response, docid, qformat, abstrpath,
1071
                              user, groups, zip, zout, withInlineData);
1072
            }
1073

    
1074
          // case docid="http://.../filename"
1075
          } else {
1076
            docid = docs[i];
1077
            if ( zip ) {
1078
              addDocToZip(docid, zout, user, groups);
1079
            } else {
1080
              readFromURLConnection(response, docid);
1081
            }
1082
          }
1083

    
1084
        // case docid="ccc"
1085
        } catch (MalformedURLException mue) {
1086
          docid = docs[i];
1087
          if ( zip ) {
1088
            addDocToZip(docid, zout, user, groups);
1089
          } else {
1090
            readFromMetacat(response, docid, qformat, abstrpath,
1091
                            user, groups, zip, zout, withInlineData);
1092
          }
1093
        }
1094

    
1095
      } /* end for */
1096

    
1097
      if ( zip ) {
1098
        zout.finish(); //terminate the zip file
1099
        zout.close();  //close the zip stream
1100
      }
1101

    
1102

    
1103
    }
1104
    // To handle doc not found exception
1105
    catch (McdbDocNotFoundException notFoundE)
1106
    {
1107
      // the docid which didn't be found
1108
      String notFoundDocId = notFoundE.getUnfoundDocId();
1109
      String notFoundRevision = notFoundE.getUnfoundRevision();
1110
      MetaCatUtil.debugMessage("Missed id: "+ notFoundDocId, 30);
1111
      MetaCatUtil.debugMessage("Missed rev: "+ notFoundRevision, 30);
1112
      try
1113
      {
1114
        // read docid from remote server
1115
        readFromRemoteMetaCat(response, notFoundDocId, notFoundRevision,
1116
                                              user, passWord, out, zip, zout);
1117
        // Close zout outputstream
1118
        if ( zout != null)
1119
        {
1120
          zout.close();
1121
        }
1122
        // close output stream
1123
        if (out != null)
1124
        {
1125
          out.close();
1126
        }
1127

    
1128
      }//try
1129
      catch ( Exception exc)
1130
      {
1131
        MetaCatUtil.debugMessage("Erorr in MetacatServlet.hanldReadAction: "+
1132
                                      exc.getMessage(), 30);
1133
        try
1134
        {
1135
          if (out != null)
1136
          {
1137
            response.setContentType("text/xml");
1138
            // Send back error message by printWriter
1139
            pw = new PrintWriter(out);
1140
            pw.println("<?xml version=\"1.0\"?>");
1141
            pw.println("<error>");
1142
            pw.println(notFoundE.getMessage());
1143
            pw.println("</error>");
1144
            pw.close();
1145
            out.close();
1146

    
1147
          }
1148
          else
1149
          {
1150
           response.setContentType("text/xml"); //MIME type
1151
           // Send back error message if out = null
1152
           if (pw == null)
1153
           {
1154
             // If pw is null, open the respnose
1155
            pw = response.getWriter();
1156
           }
1157
           pw.println("<?xml version=\"1.0\"?>");
1158
           pw.println("<error>");
1159
           pw.println(notFoundE.getMessage());
1160
           pw.println("</error>");
1161
           pw.close();
1162
        }
1163
        // close zout
1164
        if ( zout != null )
1165
        {
1166
          zout.close();
1167
        }
1168
        }//try
1169
        catch (IOException ie)
1170
        {
1171
          MetaCatUtil.debugMessage("Problem with the servlet output " +
1172
                           "in MetacatServlet.handleReadAction: " +
1173
                           ie.getMessage(), 30);
1174
        }//cathch
1175
      }//catch
1176
    }// catch McdbDocNotFoundException
1177
    catch (Exception e)
1178
    {
1179
      try {
1180

    
1181
        if (out != null) {
1182
            response.setContentType("text/xml"); //MIME type
1183
            pw = new PrintWriter(out);
1184
            pw.println("<?xml version=\"1.0\"?>");
1185
            pw.println("<error>");
1186
            pw.println(e.getMessage());
1187
            pw.println("</error>");
1188
            pw.close();
1189
            out.close();
1190
        }
1191
        else
1192
        {
1193
           response.setContentType("text/xml"); //MIME type
1194
           // Send back error message if out = null
1195
           if ( pw == null)
1196
           {
1197
            pw = response.getWriter();
1198
           }
1199
           pw.println("<?xml version=\"1.0\"?>");
1200
           pw.println("<error>");
1201
           pw.println(e.getMessage());
1202
           pw.println("</error>");
1203
           pw.close();
1204

    
1205
        }
1206
        // Close zip output stream
1207
        if ( zout != null ) { zout.close(); }
1208

    
1209
      } catch (IOException ioe) {
1210
        MetaCatUtil.debugMessage("Problem with the servlet output " +
1211
                           "in MetacatServlet.handleReadAction: " +
1212
                           ioe.getMessage(), 30);
1213
        ioe.printStackTrace(System.out);
1214

    
1215
      }
1216

    
1217
      MetaCatUtil.debugMessage("Error in MetacatServlet.handleReadAction: " +
1218
                               e.getMessage(), 30);
1219
      //e.printStackTrace(System.out);
1220
    }
1221

    
1222
  }
1223

    
1224
  // read metadata or data from Metacat
1225
  private void readFromMetacat(HttpServletResponse response, String docid,
1226
                               String qformat, String abstrpath, String user,
1227
                               String[] groups, boolean zip, 
1228
                               ZipOutputStream zout, boolean withInlineData)
1229
               throws ClassNotFoundException, IOException, SQLException,
1230
                      McdbException, Exception
1231
  {
1232

    
1233
    try {
1234

    
1235

    
1236
      DocumentImpl doc = new DocumentImpl(docid);
1237

    
1238
      //check the permission for read
1239
      if (!doc.hasReadPermission(user, groups, docid))
1240
      {
1241
        Exception e = new Exception("User " + user + " does not have permission"
1242
                       +" to read the document with the docid " + docid);
1243

    
1244
        throw e;
1245
      }
1246

    
1247
      if ( doc.getRootNodeID() == 0 ) {
1248
        // this is data file
1249
        String filepath = util.getOption("datafilepath");
1250
        if(!filepath.endsWith("/")) {
1251
          filepath += "/";
1252
        }
1253
        String filename = filepath + docid;
1254
        FileInputStream fin = null;
1255
        fin = new FileInputStream(filename);
1256

    
1257
        //MIME type
1258
        String contentType = getServletContext().getMimeType(filename);
1259
        if (contentType == null) 
1260
        {
1261
          ContentTypeProvider provider = new ContentTypeProvider(docid);
1262
          contentType = provider.getContentType();
1263
          MetaCatUtil.debugMessage("Final contenttype is: "+ contentType, 30);
1264
        }
1265
       
1266
        response.setContentType(contentType);
1267
        // if we decide to use "application/octet-stream" for all data returns
1268
        // response.setContentType("application/octet-stream");
1269

    
1270
        try {
1271

    
1272
          ServletOutputStream out = response.getOutputStream();
1273
          byte[] buf = new byte[4 * 1024]; // 4K buffer
1274
          int b = fin.read(buf);
1275
          while (b != -1) {
1276
            out.write(buf, 0, b);
1277
            b = fin.read(buf);
1278
          }
1279
        } finally {
1280
          if (fin != null) fin.close();
1281
        }
1282

    
1283
      } else {
1284
        // this is metadata doc
1285
        if ( qformat.equals("xml") ) {
1286

    
1287
          // set content type first
1288
          response.setContentType("text/xml");   //MIME type
1289
          PrintWriter out = response.getWriter();
1290
          doc.toXml(out, user, groups, withInlineData);
1291
        } else {
1292
          response.setContentType("text/html");  //MIME type
1293
          PrintWriter out = response.getWriter();
1294

    
1295
          // Look up the document type
1296
          String doctype = doc.getDoctype();
1297
          // Transform the document to the new doctype
1298
          DBTransform dbt = new DBTransform();
1299
          dbt.transformXMLDocument(doc.toString(user, groups, withInlineData),
1300
                                   doctype,"-//W3C//HTML//EN", qformat, out);
1301
        }
1302

    
1303
      }
1304
    }
1305
    catch (Exception except)
1306
    {
1307
      throw except;
1308

    
1309
    }
1310

    
1311
  }
1312

    
1313
  // read data from URLConnection
1314
  private void readFromURLConnection(HttpServletResponse response, String docid)
1315
               throws IOException, MalformedURLException
1316
  {
1317
    ServletOutputStream out = response.getOutputStream();
1318
    String contentType = getServletContext().getMimeType(docid); //MIME type
1319
    if (contentType == null) {
1320
      if (docid.endsWith(".xml")) {
1321
        contentType="text/xml";
1322
      } else if (docid.endsWith(".css")) {
1323
        contentType="text/css";
1324
      } else if (docid.endsWith(".dtd")) {
1325
        contentType="text/plain";
1326
      } else if (docid.endsWith(".xsd")) {
1327
        contentType="text/xml";
1328
      } else if (docid.endsWith("/")) {
1329
        contentType="text/html";
1330
      } else {
1331
        File f = new File(docid);
1332
        if ( f.isDirectory() ) {
1333
          contentType="text/html";
1334
        } else {
1335
          contentType="application/octet-stream";
1336
        }
1337
      }
1338
    }
1339
    response.setContentType(contentType);
1340
    // if we decide to use "application/octet-stream" for all data returns
1341
    // response.setContentType("application/octet-stream");
1342

    
1343
    // this is http url
1344
    URL url = new URL(docid);
1345
    BufferedInputStream bis = null;
1346
    try {
1347
      bis = new BufferedInputStream(url.openStream());
1348
      byte[] buf = new byte[4 * 1024]; // 4K buffer
1349
      int b = bis.read(buf);
1350
      while (b != -1) {
1351
        out.write(buf, 0, b);
1352
        b = bis.read(buf);
1353
      }
1354
    } finally {
1355
      if (bis != null) bis.close();
1356
    }
1357

    
1358
  }
1359

    
1360
  // read file/doc and write to ZipOutputStream
1361
  private void addDocToZip(String docid, ZipOutputStream zout,
1362
                              String user, String[] groups)
1363
               throws ClassNotFoundException, IOException, SQLException,
1364
                      McdbException, Exception
1365
  {
1366
    byte[] bytestring = null;
1367
    ZipEntry zentry = null;
1368

    
1369
    try {
1370
      URL url = new URL(docid);
1371

    
1372
      // this http url; read from URLConnection; add to zip
1373
      zentry = new ZipEntry(docid);
1374
      zout.putNextEntry(zentry);
1375
      BufferedInputStream bis = null;
1376
      try {
1377
        bis = new BufferedInputStream(url.openStream());
1378
        byte[] buf = new byte[4 * 1024]; // 4K buffer
1379
        int b = bis.read(buf);
1380
        while(b != -1) {
1381
          zout.write(buf, 0, b);
1382
          b = bis.read(buf);
1383
        }
1384
      } finally {
1385
        if (bis != null) bis.close();
1386
      }
1387
      zout.closeEntry();
1388

    
1389
    } catch (MalformedURLException mue) {
1390

    
1391
      // this is metacat doc (data file or metadata doc)
1392

    
1393
      try {
1394

    
1395
        DocumentImpl doc = new DocumentImpl(docid);
1396

    
1397
        //check the permission for read
1398
        if (!doc.hasReadPermission(user, groups, docid))
1399
        {
1400
          Exception e = new Exception("User " + user + " does not have "
1401
                    +"permission to read the document with the docid " + docid);
1402

    
1403
          throw e;
1404
        }
1405

    
1406
        if ( doc.getRootNodeID() == 0 ) {
1407
          // this is data file; add file to zip
1408
          String filepath = util.getOption("datafilepath");
1409
          if(!filepath.endsWith("/")) {
1410
            filepath += "/";
1411
          }
1412
          String filename = filepath + docid;
1413
          FileInputStream fin = null;
1414
          fin = new FileInputStream(filename);
1415
          try {
1416

    
1417
            zentry = new ZipEntry(docid);
1418
            zout.putNextEntry(zentry);
1419
            byte[] buf = new byte[4 * 1024]; // 4K buffer
1420
            int b = fin.read(buf);
1421
            while (b != -1) {
1422
              zout.write(buf, 0, b);
1423
              b = fin.read(buf);
1424
            }
1425
          } finally {
1426
            if (fin != null) fin.close();
1427
          }
1428
          zout.closeEntry();
1429

    
1430
        } else {
1431
          // this is metadata doc; add doc to zip
1432
          bytestring = doc.toString().getBytes();
1433
          zentry = new ZipEntry(docid + ".xml");
1434
          zentry.setSize(bytestring.length);
1435
          zout.putNextEntry(zentry);
1436
          zout.write(bytestring, 0, bytestring.length);
1437
          zout.closeEntry();
1438
        }
1439
      } catch (Exception except) {
1440
        throw except;
1441

    
1442
      }
1443

    
1444
    }
1445

    
1446
  }
1447

    
1448
  // view abstract within document
1449
  private void viewAbstract(HttpServletResponse response,
1450
                            String abstractpath, String docid)
1451
               throws ClassNotFoundException, IOException, SQLException,
1452
                      McdbException, Exception
1453
  {
1454

    
1455
    PrintWriter out =null;
1456
    try {
1457

    
1458
      response.setContentType("text/html");  //MIME type
1459
      out = response.getWriter();
1460
      Object[] abstracts = DBQuery.getNodeContent(abstractpath, docid);
1461
      out.println("<html><head><title>Abstract</title></head>");
1462
      out.println("<body bgcolor=\"white\"><h1>Abstract</h1>");
1463
      for (int i=0; i<abstracts.length; i++) {
1464
        out.println("<p>" + (String)abstracts[i] + "</p>");
1465
      }
1466
      out.println("</body></html>");
1467

    
1468
    } catch (Exception e) {
1469
       out.println("<?xml version=\"1.0\"?>");
1470
       out.println("<error>");
1471
       out.println(e.getMessage());
1472
       out.println("</error>");
1473

    
1474

    
1475
    }
1476
  }
1477
  /**
1478
   * If metacat couldn't find a data file or document locally, it will read this
1479
   * docid from its home server. This is for the replication feature
1480
   */
1481
  private void readFromRemoteMetaCat(HttpServletResponse response, String docid,
1482
                     String rev, String user, String password,
1483
                     ServletOutputStream out, boolean zip, ZipOutputStream zout)
1484
                        throws Exception
1485
 {
1486
   // Create a object of RemoteDocument, "" is for zipEntryPath
1487
   RemoteDocument remoteDoc =
1488
                        new RemoteDocument (docid, rev,user, password, "");
1489
   String docType = remoteDoc.getDocType();
1490
   // Only read data file
1491
   if (docType.equals("BIN"))
1492
   {
1493
    // If it is zip format
1494
    if (zip)
1495
    {
1496
      remoteDoc.readDocumentFromRemoteServerByZip(zout);
1497
    }//if
1498
    else
1499
    {
1500
      if (out == null)
1501
      {
1502
        out = response.getOutputStream();
1503
      }//if
1504
      response.setContentType("application/octet-stream");
1505
      remoteDoc.readDocumentFromRemoteServer(out);
1506
    }//else (not zip)
1507
   }//if doctype=bin
1508
   else
1509
   {
1510
     throw new Exception("Docid: "+docid+"."+rev+" couldn't find");
1511
   }//else
1512
 }//readFromRemoteMetaCat
1513

    
1514
  // END OF READ SECTION
1515
  
1516
 
1517
  
1518
  // INSERT/UPDATE SECTION
1519
  /**
1520
   * Handle the database putdocument request and write an XML document
1521
   * to the database connection
1522
   */
1523
  private void handleInsertOrUpdateAction(PrintWriter out, Hashtable params,
1524
               String user, String[] groups) {
1525

    
1526
    DBConnection dbConn = null;
1527
    int serialNumber = -1;
1528

    
1529
    try {
1530
      // Get the document indicated
1531
      String[] doctext = (String[])params.get("doctext");
1532

    
1533
      String pub = null;
1534
      if (params.containsKey("public")) {
1535
        pub = ((String[])params.get("public"))[0];
1536
      }
1537

    
1538
      StringReader dtd = null;
1539
      if (params.containsKey("dtdtext")) {
1540
        String[] dtdtext = (String[])params.get("dtdtext");
1541
        try {
1542
          if ( !dtdtext[0].equals("") ) {
1543
            dtd = new StringReader(dtdtext[0]);
1544
          }
1545
        } catch (NullPointerException npe) {}
1546
      }
1547

    
1548
      StringReader xml = null;
1549
      boolean validate = false;
1550
      DocumentImplWrapper documentWrapper = null;
1551
      try {
1552
        // look inside XML Document for <!DOCTYPE ... PUBLIC/SYSTEM ... >
1553
        // in order to decide whether to use validation parser
1554
        validate = needDTDValidation(doctext[0]);
1555
        if (validate)
1556
        {
1557
          // set a dtd base validation parser
1558
          String rule = DocumentImpl.DTD;
1559
          documentWrapper = new DocumentImplWrapper(rule, validate);
1560
        }
1561
        else if (needSchemaValidation(doctext[0]))
1562
        {
1563
          // for eml2
1564
          if (needEml2Validation(doctext[0]))
1565
          {
1566
             // set eml2 base validation parser
1567
            String rule = DocumentImpl.EML2;
1568
            // using emlparser to check id validation
1569
            EMLParser parser = new EMLParser(doctext[0]);
1570
            documentWrapper = new DocumentImplWrapper(rule, true);
1571
          }
1572
          else
1573
          {
1574
            // set schema base validation parser
1575
            String rule = DocumentImpl.SCHEMA;
1576
            documentWrapper = new DocumentImplWrapper(rule, true);
1577
          }
1578
        }
1579
        else
1580
        {
1581
          documentWrapper = new DocumentImplWrapper("", false);
1582
        }
1583
        
1584
        xml = new StringReader(doctext[0]);
1585

    
1586
        String[] action = (String[])params.get("action");
1587
        String[] docid = (String[])params.get("docid");
1588
        String newdocid = null;
1589

    
1590
        String doAction = null;
1591
        if (action[0].equals("insert")) {
1592
          doAction = "INSERT";
1593
        } else if (action[0].equals("update")) {
1594
          doAction = "UPDATE";
1595
        }
1596

    
1597
        try
1598
        {
1599
          // get a connection from the pool
1600
          dbConn=DBConnectionPool.
1601
                  getDBConnection("MetaCatServlet.handleInsertOrUpdateAction");
1602
          serialNumber=dbConn.getCheckOutSerialNumber();
1603

    
1604

    
1605
          // write the document to the database
1606
          try
1607
          {
1608
            String accNumber = docid[0];
1609
            MetaCatUtil.debugMessage(""+ doAction + " " + accNumber +"...", 10);
1610
            if (accNumber.equals(""))
1611
            {
1612
              accNumber = null;
1613
            }//if
1614
            newdocid = documentWrapper.write(dbConn, xml, pub, dtd, doAction,
1615
                                          accNumber, user, groups);
1616

    
1617
          }//try
1618
          catch (NullPointerException npe)
1619
          {
1620
            newdocid = documentWrapper.write(dbConn, xml, pub, dtd, doAction,
1621
                                          null, user, groups);
1622
          }//catch
1623
        
1624
        }//try
1625
        finally
1626
        {
1627
          // Return db connection
1628
          DBConnectionPool.returnDBConnection(dbConn, serialNumber);
1629
        }
1630

    
1631
        // set content type and other response header fields first
1632
        //response.setContentType("text/xml");
1633
        out.println("<?xml version=\"1.0\"?>");
1634
        out.println("<success>");
1635
        out.println("<docid>" + newdocid + "</docid>");
1636
        out.println("</success>");
1637

    
1638
      } 
1639
      catch (NullPointerException npe) 
1640
      {
1641
        //response.setContentType("text/xml");
1642
        out.println("<?xml version=\"1.0\"?>");
1643
        out.println("<error>");
1644
        out.println(npe.getMessage());
1645
        out.println("</error>");
1646
      }
1647
    } 
1648
    catch (Exception e) 
1649
    {
1650
      //response.setContentType("text/xml");
1651
      out.println("<?xml version=\"1.0\"?>");
1652
      out.println("<error>");
1653
      out.println(e.getMessage());
1654
      out.println("</error>");
1655
    }
1656
  }
1657

    
1658
  /**
1659
   * Parse XML Document to look for <!DOCTYPE ... PUBLIC/SYSTEM ... >
1660
   * in order to decide whether to use validation parser
1661
   */
1662
  private static boolean needDTDValidation(String xmltext) throws IOException {
1663

    
1664
    StringReader xmlreader = new StringReader(xmltext);
1665
    StringBuffer cbuff = new StringBuffer();
1666
    java.util.Stack st = new java.util.Stack();
1667
    boolean validate = false;
1668
    int c;
1669
    int inx;
1670

    
1671
    // read from the stream until find the keywords
1672
    while ( (st.empty() || st.size()<4) && ((c = xmlreader.read()) != -1) ) {
1673
      cbuff.append((char)c);
1674

    
1675
      // "<!DOCTYPE" keyword is found; put it in the stack
1676
      if ( (inx = cbuff.toString().indexOf("<!DOCTYPE")) != -1 ) {
1677
        cbuff = new StringBuffer();
1678
        st.push("<!DOCTYPE");
1679
      }
1680
      // "PUBLIC" keyword is found; put it in the stack
1681
      if ( (inx = cbuff.toString().indexOf("PUBLIC")) != -1 ) {
1682
        cbuff = new StringBuffer();
1683
        st.push("PUBLIC");
1684
      }
1685
      // "SYSTEM" keyword is found; put it in the stack
1686
      if ( (inx = cbuff.toString().indexOf("SYSTEM")) != -1 ) {
1687
        cbuff = new StringBuffer();
1688
        st.push("SYSTEM");
1689
      }
1690
      // ">" character is found; put it in the stack
1691
      // ">" is found twice: fisrt from <?xml ...?>
1692
      // and second from <!DOCTYPE ... >
1693
      if ( (inx = cbuff.toString().indexOf(">")) != -1 ) {
1694
        cbuff = new StringBuffer();
1695
        st.push(">");
1696
      }
1697
    }
1698

    
1699
    // close the stream
1700
    xmlreader.close();
1701

    
1702
    // check the stack whether it contains the keywords:
1703
    // "<!DOCTYPE", "PUBLIC" or "SYSTEM", and ">" in this order
1704
    if ( st.size() == 4 ) {
1705
      if ( ((String)st.pop()).equals(">") &&
1706
           ( ((String)st.peek()).equals("PUBLIC") |
1707
             ((String)st.pop()).equals("SYSTEM") ) &&
1708
           ((String)st.pop()).equals("<!DOCTYPE") )  {
1709
        validate = true;
1710
      }
1711
    }
1712

    
1713
    MetaCatUtil.debugMessage("Validation for dtd is " + validate, 10);
1714
    return validate;
1715
  }
1716
  // END OF INSERT/UPDATE SECTION
1717
  
1718
  /* check if the xml string contains key words to specify schema loocation*/
1719
  private boolean needSchemaValidation(String xml)
1720
  {
1721
    boolean needSchemaValidate =false;
1722
    if (xml == null)
1723
    {
1724
      MetaCatUtil.debugMessage("Validation for schema is " +
1725
                               needSchemaValidate, 10);
1726
      return needSchemaValidate;
1727
    }
1728
    String targetLine = getSchemaLine(xml);
1729
    // to see if the second line contain some keywords 
1730
    if (targetLine != null && (targetLine.indexOf(SCHEMALOCATIONKEYWORD) != -1||
1731
             targetLine.indexOf(NONAMESPACELOCATION) != -1 ))
1732
    {
1733
      // if contains schema location key word, should be validate
1734
      needSchemaValidate = true;
1735
    }
1736
    
1737
    MetaCatUtil.debugMessage("Validation for schema is " + 
1738
                             needSchemaValidate, 10);
1739
    return needSchemaValidate;
1740
   
1741
  }
1742
  
1743
   /* check if the xml string contains key words to specify schema loocation*/
1744
  private boolean needEml2Validation(String xml)
1745
  {
1746
    boolean needEml2Validate =false;
1747
    String emlNameSpace = "=\""+DocumentImpl.EMLNAMESPACE+"\"";
1748
    if (xml == null)
1749
    {
1750
      MetaCatUtil.debugMessage("Validation for schema is " +
1751
                               needEml2Validate, 10);
1752
      return needEml2Validate;
1753
    }
1754
    String targetLine = getSchemaLine(xml);
1755
    
1756
    if (targetLine != null && targetLine.indexOf(EML2KEYWORD) != -1 && 
1757
        targetLine.indexOf(emlNameSpace) != -1)
1758
    {
1759
      // if contains schema location key word, should be validate
1760
      needEml2Validate = true;
1761
    }
1762
    
1763
    MetaCatUtil.debugMessage("Validation for eml is " + 
1764
                             needEml2Validate, 10);
1765
    return needEml2Validate;
1766
   
1767
  }
1768
  
1769
  private String getSchemaLine(String xml)
1770
  {
1771
    // find the line
1772
    String secondLine = null;
1773
    int count =0;
1774
    int endIndex = 0;
1775
    int startIndex = 0;
1776
    final int TARGETNUM = 2;
1777
    for (int i=0; i<xml.length(); i++)
1778
    {
1779
      // didn't count comment
1780
      if (xml.charAt(i) =='<' && xml.charAt(i+1) != '!')
1781
      {
1782
        count ++;
1783
        //find start index
1784
        if (count == TARGETNUM)
1785
        {
1786
          startIndex = i;
1787
        }
1788
      }//if
1789
      // find the end index
1790
      if (count== TARGETNUM && xml.charAt(i) =='>')
1791
      {
1792
        endIndex = i;
1793
        break;
1794
      }//if
1795
    }//for
1796
    // get the second line string 
1797
    MetaCatUtil.debugMessage("The start index for second line: "+startIndex, 25);
1798
    MetaCatUtil.debugMessage("The end index for second line: "+endIndex, 25);
1799
    if (startIndex != 0 && endIndex != 0)
1800
    {
1801
      secondLine = xml.substring(startIndex+1, endIndex);
1802
      
1803
    }//if
1804
    MetaCatUtil.debugMessage("the second line string is: "+secondLine, 25);
1805
    return secondLine;
1806
  }
1807
  
1808
  // DELETE SECTION
1809
  /**
1810
   * Handle the database delete request and delete an XML document
1811
   * from the database connection
1812
   */
1813
  private void handleDeleteAction(PrintWriter out, Hashtable params,
1814
               HttpServletResponse response, String user, String[] groups) {
1815

    
1816
    String[] docid = (String[])params.get("docid");
1817

    
1818
    // delete the document from the database
1819
    try {
1820

    
1821
                                      // NOTE -- NEED TO TEST HERE
1822
                                      // FOR EXISTENCE OF DOCID PARAM
1823
                                      // BEFORE ACCESSING ARRAY
1824
      try {
1825
        DocumentImpl.delete(docid[0], user, groups);
1826
        response.setContentType("text/xml");
1827
        out.println("<?xml version=\"1.0\"?>");
1828
        out.println("<success>");
1829
        out.println("Document deleted.");
1830
        out.println("</success>");
1831
      } catch (AccessionNumberException ane) {
1832
        response.setContentType("text/xml");
1833
        out.println("<?xml version=\"1.0\"?>");
1834
        out.println("<error>");
1835
        out.println("Error deleting document!!!");
1836
        out.println(ane.getMessage());
1837
        out.println("</error>");
1838
      }
1839
    } catch (Exception e) {
1840
      response.setContentType("text/xml");
1841
      out.println("<?xml version=\"1.0\"?>");
1842
      out.println("<error>");
1843
      out.println(e.getMessage());
1844
      out.println("</error>");
1845
    }
1846
  }
1847
  // END OF DELETE SECTION
1848

    
1849
  // VALIDATE SECTION
1850
  /**
1851
   * Handle the validation request and return the results to the requestor
1852
   */
1853
  private void handleValidateAction(PrintWriter out, Hashtable params) {
1854

    
1855
    // Get the document indicated
1856
    String valtext = null;
1857
    DBConnection dbConn = null;
1858
    int serialNumber = -1;
1859

    
1860
    try {
1861
      valtext = ((String[])params.get("valtext"))[0];
1862
    } catch (Exception nullpe) {
1863

    
1864

    
1865
      String docid = null;
1866
      try {
1867
        // Find the document id number
1868
        docid = ((String[])params.get("docid"))[0];
1869

    
1870

    
1871
        // Get the document indicated from the db
1872
        DocumentImpl xmldoc = new DocumentImpl(docid);
1873
        valtext = xmldoc.toString();
1874

    
1875
      } catch (NullPointerException npe) {
1876

    
1877
        out.println("<error>Error getting document ID: " + docid + "</error>");
1878
        //if ( conn != null ) { util.returnConnection(conn); }
1879
        return;
1880
      } catch (Exception e) {
1881

    
1882
        out.println(e.getMessage());
1883
      }
1884
    }
1885

    
1886

    
1887
    try {
1888
      // get a connection from the pool
1889
      dbConn=DBConnectionPool.
1890
                  getDBConnection("MetaCatServlet.handleValidateAction");
1891
      serialNumber=dbConn.getCheckOutSerialNumber();
1892
      DBValidate valobj = new DBValidate(saxparser,dbConn);
1893
      boolean valid = valobj.validateString(valtext);
1894

    
1895
      // set content type and other response header fields first
1896

    
1897
      out.println(valobj.returnErrors());
1898

    
1899
    } catch (NullPointerException npe2) {
1900
      // set content type and other response header fields first
1901

    
1902
      out.println("<error>Error validating document.</error>");
1903
    } catch (Exception e) {
1904

    
1905
      out.println(e.getMessage());
1906
    } finally {
1907
      // Return db connection
1908
      DBConnectionPool.returnDBConnection(dbConn, serialNumber);
1909
    }
1910
  }
1911
  // END OF VALIDATE SECTION
1912

    
1913
  // OTHER ACTION HANDLERS
1914

    
1915
  /**
1916
   * Handle "getrevsionanddoctype" action
1917
   * Given a docid, return it's current revision and doctype from data base
1918
   * The output is String look like "rev;doctype"
1919
   */
1920
  private void handleGetRevisionAndDocTypeAction(PrintWriter out,
1921
                                                              Hashtable params)
1922
  {
1923
    // To store doc parameter
1924
    String [] docs = new String[10];
1925
    // Store a single doc id
1926
    String givenDocId = null;
1927
    // Get docid from parameters
1928
    if (params.containsKey("docid"))
1929
    {
1930
      docs = (String[])params.get("docid");
1931
    }
1932
    // Get first docid form string array
1933
    givenDocId = docs[0];
1934

    
1935
    try
1936
    {
1937
      // Make sure there is a docid
1938
      if (givenDocId == null || givenDocId.equals(""))
1939
      {
1940
        throw new Exception("User didn't specify docid!");
1941
      }//if
1942

    
1943
      // Create a DBUtil object
1944
      DBUtil dbutil = new DBUtil();
1945
      // Get a rev and doctype
1946
      String revAndDocType =
1947
                dbutil.getCurrentRevisionAndDocTypeForGivenDocument(givenDocId);
1948
      out.println(revAndDocType);
1949

    
1950
    }//try
1951
    catch (Exception e)
1952
    {
1953
      // Handle exception
1954
      out.println("<?xml version=\"1.0\"?>");
1955
      out.println("<error>");
1956
      out.println(e.getMessage());
1957
      out.println("</error>");
1958
    }//catch
1959

    
1960
  }//handleGetRevisionAndDocTypeAction
1961

    
1962
  /**
1963
   * Handle "getaccesscontrol" action.
1964
   * Read Access Control List from db connection in XML format
1965
   */
1966
  private void handleGetAccessControlAction(PrintWriter out, Hashtable params,
1967
                                       HttpServletResponse response,
1968
                                       String username, String[] groupnames) {
1969

    
1970
    DBConnection dbConn = null;
1971
    int serialNumber = -1;
1972
    String docid = ((String[])params.get("docid"))[0];
1973

    
1974
    try {
1975

    
1976
        // get connection from the pool
1977
        dbConn=DBConnectionPool.
1978
                 getDBConnection("MetaCatServlet.handleGetAccessControlAction");
1979
        serialNumber=dbConn.getCheckOutSerialNumber();
1980
        AccessControlList aclobj = new AccessControlList(dbConn);
1981
        String acltext = aclobj.getACL(docid, username, groupnames);
1982
        out.println(acltext);
1983

    
1984
    } catch (Exception e) {
1985
      out.println("<?xml version=\"1.0\"?>");
1986
      out.println("<error>");
1987
      out.println(e.getMessage());
1988
      out.println("</error>");
1989
    } finally {
1990
      // Retrun db connection to pool
1991
      DBConnectionPool.returnDBConnection(dbConn, serialNumber);
1992
    }
1993

    
1994
  }
1995

    
1996
  /**
1997
   * Handle the "getprincipals" action.
1998
   * Read all principals from authentication scheme in XML format
1999
   */
2000
  private void handleGetPrincipalsAction(PrintWriter out, String user,
2001
                                         String password) {
2002

    
2003

    
2004
    try {
2005

    
2006

    
2007
        AuthSession auth = new AuthSession();
2008
        String principals = auth.getPrincipals(user, password);
2009
        out.println(principals);
2010

    
2011
    } catch (Exception e) {
2012
      out.println("<?xml version=\"1.0\"?>");
2013
      out.println("<error>");
2014
      out.println(e.getMessage());
2015
      out.println("</error>");
2016
    }
2017

    
2018
  }
2019

    
2020
  /**
2021
   * Handle "getdoctypes" action.
2022
   * Read all doctypes from db connection in XML format
2023
   */
2024
  private void handleGetDoctypesAction(PrintWriter out, Hashtable params,
2025
                                       HttpServletResponse response) {
2026

    
2027

    
2028
    try {
2029

    
2030

    
2031
        DBUtil dbutil = new DBUtil();
2032
        String doctypes = dbutil.readDoctypes();
2033
        out.println(doctypes);
2034

    
2035
    } catch (Exception e) {
2036
      out.println("<?xml version=\"1.0\"?>");
2037
      out.println("<error>");
2038
      out.println(e.getMessage());
2039
      out.println("</error>");
2040
    }
2041

    
2042
  }
2043

    
2044
  /**
2045
   * Handle the "getdtdschema" action.
2046
   * Read DTD or Schema file for a given doctype from Metacat catalog system
2047
   */
2048
  private void handleGetDTDSchemaAction(PrintWriter out, Hashtable params,
2049
                                        HttpServletResponse response) {
2050

    
2051

    
2052
    String doctype = null;
2053
    String[] doctypeArr = (String[])params.get("doctype");
2054

    
2055
    // get only the first doctype specified in the list of doctypes
2056
    // it could be done for all doctypes in that list
2057
    if (doctypeArr != null) {
2058
        doctype = ((String[])params.get("doctype"))[0];
2059
    }
2060

    
2061
    try {
2062

    
2063

    
2064
        DBUtil dbutil = new DBUtil();
2065
        String dtdschema = dbutil.readDTDSchema(doctype);
2066
        out.println(dtdschema);
2067

    
2068
    } catch (Exception e) {
2069
      out.println("<?xml version=\"1.0\"?>");
2070
      out.println("<error>");
2071
      out.println(e.getMessage());
2072
      out.println("</error>");
2073
    }
2074

    
2075
  }
2076

    
2077
  /**
2078
   * Handle the "getdataguide" action.
2079
   * Read Data Guide for a given doctype from db connection in XML format
2080
   */
2081
  private void handleGetDataGuideAction(PrintWriter out, Hashtable params,
2082
                                        HttpServletResponse response) {
2083

    
2084

    
2085
    String doctype = null;
2086
    String[] doctypeArr = (String[])params.get("doctype");
2087

    
2088
    // get only the first doctype specified in the list of doctypes
2089
    // it could be done for all doctypes in that list
2090
    if (doctypeArr != null) {
2091
        doctype = ((String[])params.get("doctype"))[0];
2092
    }
2093

    
2094
    try {
2095

    
2096

    
2097
        DBUtil dbutil = new DBUtil();
2098
        String dataguide = dbutil.readDataGuide(doctype);
2099
        out.println(dataguide);
2100

    
2101
    } catch (Exception e) {
2102
      out.println("<?xml version=\"1.0\"?>");
2103
      out.println("<error>");
2104
      out.println(e.getMessage());
2105
      out.println("</error>");
2106
    }
2107

    
2108
  }
2109

    
2110
  /**
2111
   * Handle the "getlastdocid" action.
2112
   * Get the latest docid with rev number from db connection in XML format
2113
   */
2114
  private void handleGetMaxDocidAction(PrintWriter out, Hashtable params,
2115
                                        HttpServletResponse response) {
2116

    
2117

    
2118
    String scope = ((String[])params.get("scope"))[0];
2119
    if (scope == null) {
2120
        scope = ((String[])params.get("username"))[0];
2121
    }
2122

    
2123
    try {
2124

    
2125

    
2126
        DBUtil dbutil = new DBUtil();
2127
        String lastDocid = dbutil.getMaxDocid(scope);
2128
        out.println("<?xml version=\"1.0\"?>");
2129
        out.println("<lastDocid>");
2130
        out.println("  <scope>" + scope + "</scope>");
2131
        out.println("  <docid>" + lastDocid + "</docid>");
2132
        out.println("</lastDocid>");
2133

    
2134
    } catch (Exception e) {
2135
      out.println("<?xml version=\"1.0\"?>");
2136
      out.println("<error>");
2137
      out.println(e.getMessage());
2138
      out.println("</error>");
2139
    }
2140

    
2141
  }
2142

    
2143
  /**
2144
   * Handle documents passed to metacat that are encoded using the
2145
   * "multipart/form-data" mime type.  This is typically used for uploading
2146
   * data files which may be binary and large.
2147
   */
2148
  private void handleMultipartForm(HttpServletRequest request,
2149
                                   HttpServletResponse response)
2150
  {
2151
    PrintWriter out = null;
2152
    String action = null;
2153

    
2154
    // Parse the multipart form, and save the parameters in a Hashtable and
2155
    // save the FileParts in a hashtable
2156

    
2157
    Hashtable params = new Hashtable();
2158
    Hashtable fileList = new Hashtable();
2159
    int sizeLimit = (new Integer(MetaCatUtil.getOption("datafilesizelimit")))
2160
                                                                   .intValue();
2161
    MetaCatUtil.debugMessage("The limit size of data file is: "+sizeLimit, 50);
2162

    
2163
    try {
2164
      // MBJ: need to put filesize limit in Metacat config (metacat.properties)
2165
      MultipartParser mp = new MultipartParser(request, sizeLimit*1024*1024); 
2166
      Part part;
2167
      while ((part = mp.readNextPart()) != null) {
2168
        String name = part.getName();
2169

    
2170
        if (part.isParam()) {
2171
          // it's a parameter part
2172
          ParamPart paramPart = (ParamPart) part;
2173
          String value = paramPart.getStringValue();
2174
          params.put(name, value);
2175
          if (name.equals("action")) {
2176
            action = value;
2177
          }
2178
        } else if (part.isFile()) {
2179
          // it's a file part
2180
          FilePart filePart = (FilePart) part;
2181
          fileList.put(name, filePart);
2182

    
2183
          // Stop once the first file part is found, otherwise going onto the
2184
          // next part prevents access to the file contents.  So...for upload
2185
          // to work, the datafile must be the last part
2186
          break;
2187
        }
2188
      }
2189
    } catch (IOException ioe) {
2190
      try {
2191
        out = response.getWriter();
2192
      } catch (IOException ioe2) {
2193
        System.err.println("Fatal Error: couldn't get response output stream.");
2194
      }
2195
      out.println("<?xml version=\"1.0\"?>");
2196
      out.println("<error>");
2197
      out.println("Error: problem reading multipart data.");
2198
      out.println("</error>");
2199
    }
2200

    
2201
    // Get the session information
2202
    String username = null;
2203
    String password = null;
2204
    String[] groupnames = null;
2205
    String sess_id = null;
2206

    
2207
    // be aware of session expiration on every request
2208
    HttpSession sess = request.getSession(true);
2209
    if (sess.isNew()) {
2210
      // session expired or has not been stored b/w user requests
2211
      username = "public";
2212
      sess.setAttribute("username", username);
2213
    } else {
2214
      username = (String)sess.getAttribute("username");
2215
      password = (String)sess.getAttribute("password");
2216
      groupnames = (String[])sess.getAttribute("groupnames");
2217
      try {
2218
        sess_id = (String)sess.getId();
2219
      } catch(IllegalStateException ise) {
2220
        System.out.println("error in  handleMultipartForm: this shouldn't " +
2221
                           "happen: the session should be valid: " +
2222
                           ise.getMessage());
2223
      }
2224
    }
2225

    
2226
    // Get the out stream
2227
    try {
2228
          out = response.getWriter();
2229
        } catch (IOException ioe2) {
2230
          util.debugMessage("Fatal Error: couldn't get response "+
2231
                             "output stream.", 30);
2232
        }
2233

    
2234
    if ( action.equals("upload")) {
2235
      if (username != null &&  !username.equals("public")) {
2236
        handleUploadAction(request, out, params, fileList,
2237
                           username, groupnames);
2238
      } else {
2239

    
2240
        out.println("<?xml version=\"1.0\"?>");
2241
        out.println("<error>");
2242
        out.println("Permission denied for " + action);
2243
        out.println("</error>");
2244
      }
2245
    } else {
2246
      /*try {
2247
        out = response.getWriter();
2248
      } catch (IOException ioe2) {
2249
        System.err.println("Fatal Error: couldn't get response output stream.");
2250
      }*/
2251
      out.println("<?xml version=\"1.0\"?>");
2252
      out.println("<error>");
2253
      out.println("Error: action not registered.  Please report this error.");
2254
      out.println("</error>");
2255
    }
2256
    out.close();
2257
  }
2258

    
2259
  /**
2260
   * Handle the upload action by saving the attached file to disk and
2261
   * registering it in the Metacat db
2262
   */
2263
  private void handleUploadAction(HttpServletRequest request,
2264
                                  PrintWriter out,
2265
                                  Hashtable params, Hashtable fileList,
2266
                                  String username, String[] groupnames)
2267
  {
2268
    //PrintWriter out = null;
2269
    //Connection conn = null;
2270
    String action = null;
2271
    String docid = null;
2272

    
2273
    /*response.setContentType("text/xml");
2274
    try
2275
    {
2276
      out = response.getWriter();
2277
    }
2278
    catch (IOException ioe2)
2279
    {
2280
      System.err.println("Fatal Error: couldn't get response output stream.");
2281
    }*/
2282

    
2283
    if (params.containsKey("docid"))
2284
    {
2285
      docid = (String)params.get("docid");
2286
    }
2287

    
2288
    // Make sure we have a docid and datafile
2289
    if (docid != null && fileList.containsKey("datafile")) {
2290

    
2291
      // Get a reference to the file part of the form
2292
      FilePart filePart = (FilePart)fileList.get("datafile");
2293
      String fileName = filePart.getFileName();
2294
      MetaCatUtil.debugMessage("Uploading filename: " + fileName, 10);
2295

    
2296
      // Check if the right file existed in the uploaded data
2297
      if (fileName != null) {
2298

    
2299
        try
2300
        {
2301
           //MetaCatUtil.debugMessage("Upload datafile " + docid +"...", 10);
2302
           //If document get lock data file grant
2303
           if (DocumentImpl.getDataFileLockGrant(docid))
2304
           {
2305
              // register the file in the database (which generates an exception
2306
              //if the docid is not acceptable or other untoward things happen
2307
              DocumentImpl.registerDocument(fileName, "BIN", docid, username);
2308

    
2309
              // Save the data file to disk using "docid" as the name
2310
              dataDirectory.mkdirs();
2311
              File newFile = new File(dataDirectory, docid);
2312
              long size = filePart.writeTo(newFile);
2313

    
2314
              // Force replication this data file
2315
              // To data file, "insert" and update is same
2316
              // The fourth parameter is null. Because it is notification server
2317
              // and this method is in MetaCatServerlet. It is original command,
2318
              // not get force replication info from another metacat
2319
              ForceReplicationHandler frh = new ForceReplicationHandler
2320
                                                (docid, "insert", false, null);
2321

    
2322
              // set content type and other response header fields first
2323
              out.println("<?xml version=\"1.0\"?>");
2324
              out.println("<success>");
2325
              out.println("<docid>" + docid + "</docid>");
2326
              out.println("<size>" + size + "</size>");
2327
              out.println("</success>");
2328
          }//if
2329

    
2330
        } //try
2331
        catch (Exception e)
2332
        {
2333
          out.println("<?xml version=\"1.0\"?>");
2334
          out.println("<error>");
2335
          out.println(e.getMessage());
2336
          out.println("</error>");
2337
        }
2338

    
2339
      }
2340
      else
2341
      {
2342
        // the field did not contain a file
2343
        out.println("<?xml version=\"1.0\"?>");
2344
        out.println("<error>");
2345
        out.println("The uploaded data did not contain a valid file.");
2346
        out.println("</error>");
2347
      }
2348
    }
2349
    else
2350
    {
2351
      // Error bcse docid missing or file missing
2352
      out.println("<?xml version=\"1.0\"?>");
2353
      out.println("<error>");
2354
      out.println("The uploaded data did not contain a valid docid " +
2355
                  "or valid file.");
2356
      out.println("</error>");
2357
    }
2358
  }
2359
  
2360
  /*
2361
   * A method to handle set access action
2362
   */
2363
  private void handleSetAccessAction(PrintWriter out,
2364
                                   Hashtable params,
2365
                                   String username)
2366
  {
2367
    String [] docList        = null;
2368
    String [] principalList  = null;
2369
    String [] permissionList = null;
2370
    String [] permTypeList   = null;
2371
    String [] permOrderList  = null;
2372
    String permission = null;
2373
    String permType   = null;
2374
    String permOrder  = null;
2375
    Vector errorList  = new Vector();
2376
    String error      = null;
2377
    Vector successList = new Vector();
2378
    String success    = null;
2379
   
2380
    
2381
    // Get parameters
2382
    if (params.containsKey("docid")) 
2383
    {
2384
      docList = (String[])params.get("docid");
2385
    }
2386
    if (params.containsKey("principal"))
2387
    {
2388
      principalList = (String[])params.get("principal"); 
2389
    }
2390
    if (params.containsKey("permission"))
2391
    {
2392
      permissionList = (String[])params.get("permission");
2393
      
2394
    }
2395
    if (params.containsKey("permType"))
2396
    {
2397
      permTypeList = (String[])params.get("permType");
2398
    
2399
    }
2400
    if (params.containsKey("permOrder"))
2401
    {
2402
      permOrderList = (String[])params.get("permOrder");
2403
     
2404
    }
2405
   
2406
    // Make sure the parameter is not null
2407
    if (docList == null || principalList == null || permTypeList == null ||
2408
        permissionList == null)
2409
    {
2410
      error = "Please check your parameter list, it should look like: "+
2411
              "?action=setaccess&docid=pipeline.1.1&principal=public" +
2412
              "&permission=read&permType=allow&permOrder=allowFirst";
2413
      errorList.addElement(error);
2414
      outputResponse(successList, errorList, out);
2415
      return;
2416
    }
2417
    
2418
    // Only select first element for permission, type and order
2419
    permission = permissionList[0];
2420
    permType = permTypeList[0];
2421
    if (permOrderList != null)
2422
    {
2423
       permOrder = permOrderList[0];
2424
    }
2425
    
2426
    // Get package doctype set
2427
    Vector packageSet =MetaCatUtil.getOptionList(
2428
                                    MetaCatUtil.getOption("packagedoctypeset"));
2429
    //debug
2430
    if (packageSet != null)
2431
    {
2432
      for (int i = 0; i<packageSet.size(); i++)
2433
      {
2434
        MetaCatUtil.debugMessage("doctype in package set: " + 
2435
                              (String)packageSet.elementAt(i), 34);
2436
      }
2437
    }//if
2438
    
2439
    // handle every accessionNumber
2440
    for (int i=0; i <docList.length; i++)
2441
    {
2442
      String accessionNumber = docList[i];
2443
      String owner = null;
2444
      String publicId = null;
2445
      // Get document owner and public id
2446
      try
2447
      {
2448
        owner = getFieldValueForDoc(accessionNumber, "user_owner");
2449
        publicId = getFieldValueForDoc(accessionNumber, "doctype");
2450
      }//try
2451
      catch (Exception e)
2452
      {
2453
        MetaCatUtil.debugMessage("Error in handleSetAccessAction: " +
2454
                                  e.getMessage(), 30);
2455
        error = "Error in set access control for document - " + accessionNumber+
2456
                 e.getMessage();
2457
        errorList.addElement(error);
2458
        continue;
2459
      }
2460
      //check if user is the owner. Only owner can do owner                            
2461
      if (username == null || owner == null || !username.equals(owner))
2462
      {
2463
        error = "User - " + username + " does not have permission to set " +
2464
                "access control for docid - " + accessionNumber;
2465
        errorList.addElement(error);
2466
        continue;
2467
      }
2468
      
2469
      // If docid publicid is BIN data file or other beta4, 6 package document
2470
      // we could not do set access control. Because we don't want inconsistent
2471
      // to its access docuemnt
2472
      if (publicId!=null && packageSet!=null && packageSet.contains(publicId))
2473
      {
2474
        error = "Could not set access control to document "+ accessionNumber +
2475
                "because it is in a pakcage and it has a access file for it";
2476
        errorList.addElement(error);
2477
        continue;
2478
      }
2479
      
2480
      // for every principle
2481
      for (int j = 0; j<principalList.length; j++)
2482
      {
2483
        String principal = principalList[j];
2484
        try
2485
        {
2486
          //insert permission
2487
          AccessControlForSingleFile accessControl = new 
2488
                           AccessControlForSingleFile(accessionNumber,
2489
                                    principal, permission, permType, permOrder);
2490
          accessControl.insertPermissions();
2491
          success = "Set access control to document "+ accessionNumber +
2492
                    " successfully";
2493
          successList.addElement(success);
2494
        }
2495
        catch (Exception ee)
2496
        {
2497
          MetaCatUtil.debugMessage("Erorr in handleSetAccessAction2: " +
2498
                                   ee.getMessage(), 30);
2499
          error = "Faild to set access control for document " + 
2500
                  accessionNumber + " because " + ee.getMessage();
2501
          errorList.addElement(error);
2502
          continue;
2503
        }
2504
      }//for every principle
2505
    }//for every document 
2506
    outputResponse(successList, errorList, out);
2507
  }//handleSetAccessAction
2508
  
2509
 
2510
  /*
2511
   * A method try to determin a docid's public id, if couldn't find null
2512
   * will be returned.
2513
   */
2514
  private String getFieldValueForDoc(String accessionNumber, String fieldName) 
2515
                                      throws Exception
2516
  {
2517
    if (accessionNumber==null || accessionNumber.equals("") ||fieldName == null
2518
        || fieldName.equals(""))
2519
    {
2520
      throw new Exception("Docid or field name was not specified");
2521
    }
2522
    
2523
    PreparedStatement pstmt = null;
2524
    ResultSet rs = null;
2525
    String fieldValue = null;
2526
    String docId = null;
2527
    DBConnection conn = null;
2528
    int serialNumber = -1;
2529
    
2530
    // get rid of revision if access number has
2531
    docId = MetaCatUtil.getDocIdFromString(accessionNumber);
2532
    try
2533
    {
2534
      //check out DBConnection
2535
      conn=DBConnectionPool.getDBConnection("MetaCatServlet.getPublicIdForDoc");
2536
      serialNumber=conn.getCheckOutSerialNumber();
2537
      pstmt = conn.prepareStatement(
2538
            "SELECT " + fieldName + " FROM xml_documents " +
2539
            "WHERE docid = ? ");
2540
          
2541
      pstmt.setString(1, docId);
2542
      pstmt.execute();
2543
      rs = pstmt.getResultSet();
2544
      boolean hasRow = rs.next();
2545
      int perm = 0;
2546
      if ( hasRow ) 
2547
      {
2548
        fieldValue = rs.getString(1);
2549
      }
2550
      else
2551
      {
2552
        throw new Exception("Could not find document: "+accessionNumber);
2553
      }
2554
    }//try
2555
    catch (Exception e)
2556
    {
2557
      MetaCatUtil.debugMessage("Exception in MetacatServlet.getPublicIdForDoc: "
2558
                               + e.getMessage(), 30);
2559
      throw e;
2560
    }
2561
    finally
2562
    {
2563
      try
2564
      {
2565
        rs.close();
2566
        pstmt.close();
2567
        
2568
      }
2569
      finally
2570
      {
2571
        DBConnectionPool.returnDBConnection(conn, serialNumber);
2572
      }
2573
    }
2574
    return fieldValue;
2575
  }//getFieldValueForDoc
2576
  
2577
  /*
2578
   * A method to output setAccess action result
2579
   */
2580
  private void outputResponse(Vector successList, 
2581
                              Vector errorList,
2582
                              PrintWriter out)
2583
  {
2584
    boolean error = false;
2585
    boolean success = false;
2586
    // Output prolog
2587
    out.println(PROLOG);
2588
    // output success message
2589
    if ( successList != null)
2590
    {
2591
      for (int i = 0; i<successList.size(); i++)
2592
      {
2593
        out.println(SUCCESS);
2594
        out.println((String)successList.elementAt(i));
2595
        out.println(SUCCESSCLOSE);
2596
        success = true;
2597
      }//for
2598
    }//if
2599
    // output error message
2600
    if (errorList != null)
2601
    {
2602
      for (int i = 0; i<errorList.size(); i++)
2603
      {
2604
        out.println(ERROR);
2605
        out.println((String)errorList.elementAt(i));
2606
        out.println(ERRORCLOSE);
2607
        error = true;
2608
      }//for
2609
    }//if
2610
    
2611
    // if no error and no success info, send a error that nothing happened
2612
    if( !error && !success)
2613
    {
2614
      out.println(ERROR);
2615
      out.println("Nothing happend for setaccess action");
2616
      out.println(ERRORCLOSE);
2617
    }
2618
    
2619
  }//outputResponse
2620
}
(40-40/57)