Session.java

Go to the documentation of this file.

package org.libcsdbg.jtracer;

import java.beans.PropertyChangeListener;

import java.util.Hashtable;
import java.util.Enumeration;
import java.util.Vector;

import java.io.IOException;
import java.io.InputStreamReader;
import java.io.BufferedReader;

import java.net.Socket;
import java.net.ProtocolException;

import java.awt.event.ActionEvent;
import java.awt.event.ActionListener;
import java.awt.Color;
import java.awt.Font;
import java.awt.Insets;
import java.awt.Dimension;
import java.awt.Container;
import java.awt.BorderLayout;

import javax.swing.event.ChangeEvent;
import javax.swing.event.ChangeListener;
import javax.swing.JFrame;
import javax.swing.JToolBar;
import javax.swing.JTabbedPane;
import javax.swing.JPanel;
import javax.swing.JScrollPane;

public class Session extends JFrame implements
  ActionListener,
  ChangeListener,
  Runnable
{
  private static final long serialVersionUID = 0x0400;

  private JToolBar tools = null;

  private JTabbedPane tabs = null;

  private Vector<TracePane> traces = null;

  private TraceBar details = null;

  private JFrame owner = null;

  private Socket link = null;

  private BufferedReader rx = null;

  private Thread server = null;

  private boolean isLocked = false;


  Session(JFrame f, Socket sock) throws IOException
  {
    super("Session");
    owner = f;

    Registry conf = Registry.getCurrent();
    setIconImages(conf.getProjectIcons());

    link = sock;
    //link.shutdownOutput();
    rx = new BufferedReader(new InputStreamReader(link.getInputStream()));

    /*Integer timeout = (Integer) conf.get("session", "timeout");
    if (timeout != null)
      link.setSoTimeout(timeout);*/

    tools = new JToolBar();
    tools.add(createTool("Remove trace", (ActionListener) owner));
    tools.add(createTool("Lock", this));
    tools.addSeparator();

    tools.add(createTool("Select previous", (ActionListener) owner));
    tools.add(createTool("Select next", (ActionListener) owner));
    tools.add(createTool("Close", (ActionListener) owner));
    tools.addSeparator();

    tools.add(createTool("Find", this));
    add(tools, BorderLayout.NORTH);

    Button tool = (Button) tools.getComponent(0);
    tool.addActionListener(this);

    traces = new Vector<TracePane>();
    tabs = new JTabbedPane();
    tabs.addChangeListener(this);
    tabs.setFont((Font) conf.get("component", "font"));
    tabs.setForeground((Color) conf.get("component", "fgcolor"));
    add(tabs, BorderLayout.CENTER);

    details = new TraceBar();
    add(details, BorderLayout.SOUTH);

    setSize((Dimension) conf.get("session", "size"));
    addPropertyChangeListener("sessionRequest", (PropertyChangeListener) owner);
    addPropertyChangeListener("traceCount", (PropertyChangeListener) owner);

    server = new Thread(this);
    server.start();
  }


  public void addTrace(Hashtable<String, String> rqst)
  {
    Registry conf = Registry.getCurrent();
    TracePane pane = new TracePane(rqst);
    traces.add(pane);

    /*String token = rqst.get("tstamp");
    details.setTimestamp(Long.parseLong(token, 16));

    token = rqst.get("exception");
    if (token != null)
      details.setMessage(token);
    else
      details.setMessage("Thread stack trace"); */

    String host = link.getInetAddress().getCanonicalHostName();
    String addr = link.getInetAddress().getHostAddress();
    int port = link.getPort();
    //details.setAddress(host, port);

    pane.append(rqst.get("trace"));
    JPanel view = new JPanel(new BorderLayout());
    view.add(pane);
    pane.setMargin(new Insets(6, 2, 0, 2));

    JScrollPane scroll = new JScrollPane(view);
    scroll.setVerticalScrollBarPolicy(JScrollPane.VERTICAL_SCROLLBAR_ALWAYS);
    tabs.addTab("Thread " + rqst.get("tid"), conf.loadIcon("new.png"), scroll);

    String path = rqst.get("path");
    int pid = Integer.parseInt(rqst.get("pid"), 16);
    setTitle(path + " (" + pid + "@" + addr + ")");

    int len = rqst.get("request").length();
    StringBuffer msg = new StringBuffer();
    msg.append("Read " + len + " bytes ");
    msg.append("for " + path + " (" + pid + "@");
    msg.append(addr + ":" + port + ")");
    firePropertyChange("sessionRequest", null, msg.toString());
    firePropertyChange("traceCount", null, tabs.getTabCount());

    tools.getComponent(0).setEnabled(true);
    tools.getComponent(7).setEnabled(true);
  }


  public boolean isIconified()
  {
    return (getExtendedState() & ICONIFIED) != 0;
  }


  public int setIconified(boolean how)
  {
    int state = getExtendedState();
    if (how)
      state |= ICONIFIED;
    else
      state &= ~ICONIFIED;

    setExtendedState(state);
    return state;
  }


  public Hashtable<String, String> getRequest() throws IOException
  {
    Hashtable<String, String> retval = new Hashtable<String, String>();

    int cnt = 0;
    boolean text = false;
    StringBuffer buffer = new StringBuffer();
    StringBuffer trace = new StringBuffer();

    /* Read a line at a time and parse it */
    do {
      String ln = rx.readLine();
      if (ln == null)
        throw new IOException("The peer disconnected prematurely");

      cnt++;
      buffer.append(ln).append("\n");
      if (ln.length() == 0) {
        text = true;
        continue;
      }

      if (text) {
        trace.append(ln).append("\n");
        continue;
      }

      String hdr[] = ln.split(":");
      if (hdr.length < 2 || hdr[0].trim().length() == 0) {
        String msg = "LDP request format error (" + cnt + ": " + ln + ")";
        throw new ProtocolException(msg);
      }

      String field = hdr[1];
      for (int i = 2; i < hdr.length; i++)
        field += ":" + hdr[i];

      retval.put(hdr[0].trim(), field.trim());
    }
    while (!buffer.toString().endsWith("}\n\n"));

    retval.put("trace", trace.toString().trim());
    retval.put("request", buffer.toString().trim());
    return retval;
  }


  public void quit()
  {
    try {
      link.close();
      rx.close();

      /*Thread thr = server;
      server = null;
      thr.interrupt();*/
    }

    catch (Throwable t) {
      Registry.debug(t);
    }
  }


  public int getTraceCount()
  {
    return tabs.getTabCount();
  }


  private Button createTool(String cmd, ActionListener handler)
  {
    String nm = cmd.toLowerCase().replace(' ', '_') + "24.png";
    Button retval = new Button(nm, cmd, handler);
    retval.setMargin(new Insets(2, 2, 2, 2));
    return retval;
  }


  public void run()
  {
    /* Listen for incoming connections */
    while (server == Thread.currentThread()) {
      try {
        Hashtable<String, String> rqst = getRequest();
        addTrace(rqst);
      }
      
      catch (Throwable t) {
        //Registry.debug(t);
        server = null;
      }
    }
  }


  public void actionPerformed(ActionEvent event)
  {
    try {
      String cmd = event.getActionCommand();
      
      if (cmd.equals("Remove trace")) {
        int i = tabs.getSelectedIndex();
        if (i < 0)
          return;
        
        traces.remove(i);
        tabs.remove(i);
        int cnt = tabs.getTabCount();
        firePropertyChange("traceCount", null, cnt);
        if (cnt != 0)
          return;

        if (!isLocked) {
          owner.toFront();
          dispose();
          return;
        }

        tools.getComponent(0).setEnabled(false);
        tools.getComponent(7).setEnabled(false);
      }

      else if (cmd.equals("Lock")) {
        isLocked = !isLocked;

        Registry conf = Registry.getCurrent();
        Button tool = (Button) tools.getComponent(1);

        if (isLocked) {
          tool.setIcon(conf.loadIcon("unlock24.png"));
          tool.setToolTipText("Unlock");
        }

        else {
          tool.setIcon(conf.loadIcon("lock24.png"));
          tool.setToolTipText("Lock");
        }
      }

      else if (cmd.equals("Find"))
        Alert.error(this, "Not implemented yet", false);
    }

    catch (Throwable t) {
      Registry.debug(t);
    }
  }


  public void stateChanged(ChangeEvent event)
  {
    try {
      int i = tabs.getSelectedIndex();
      if (i < 0) {
        details.clear();
        return;
      }

      tabs.setIconAt(i, Registry.getCurrent().loadIcon("void.png"));

      /*for (int j = 0, cnt = traces.size(); j < cnt; j++) {
        Hashtable<String, String> rqst = traces.get(j);

        System.out.println("Trace " + j + ": ");
        Enumeration keys = rqst.keys();
        while (keys.hasMoreElements()) {
          String key = (String) keys.nextElement();
          String field = rqst.get(key);

          System.out.println("\t" + key + ": " + field);
        }
      }*/
      
      TracePane pane = traces.get(i);
      String field = pane.getField("exception");
      if (field != null)
        details.setMessage(field);
      else
        details.setMessage("Thread stack trace");

      field = pane.getField("tstamp");
      details.setTimestamp(Long.parseLong(field, 16));

      String host = link.getInetAddress().getCanonicalHostName();
      int port = link.getPort();
      details.setAddress(host, port);
    }

    catch (Throwable t) {
      Registry.debug(t);
    }
  }
}