root/trunk/LogicMail/src/org/logicprobe/LogicMail/util/Connection.java

/*-
 * Copyright (c) 2006, Derek Konigsberg
 * All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 *
 * 1. Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 * 2. Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in the
 *    documentation and/or other materials provided with the distribution.
 * 3. Neither the name of the project nor the names of its
 *    contributors may be used to endorse or promote products derived
 *    from this software without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
 * FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE
 * COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING,
 * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
 * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT,
 * STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED
 * OF THE POSSIBILITY OF SUCH DAMAGE.
 */

/******************************************************************************
 * This class has been modified to integrate J2ME socket connection code,
 * since LogicMail has no need for a dual-use implementation.  The bulk of
 * this code is a combination of Connection.java and the J2ME ConnectionImpl.java
 * from Mail4ME, and is covered under the copyright shown below.  Any code not
 * originating from Mail4ME is covered under the copyright shown at the top of
 * this source file.
 *
 * Mail4ME - Mail for the Java 2 Micro Edition
 *
 * A lightweight, J2ME- (and also J2SE-) compatible package for sending and
 * receiving Internet mail messages using the SMTP and POP3 protocols.
 *
 * Copyright (c) 2000-2002 Jorg Pleumann <joerg@pleumann.de>
 *
 * Mail4ME is part of the EnhydraME family of projects. See the following web
 * sites for more information:
 *
 * -> http://mail4me.enhydra.org
 * -> http://me.enhydra.org
 *
 * Mail4ME is distributed under the Enhydra Public License (EPL), which is
 * discussed in great detail here:
 *
 * -> http://www.enhydra.org/software/license/index.html
 *
 * Have fun!
 ******************************************************************************/
package org.logicprobe.LogicMail.util;

import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.util.Vector;
import javax.microedition.io.SocketConnection;
import javax.microedition.io.StreamConnection;
import javax.microedition.io.Connector;
import net.rim.device.api.system.EventLogger;
import net.rim.device.api.ui.UiApplication;
import net.rim.device.api.ui.component.Dialog;
import org.logicprobe.LogicMail.AppInfo;
import org.logicprobe.LogicMail.conf.GlobalConfig;
import org.logicprobe.LogicMail.conf.MailSettings;

/**
 * Is the abstract base class for socket connections used inside the SMTP,
 * POP3 and IMAP protocols of the mail package. The rationale for using this class
 * is the difference in how networking is handled in J2ME and J2SE: While J2SE
 * has java.net.Socket, J2ME uses the Generic Connection Framework with its
 * javax.io.Connector class as a central means to open sockets or HTTP
 * connection. Unfortunately, both methods are totally incompatible with each
 * other, so a common abstraction has to be found to make the mail package
 * work in both environments.
 * <p>
 * The Connection class is such an abstraction. It provides abstract
 * versions of methods that open and close sockets, read from them and write
 * to them. It also provides a static factory method that is able to instantiate
 * one of several sub-classes of Connection that match the different run-time
 * environments. These sub-classes are all called ConnectionImpl, and they are
 * loaded "by name" from the j2me, j2se, or http packages, respectively. This
 * is the only way to get rid of compile-time dependencies on these classes.
 */
public class Connection {
    private String serverName;
    private int serverPort;
    private boolean useSSL;
    private boolean deviceSide;
    private StreamConnection socket;
    private String localAddress;
    private GlobalConfig globalConfig;
    protected InputStream input;
    protected OutputStream output;
    private boolean useWiFi;
    private int fakeAvailable = -1;
    
    /**
     * Byte array holding carriage return and line feed
     */
    private static final byte[] CRLF = new byte[] {13, 10};
    
    /**
     * Provides a buffer used for incoming data.
     */
    private byte[] buffer = new byte[128];
    
    /**
     * Holds a list of open connections
     */
    private static Vector openConnections = new Vector();
    
    public Connection(String serverName, int serverPort, boolean useSSL, boolean deviceSide) {
        this.serverName = serverName;
        this.serverPort = serverPort;
        this.useSSL = useSSL;
        this.deviceSide = deviceSide;
        this.input = null;
        this.output = null;
        this.socket = null;
        this.globalConfig = MailSettings.getInstance().getGlobalConfig();
    }
    
    /**
     * Opens a connection.
     */
    public void open() throws IOException {
        if(input != null || output != null || socket != null) {
            close();
        }

        synchronized(openConnections) {
            if(!openConnections.contains(this)) {
                openConnections.addElement(this);
            }
        }
        
        String protocolStr = (useSSL ? "ssl" : "socket");
        // This parameter, which allows bypassing the MDS proxy, should probably
        // be a global user configurable option
        String paramStr = (deviceSide ? ";deviceside=true" : "");

        useWiFi = false;
        if(globalConfig.getWifiMode() == GlobalConfig.WIFI_PROMPT) {
            UiApplication.getUiApplication().invokeAndWait(new Runnable() {
                public void run() {
                    useWiFi = (Dialog.ask(Dialog.D_YES_NO, "Connect through WiFi?") == Dialog.YES);
                }
            });
        }
        else if(globalConfig.getWifiMode() == GlobalConfig.WIFI_ALWAYS) {
            useWiFi = true;
        }
        
        if(useWiFi) {
            paramStr = paramStr + ";ConnectionUID=S TCP-WiFi;ConnectionSetup=delayed;retrynocontext=true";
        }
        
        String connectStr = protocolStr + "://" + serverName + ":" + serverPort + paramStr;
        
        if(EventLogger.getMinimumLevel() >= EventLogger.INFORMATION) {
            String msg = "Opening connection:\r\n"+connectStr+"\r\n";
            EventLogger.logEvent(AppInfo.GUID, msg.getBytes(), EventLogger.INFORMATION);
        }
        
        socket = (StreamConnection)Connector.open(connectStr, Connector.READ_WRITE, true);
        input = socket.openDataInputStream();
        output = socket.openDataOutputStream();
        localAddress = ((SocketConnection)socket).getLocalAddress();

        if(EventLogger.getMinimumLevel() >= EventLogger.INFORMATION) {
            String msg =
                "Connection established:\r\n"+
                "Socket: "+socket.getClass().toString()+"\r\n"+
                "Local address: "+localAddress+"\r\n";
            EventLogger.logEvent(AppInfo.GUID, msg.getBytes(), EventLogger.INFORMATION);
        }
    }
    
    /**
     * Closes a connection.
     */
    public void close() throws IOException {
        try {
            if(input != null) {
                input.close();
                input = null;
            }
        } catch (Exception exp) {
            input = null;
        }
        try {
            if(output != null) {
                output.close();
                output = null;
            }
        } catch (Exception exp) {
            output = null;
        }
        try {
            if(socket != null) {
                socket.close();
                socket = null;
            }
        } catch (Exception exp) {
            socket = null;
        }

        synchronized(openConnections) {
            if(openConnections.contains(this)) {
                openConnections.removeElement(this);
            }
        }
        
        EventLogger.logEvent(AppInfo.GUID, "Connection closed".getBytes(), EventLogger.INFORMATION);
    }
    
    /**
     * Determine whether open connections exist
     *
     * @return True if there are open connections
     */
    public static boolean hasOpenConnections() {
        boolean result;
        synchronized(openConnections) {
            result = !openConnections.isEmpty();
        }
        return result;
    }
    
    /**
     * Close all open connections
     */
    public static void closeAllConnections() {
        synchronized(openConnections) {
            int size = openConnections.size();
            for(int i = 0; i < size; i++) {
                try {
                    ((Connection)openConnections.elementAt(i)).close();
                } catch (IOException e) { }
            }
            openConnections.removeAllElements();
        }
    }
    
    /**
     * Determine whether we are currently connected
     * @return True if connected
     */
    public boolean isConnected() {
        if(socket != null)
            return true;
        else
            return false;
    }
    
    /**
     * Get the local address to which this connection is bound
     * @return Local address
     */
    public String getLocalAddress() {
        return localAddress;
    }
    
    /**
     * Get the server name used when this connection was created
     * @return Server name
     */
    public String getServerName() {
        return serverName;
    }
    
    /**
     * Sends a string to the server. This method is used internally for
     * all outgoing communication to the server. The main thing it does
     * it terminate the line with a CR/LF. If there are occurrences of CR or
     * LF inside the string, the method ensures that proper CR/LF sequences
     * are sent for them, since this is what most Internet protocols expect.
     *
     * @see #receive
     */
    public void send(String s) throws IOException {
        byte[] bytes = s.getBytes();
        int length = bytes.length;
        
        /**
         * Special case for empty strings: Only CR/LF is sent.
         */
        if (s.length() == 0) {
            if(globalConfig.getConnDebug()) {
                EventLogger.logEvent(AppInfo.GUID, "[SEND]".getBytes(), EventLogger.DEBUG_INFO);
            }
            
            output.write(CRLF, 0, 2);
        }
        /**
         * The usual case goes here.
         */
        else {
            int i = 0;
            while (i < length) {
                int j = i;
                
                /**
                 * Find next occurrence of a line separator or the end of the
                 * string.
                 */
                while ((j < length) && (bytes[j] != 0x0A) && (bytes[j] != 0x0D)) {
                    j++;
                }
                
                if(globalConfig.getConnDebug()) {
                    EventLogger.logEvent(AppInfo.GUID, ("[SEND] " + s.substring(i, j)).getBytes(), EventLogger.DEBUG_INFO);
                }
                
                /**
                 * Write the string up to there and terminate it properly.
                 */
                output.write((s.substring(i, j)+"\r\n").getBytes());
                
                /**
                 * If we stopped at a CR, ignore a possibly following LF.
                 */
                if ((j < length - 1) && (bytes[j] == 0x0D) && (bytes[j + 1] == 0x0A)) {
                    j++;
                }
                
                i = j + 1;
            }
        }
        output.flush();
    }
    
    /**
     * Sends a string to the server, terminating it with a CRLF.
     * No cleanup is performed, as it is expected that the string
     * is a prepared protocol command.
     */
    public void sendCommand(String s) throws IOException {
        if(globalConfig.getConnDebug()) {
            EventLogger.logEvent(AppInfo.GUID, ("[SEND CMD] " + s).getBytes(), EventLogger.DEBUG_INFO);
        }

        if(s == null) {
            output.write(CRLF, 0, 2);
        }
        else {
            output.write((s+"\r\n").getBytes());
        }
        output.flush();
    }
    
    /**
     * Sends a string to the server. This method is used to bypass all
     * the processing done by the normal send method, and is most useful
     * for bulk transmissions.  It writes the provided string to the socket
     * in a single command, followed by a flush.
     *
     * @see #send
     */
    public void sendRaw(String s) throws IOException {
        byte[] bytes = s.getBytes();
        
        if(globalConfig.getConnDebug()) {
            EventLogger.logEvent(AppInfo.GUID, ("[SEND RAW]\r\n" + s).getBytes(), EventLogger.DEBUG_INFO);
        }
        
        output.write(bytes, 0, bytes.length);
        
        output.flush();
    }

    /**
     * Returns the number of bytes available for reading.
     * Used to poll the connection without blocking.
     * 
     * @see InputStream#available()
     */
    public int available() throws IOException {
        if(fakeAvailable == -1) {
                return input.available();
        }
        else {
                return fakeAvailable;
        }
    }
    
    /**
     * Receives a string from the server. This method is used internally for
     * incoming communication from the server. The main thing it does is
     * ensuring that only complete lines are returned to the application, that is,
     * lines that were terminated at least by a CR. LFs are ignored completely.
     * Neither CRs nor LFs are returned as part of the result.
     *
     * @see #send
     */
    public String receive() throws IOException {
        /**
         * A note on how this method works and why it is designed the
         * way it is: A previous implementation tried to read multiple
         * bytes from the InputStream, searched a possible line terminator
         * in them, and appended characters to a temporary result string.
         * That didn't work very well, because the code had to deal
         * with a large number of temporary strings that grew as more
         * characters belonging to a line were read. The result was
         * a high level of heap fragmentation, which made applications
         * unstable on MIDP devices (that don't provide compacting GC,
         * that is).
         *
         * This new implementation seems to work better: Is reads
         * characters from the InputStream one-by-one, putting them
         * in a buffer of fixed size that is large enough to hold a
         * "normal" line belonging to an e-mail. When a terminator
         * is found, a new String object is created from that buffer.
         * Since the buffer is of fixed size and used during the whole
         * lifetime of the Connection object, and the String is the
         * actual String that makes it into the final Message object,
         * there are no temporary Strings needed -- at least not in the
         * common case: If a mail server tends to send whole messages
         * without line terminators (and thus disregards for example
         * the POP3 specification that dictates a maximum response of
         * 512 characters), temporary Strings will again be necessary,
         * but there's not much one can about that.
         */
        StringBuffer resultBuffer = new StringBuffer();
        boolean stop = false;
        int actualAvailable = input.available();
        int readBytes = 0;
        int count;
        
        /**
         * The "stop" flag will be set as soon as we have received
         * a complete line, that is, a line terminated by CR/LF. Until
         * this is the case, the following block is repeated.
         */
        while (!stop) {
            count = 0;
            
            /**
             * The inner block reads single bytes from the InputStream
             * until, again, a line terminator is read or the buffer is
             * full.
             */
            while (true) {
                int actual = input.read(buffer, count, 1);
                
                /**
                 * If -1 is returned, the InputStream is already closed,
                 * probably because the connection is broken, or the server
                 * didn't like is. Try to close the connection, but ignore
                 * any errors that might result from this.
                 */
                if (actual == -1) {
                    EventLogger.logEvent(AppInfo.GUID, "Unable to read from socket, closing connection".getBytes(), EventLogger.INFORMATION);
                    try {
                        close();
                    } catch (IOException e) { }
                    
                    throw new IOException("Connection closed");
                }
                
                /**
                 * If no bytes have been received, we wait a little
                 * while (by yielding processing time to other threads).
                 */
                else if (actual == 0) {
                    try {
                        Thread.yield();
                    } catch (Exception e) { }
                }
                
                /**
                 * If a byte has been read, examine it and put it in the
                 * buffer.
                 */
                // Note: We really should look for CRLF, and not use this
                // approach which screws up on mid-line LFs. (DK)
                else {
                    byte b = buffer[count];
                    readBytes++;
                    
                    /**
                     * Ignore all CRs.
                     */
                    if (b == 0x0D) {
                        /* Ignore CRs */
                    }
                    /**
                     * Take LF as line separator.
                     */
                    else if (b == 0x0A) {
                        stop = true;
                        break;
                    }
                    /**
                     * Everything else makes it into the buffer.
                     */
                    else {
                        count++;
                        if (count == buffer.length) {
                            break;
                        }
                    }
                }
            }
            resultBuffer.append(new String(buffer, 0, count));
        }
        
        if(globalConfig.getConnDebug()) {
            EventLogger.logEvent(AppInfo.GUID, ("[RECV] " + resultBuffer.toString()).getBytes(), EventLogger.DEBUG_INFO);
        }
        if(actualAvailable > readBytes) {
                fakeAvailable = actualAvailable - readBytes;
        }
        else {
                fakeAvailable = -1;
        }
        
        return resultBuffer.toString();
    }
}