Logo Search packages:      
Sourcecode: paros version File versions  Download package

Session.java

/*
 *
 * Paros and its related class files.
 * 
 * Paros is an HTTP/HTTPS proxy for assessing web application security.
 * Copyright (C) 2003-2004 Chinotec Technologies Company
 * 
 * This program is free software; you can redistribute it and/or
 * modify it under the terms of the Clarified Artistic License
 * as published by the Free Software Foundation.
 * 
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * Clarified Artistic License for more details.
 * 
 * You should have received a copy of the Clarified Artistic License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
 */
package org.parosproxy.paros.model;

import java.io.File;
import java.io.IOException;
import java.sql.SQLException;
import java.util.List;

import org.parosproxy.paros.Constant;
import org.parosproxy.paros.common.FileXML;
import org.parosproxy.paros.network.HttpMessage;
import org.xml.sax.SAXException;


/**
 *
 * To change the template for this generated type comment go to
 * Window - Preferences - Java - Code Generation - Code and Comments
 */
00039 public class Session extends FileXML {
      private static final String ROOT = "session";
      
      private static final String SESSION_DESC = "sessionDesc";
      private static final String SESSION_ID = "sessionId";
      private static final String SESSION_NAME = "sessionName";
      
      private static final String[] PATH_SESSION_DESC = {ROOT, SESSION_DESC}; 
      private static final String[] PATH_SESSION_ID = {ROOT, SESSION_ID};
      private static final String[] PATH_SESSION_NAME = {ROOT, SESSION_NAME};


      // other runtime members
      private Model model = null;
      private String fileName = "";
      private String sessionDesc = "";
      
      // parameters in XML
      private long sessionId = 0;
      private String sessionName = "";
      private SiteMap siteTree = null;
      
      /**
       * Constructor for the current session.  The current system time will be used as the session ID.
       * @param sessionId
       */
00065       public Session(Model model) {
            super(ROOT);

            /*try {
                  parseFile("xml/untitledsession.xml");
            } catch (SAXException e) {
                  e.printStackTrace();
            } catch (IOException e) {
                  e.printStackTrace();
            }*/
            
            // add session variable here
            setSessionId(System.currentTimeMillis());
            setSessionName("Untitled Session");
            setSessionDesc("");
            
            // create default object
            this.siteTree = SiteMap.createTree(model);
            
            this.model = model;
            
      }
      
      public void discard() {
          try {
              model.getDb().getTableHistory().deleteHistorySession(getSessionId());
        } catch (SQLException e) {
            e.printStackTrace();
        }
      }
      
      
    /**
     * @return Returns the sessionDesc.
     */
00100     public String getSessionDesc() {
        return sessionDesc;
    }
      
      /**
       * @return Returns the sessionId.
       */
00107       public long getSessionId() {
            return sessionId;
      }
      /**
       * @return Returns the name.
       */
00113       public String getSessionName() {
            return sessionName;
      }
    /**
     * @return Returns the siteTree.
     */
00119     public SiteMap getSiteTree() {
        return siteTree;
    }
      
      

    /**
     * @return Returns the newState.
     */
00128     public boolean isNewState() {
        return fileName.equals("");
    }

    
    public void open(final File file, final SessionListener callback) {
        Thread t = new Thread(new Runnable() {
            public void run() {
                Exception thrownException = null;
                try {
                    open(file.getAbsolutePath());
                } catch (Exception e) {
                    thrownException = e;
                }
                if (callback != null) {
                    callback.sessionOpened(file, thrownException);
                }
            }
        });
        t.setPriority(Thread.NORM_PRIORITY-2);
        t.start();
    }

      public void open(String fileName) throws SQLException, SAXException, IOException, Exception {

            readAndParseFile(fileName);
            model.getDb().close(false);
            model.getDb().open(fileName);
            this.fileName = fileName;
            
            //historyList.removeAllElements();

            SiteNode newRoot = new SiteNode("Sites");
            siteTree.setRoot(newRoot);

            // update history reference
            List list = model.getDb().getTableHistory().getHistoryList(getSessionId(), HistoryReference.TYPE_MANUAL);
            HistoryReference historyRef = null;
            
            for (int i=0; i<list.size(); i++) {
                  int historyId = ((Integer) list.get(i)).intValue();

                  try {
                        historyRef = new HistoryReference(historyId);
                        getSiteTree().addPath(historyRef);

                        if (i % 100 == 99) Thread.yield();
                  } catch (Exception e) {};
                  
            }
            
            // update siteTree reference
            list = model.getDb().getTableHistory().getHistoryList(getSessionId(), HistoryReference.TYPE_SPIDER);
            
            for (int i=0; i<list.size(); i++) {
                  int historyId = ((Integer) list.get(i)).intValue();

                  try {
                        historyRef = new HistoryReference(historyId);
                        getSiteTree().addPath(historyRef);

                        if (i % 100 == 99) Thread.yield();

                  } catch (Exception e) {};
                  
                  
            }
            
            System.gc();
            
      }
      
      protected void parse() throws Exception {
          
          long tempSessionId = 0;
          String tempSessionName = "";
          String tempSessionDesc = "";
          
          // use temp variable to check.  Exception will be flagged if any error.
            tempSessionId = Long.parseLong(getValue(SESSION_ID));
            tempSessionName = getValue(SESSION_NAME);
            tempSessionName = getValue(SESSION_NAME);

            // set member variable after here
            sessionId = tempSessionId;
            sessionName = tempSessionName;
            sessionDesc = tempSessionDesc;
            


      }

      /**
       * Asynchronous call to save a session.
       * @param fileName
       * @param callback
       */
00225     public void save(final String fileName, final SessionListener callback) {
        Thread t = new Thread(new Runnable() {
            public void run() {
                Exception thrownException = null;
                try {
                    save(fileName);
                } catch (Exception e) {
                    thrownException = e;
                }
                if (callback != null) {
                    callback.sessionSaved(thrownException);
                }
            }
        });
        t.setPriority(Thread.NORM_PRIORITY-2);
        t.start();
    }
    
    /**
     * Synchronous call to save a session.
     * @param fileName
     * @throws Exception
     */
00248       public void save(String fileName) throws Exception {
          saveFile(fileName);
            if (isNewState()) {
                model.moveSessionDb(fileName);
            } else {
                if (!this.fileName.equals(fileName)) {
                    // copy file to new fileName
                    model.copySessionDb(this.fileName, fileName);
                }
            }
          this.fileName = fileName;
            
            synchronized (siteTree) {
                saveSiteTree((SiteNode) siteTree.getRoot());
            }
            
            model.getDb().getTableSession().update(getSessionId(), getSessionName());

      }
      
    /**
     * @param sessionDesc The sessionDesc to set.
     */
00271     public void setSessionDesc(String sessionDesc) {
        this.sessionDesc = sessionDesc;
            setValue(PATH_SESSION_DESC, sessionDesc);
    }
      
      /**
       * @param sessionId The sessionId to set.
       */
00279       public void setSessionId(long sessionId) {
            this.sessionId = sessionId;
            //setText(SESSION_ID, Long.toString(sessionId));
            setValue(PATH_SESSION_ID, Long.toString(sessionId));

      }
      /**
       * @param name The name to set.
       */
00288       public void setSessionName(String name) {
            this.sessionName = name;
            //setText(SESSION_NAME, name);
            setValue(PATH_SESSION_NAME, name);
            
      }

    
    public String getFileName() {
        return fileName;
    }
    
    private void saveSiteTree(SiteNode node) {
        HttpMessage msg = null;

        if (!node.isRoot()) {
            if (node.getHistoryReference().getHistoryType() < 0) {
                // -ve means to be saved
                saveNodeMsg(msg);
            }
        }
        
        for (int i=0; i<node.getChildCount(); i++) {
            try {
                saveSiteTree((SiteNode) node.getChildAt(i));
            } catch (Exception e) {}
        }
        
    }
    
    private void saveNodeMsg(HttpMessage msg) {
        // nothing need to be done
    }
    
    public String getSessionFolder() {
        String result = "";
        if (fileName.equals("")) {
//            result = Constant.FOLDER_SESSION;
            result = Constant.getInstance().FOLDER_SESSION;
        } else {
            File file = new File(fileName);
            result = file.getParent();
        }
        return result;
    }

}

Generated by  Doxygen 1.6.0   Back to index