1/*
2 * Copyright (C) 2012 The Android Open Source Project
3 *
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at
7 *
8 *      http://www.apache.org/licenses/LICENSE-2.0
9 *
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
15 */
16
17package org.conscrypt;
18
19import java.io.FilterInputStream;
20import java.io.IOException;
21import java.io.InputStream;
22
23/**
24 * Provides an interface to OpenSSL's BIO system directly from a Java
25 * InputStream. It allows an OpenSSL API to read directly from something more
26 * flexible interface than a byte array.
27 */
28public class OpenSSLBIOInputStream extends FilterInputStream {
29    private long ctx;
30
31    public OpenSSLBIOInputStream(InputStream is) {
32        super(is);
33
34        ctx = NativeCrypto.create_BIO_InputStream(this);
35    }
36
37    public long getBioContext() {
38        return ctx;
39    }
40
41    public void release() {
42        NativeCrypto.BIO_free_all(ctx);
43    }
44
45    /**
46     * Similar to a {@code readLine} method, but matches what OpenSSL expects
47     * from a {@code BIO_gets} method.
48     */
49    public int gets(byte[] buffer) throws IOException {
50        if (buffer == null || buffer.length == 0) {
51            return 0;
52        }
53
54        int offset = 0;
55        int inputByte = 0;
56        while (offset < buffer.length) {
57            inputByte = read();
58            if (inputByte == -1) {
59                // EOF
60                break;
61            }
62            if (inputByte == '\n') {
63                if (offset == 0) {
64                    // If we haven't read anything yet, ignore CRLF.
65                    continue;
66                } else {
67                    break;
68                }
69            }
70
71            buffer[offset++] = (byte) inputByte;
72        }
73
74        return offset;
75    }
76}
77