1 // Copyright (c) 2023 Huawei Device Co., Ltd. 2 // Licensed under the Apache License, Version 2.0 (the "License"); 3 // you may not use this file except in compliance with the License. 4 // You may obtain a copy of the License at 5 // 6 // http://www.apache.org/licenses/LICENSE-2.0 7 // 8 // Unless required by applicable law or agreed to in writing, software 9 // distributed under the License is distributed on an "AS IS" BASIS, 10 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 11 // See the License for the specific language governing permissions and 12 // limitations under the License. 13 14 use std::io::{Read, Write}; 15 16 #[cfg(feature = "__c_openssl")] 17 use crate::util::c_openssl::ssl::SslStream; 18 19 /// A stream which may be wrapped with TLS. 20 pub enum MixStream<T> { 21 /// A raw HTTP stream. 22 Http(T), 23 /// An SSL-wrapped HTTP stream. 24 Https(SslStream<T>), 25 } 26 27 impl<T> Read for MixStream<T> 28 where 29 T: Read + Write, 30 { read(&mut self, buf: &mut [u8]) -> std::io::Result<usize>31 fn read(&mut self, buf: &mut [u8]) -> std::io::Result<usize> { 32 match &mut *self { 33 MixStream::Http(s) => s.read(buf), 34 MixStream::Https(s) => s.read(buf), 35 } 36 } 37 } 38 impl<T> Write for MixStream<T> 39 where 40 T: Read + Write, 41 { write(&mut self, buf: &[u8]) -> std::io::Result<usize>42 fn write(&mut self, buf: &[u8]) -> std::io::Result<usize> { 43 match &mut *self { 44 MixStream::Http(s) => s.write(buf), 45 MixStream::Https(s) => s.write(buf), 46 } 47 } 48 flush(&mut self) -> std::io::Result<()>49 fn flush(&mut self) -> std::io::Result<()> { 50 match &mut *self { 51 MixStream::Http(s) => s.flush(), 52 MixStream::Https(s) => s.flush(), 53 } 54 } 55 } 56