com.myftpserver.handler.SendTextFileHandler.java Source code

Introduction

Here is the source code for com.myftpserver.handler.SendTextFileHandler.java

Source

package com.myftpserver.handler;

import java.io.BufferedReader;
import java.io.FileInputStream;
import java.io.InputStreamReader;
import java.nio.ByteBuffer;

import com.util.Utility;
import com.myftpserver.PassiveServer;
import com.myftpserver.interfaces.SendHandler;

import io.netty.buffer.Unpooled;
import io.netty.util.CharsetUtil;
import io.netty.channel.ChannelFuture;
import io.netty.channel.ChannelHandlerContext;

import org.apache.logging.log4j.Logger;

/*
* Copyright 2004-2005 the original author or authors.
*
* Licensed 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.
*/
/**
* 
* @author SITO3
*
*/
public class SendTextFileHandler extends SendHandler {
    private Logger logger;

    boolean isCompleted = false;
    private BufferedReader br;
    private String remoteIp, line;
    private FtpSessionHandler fs;
    private ChannelHandlerContext ctx;
    ByteBuffer buffer = ByteBuffer.allocate(1024);

    /**
     * Send text file handler
     * It send file listing to client and then close the channel.
     * @param fs {@link FtpSessionHandler} FtpSessionHandler object 
     */
    public SendTextFileHandler(FtpSessionHandler fs, ChannelHandlerContext ctx) {
        this.fs = fs;
        this.ctx = ctx;
        this.logger = fs.getLogger();
        this.remoteIp = fs.getClientIp();
    }

    @Override
    public void handlerAdded(ChannelHandlerContext ctx) throws Exception {
        if (fs.isPassiveModeTransfer)
            startToSend(ctx);
    }

    @Override
    public void channelActive(ChannelHandlerContext ctx) throws Exception {
        if (!fs.isPassiveModeTransfer)
            startToSend(ctx);
    }

    @Override
    public void startToSend(ChannelHandlerContext ctx) throws Exception {
        br = new BufferedReader(new InputStreamReader(new FileInputStream(fs.getDownloadFile()), "ISO-8859-1"));
        line = br.readLine();
        ctx.writeAndFlush(Unpooled.copiedBuffer(line + "\r\n", CharsetUtil.ISO_8859_1));
    }

    @Override
    public void channelWritabilityChanged(ChannelHandlerContext ctx) throws Exception {
        if (isCompleted) {
            br.close();
            br = null;
            closeChannel(ctx);
        } else {
            if (br.ready()) {
                if (ctx.channel().isWritable()) {
                    line = br.readLine();
                    if (line == null)
                        isCompleted = true;
                    else
                        ctx.writeAndFlush(Unpooled.copiedBuffer(line + "\r\n", CharsetUtil.ISO_8859_1));
                }
            } else
                isCompleted = true;
        }
    }

    @Override
    public void closeChannel(ChannelHandlerContext ctx) throws Exception {
        if (fs.isPassiveModeTransfer) {
            PassiveServer passiveServer = fs.getPassiveServer();
            if (passiveServer != null) {
                passiveServer.stop();
                passiveServer = null;
                fs.setPassiveServer(passiveServer);
            }
        } else
            ctx.channel().close();
    }

    @Override
    public void operationComplete(ChannelFuture cf) throws Exception {
        String message = fs.getFtpMessage("226_Transfer_Ok");
        Utility.sendMessageToClient(this.ctx.channel(), logger, remoteIp, message);
    }

    @Override
    public void exceptionCaught(ChannelHandlerContext ctx, Throwable cause) throws Exception {
        cause.printStackTrace();
    }

    @Override
    public void handlerRemoved(ChannelHandlerContext ctx) throws Exception {
        // TODO Auto-generated method stub
    }

    @Override
    public void channelRegistered(ChannelHandlerContext ctx) throws Exception {
        // TODO Auto-generated method stub
    }

    @Override
    public void channelUnregistered(ChannelHandlerContext ctx) throws Exception {
        // TODO Auto-generated method stub
    }

    @Override
    public void channelInactive(ChannelHandlerContext ctx) throws Exception {
        // TODO Auto-generated method stub
    }

    @Override
    public void channelRead(ChannelHandlerContext ctx, Object msg) throws Exception {
        // TODO Auto-generated method stub
    }

    @Override
    public void channelReadComplete(ChannelHandlerContext ctx) throws Exception {
        // TODO Auto-generated method stub
    }

    @Override
    public void userEventTriggered(ChannelHandlerContext ctx, Object evt) throws Exception {
        // TODO Auto-generated method stub
    }
}