Subversion Repositories Programming Utils

Rev

Details | Last modification | View Log | RSS feed

Rev Author Line No. Line
86 rm5248 1
/*
2
 * Licensed to the Apache Software Foundation (ASF) under one
3
 * or more contributor license agreements.  See the NOTICE file
4
 * distributed with this work for additional information
5
 * regarding copyright ownership.  The ASF licenses this file
6
 * to you under the Apache License, Version 2.0 (the
7
 * "License"); you may not use this file except in compliance
8
 * with the License.  You may obtain a copy of the License at
9
 *
10
 *   http://www.apache.org/licenses/LICENSE-2.0
11
 *
12
 * Unless required by applicable law or agreed to in writing,
13
 * software distributed under the License is distributed on an
14
 * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
15
 * KIND, either express or implied.  See the License for the
16
 * specific language governing permissions and limitations
17
 * under the License.
18
 */
19
package org.apache.sshd.sftp.subsystem;
20
 
21
import org.apache.sshd.common.util.IoUtils;
22
import org.apache.sshd.common.file.SshFile;
23
 
24
import java.io.IOException;
25
import java.io.InputStream;
26
import java.io.OutputStream;
27
 
28
public class FileHandle extends BaseHandle {
29
 
30
    int flags;
31
    OutputStream output;
32
    long outputPos;
33
    InputStream input;
34
    long inputPos;
35
    long length;
36
 
37
    public FileHandle(String id, SshFile sshFile, int flags) {
38
        super(id, sshFile);
39
        this.flags = flags;
40
    }
41
 
42
    public int getFlags() {
43
        return flags;
44
    }
45
 
46
    public int read(byte[] data, long offset) throws IOException {
47
        if (input != null && offset >= length) {
48
            return -1;
49
        }
50
        if (input != null && offset != inputPos) {
51
            IoUtils.closeQuietly(input);
52
            input = null;
53
        }
54
        if (input == null) {
55
            input = getFile().createInputStream(offset);
56
            length = getFile().getSize();
57
            inputPos = offset;
58
        }
59
        if (offset >= length) {
60
            return -1;
61
        }
62
        int read = input.read(data);
63
        inputPos += read;
64
        return read;
65
    }
66
 
67
    public void write(byte[] data, long offset) throws IOException {
68
        if (output != null && offset != outputPos) {
69
            IoUtils.closeQuietly(output);
70
            output = null;
71
        }
72
        if (output == null) {
73
            output = getFile().createOutputStream(offset);
74
        }
75
        output.write(data);
76
        outputPos += data.length;
77
    }
78
 
79
    @Override
80
    public void close() throws IOException {
81
        IoUtils.closeQuietly(output, input);
82
        output = null;
83
        input = null;
84
        super.close();
85
    }
86
}