001 /*
002 * Copyright (C) 2012 eXo Platform SAS.
003 *
004 * This is free software; you can redistribute it and/or modify it
005 * under the terms of the GNU Lesser General Public License as
006 * published by the Free Software Foundation; either version 2.1 of
007 * the License, or (at your option) any later version.
008 *
009 * This software is distributed in the hope that it will be useful,
010 * but WITHOUT ANY WARRANTY; without even the implied warranty of
011 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
012 * Lesser General Public License for more details.
013 *
014 * You should have received a copy of the GNU Lesser General Public
015 * License along with this software; if not, write to the Free
016 * Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
017 * 02110-1301 USA, or see the FSF site: http://www.fsf.org.
018 */
019
020 package org.crsh.util;
021
022 import java.io.Closeable;
023 import java.io.IOException;
024 import java.io.InputStream;
025 import java.io.OutputStream;
026 import java.net.InetSocketAddress;
027 import java.net.ServerSocket;
028 import java.net.Socket;
029
030 public abstract class AbstractSocketServer implements Closeable {
031
032 /** . */
033 private final int bindingPort;
034
035 /** . */
036 private ServerSocket socketServer;
037
038 /** . */
039 private Socket socket;
040
041 /** . */
042 private InputStream in;
043
044 /** . */
045 private OutputStream out;
046
047 /** . */
048 private int port;
049
050 public AbstractSocketServer(int bindingPort) {
051 this.bindingPort = bindingPort;
052 }
053
054 public final int getBindingPort() {
055 return socketServer.getLocalPort();
056 }
057
058 public final int getPort() {
059 return port;
060 }
061
062 public final int bind() throws IOException {
063 ServerSocket socketServer = new ServerSocket();
064 socketServer.bind(new InetSocketAddress(bindingPort));
065 int port = socketServer.getLocalPort();
066
067 //
068 this.socketServer = socketServer;
069 this.port = port;
070
071 //
072 return port;
073 }
074
075 public final void accept() throws IOException {
076 if (socketServer == null) {
077 throw new IllegalStateException();
078 }
079
080 //
081 this.socket = socketServer.accept();
082 this.in = socket.getInputStream();
083 this.out = socket.getOutputStream();
084
085 //
086 handle(in, out);
087 }
088
089 protected abstract void handle(InputStream in, OutputStream out) throws IOException;
090
091 public final void close() {
092 try {
093 Safe.close(socket);
094 Safe.close(in);
095 Safe.close(out);
096 }
097 finally {
098 this.socket = null;
099 this.in = null;
100 this.out = null;
101 }
102 }
103 }