mblanklineremover.java

来自「httptunnel.jar httptunnel java 源码」· Java 代码 · 共 75 行

package net.jumperz.app.MGuardian.plugin;

import net.jumperz.net.*;
import java.util.*;
import net.jumperz.util.*;
import net.jumperz.io.*;
import java.io.*;

public class MBlankLineRemover
extends MGuardianPlugin
{
//--------------------------------------------------------------------------------
public Map execute( Map sessionInfo )
{
MHttpResponse response = ( MHttpResponse )sessionInfo.get( "response" );
if( response == null )
	{
	Exception e = new Exception( "Configuration Error. You need to call this plugin from 'RESPONSE-TYPE' rule like 'type=responseHeader'." );
	e.printStackTrace();
	return null;
	}

try
	{
	execute2( response );
	}
catch( IOException e )
	{
	e.printStackTrace();
	return null;
	}

return null;
}
// --------------------------------------------------------------------------------
private void execute2( MHttpResponse response )
throws IOException
{
if( !response.hasBody() )
	{
	return;
	}
response.chunkToNormal();
MBuffer buffer = new MBuffer();
MLineReader reader = new MAdvancedLineReader();
reader.setInputStream( response.getBodyInputStream() );

boolean isTop = true;
while( true )
	{
	String line = reader.readLine();
	if( line == null )
		{
		break;
		}
	if( isTop )
		{
		if( line.length() > 0 )
			{
			isTop = false;
			buffer.write( MStringUtil.getBytes( line ) );
			buffer.write( MStringUtil.getBytes( reader.getLastDelimiterString() ) );
			}	
		}
	else
		{
		buffer.write( MStringUtil.getBytes( line ) );
		buffer.write( MStringUtil.getBytes( reader.getLastDelimiterString() ) );		
		}
	}
response.setBody( buffer );
response.setHeaderValue( "Content-Length", Integer.toString( buffer.getSize() ) );
}
//--------------------------------------------------------------------------------
}