netservlet.java

derby database source code.good for you.

/*

   Derby - Class org.apache.derby.drda.NetServlet

   Copyright 2002, 2004 The Apache Software Foundation or its licensors, as applicable.

   Licensed under the Apache License, Version 2.0 (the "License");
   you may not use this file except in compliance with the License.
   You may obtain a copy of the License at

      http://www.apache.org/licenses/LICENSE-2.0

   Unless required by applicable law or agreed to in writing, software
   distributed under the License is distributed on an "AS IS" BASIS,
   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
   See the License for the specific language governing permissions and
   limitations under the License.

 */

package org.apache.derby.drda;

import java.io.*;
import java.util.*;

import java.sql.*;
import java.net.*;

import java.security.AccessController;
import java.security.PrivilegedExceptionAction;
import java.security.PrivilegedActionException;

import javax.servlet.*;
import javax.servlet.http.*;

import org.apache.derby.iapi.tools.i18n.LocalizedResource;
import org.apache.derby.iapi.reference.Property;

/**
	This servlet can be used to start Derby Network Server from a remote location.
	<P>
	These servlet configuration parameters are understood by this servlet.
	<UL>
	<LI><PRE>portNumber</PRE> - Port number to use. The default is 1527.
	<LI><PRE>startNetworkServerOnInit</PRE> - Starts the Derby Network Server at servlet 
			initialization if 'true'.
	<LI><PRE>tracingDirectory</PRE> - Directory for trace files
	</UL>

*/
public class NetServlet extends HttpServlet {
	private final static int MAX_CONNECT_TRYS = 20;
	private final static String SERVLET_PROP_MESSAGES =  "org.apache.derby.loc.drda.servlet";
	private final static String SERVLET_ADDRESS = "derbynet";

	private String formHeader = null;
	private String host = "localhost";
	private int portNumber=1527;
	private String tracingDirectory;
	private boolean logStatus= false;	/* Logging off */
	private boolean traceStatus = false;	/* Tracing off */
	private String[] knownLang = {"en","es","de_DE", "fr", "it", "ja_JP",
			"ko_KR", "zh_CN", "zh_TW"};
	private String locale;

	private final static int NOT_GIVEN = -2;
	private final static int INVALID = -3;

	private NetworkServerControl server;
	private PrintWriter out;

	// for doPri block
	private Runnable service;

	/**
		Initialize the servlet.
		Configuration parameters:
		<UL>
		<LI><PRE>portNumber</PRE> - Port number
		<LI><PRE>traceDirectory</PRE> - location of trace directory
		<LI><PRE>startNetworkServerOnInit</PRE> - start the server on initialization
		</UL>
	*/
	public void init(ServletConfig config)
		throws ServletException
	{

		LocalizedResource langUtil = new LocalizedResource(null,null,SERVLET_PROP_MESSAGES);
				
		String port = config.getInitParameter("portNumber");
		if (port != null) {
			int p = Integer.valueOf(port).intValue();
			if (p > 0)
				portNumber = p;
		}

		String tracingDirectory = config.getInitParameter("tracingDirectory");

		String startup = config.getInitParameter("startNetworkServerOnInit");

		// test if the server is already running
		try {
			//don't send output to console
			if (server == null)
				server = new NetworkServerControl(InetAddress.getByName(host), portNumber);
			if (isServerStarted(server,1))
				return;
		} catch (Exception e) {}

		if (startup != null) {
			boolean start = Boolean.valueOf(startup).booleanValue();
			if (start)
			{
				runServer(langUtil, null, null);
				return;
			}
		}
	}

	/**
		Get the form of NetServlet. Provides buttons and forms to control the
		Network server.
	*/
	public void doGet (HttpServletRequest request, HttpServletResponse response)
	        throws ServletException, IOException
	{
		String logOnMessage;
		String logOffMessage;
		String traceOnMessage;
		String traceOffMessage;
		String traceOnOffMessage;
		String startMessage;
		String stopMessage;
		String returnMessage;
		String traceSessionMessage;
		String traceDirMessage;
		String contMessage;
		String setParamMessage;
		String setParamMessage2;
		String netParamMessage;
		LocalizedResource langUtil;
		
		langUtil = getCurrentAppUI(request);
		response.setContentType("text/html; charset=UTF-8");
		
		//prevent caching of the servlet since contents can change - beetle 4649
		response.setHeader("Cache-Control", "no-cache,no-store");

		formHeader = "<form enctype='multipart/form-data; charset=UTF-8' action='" +
			request.getRequestURI() +" '>";

		out = new PrintWriter(new
 			OutputStreamWriter(response.getOutputStream(), "UTF8"),true);
		
		//inialize messages
		logOnMessage = escapeSingleQuotes(langUtil.getTextMessage("SRV_LogOn"));
		logOffMessage = escapeSingleQuotes(langUtil.getTextMessage("SRV_LogOff"));
		traceOnMessage = escapeSingleQuotes(langUtil.getTextMessage("SRV_TraceOn"));
		traceOffMessage = escapeSingleQuotes(langUtil.getTextMessage("SRV_TraceOff"));
		startMessage = escapeSingleQuotes(langUtil.getTextMessage("SRV_Start"));
		stopMessage = escapeSingleQuotes(langUtil.getTextMessage("SRV_Stop"));
		traceSessionMessage = escapeSingleQuotes(langUtil.getTextMessage("SRV_TraceSessButton"));
		traceOnOffMessage = escapeSingleQuotes(langUtil.getTextMessage("SRV_TraceOnOff"));
		returnMessage = escapeSingleQuotes(langUtil.getTextMessage("SRV_Return"));
		traceDirMessage = escapeSingleQuotes(langUtil.getTextMessage("SRV_TraceDir"));
		contMessage = escapeSingleQuotes(langUtil.getTextMessage("SRV_Continue"));
		setParamMessage = escapeSingleQuotes(langUtil.getTextMessage("SRV_SetParam"));
		setParamMessage2 = escapeSingleQuotes(langUtil.getTextMessage("SRV_SetParam2"));
		netParamMessage = escapeSingleQuotes(langUtil.getTextMessage("SRV_NetParam"));

		printBanner(langUtil);
		// set up a server we can use
		if (server == null) {
			try {
				server = new NetworkServerControl();
			}catch (Exception e) {
				printErrorForm(langUtil, request, e, returnMessage);
				return;
			}
		}
		server.setClientLocale(locale);
		String form = getForm(request);
		String doAction = getDoAction(request);
		// if doAction is set, use it to determine form
		if (doAction != null )
		{
			if (doAction.equals(traceOnOffMessage))
				form = traceSessionMessage;
			else
				form = doAction;
		}
		// if no form, determine form based on server status
		boolean serverStatus = getServerStatus();
		if (form == null)
		{
			if (serverStatus)
				form = startMessage;
			else
				form = stopMessage;
		}
		else if (form.equals(startMessage))
		{
			if (!serverStatus)  {
				runServer(langUtil, request, returnMessage);
			}
		}
		else if (form.equals(stopMessage))
		{
			if (serverStatus)   {
				shutdownServer(langUtil, request, returnMessage);
			}
			setDefaults();
					
		}
		else if (form.equals(returnMessage) || form.equals(returnMessage))
		{
			// check if server is still running and use that to determine which form
			if (serverStatus)
			{
				form = startMessage;
			}
			else
			{
				form = stopMessage;
			}
		}

		out.println( formHeader);
		// display forms

		form = escapeSingleQuotes(form);
		doAction = escapeSingleQuotes(doAction);
	  	if (form.equals(startMessage))
		{
			String logButton = getLogging(request);
			String traceButton = getTrace(request);
			if (logButton !=  null && logButton.equals(logOnMessage))
			{
				if (logging(langUtil, true, request, returnMessage))
					logStatus = true;
			}
			if (logButton !=  null && logButton.equals(logOffMessage))
			{
				if (logging(langUtil, false, request, returnMessage))
					logStatus = false;
			}
			if (traceButton !=  null && traceButton.equals(traceOnMessage))
			{
				if (traceAll(langUtil, true, request, returnMessage))
					traceStatus = true;
			}
			if (traceButton !=  null && traceButton.equals(traceOffMessage))
			{
				if (traceAll(langUtil, false, request, returnMessage))
					traceStatus = false;
			}
			displayCurrentStatus(request, langUtil, returnMessage);
			out.println( "<h4>"+langUtil.getTextMessage("SRV_StopButton")+"</h4>" );
			out.println( "<INPUT type=submit name=form value='"+ stopMessage + "'>" );

			out.println( "<h4>"+langUtil.getTextMessage("SRV_LogButton2")+"</h4>" );

			if (logStatus)
			{
				out.println( "<INPUT type=submit name=logform value='"+logOffMessage + "'>" );
			}
			else
			{
				out.println( "<INPUT type=submit name=logform value='"+logOnMessage + "'>" );
			}
			out.println( "<h4>"+langUtil.getTextMessage("SRV_TraceButton2")+"</h4>" );
			if (traceStatus)
			{
				out.println( "<INPUT type=submit name=traceform value='"+traceOffMessage+ "'>" );
			}
			else
			{
				out.println( "<INPUT type=submit name=traceform value='"+traceOnMessage + "'>" );
			}

			out.println( "<h4>"+langUtil.getTextMessage("SRV_TraceSession")+"</h4>" );
			out.println( "<INPUT type=submit name=form value='"+ traceSessionMessage + "'>" );
			out.println( "<h4>"+langUtil.getTextMessage("SRV_TraceDirButton")+"</h4>" );
			out.println( "<INPUT type=submit name=form value='"+ traceDirMessage + "'>" );
			out.println( "<h4>"+langUtil.getTextMessage("SRV_ThreadButton")+"</h4>" );
			out.println( "<INPUT type=submit name=form value='"+ netParamMessage+ "'>" );
		}
		else if (form.equals(stopMessage))
		{

			printAsContentHeader(langUtil.getTextMessage("SRV_NotStarted"));
			String logButton = getLogging(request);
			String traceButton =  getTrace(request);
			if (logButton !=  null && logButton.equals(logOnMessage))
				logStatus = true;
			if (logButton !=  null && logButton.equals(logOffMessage))
				logStatus = false;
			if (traceButton !=  null && traceButton.equals(traceOnMessage))
				traceStatus = true;
			if (traceButton !=  null && traceButton.equals(traceOffMessage))
				traceStatus = false;
			if (logStatus)
			{
				out.println( "<h4>"+langUtil.getTextMessage("SRV_LogOffButton")+"</h4>" );
				out.println( "<INPUT type=submit name=logform value='"+logOffMessage + "'>" );
			}
			else
			{
				out.println( "<h4>"+langUtil.getTextMessage("SRV_LogOnButton")+"</h4>" );
				out.println( "<INPUT type=submit name=logform value='"+logOnMessage + "'>" );
			}
			if (traceStatus)
			{
				out.println( "<h4>"+langUtil.getTextMessage("SRV_TraceOffButton")+"</h4>" );
				out.println( "<INPUT type=submit name=traceform value='"+traceOffMessage + "'>" );
			}
			else
			{
				out.println( "<h4>"+langUtil.getTextMessage("SRV_TraceOnButton")+"</h4>" );
				out.println( "<INPUT type=submit name=traceform value='"+traceOnMessage + "'>" );
			}
			out.println( "<h4>"+langUtil.getTextMessage("SRV_StartButton")+"</h4>" );
			out.println( "<INPUT type=submit name=form value='"+startMessage+ "'>" );
		}
		else if (form.equals(traceSessionMessage))
		{
			if (doAction != null)
			{
				if (doAction.equals(traceOnOffMessage))
				{
					String sessionid = request.getParameter("sessionid");
					int session = 0;
					try {
					 	session = (new Integer(sessionid)).intValue();
					} catch (Exception e) {
						printErrorForm(langUtil, request,
							langUtil.getTextMessage("SRV_InvalidVal",
							sessionid, langUtil.getTextMessage("SRV_SessionID")),
							returnMessage);
						return;
					}
					Properties p = null;
					try {
						p = server.getCurrentProperties();
					} catch (Exception e) {
						printErrorForm(langUtil, request, e, returnMessage);
						return;
					}