CodeCoverage/JavaServer/flex/src/CodeCoverageClientSocket.as

////////////////////////////////////////////////////////////////////////////////
//
//  Licensed to the Apache Software Foundation (ASF) under one or more
//  contributor license agreements.  See the NOTICE file distributed with
//  this work for additional information regarding copyright ownership.
//  The ASF licenses this file to You 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
{
import flash.events.Event;
import flash.events.IOErrorEvent;
import flash.events.SecurityErrorEvent;
import flash.net.Socket;
import flash.system.Security;

/**
 * Wrapper around a Flash Player Socket.
 */
public class CodeCoverageClientSocket extends Socket
{
    //--------------------------------------------------------------------------
    //
    //  Constructor
    //
    //--------------------------------------------------------------------------
    
    /**
     *  Constructor
	 * 
	 *  @param host The host to connect to.
	 *  @param port The port to send trace data to.
	 *  @param policyFilePort The port used to load the policy file.
     */
    public function CodeCoverageClientSocket(host:String, port:int, policyFilePort:int)
    {
        super();
        addListeners();
		
		var policyFile:String = "xmlsocket://" + host + ":" + policyFilePort;
        Security.loadPolicyFile(policyFile);
        connect(host, port);
    }
    
    //--------------------------------------------------------------------------
    //
    //  Variables
    //
    //--------------------------------------------------------------------------

    /**
     *  Buffer output from trace data received before a connect is avaiable from
     *  the server. 
     */
    private var preConnectBuffer:Array = [];
    private var connectionAvailable:Boolean;
	
    //--------------------------------------------------------------------------
    //
    //  Methods
    //
    //--------------------------------------------------------------------------
    
	/**
	 * Send data to the sever.
	 * 
	 * If the socket is not connected the data will be buffered until the 
	 * connection is made.
	 * 
	 * @param data The data to send.
	 */
    public function sendData(data:String):void 
    {
        if (connectionAvailable)
        {
            writeUTFBytes(data);
	        flush();
        }
        else
        {
            preConnectBuffer.push(data);
        }
    }
    
	private function addListeners():void
	{
		addEventListener(Event.CLOSE, closeHandler);
		addEventListener(Event.CONNECT, connectHandler);
		addEventListener(IOErrorEvent.IO_ERROR, ioErrorHandler);
		addEventListener(SecurityErrorEvent.SECURITY_ERROR, securityErrorHandler);
	}
	
    //--------------------------------------------------------------------------
    //
    //  Event handler
    //
    //--------------------------------------------------------------------------
    
    private function closeHandler(event:Event):void 
    {
        connectionAvailable = false;
    }
    
	/**
	 * Call when the socket connection is established.
	 */
    private function connectHandler(event:Event):void
    {
        removeEventListener(Event.CONNECT, connectHandler);
		
		// Write out buffered data.
        if (preConnectBuffer.length > 0)
        {
            var data:String = preConnectBuffer.join("");
            writeUTFBytes(data);
			flush();
        }
        
        connectionAvailable = true;
        preConnectBuffer = [];
    }
    
    private function ioErrorHandler(event:IOErrorEvent):void
    {
        trace("CodeCoveragePreloadSWF ioErrorHandler: " + event.errorID);
    }
    
    private function securityErrorHandler(event:SecurityErrorEvent):void
    {
        trace("CodeCoveragePreloadSWF securityErrorHandler: " + event.errorID);
    }
    
}    
}