g3/lib/g3-dpi/tests/ssh.rs
2023-03-09 17:55:45 +08:00

30 lines
1 KiB
Rust

/*
* Copyright 2023 ByteDance and/or its affiliates.
*
* 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.
*/
use g3_dpi::{Protocol, ProtocolInspectionConfig, ProtocolInspector};
#[test]
fn port22_openssh() {
let mut inspector = ProtocolInspector::default();
let config = ProtocolInspectionConfig::default();
const DATA: &[u8] = b"SSH-2.0-OpenSSH_7.9p1 Debian-10+deb10u2\r\n";
let protocol = inspector
.check_server_initial_data(&config, 22, DATA)
.unwrap();
assert_eq!(protocol, Protocol::Ssh);
}