7 files changed, 560 insertions, 0 deletions
diff --git a/java/com/rsyslog/lib/DiagSess.java b/java/com/rsyslog/lib/DiagSess.java
new file mode 100644
index 00000000..799b9a4a
--- /dev/null
+++ b/java/com/rsyslog/lib/DiagSess.java
@@ -0,0 +1,78 @@
+/* The diagnostic session to an imdiag module (running inside rsyslogd).
+ *
+ * Copyright 2009 Rainer Gerhards and Adiscon GmbH.
+ *
+ * This file is part of rsyslog.
+ *
+ * Rsyslog is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation, either version 3 of the License, or
+ * (at your option) any later version.
+ *
+ * Rsyslog 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
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with Rsyslog.  If not, see <http://www.gnu.org/licenses/>.
+ *
+ * A copy of the GPL can be found in the file "COPYING" in this distribution.
+ */
+package com.rsyslog.lib;
+import java.io.*;
+import java.net.*;
+
+public class DiagSess {
+
+	private String host = new String("127.0.0.1");
+	private int port = 13500;
+	int timeout = 0;
+	private PrintWriter out = null;
+    private BufferedReader in = null;
+	private Socket diagSocket = null;
+
+	/** set connection timeout */
+	public void setTimeout(int timeout_) {
+		timeout = timeout_;
+	}
+	
+	public DiagSess(String host_, int port_) {
+		host = host_;
+		port = port_;
+	}
+	
+	/** connect to remote server. Initializes everything for request-reply
+	 * processing.
+	 * 
+	 * @throws IOException
+	 */
+	public void connect() throws IOException {
+        diagSocket = new Socket(host, port);
+	    diagSocket.setSoTimeout(timeout);
+        out = new PrintWriter(diagSocket.getOutputStream(), true);
+        in = new BufferedReader(new InputStreamReader(
+                                    diagSocket.getInputStream()));
+	
+	}
+	
+	/** end session with remote server. */
+	public void disconnect() throws IOException {
+		out.close();
+		in.close();
+		diagSocket.close();
+	}
+
+	/** issue a request to imdiag and return its response.
+	 * 
+	 * @param req request string
+	 * @return response string (unparsed)
+	 * @throws IOException
+	 */
+	public String request(String req) throws IOException {
+	    out.println(req);
+	    String resp = in.readLine();
+	    return resp;
+	}
+	
+}
diff --git a/java/com/rsyslog/lib/SyslogMessage.java b/java/com/rsyslog/lib/SyslogMessage.java
new file mode 100644
index 00000000..b544a6db
--- /dev/null
+++ b/java/com/rsyslog/lib/SyslogMessage.java
@@ -0,0 +1,75 @@
+/**
+ * Implementation of the syslog message object.
+ *
+ * This is a limited-capability implementation of a syslog message together
+ * with all its properties. It is limit to what is currently needed and may
+ * be extended as further need arises.
+ *
+ * @author Rainer Gerhards
+ * 
+ * Copyright 2009 Rainer Gerhards and Adiscon GmbH.
+ *
+ * This file is part of rsyslog.
+ *
+ * Rsyslog is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation, either version 3 of the License, or
+ * (at your option) any later version.
+ *
+ * Rsyslog 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
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with Rsyslog.  If not, see http://www.gnu.org/licenses/.
+ *
+ * A copy of the GPL can be found in the file "COPYING" in this distribution.
+ */
+package com.rsyslog.lib;
+
+public class SyslogMessage {
+
+	/** message as received from the wire */
+	private String rawmsg;
+	/** the rawmsg without the PRI part */
+	private String rawMsgAfterPRI;
+	/** PRI part */
+	private int pri;
+
+	/** a very simple syslog parser. So far, it only parses out the
+	 * PRI part of the message. May be extended later. Rawmsg must have
+	 * been set before the parser is called. It will populate "all" other
+	 * fields.
+	 */
+	private void parse() {
+		int i;
+		if(rawmsg.charAt(0) == '<') {
+			pri = 0;
+			for(i = 1 ; Character.isDigit(rawmsg.charAt(i)) && i < 4 ; ++i) {
+				pri = pri * 10 + rawmsg.charAt(i) - '0';
+			}
+			if(rawmsg.charAt(i) != '>')
+				/* not a real cure, but sufficient for the current
+				 * mini-parser... */
+				--i;
+			rawMsgAfterPRI = rawmsg.substring(i + 1);
+		} else {
+			pri = 116;
+			rawMsgAfterPRI = rawmsg;
+		}
+	}
+
+	public SyslogMessage(String _rawmsg) {
+		rawmsg = _rawmsg;
+		parse();
+	}
+
+	public String getRawMsg() {
+		return rawmsg;
+	}
+
+	public String getRawMsgAfterPRI() {
+		return rawMsgAfterPRI;
+	}
+}
diff --git a/java/com/rsyslog/lib/SyslogMsgConsumer.java b/java/com/rsyslog/lib/SyslogMsgConsumer.java
new file mode 100644
index 00000000..42c9931a
--- /dev/null
+++ b/java/com/rsyslog/lib/SyslogMsgConsumer.java
@@ -0,0 +1,29 @@
+/**
+ * Interface for SyslogConsumers.
+ * 
+ * @author Rainer Gerhards
+ * 
+ * Copyright 2009 Rainer Gerhards and Adiscon GmbH.
+ *
+ * This file is part of rsyslog.
+ *
+ * Rsyslog is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation, either version 3 of the License, or
+ * (at your option) any later version.
+ *
+ * Rsyslog 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
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with Rsyslog.  If not, see http://www.gnu.org/licenses/.
+ *
+ * A copy of the GPL can be found in the file "COPYING" in this distribution.
+ */
+package com.rsyslog.lib;
+
+public interface SyslogMsgConsumer {
+	public void consumeMsg(String msg);
+}
diff --git a/java/com/rsyslog/lib/SyslogSender.java b/java/com/rsyslog/lib/SyslogSender.java
new file mode 100644
index 00000000..fc0e3fec
--- /dev/null
+++ b/java/com/rsyslog/lib/SyslogSender.java
@@ -0,0 +1,96 @@
+/**
+ * This class specifies all methods common to syslog senders. It also implements
+ * some generic ways to send data. Actual syslog senders (e.g. UDP, TCP, ...) shall
+ * be derived from this class.
+ *
+ * This is a limited-capability implementation of a syslog message together
+ * with all its properties. It is limit to what is currently needed and may
+ * be extended as further need arises.
+ *
+ * @author Rainer Gerhards
+ * 
+ * Copyright 2009 Rainer Gerhards and Adiscon GmbH.
+ *
+ * This file is part of rsyslog.
+ *
+ * Rsyslog is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation, either version 3 of the License, or
+ * (at your option) any later version.
+ *
+ * Rsyslog 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
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with Rsyslog.  If not, see http://www.gnu.org/licenses/.
+ *
+ * A copy of the GPL can be found in the file "COPYING" in this distribution.
+ */
+package com.rsyslog.lib;
+
+public abstract class SyslogSender {
+
+	/** the rawmsg without the PRI part */
+	private String target;
+
+	/** the rawmsg without the PRI part */
+	private boolean isConnected = false;
+
+	/** Constructs Sender, sets target system.
+	 * @param target the system to connect to. Syntax of target is depending
+	 * 		 on the underlying transport.
+	 */
+	public SyslogSender(String target) {
+		this.target = target;
+	}
+
+
+	/** send a message on the wire.
+	 * This needs a complete formatted message, which will be extended by
+	 * the transport framing, if necessary.
+	 *
+	 * @param MSG a validly formatted syslog message as of the RFC (all parts)
+	 * @throws Exception (depending on transport)
+	 */
+	protected abstract void sendTransport(String MSG) throws Exception;
+
+	/** send an alread-formatted message.
+	 * Sends a preformatted syslog message payload to the target. Connects
+	 * to the target if not already connected.
+	 *
+	 * @param MSG a validly formatted syslog message as of the RFC (all parts)
+	 * @throws Exception (depending on transport)
+	 */
+	public void sendMSG(String MSG) throws Exception {
+		if(!isConnected)
+			connect();
+		sendTransport(MSG);
+	}
+
+	/** connect to the target. 
+	 * Note that this may be a null operation if there is no session-like entity
+	 * in the underlying transport (as is for example in UDP).
+	 */
+	public void connect() throws Exception {
+		/* the default implementation does (almost) nothing */
+		isConnected = true;
+	}
+
+	/** disconnects from the target. 
+	 * Note that this may be a null operation if there is no session-like entity
+	 * in the underlying transport (as is for example in UDP).
+	 */
+	public void disconnect() {
+		/* the default implementation does (almost) nothing */
+		isConnected = false;
+	}
+
+	/** return target of this Sender.
+	 * @returns target as initially set
+	 */
+	public String getTarget() {
+		return target;
+	}
+}
diff --git a/java/com/rsyslog/lib/SyslogServerTCP.java b/java/com/rsyslog/lib/SyslogServerTCP.java
new file mode 100644
index 00000000..d5376a32
--- /dev/null
+++ b/java/com/rsyslog/lib/SyslogServerTCP.java
@@ -0,0 +1,126 @@
+/**
+ * Implementation of a tcp-based syslog server.
+ *
+ * This is a limited-capability implementation of a syslog tcp server.
+ * 
+ * @author Rainer Gerhards
+ * 
+ * Copyright 2009 Rainer Gerhards and Adiscon GmbH.
+ *
+ * This file is part of rsyslog.
+ *
+ * Rsyslog is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation, either version 3 of the License, or
+ * (at your option) any later version.
+ *
+ * Rsyslog 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
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with Rsyslog.  If not, see http://www.gnu.org/licenses/.
+ *
+ * A copy of the GPL can be found in the file "COPYING" in this distribution.
+ */
+package com.rsyslog.lib;
+
+import com.rsyslog.lib.SyslogMsgConsumer;
+import java.io.*;
+import java.net.*;
+
+
+/** a small test consumer */
+/*
+class TestConsumer implements SyslogMsgConsumer {
+	public void consumeMsg(String ln) {
+		System.out.println("Line received '" + ln + "'\n");
+	}
+}
+*/
+
+public class SyslogServerTCP extends Thread {
+	private ServerSocket lstnSock;
+	private boolean contRun;	/* continue processing requests? */
+	public SyslogMsgConsumer consumer;
+
+	/** Process a single connection */
+	class Session extends Thread {
+		private Socket sock;
+		private SyslogServerTCP srvr;
+
+		public Session(Socket so, SyslogServerTCP _srvr) {
+			sock = so;
+			srvr = _srvr;
+		}
+
+		public void run() {
+			try {
+				BufferedReader data = new BufferedReader(
+					new InputStreamReader(sock.getInputStream()));
+
+				String ln = data.readLine();
+				while(ln != null) {
+					srvr.getConsumer().consumeMsg(ln);
+					ln = data.readLine();
+				}
+				System.out.println("End of Session.\n");
+				sock.close();
+			}
+			catch(Exception e) {
+				/* we ignore any errors we may have... */
+				System.out.println("Session exception " + e.toString());
+			}
+		}
+	}
+
+	/** a small test driver */
+/*
+	public static void main(String args[]) {
+		try {
+			SyslogMsgConsumer cons = new TestConsumer();
+			System.out.println("Starting server on port " + args[0] + "\n");
+			SyslogServerTCP myServ = new
+				SyslogServerTCP(Integer.parseInt(args[0]), cons);
+			myServ.start();
+			System.out.println("Press ctl-c to terminate\n");
+		}
+		catch(Exception e) {
+			System.out.println("Fehler! " + e.toString());
+		}
+	}
+*/
+
+	public SyslogServerTCP(int port, SyslogMsgConsumer cons) throws java.io.IOException {
+		if(lstnSock != null)
+			terminate();
+		lstnSock = new ServerSocket(port);
+		consumer = cons;
+		contRun = true;
+	}
+
+	public void terminate() {
+		contRun = false;
+	}
+
+	public SyslogMsgConsumer getConsumer() {
+		return consumer;
+	}
+
+	public void run() {
+		try {
+			while(contRun) {
+				Socket sock = lstnSock.accept();
+				System.out.println("New connection request! " + sock.toString());
+				Thread sess = new Session(sock, this);
+				sock = null;
+				sess.start();
+			}
+		}
+		catch(Exception e) {
+			System.out.println("Error during server run " + e.toString());
+		}
+			
+	}
+}
diff --git a/java/com/rsyslog/lib/SyslogTrafficGenerator.java b/java/com/rsyslog/lib/SyslogTrafficGenerator.java
new file mode 100644
index 00000000..79a99495
--- /dev/null
+++ b/java/com/rsyslog/lib/SyslogTrafficGenerator.java
@@ -0,0 +1,81 @@
+/**
+ * This class is a syslog traffic generator. It is primarily intended to be used
+ * together with testing tools, but may have some use cases outside that domain.
+ *
+ * @author Rainer Gerhards
+ * 
+ * Copyright 2009 Rainer Gerhards and Adiscon GmbH.
+ *
+ * This file is part of rsyslog.
+ *
+ * Rsyslog is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation, either version 3 of the License, or
+ * (at your option) any later version.
+ *
+ * Rsyslog 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
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with Rsyslog.  If not, see http://www.gnu.org/licenses/.
+ *
+ * A copy of the GPL can be found in the file "COPYING" in this distribution.
+ */
+package com.rsyslog.lib;
+
+public class SyslogTrafficGenerator extends Thread {
+
+	/** the target host to receive traffic */
+	private String target;
+
+	/** the message (template) to be sent */
+	private String message;
+
+	/** number of messages to be sent */
+	private long nummsgs;
+
+	/** Constructs Sender, sets target system.
+	 * @param target the system to connect to. Syntax of target is depending
+	 * 		 on the underlying transport.
+	 */
+	public SyslogTrafficGenerator(String target, String message, long nummsgs) {
+		this.target = target;
+		this.message = message;
+		this.nummsgs = nummsgs;
+	}
+
+	/** Generates the traffic. Stops when either called to terminate
+	 * or the max number of messages have been sent. Note that all
+	 * necessary properties must have been set up before starting the
+	 * generator thread!
+	 */
+	private void performTest() throws Exception {
+		int doDisp = 0;
+		UDPSyslogSender sender = new UDPSyslogSender(target);
+		for(long i = 0 ; i < nummsgs ; ++i) {
+			sender.sendMSG(message + " " + Long.toString(i) + " " + this.toString() + "\0");
+			if((doDisp++ % 1000) == 0)
+				System.out.println(this.toString() + " send message " + Long.toString(i));
+			sleep(1);
+		}	
+	}
+
+
+/** Wrapper around the real traffic generator, catches exceptions.
+ */
+	public void run() {
+		System.out.println("traffic generator " + this.toString() + " thread started");
+		try {
+			performTest();
+		}
+		catch(Exception e) {
+			/* at some time, we may find a more intelligent way to
+			 * handle this! ;)
+			 */
+			System.out.println(e.toString());
+		}
+		System.out.println("traffic generator " + this.toString() + " thread finished");
+	}
+}
diff --git a/java/com/rsyslog/lib/UDPSyslogSender.java b/java/com/rsyslog/lib/UDPSyslogSender.java
new file mode 100644
index 00000000..1a2c4726
--- /dev/null
+++ b/java/com/rsyslog/lib/UDPSyslogSender.java
@@ -0,0 +1,75 @@
+/**
+ * A UDP transport implementation of a syslog sender.
+ *
+ * Note that there is an anomaly in this version of the code: we query the remote system
+ * address only once during the connection setup and resue it. If we potentially run for
+ * an extended period of time, the remote address may change, what we do not reflect. For
+ * the current use case, this is acceptable, but if this code is put into more wide-spread
+ * use outside of debugging, a periodic requery should be added.
+ *
+ * @author Rainer Gerhards
+ * 
+ * Copyright 2009 Rainer Gerhards and Adiscon GmbH.
+ *
+ * This file is part of rsyslog.
+ *
+ * Rsyslog is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation, either version 3 of the License, or
+ * (at your option) any later version.
+ *
+ * Rsyslog 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
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with Rsyslog.  If not, see http://www.gnu.org/licenses/.
+ *
+ * A copy of the GPL can be found in the file "COPYING" in this distribution.
+ */
+package com.rsyslog.lib;
+import java.net.*;
+
+public class UDPSyslogSender extends SyslogSender {
+
+	private final int port = 514; // TODO: take from target!
+	private InetAddress targetAddr;
+
+	/** the socket to communicate over with the remote system. */
+	private DatagramSocket sock;
+
+	/** Constructs Sender, sets target system.
+	 * @param target the system to connect to. Syntax of target is depending
+	 * 		 on the underlying transport.
+	 */
+	public UDPSyslogSender(String target) throws Exception {
+		super(target);
+	}
+
+	/** send a message on the wire.
+	 * This needs a complete formatted message, which will be extended by
+	 * the transport framing, if necessary.
+	 *
+	 * @param MSG a validly formatted syslog message as of the RFC (all parts)
+	 * @throws Exception (depending on transport)
+	 */
+	protected void sendTransport(String MSG) throws Exception {
+		byte msg[] = MSG.getBytes();
+		DatagramPacket pkt = new DatagramPacket(msg, msg.length, targetAddr, port);
+		sock.send(pkt);
+	}
+
+
+	/** connect to the target. 
+	 * For UDP, this means we create the socket.
+	 */
+	public void connect() throws Exception {
+		super.connect();
+		sock = new DatagramSocket();
+
+		// TODO: we should extract the actual hostname & port!
+		targetAddr = InetAddress.getByName(getTarget());
+	}
+
+}