| /* |
| * 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 org.apache.ftpserver.command.impl; |
| |
| import java.io.IOException; |
| import java.io.OutputStream; |
| import java.net.InetAddress; |
| import java.net.SocketException; |
| |
| import org.apache.ftpserver.command.AbstractCommand; |
| import org.apache.ftpserver.ftplet.DataConnection; |
| import org.apache.ftpserver.ftplet.DataConnectionFactory; |
| import org.apache.ftpserver.ftplet.DefaultFtpReply; |
| import org.apache.ftpserver.ftplet.FileSystemView; |
| import org.apache.ftpserver.ftplet.FtpException; |
| import org.apache.ftpserver.ftplet.FtpFile; |
| import org.apache.ftpserver.ftplet.FtpReply; |
| import org.apache.ftpserver.ftplet.FtpRequest; |
| import org.apache.ftpserver.impl.FtpIoSession; |
| import org.apache.ftpserver.impl.FtpServerContext; |
| import org.apache.ftpserver.impl.IODataConnectionFactory; |
| import org.apache.ftpserver.impl.LocalizedDataTransferFtpReply; |
| import org.apache.ftpserver.impl.ServerFtpStatistics; |
| import org.apache.ftpserver.util.IoUtils; |
| import org.slf4j.Logger; |
| import org.slf4j.LoggerFactory; |
| |
| /** |
| * <strong>Internal class, do not use directly.</strong> |
| * |
| * <code>STOU <CRLF></code><br> |
| * |
| * This command behaves like STOR except that the resultant file is to be |
| * created in the current directory under a name unique to that directory. The |
| * 150 Transfer Started response must include the name generated, See RFC1123 |
| * section 4.1.2.9 |
| * |
| * @author <a href="http://mina.apache.org">Apache MINA Project</a> |
| */ |
| public class STOU extends AbstractCommand { |
| |
| private final Logger LOG = LoggerFactory.getLogger(STOU.class); |
| |
| /** |
| * Execute command. |
| */ |
| public void execute(final FtpIoSession session, |
| final FtpServerContext context, final FtpRequest request) |
| throws IOException, FtpException { |
| |
| try { |
| // 24-10-2007 - added check if PORT or PASV is issued, see |
| // https://issues.apache.org/jira/browse/FTPSERVER-110 |
| DataConnectionFactory connFactory = session.getDataConnection(); |
| if (connFactory instanceof IODataConnectionFactory) { |
| InetAddress address = ((IODataConnectionFactory) connFactory) |
| .getInetAddress(); |
| if (address == null) { |
| session.write(new DefaultFtpReply( |
| FtpReply.REPLY_503_BAD_SEQUENCE_OF_COMMANDS, |
| "PORT or PASV must be issued first")); |
| return; |
| } |
| } |
| |
| // reset state variables |
| session.resetState(); |
| |
| String pathName = request.getArgument(); |
| |
| // get filenames |
| FtpFile file = null; |
| try { |
| String filePrefix; |
| if (pathName == null) { |
| filePrefix = "ftp.dat"; |
| } else { |
| FtpFile dir = session.getFileSystemView().getFile( |
| pathName); |
| if (dir.isDirectory()) { |
| filePrefix = pathName + "/ftp.dat"; |
| } else { |
| filePrefix = pathName; |
| } |
| } |
| |
| file = session.getFileSystemView().getFile(filePrefix); |
| if (file != null) { |
| file = getUniqueFile(session, file); |
| } |
| } catch (Exception ex) { |
| LOG.debug("Exception getting file object", ex); |
| } |
| |
| if (file == null) { |
| session.write(LocalizedDataTransferFtpReply.translate(session, request, context, |
| FtpReply.REPLY_550_REQUESTED_ACTION_NOT_TAKEN, "STOU", |
| null, null)); |
| return; |
| } |
| String fileName = file.getAbsolutePath(); |
| |
| // check permission |
| if (!file.isWritable()) { |
| session.write(LocalizedDataTransferFtpReply.translate(session, request, context, |
| FtpReply.REPLY_550_REQUESTED_ACTION_NOT_TAKEN, |
| "STOU.permission", fileName, file)); |
| return; |
| } |
| |
| // get data connection |
| session.write(new DefaultFtpReply( |
| FtpReply.REPLY_150_FILE_STATUS_OKAY, "FILE: " + fileName)); |
| |
| // get data from client |
| boolean failure = false; |
| OutputStream os = null; |
| |
| DataConnection dataConnection; |
| try { |
| dataConnection = session.getDataConnection().openConnection(); |
| } catch (Exception e) { |
| LOG.debug("Exception getting the input data stream", e); |
| session.write(LocalizedDataTransferFtpReply.translate(session, request, context, |
| FtpReply.REPLY_425_CANT_OPEN_DATA_CONNECTION, "STOU", |
| fileName, file)); |
| return; |
| } |
| |
| long transSz = 0L; |
| try { |
| |
| // open streams |
| os = file.createOutputStream(0L); |
| |
| // transfer data |
| transSz = dataConnection.transferFromClient(session.getFtpletSession(), os); |
| |
| // attempt to close the output stream so that errors in |
| // closing it will return an error to the client (FTPSERVER-119) |
| if(os != null) { |
| os.close(); |
| } |
| |
| LOG.info("File uploaded {}", fileName); |
| |
| // notify the statistics component |
| ServerFtpStatistics ftpStat = (ServerFtpStatistics) context |
| .getFtpStatistics(); |
| if (ftpStat != null) { |
| ftpStat.setUpload(session, file, transSz); |
| } |
| |
| } catch (SocketException ex) { |
| LOG.debug("Socket exception during data transfer", ex); |
| failure = true; |
| session.write(LocalizedDataTransferFtpReply.translate(session, request, context, |
| FtpReply.REPLY_426_CONNECTION_CLOSED_TRANSFER_ABORTED, |
| "STOU", fileName, file)); |
| } catch (IOException ex) { |
| LOG.debug("IOException during data transfer", ex); |
| failure = true; |
| session |
| .write(LocalizedDataTransferFtpReply |
| .translate( |
| session, |
| request, |
| context, |
| FtpReply.REPLY_551_REQUESTED_ACTION_ABORTED_PAGE_TYPE_UNKNOWN, |
| "STOU", fileName, file)); |
| } finally { |
| // make sure we really close the output stream |
| IoUtils.close(os); |
| } |
| |
| // if data transfer ok - send transfer complete message |
| if (!failure) { |
| session.write(LocalizedDataTransferFtpReply.translate(session, request, context, |
| FtpReply.REPLY_226_CLOSING_DATA_CONNECTION, "STOU", |
| fileName, file, transSz)); |
| |
| } |
| } finally { |
| session.getDataConnection().closeDataConnection(); |
| } |
| |
| } |
| |
| /** |
| * Get unique file object. |
| */ |
| //TODO may need synchronization |
| protected FtpFile getUniqueFile(FtpIoSession session, FtpFile oldFile) |
| throws FtpException { |
| FtpFile newFile = oldFile; |
| FileSystemView fsView = session.getFileSystemView(); |
| String fileName = newFile.getAbsolutePath(); |
| while (newFile.doesExist()) { |
| newFile = fsView.getFile(fileName + '.' |
| + System.currentTimeMillis()); |
| if (newFile == null) { |
| break; |
| } |
| } |
| return newFile; |
| } |
| |
| } |