summaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
-rw-r--r--.appveyor.yml33
-rw-r--r--.dockerignore3
-rw-r--r--.github/workflows/tox.yaml24
-rw-r--r--.gitignore27
-rw-r--r--.travis.yml25
-rw-r--r--CONTRIBUTING.md26
-rw-r--r--Dockerfile10
-rw-r--r--LICENSE23
-rw-r--r--Makefile.docker13
-rw-r--r--Makefile.pypi15
-rw-r--r--Makefile.snap6
-rw-r--r--PACKAGING.md97
-rw-r--r--README.md321
-rwxr-xr-xbuild_windows_executable.sh130
-rwxr-xr-xdocker_test.sh716
-rw-r--r--pyproject.toml7
-rw-r--r--setup.cfg41
-rw-r--r--setup.py23
-rw-r--r--snapcraft.yaml21
-rw-r--r--src/ssh_audit/__init__.py0
-rw-r--r--src/ssh_audit/__main__.py16
-rw-r--r--src/ssh_audit/algorithm.py61
-rw-r--r--src/ssh_audit/algorithms.py223
-rw-r--r--src/ssh_audit/auditconf.py98
-rw-r--r--src/ssh_audit/banner.py92
-rw-r--r--src/ssh_audit/exitcodes.py6
-rw-r--r--src/ssh_audit/fingerprint.py43
-rw-r--r--src/ssh_audit/gextest.py162
-rw-r--r--src/ssh_audit/globals.py27
-rw-r--r--src/ssh_audit/hostkeytest.py188
-rw-r--r--src/ssh_audit/kexdh.py365
-rw-r--r--src/ssh_audit/outputbuffer.py185
-rw-r--r--src/ssh_audit/policy.py521
-rw-r--r--src/ssh_audit/product.py31
-rw-r--r--src/ssh_audit/protocol.py36
-rw-r--r--src/ssh_audit/readbuf.py92
-rw-r--r--src/ssh_audit/software.py227
-rw-r--r--src/ssh_audit/ssh1.py40
-rw-r--r--src/ssh_audit/ssh1_crc32.py47
-rw-r--r--src/ssh_audit/ssh1_kexdb.py58
-rw-r--r--src/ssh_audit/ssh1_publickeymessage.py144
-rw-r--r--src/ssh_audit/ssh2_kex.py136
-rw-r--r--src/ssh_audit/ssh2_kexdb.py280
-rw-r--r--src/ssh_audit/ssh2_kexparty.py50
-rwxr-xr-xsrc/ssh_audit/ssh_audit.py1132
-rw-r--r--src/ssh_audit/ssh_socket.py343
-rw-r--r--src/ssh_audit/timeframe.py77
-rw-r--r--src/ssh_audit/utils.py165
-rw-r--r--src/ssh_audit/versionvulnerabilitydb.py151
-rw-r--r--src/ssh_audit/writebuf.py108
-rw-r--r--ssh-audit.1245
-rwxr-xr-xssh-audit.py21
-rw-r--r--test/conftest.py145
-rw-r--r--test/docker/.ed25519.sk1
-rw-r--r--test/docker/Dockerfile32
-rwxr-xr-xtest/docker/debug.sh9
-rw-r--r--test/docker/dropbear_dss_host_keybin0 -> 458 bytes
-rw-r--r--test/docker/dropbear_ecdsa_host_keybin0 -> 141 bytes
-rw-r--r--test/docker/dropbear_rsa_host_key_1024bin0 -> 421 bytes
-rw-r--r--test/docker/dropbear_rsa_host_key_3072bin0 -> 1189 bytes
-rw-r--r--test/docker/ed25519.pk1
-rw-r--r--test/docker/expected_results/dropbear_2019.78_test1.json1
-rw-r--r--test/docker/expected_results/dropbear_2019.78_test1.txt88
-rw-r--r--test/docker/expected_results/openssh_4.0p1_test1.json1
-rw-r--r--test/docker/expected_results/openssh_4.0p1_test1.txt145
-rw-r--r--test/docker/expected_results/openssh_5.6p1_custom_policy_test1.json1
-rw-r--r--test/docker/expected_results/openssh_5.6p1_custom_policy_test1.txt3
-rw-r--r--test/docker/expected_results/openssh_5.6p1_custom_policy_test10.json1
-rw-r--r--test/docker/expected_results/openssh_5.6p1_custom_policy_test10.txt13
-rw-r--r--test/docker/expected_results/openssh_5.6p1_custom_policy_test2.json1
-rw-r--r--test/docker/expected_results/openssh_5.6p1_custom_policy_test2.txt9
-rw-r--r--test/docker/expected_results/openssh_5.6p1_custom_policy_test3.json1
-rw-r--r--test/docker/expected_results/openssh_5.6p1_custom_policy_test3.txt9
-rw-r--r--test/docker/expected_results/openssh_5.6p1_custom_policy_test4.json1
-rw-r--r--test/docker/expected_results/openssh_5.6p1_custom_policy_test4.txt9
-rw-r--r--test/docker/expected_results/openssh_5.6p1_custom_policy_test5.json1
-rw-r--r--test/docker/expected_results/openssh_5.6p1_custom_policy_test5.txt9
-rw-r--r--test/docker/expected_results/openssh_5.6p1_custom_policy_test7.json1
-rw-r--r--test/docker/expected_results/openssh_5.6p1_custom_policy_test7.txt3
-rw-r--r--test/docker/expected_results/openssh_5.6p1_custom_policy_test8.json1
-rw-r--r--test/docker/expected_results/openssh_5.6p1_custom_policy_test8.txt9
-rw-r--r--test/docker/expected_results/openssh_5.6p1_custom_policy_test9.json1
-rw-r--r--test/docker/expected_results/openssh_5.6p1_custom_policy_test9.txt9
-rw-r--r--test/docker/expected_results/openssh_5.6p1_test1.json1
-rw-r--r--test/docker/expected_results/openssh_5.6p1_test1.txt153
-rw-r--r--test/docker/expected_results/openssh_5.6p1_test2.json1
-rw-r--r--test/docker/expected_results/openssh_5.6p1_test2.txt153
-rw-r--r--test/docker/expected_results/openssh_5.6p1_test3.json1
-rw-r--r--test/docker/expected_results/openssh_5.6p1_test3.txt153
-rw-r--r--test/docker/expected_results/openssh_5.6p1_test4.json1
-rw-r--r--test/docker/expected_results/openssh_5.6p1_test4.txt152
-rw-r--r--test/docker/expected_results/openssh_5.6p1_test5.json1
-rw-r--r--test/docker/expected_results/openssh_5.6p1_test5.txt151
-rw-r--r--test/docker/expected_results/openssh_8.0p1_builtin_policy_test1.json1
-rw-r--r--test/docker/expected_results/openssh_8.0p1_builtin_policy_test1.txt3
-rw-r--r--test/docker/expected_results/openssh_8.0p1_builtin_policy_test2.json1
-rw-r--r--test/docker/expected_results/openssh_8.0p1_builtin_policy_test2.txt9
-rw-r--r--test/docker/expected_results/openssh_8.0p1_custom_policy_test11.json1
-rw-r--r--test/docker/expected_results/openssh_8.0p1_custom_policy_test11.txt3
-rw-r--r--test/docker/expected_results/openssh_8.0p1_custom_policy_test12.json1
-rw-r--r--test/docker/expected_results/openssh_8.0p1_custom_policy_test12.txt17
-rw-r--r--test/docker/expected_results/openssh_8.0p1_custom_policy_test13.json1
-rw-r--r--test/docker/expected_results/openssh_8.0p1_custom_policy_test13.txt3
-rw-r--r--test/docker/expected_results/openssh_8.0p1_custom_policy_test14.json1
-rw-r--r--test/docker/expected_results/openssh_8.0p1_custom_policy_test14.txt9
-rw-r--r--test/docker/expected_results/openssh_8.0p1_custom_policy_test6.json1
-rw-r--r--test/docker/expected_results/openssh_8.0p1_custom_policy_test6.txt3
-rw-r--r--test/docker/expected_results/openssh_8.0p1_test1.json1
-rw-r--r--test/docker/expected_results/openssh_8.0p1_test1.txt85
-rw-r--r--test/docker/expected_results/openssh_8.0p1_test2.json1
-rw-r--r--test/docker/expected_results/openssh_8.0p1_test2.txt77
-rw-r--r--test/docker/expected_results/openssh_8.0p1_test3.json1
-rw-r--r--test/docker/expected_results/openssh_8.0p1_test3.txt38
-rw-r--r--test/docker/expected_results/tinyssh_20190101_test1.json1
-rw-r--r--test/docker/expected_results/tinyssh_20190101_test1.txt25
-rw-r--r--test/docker/host_ca_ed255197
-rw-r--r--test/docker/host_ca_ed25519.pub1
-rw-r--r--test/docker/host_ca_rsa_102415
-rw-r--r--test/docker/host_ca_rsa_1024.pub1
-rw-r--r--test/docker/host_ca_rsa_307239
-rw-r--r--test/docker/host_ca_rsa_3072.pub1
-rw-r--r--test/docker/moduli_102444
-rw-r--r--test/docker/policies/policy_test1.txt10
-rw-r--r--test/docker/policies/policy_test10.txt39
-rw-r--r--test/docker/policies/policy_test11.txt35
-rw-r--r--test/docker/policies/policy_test12.txt35
-rw-r--r--test/docker/policies/policy_test13.txt38
-rw-r--r--test/docker/policies/policy_test14.txt38
-rw-r--r--test/docker/policies/policy_test2.txt10
-rw-r--r--test/docker/policies/policy_test3.txt10
-rw-r--r--test/docker/policies/policy_test4.txt10
-rw-r--r--test/docker/policies/policy_test5.txt10
-rw-r--r--test/docker/policies/policy_test6.txt12
-rw-r--r--test/docker/policies/policy_test7.txt39
-rw-r--r--test/docker/policies/policy_test8.txt39
-rw-r--r--test/docker/policies/policy_test9.txt39
-rw-r--r--test/docker/ssh1_host_keybin0 -> 536 bytes
-rw-r--r--test/docker/ssh1_host_key.pub1
-rw-r--r--test/docker/ssh_host_dsa_key12
-rw-r--r--test/docker/ssh_host_dsa_key.pub1
-rw-r--r--test/docker/ssh_host_ecdsa_key5
-rw-r--r--test/docker/ssh_host_ecdsa_key.pub1
-rw-r--r--test/docker/ssh_host_ed25519_key7
-rw-r--r--test/docker/ssh_host_ed25519_key-cert.pub1
-rw-r--r--test/docker/ssh_host_ed25519_key.pub1
-rw-r--r--test/docker/ssh_host_rsa_key_102415
-rw-r--r--test/docker/ssh_host_rsa_key_1024-cert_1024.pub1
-rw-r--r--test/docker/ssh_host_rsa_key_1024-cert_3072.pub1
-rw-r--r--test/docker/ssh_host_rsa_key_1024.pub1
-rw-r--r--test/docker/ssh_host_rsa_key_307239
-rw-r--r--test/docker/ssh_host_rsa_key_3072-cert_1024.pub1
-rw-r--r--test/docker/ssh_host_rsa_key_3072-cert_3072.pub1
-rw-r--r--test/docker/ssh_host_rsa_key_3072.pub1
-rw-r--r--test/stubs/colorama.pyi4
-rw-r--r--test/test_algorithm.py166
-rw-r--r--test/test_auditconf.py178
-rw-r--r--test/test_banner.py69
-rw-r--r--test/test_buffer.py145
-rw-r--r--test/test_build_struct.py43
-rw-r--r--test/test_errors.py172
-rw-r--r--test/test_outputbuffer.py183
-rw-r--r--test/test_policy.py358
-rw-r--r--test/test_resolve.py81
-rw-r--r--test/test_socket.py42
-rw-r--r--test/test_software.py289
-rw-r--r--test/test_ssh1.py173
-rw-r--r--test/test_ssh2.py165
-rw-r--r--test/test_utils.py76
-rw-r--r--test/test_version_compare.py219
-rw-r--r--test/tools/ci-win.cmd131
-rw-r--r--tox.ini141
-rwxr-xr-xupdate_windows_man_page.sh130
-rw-r--r--windows_icon.icobin0 -> 16958 bytes
173 files changed, 11812 insertions, 0 deletions
diff --git a/.appveyor.yml b/.appveyor.yml
new file mode 100644
index 0000000..f1c0cae
--- /dev/null
+++ b/.appveyor.yml
@@ -0,0 +1,33 @@
+version: 'v2.2.1-dev.{build}'
+
+build: off
+branches:
+ only:
+ - master
+ - develop
+
+environment:
+ matrix:
+ - PYTHON: "C:\\Python35"
+ - PYTHON: "C:\\Python35-x64"
+ - PYTHON: "C:\\Python36"
+ - PYTHON: "C:\\Python36-x64"
+ - PYTHON: "C:\\Python37"
+ - PYTHON: "C:\\Python37-x64"
+ - PYTHON: "C:\\Python38"
+ - PYTHON: "C:\\Python38-x64"
+matrix:
+ fast_finish: true
+
+cache:
+ - '%LOCALAPPDATA%\pip\Cache'
+ - .downloads -> .appveyor.yml
+
+install:
+ - "cmd /c .\\test\\tools\\ci-win.cmd install"
+
+test_script:
+ - "cmd /c .\\test\\tools\\ci-win.cmd test"
+
+on_failure:
+ - ps: get-content .tox\*\log\*
diff --git a/.dockerignore b/.dockerignore
new file mode 100644
index 0000000..e7e7b48
--- /dev/null
+++ b/.dockerignore
@@ -0,0 +1,3 @@
+src/ssh_audit/__pycache__/
+src/ssh_audit.egg-info/
+src/ssh_audit/*~
diff --git a/.github/workflows/tox.yaml b/.github/workflows/tox.yaml
new file mode 100644
index 0000000..3c78ec5
--- /dev/null
+++ b/.github/workflows/tox.yaml
@@ -0,0 +1,24 @@
+name: ssh-audit
+on: [push, pull_request]
+
+jobs:
+ build:
+
+ runs-on: ubuntu-latest
+ strategy:
+ matrix:
+ python-version: [3.6, 3.7, 3.8, 3.9]
+
+ steps:
+ - uses: actions/checkout@v2
+ - name: Set up Python ${{ matrix.python-version }}
+ uses: actions/setup-python@v2
+ with:
+ python-version: ${{ matrix.python-version }}
+ - name: Install dependencies
+ run: |
+ python -m pip install --upgrade pip
+ pip install codecov coveralls flake8 mypy pylint tox vulture
+ - name: Run Tox
+ run: |
+ tox
diff --git a/.gitignore b/.gitignore
new file mode 100644
index 0000000..48fb7d4
--- /dev/null
+++ b/.gitignore
@@ -0,0 +1,27 @@
+*~
+*.py[cod]
+*.exe
+*.asc
+venv*/
+.cache/
+.mypy_cache/
+.tox
+.coverage*
+reports/
+.scannerwork/
+
+# PyPI packaging
+/build/
+/dist/
+*.egg-info/
+*.egg
+
+# Snap packaging
+/parts/
+/prime/
+/snap/
+/stage/
+/ssh-audit_*.snap
+
+# Your local server config
+servers.txt
diff --git a/.travis.yml b/.travis.yml
new file mode 100644
index 0000000..3efe2fc
--- /dev/null
+++ b/.travis.yml
@@ -0,0 +1,25 @@
+language: python
+
+arch:
+ - arm64
+ - amd64
+ - ppc64le
+
+python:
+ - "3.6"
+ - "3.7"
+ - "3.8"
+ - "3.9"
+ - "3.10"
+
+cache:
+ - pip
+
+install:
+ - pip install -U pip tox tox-travis coveralls codecov
+
+script:
+ - tox
+
+after_success:
+ - codecov
diff --git a/CONTRIBUTING.md b/CONTRIBUTING.md
new file mode 100644
index 0000000..7e93fb8
--- /dev/null
+++ b/CONTRIBUTING.md
@@ -0,0 +1,26 @@
+# Contributing to ssh-audit
+
+We are very much open to receiving patches from the community! To encourage participation, passing Travis tests, unit tests, etc., *is OPTIONAL*. As long as the patch works properly, it can be merged.
+
+However, if you can submit patches that pass all of our automated tests, then you'll lighten the load for the project maintainer (who already has enough to do!). This document describes what tests are done and what documentation is maintained.
+
+*Anything extra you can do is appreciated!*
+
+
+## Tox Tests
+
+Tox is used to do unit testing, linting with [pylint](http://pylint.pycqa.org/en/latest/) & [flake8](https://flake8.pycqa.org/en/latest/), and static type-checking with [mypy](https://mypy.readthedocs.io/en/stable/).
+
+For Ubuntu 18.04 or later, install tox with `apt install tox`, then simply run `tox` in the top-level directory. Look for any error messages in the (verbose) output.
+
+
+## Docker Tests
+
+Docker is used to run ssh-audit against various real SSH servers (OpenSSH, Dropbear, and TinySSH). The output is then diff'ed against the expected result. Any differences result in failure.
+
+The docker tests are run with `./docker_test.sh`. The first time it is run, it will download and compile the SSH servers; this may take awhile. Subsequent runs, however, will take only a minute to complete, as the docker image will already be up-to-date.
+
+
+## Man Page
+
+The `ssh-audit.1` man page documents the various features of ssh-audit. If features are added, or significant behavior is modified, the man page needs to be updated.
diff --git a/Dockerfile b/Dockerfile
new file mode 100644
index 0000000..357f63b
--- /dev/null
+++ b/Dockerfile
@@ -0,0 +1,10 @@
+FROM python:3.9-slim
+
+WORKDIR /
+
+COPY ssh-audit.py .
+COPY src/ .
+
+ENTRYPOINT ["python3", "/ssh-audit.py"]
+
+EXPOSE 2222
diff --git a/LICENSE b/LICENSE
new file mode 100644
index 0000000..0c66e51
--- /dev/null
+++ b/LICENSE
@@ -0,0 +1,23 @@
+The MIT License (MIT)
+
+Copyright (C) 2017-2020 Joe Testa (jtesta@positronsecurity.com)
+Copyright (C) 2017 Andris Raugulis (moo@arthepsy.eu)
+
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.
diff --git a/Makefile.docker b/Makefile.docker
new file mode 100644
index 0000000..3028f2c
--- /dev/null
+++ b/Makefile.docker
@@ -0,0 +1,13 @@
+VERSION = $(shell grep VERSION src/ssh_audit/globals.py | grep -E -o "'(v.*)'" | tr -d "'")
+ifeq ($(VERSION),)
+ $(error "could not determine version!")
+endif
+
+all:
+ docker build -t positronsecurity/ssh-audit:${VERSION} .
+ docker tag positronsecurity/ssh-audit:${VERSION} positronsecurity/ssh-audit:latest
+
+upload:
+ docker login
+ docker push positronsecurity/ssh-audit:${VERSION}
+ docker push positronsecurity/ssh-audit:latest
diff --git a/Makefile.pypi b/Makefile.pypi
new file mode 100644
index 0000000..f505fbd
--- /dev/null
+++ b/Makefile.pypi
@@ -0,0 +1,15 @@
+all:
+ rm -rf /tmp/pypi_upload
+ virtualenv -p /usr/bin/python3 /tmp/pypi_upload/
+ cp -R src /tmp/pypi_upload/
+ cp setup.py setup.cfg README.md LICENSE /tmp/pypi_upload/
+ /bin/bash -c "pushd /tmp/pypi_upload/; source bin/activate; pip3 install setuptools twine; python3 setup.py sdist bdist_wheel"
+
+uploadtest:
+ /bin/bash -c "pushd /tmp/pypi_upload; source bin/activate; twine upload --repository-url https://test.pypi.org/legacy/ /tmp/pypi_upload/dist/*"
+
+uploadprod:
+ /bin/bash -c "pushd /tmp/pypi_upload; source bin/activate; twine upload /tmp/pypi_upload/dist/*"
+
+clean:
+ rm -rf /tmp/pypi_upload/
diff --git a/Makefile.snap b/Makefile.snap
new file mode 100644
index 0000000..a5e0292
--- /dev/null
+++ b/Makefile.snap
@@ -0,0 +1,6 @@
+all:
+ echo -e "\n\nDid you remember to bump the version number in snapcraft.yaml?\n\n"
+ snapcraft --use-lxd
+
+clean:
+ rm -rf parts/ prime/ snap/ stage/ build/ dist/ src/*.egg-info/ ssh-audit*.snap
diff --git a/PACKAGING.md b/PACKAGING.md
new file mode 100644
index 0000000..c862714
--- /dev/null
+++ b/PACKAGING.md
@@ -0,0 +1,97 @@
+# Windows
+
+An executable can only be made on a Windows host because the PyInstaller tool (https://www.pyinstaller.org/) does not support cross-compilation.
+
+1.) Install Python v3.9.x from https://www.python.org/. To make life easier, check the option to add Python to the PATH environment variable.
+
+2.) Using pip, install pyinstaller and colorama:
+
+```
+ pip install pyinstaller colorama
+```
+
+3.) Install Cygwin (https://www.cygwin.com/).
+
+4.) Create the executable with:
+
+```
+ $ ./build_windows_executable.sh
+```
+
+
+# PyPI
+
+To create package and upload to test server:
+
+```
+ $ sudo apt install python3-virtualenv
+ $ make -f Makefile.pypi
+ $ make -f Makefile.pypi uploadtest
+```
+
+To download from test server and verify:
+
+```
+ $ virtualenv -p /usr/bin/python3 /tmp/pypi_test
+ $ cd /tmp/pypi_test; source bin/activate
+ $ pip3 install --index-url https://test.pypi.org/simple ssh-audit
+```
+
+To upload to production server (hint: use username '\_\_token\_\_' and API token):
+
+```
+ $ make -f Makefile.pypi uploadprod
+```
+
+To download from production server and verify:
+
+```
+ $ virtualenv -p /usr/bin/python3 /tmp/pypi_prod
+ $ cd /tmp/pypi_prod; source bin/activate
+ $ pip3 install ssh-audit
+```
+
+# Snap
+
+To create the snap package, run a fully-updated Ubuntu Server 20.04 VM.
+
+Install pre-requisites with:
+
+```
+ $ sudo apt install make snapcraft
+ $ sudo snap install review-tools lxd
+```
+
+Initialize LXD (leave all options default):
+
+```
+ $ sudo lxd init
+```
+
+Bump the version number in snapcraft.yaml. Then run:
+
+```
+ $ make -f Makefile.snap
+```
+
+Upload the snap with:
+
+```
+ $ snapcraft login
+ $ snapcraft upload --release=stable ssh-audit_*.snap
+```
+
+
+# Docker
+
+Build image with:
+
+```
+ $ make -f Makefile.docker
+```
+
+Then upload it to Dockerhub with:
+
+```
+ $ make -f Makefile.docker upload
+```
diff --git a/README.md b/README.md
new file mode 100644
index 0000000..423408a
--- /dev/null
+++ b/README.md
@@ -0,0 +1,321 @@
+# ssh-audit
+[![License](https://img.shields.io/badge/license-MIT-brightgreen.svg)](https://github.com/jtesta/ssh-audit/blob/master/LICENSE)
+[![PyPI Downloads](https://img.shields.io/pypi/dm/ssh-audit)](https://pypi.org/project/ssh-audit/)
+[![Docker Pulls](https://img.shields.io/docker/pulls/positronsecurity/ssh-audit)](https://hub.docker.com/r/positronsecurity/ssh-audit)
+[![Build Status](https://github.com/jtesta/ssh-audit/actions/workflows/tox.yaml/badge.svg)](https://github.com/jtesta/ssh-audit/actions)
+[![PRs Welcome](https://img.shields.io/badge/PRs-welcome-brightgreen.svg)](https://github.com/jtesta/ssh-audit/blob/master/CONTRIBUTING.md)
+
+**ssh-audit** is a tool for ssh server & client configuration auditing.
+
+[jtesta/ssh-audit](https://github.com/jtesta/ssh-audit/) (v2.0+) is the updated and maintained version of ssh-audit forked from [arthepsy/ssh-audit](https://github.com/arthepsy/ssh-audit) (v1.x) due to inactivity.
+
+- [Features](#features)
+- [Usage](#usage)
+- [Screenshots](#screenshots)
+ - [Server Standard Audit Example](#server-standard-audit-example)
+ - [Server Policy Audit Example](#server-policy-audit-example)
+ - [Client Standard Audit Example](#client-standard-audit-example)
+- [Hardening Guides](#hardening-guides)
+- [Pre-Built Packages](#pre-built-packages)
+- [Web Front-End](#web-front-end)
+- [ChangeLog](#changelog)
+
+## Features
+- SSH1 and SSH2 protocol server support;
+- analyze SSH client configuration;
+- grab banner, recognize device or software and operating system, detect compression;
+- gather key-exchange, host-key, encryption and message authentication code algorithms;
+- output algorithm information (available since, removed/disabled, unsafe/weak/legacy, etc);
+- output algorithm recommendations (append or remove based on recognized software version);
+- output security information (related issues, assigned CVE list, etc);
+- analyze SSH version compatibility based on algorithm information;
+- historical information from OpenSSH, Dropbear SSH and libssh;
+- policy scans to ensure adherence to a hardened/standard configuration;
+- runs on Linux and Windows;
+- supports Python 3.6 - 3.9;
+- no dependencies
+
+## Usage
+```
+usage: ssh-audit.py [options] <host>
+
+ -h, --help print this help
+ -1, --ssh1 force ssh version 1 only
+ -2, --ssh2 force ssh version 2 only
+ -4, --ipv4 enable IPv4 (order of precedence)
+ -6, --ipv6 enable IPv6 (order of precedence)
+ -b, --batch batch output
+ -c, --client-audit starts a server on port 2222 to audit client
+ software config (use -p to change port;
+ use -t to change timeout)
+ -d, --debug Enable debug output.
+ -j, --json JSON output (use -jj to enable indents)
+ -l, --level=<level> minimum output level (info|warn|fail)
+ -L, --list-policies list all the official, built-in policies
+ --lookup=<alg1,alg2,...> looks up an algorithm(s) without
+ connecting to a server
+ -m, --manual print the man page (Windows only)
+ -M, --make-policy=<policy.txt> creates a policy based on the target server
+ (i.e.: the target server has the ideal
+ configuration that other servers should
+ adhere to)
+ -n, --no-colors disable colors
+ -p, --port=<port> port to connect
+ -P, --policy=<"policy name" | policy.txt> run a policy test using the
+ specified policy
+ -t, --timeout=<secs> timeout (in seconds) for connection and reading
+ (default: 5)
+ -T, --targets=<hosts.txt> a file containing a list of target hosts (one
+ per line, format HOST[:PORT])
+ --threads=<threads> number of threads to use when scanning multiple
+ targets (-T/--targets) (default: 32)
+ -v, --verbose verbose output
+```
+* if both IPv4 and IPv6 are used, order of precedence can be set by using either `-46` or `-64`.
+* batch flag `-b` will output sections without header and without empty lines (implies verbose flag).
+* verbose flag `-v` will prefix each line with section type and algorithm name.
+* an exit code of 0 is returned when all algorithms are considered secure (for a standard audit), or when a policy check passes (for a policy audit).
+
+Basic server auditing:
+```
+ssh-audit localhost
+ssh-audit 127.0.0.1
+ssh-audit 127.0.0.1:222
+ssh-audit ::1
+ssh-audit [::1]:222
+```
+
+To run a standard audit against many servers (place targets into servers.txt, one on each line in the format of `HOST[:PORT]`):
+
+```
+ssh-audit -T servers.txt
+```
+
+To audit a client configuration (listens on port 2222 by default; connect using `ssh -p 2222 anything@localhost`):
+
+```
+ssh-audit -c
+```
+
+To audit a client configuration, with a listener on port 4567:
+```
+ssh-audit -c -p 4567
+```
+
+To list all official built-in policies (hint: use resulting policy names with `-P`/`--policy`):
+```
+ssh-audit -L
+```
+
+To run a policy audit against a server:
+```
+ssh-audit -P ["policy name" | path/to/server_policy.txt] targetserver
+```
+
+To run a policy audit against a client:
+```
+ssh-audit -c -P ["policy name" | path/to/client_policy.txt]
+```
+
+To run a policy audit against many servers:
+```
+ssh-audit -T servers.txt -P ["policy name" | path/to/server_policy.txt]
+```
+
+To create a policy based on a target server (which can be manually edited):
+```
+ssh-audit -M new_policy.txt targetserver
+```
+
+## Screenshots
+
+### Server Standard Audit Example
+Below is a screen shot of the standard server-auditing output when connecting to an unhardened OpenSSH v5.3 service:
+![screenshot](https://user-images.githubusercontent.com/2982011/64388792-317e6f80-d00e-11e9-826e-a4934769bb07.png)
+
+### Server Policy Audit Example
+Below is a screen shot of the policy auditing output when connecting to an un-hardened Ubuntu Server 20.04 machine (hint: use `-L`/`--list-policies` to see names of built-in policies to use with `-P`/`--policy`):
+![screenshot](https://user-images.githubusercontent.com/2982011/94370881-95178700-00c0-11eb-8705-3157a4669dc0.png)
+
+After applying the steps in the hardening guide (see below), the output changes to the following:
+![screenshot](https://user-images.githubusercontent.com/2982011/94370873-87620180-00c0-11eb-9a59-469f61a56ce1.png)
+
+### Client Standard Audit Example
+Below is a screen shot of the client-auditing output when an unhardened OpenSSH v7.2 client connects:
+![client_screenshot](https://user-images.githubusercontent.com/2982011/68867998-b946c100-06c4-11ea-975f-1f47e4178a74.png)
+
+## Hardening Guides
+Guides to harden server & client configuration can be found here: [https://www.ssh-audit.com/hardening_guides.html](https://www.ssh-audit.com/hardening_guides.html)
+
+## Pre-Built Packages
+Pre-built packages are available for Windows (see the releases page), on PyPI, Snap, and Homebrew.
+
+To install from PyPI:
+```
+$ pip3 install ssh-audit
+```
+
+To install the Snap package:
+```
+$ snap install ssh-audit
+```
+
+To install on Homebrew:
+```
+$ brew install ssh-audit
+```
+
+To install from Dockerhub:
+```
+$ docker pull positronsecurity/ssh-audit
+```
+(Then run with: `docker run -it -p 2222:2222 positronsecurity/ssh-audit 10.1.1.1`)
+
+## Web Front-End
+For convenience, a web front-end on top of the command-line tool is available at [https://www.ssh-audit.com/](https://www.ssh-audit.com/).
+
+## ChangeLog
+### v2.5.0 (2021-08-26)
+ - Fixed crash when running host key tests.
+ - Handles server connection failures more gracefully.
+ - Now prints JSON with indents when `-jj` is used (useful for debugging).
+ - Added MD5 fingerprints to verbose output.
+ - Added `-d`/`--debug` option for getting debugging output; credit [Adam Russell](https://github.com/thecliguy).
+ - Updated JSON output to include MD5 fingerprints. Note that this results in a breaking change in the 'fingerprints' dictionary format.
+ - Updated OpenSSH 8.1 (and earlier) policies to include `rsa-sha2-512` and `rsa-sha2-256`.
+ - Added OpenSSH v8.6 & v8.7 policies.
+ - Added 3 new key exchanges: `gss-gex-sha1-eipGX3TCiQSrx573bT1o1Q==`, `gss-group1-sha1-eipGX3TCiQSrx573bT1o1Q==`, and `gss-group14-sha1-eipGX3TCiQSrx573bT1o1Q==`.
+ - Added 3 new MACs: `hmac-ripemd160-96`, `AEAD_AES_128_GCM`, and `AEAD_AES_256_GCM`.
+
+### v2.4.0 (2021-02-23)
+ - Added multi-threaded scanning support.
+ - Added built-in Windows manual page (see `-m`/`--manual`); credit [Adam Russell](https://github.com/thecliguy).
+ - Added version check for OpenSSH user enumeration (CVE-2018-15473).
+ - Added deprecation note to host key types based on SHA-1.
+ - Added extra warnings for SSHv1.
+ - Added built-in hardened OpenSSH v8.5 policy.
+ - Upgraded warnings to failures for host key types based on SHA-1.
+ - Fixed crash when receiving unexpected response during host key test.
+ - Fixed hang against older Cisco devices during host key test & gex test.
+ - Fixed improper termination while scanning multiple targets when one target returns an error.
+ - Dropped support for Python 3.5 (which reached EOL in Sept. 2020).
+ - Added 1 new key exchange: `sntrup761x25519-sha512@openssh.com`.
+
+### v2.3.1 (2020-10-28)
+ - Now parses public key sizes for `rsa-sha2-256-cert-v01@openssh.com` and `rsa-sha2-512-cert-v01@openssh.com` host key types.
+ - Flag `ssh-rsa-cert-v01@openssh.com` as a failure due to SHA-1 hash.
+ - Fixed bug in recommendation output which suppressed some algorithms inappropriately.
+ - Built-in policies now include CA key requirements (if certificates are in use).
+ - Lookup function (`--lookup`) now performs case-insensitive lookups of similar algorithms; credit [Adam Russell](https://github.com/thecliguy).
+ - Migrated pre-made policies from external files to internal database.
+ - Split single 3,500 line script into many files (by class).
+ - Added setup.py support; credit [Ganden Schaffner](https://github.com/gschaffner).
+ - Added 1 new cipher: `des-cbc@ssh.com`.
+
+### v2.3.0 (2020-09-27)
+ - Added new policy auditing functionality to test adherence to a hardening guide/standard configuration (see `-L`/`--list-policies`, `-M`/`--make-policy` and `-P`/`--policy`). For an in-depth tutorial, see <https://www.positronsecurity.com/blog/2020-09-27-ssh-policy-configuration-checks-with-ssh-audit/>.
+ - Created new man page (see `ssh-audit.1` file).
+ - 1024-bit moduli upgraded from warnings to failures.
+ - Many Python 2 code clean-ups, testing framework improvements, pylint & flake8 fixes, and mypy type comments; credit [Jürgen Gmach](https://github.com/jugmac00).
+ - Added feature to look up algorithms in internal database (see `--lookup`); credit [Adam Russell](https://github.com/thecliguy).
+ - Suppress recommendation of token host key types.
+ - Added check for use-after-free vulnerability in PuTTY v0.73.
+ - Added 11 new host key types: `ssh-rsa1`, `ssh-dss-sha256@ssh.com`, `ssh-gost2001`, `ssh-gost2012-256`, `ssh-gost2012-512`, `spki-sign-rsa`, `ssh-ed448`, `x509v3-ecdsa-sha2-nistp256`, `x509v3-ecdsa-sha2-nistp384`, `x509v3-ecdsa-sha2-nistp521`, `x509v3-rsa2048-sha256`.
+ - Added 8 new key exchanges: `diffie-hellman-group1-sha256`, `kexAlgoCurve25519SHA256`, `Curve25519SHA256`, `gss-group14-sha256-`, `gss-group15-sha512-`, `gss-group16-sha512-`, `gss-nistp256-sha256-`, `gss-curve25519-sha256-`.
+ - Added 5 new ciphers: `blowfish`, `AEAD_AES_128_GCM`, `AEAD_AES_256_GCM`, `crypticore128@ssh.com`, `seed-cbc@ssh.com`.
+ - Added 3 new MACs: `chacha20-poly1305@openssh.com`, `hmac-sha3-224`, `crypticore-mac@ssh.com`.
+
+### v2.2.0 (2020-03-11)
+ - Marked host key type `ssh-rsa` as weak due to [practical SHA-1 collisions](https://eprint.iacr.org/2020/014.pdf).
+ - Added Windows builds.
+ - Added 10 new host key types: `ecdsa-sha2-1.3.132.0.10`, `x509v3-sign-dss`, `x509v3-sign-rsa`, `x509v3-sign-rsa-sha256@ssh.com`, `x509v3-ssh-dss`, `x509v3-ssh-rsa`, `sk-ecdsa-sha2-nistp256-cert-v01@openssh.com`, `sk-ecdsa-sha2-nistp256@openssh.com`, `sk-ssh-ed25519-cert-v01@openssh.com`, and `sk-ssh-ed25519@openssh.com`.
+ - Added 18 new key exchanges: `diffie-hellman-group14-sha256@ssh.com`, `diffie-hellman-group15-sha256@ssh.com`, `diffie-hellman-group15-sha384@ssh.com`, `diffie-hellman-group16-sha384@ssh.com`, `diffie-hellman-group16-sha512@ssh.com`, `diffie-hellman-group18-sha512@ssh.com`, `ecdh-sha2-curve25519`, `ecdh-sha2-nistb233`, `ecdh-sha2-nistb409`, `ecdh-sha2-nistk163`, `ecdh-sha2-nistk233`, `ecdh-sha2-nistk283`, `ecdh-sha2-nistk409`, `ecdh-sha2-nistp192`, `ecdh-sha2-nistp224`, `ecdh-sha2-nistt571`, `gss-gex-sha1-`, and `gss-group1-sha1-`.
+ - Added 9 new ciphers: `camellia128-cbc`, `camellia128-ctr`, `camellia192-cbc`, `camellia192-ctr`, `camellia256-cbc`, `camellia256-ctr`, `aes128-gcm`, `aes256-gcm`, and `chacha20-poly1305`.
+ - Added 2 new MACs: `aes128-gcm` and `aes256-gcm`.
+
+### v2.1.1 (2019-11-26)
+ - Added 2 new host key types: `rsa-sha2-256-cert-v01@openssh.com`, `rsa-sha2-512-cert-v01@openssh.com`.
+ - Added 2 new ciphers: `des`, `3des`.
+ - Added 3 new PuTTY vulnerabilities.
+ - During client testing, client IP address is now listed in output.
+
+### v2.1.0 (2019-11-14)
+ - Added client software auditing functionality (see `-c` / `--client-audit` option).
+ - Added JSON output option (see `-j` / `--json` option; credit [Andreas Jaggi](https://github.com/x-way)).
+ - Fixed crash while scanning Solaris Sun_SSH.
+ - Added 9 new key exchanges: `gss-group1-sha1-toWM5Slw5Ew8Mqkay+al2g==`, `gss-gex-sha1-toWM5Slw5Ew8Mqkay+al2g==`, `gss-group14-sha1-`, `gss-group14-sha1-toWM5Slw5Ew8Mqkay+al2g==`, `gss-group14-sha256-toWM5Slw5Ew8Mqkay+al2g==`, `gss-group15-sha512-toWM5Slw5Ew8Mqkay+al2g==`, `diffie-hellman-group15-sha256`, `ecdh-sha2-1.3.132.0.10`, `curve448-sha512`.
+ - Added 1 new host key type: `ecdsa-sha2-1.3.132.0.10`.
+ - Added 4 new ciphers: `idea-cbc`, `serpent128-cbc`, `serpent192-cbc`, `serpent256-cbc`.
+ - Added 6 new MACs: `hmac-sha2-256-96-etm@openssh.com`, `hmac-sha2-512-96-etm@openssh.com`, `hmac-ripemd`, `hmac-sha256-96@ssh.com`, `umac-32@openssh.com`, `umac-96@openssh.com`.
+
+### v2.0.0 (2019-08-29)
+ - Forked from https://github.com/arthepsy/ssh-audit (development was stalled, and developer went MIA).
+ - Added RSA host key length test.
+ - Added RSA certificate key length test.
+ - Added Diffie-Hellman modulus size test.
+ - Now outputs host key fingerprints for RSA and ED25519.
+ - Added 5 new key exchanges: `sntrup4591761x25519-sha512@tinyssh.org`, `diffie-hellman-group-exchange-sha256@ssh.com`, `diffie-hellman-group-exchange-sha512@ssh.com`, `diffie-hellman-group16-sha256`, `diffie-hellman-group17-sha512`.
+ - Added 3 new encryption algorithms: `des-cbc-ssh1`, `blowfish-ctr`, `twofish-ctr`.
+ - Added 10 new MACs: `hmac-sha2-56`, `hmac-sha2-224`, `hmac-sha2-384`, `hmac-sha3-256`, `hmac-sha3-384`, `hmac-sha3-512`, `hmac-sha256`, `hmac-sha256@ssh.com`, `hmac-sha512`, `hmac-512@ssh.com`.
+ - Added command line argument (`-t` / `--timeout`) for connection & reading timeouts.
+ - Updated CVEs for libssh & Dropbear.
+
+### v1.7.0 (2016-10-26)
+ - implement options to allow specify IPv4/IPv6 usage and order of precedence
+ - implement option to specify remote port (old behavior kept for compatibility)
+ - add colors support for Microsoft Windows via optional colorama dependency
+ - fix encoding and decoding issues, add tests, do not crash on encoding errors
+ - use mypy-lang for static type checking and verify all code
+
+### v1.6.0 (2016-10-14)
+ - implement algorithm recommendations section (based on recognized software)
+ - implement full libssh support (version history, algorithms, security, etc)
+ - fix SSH-1.99 banner recognition and version comparison functionality
+ - do not output empty algorithms (happens for misconfigured servers)
+ - make consistent output for Python 3.x versions
+ - add a lot more tests (conf, banner, software, SSH1/SSH2, output, etc)
+ - use Travis CI to test for multiple Python versions (2.6-3.5, pypy, pypy3)
+
+### v1.5.0 (2016-09-20)
+ - create security section for related security information
+ - match and output assigned CVE list and security issues for Dropbear SSH
+ - implement full SSH1 support with fingerprint information
+ - automatically fallback to SSH1 on protocol mismatch
+ - add new options to force SSH1 or SSH2 (both allowed by default)
+ - parse banner information and convert it to specific software and OS version
+ - do not use padding in batch mode
+ - several fixes (Cisco sshd, rare hangs, error handling, etc)
+
+### v1.0.20160902
+ - implement batch output option
+ - implement minimum output level option
+ - fix compatibility with Python 2.6
+
+### v1.0.20160812
+ - implement SSH version compatibility feature
+ - fix wrong mac algorithm warning
+ - fix Dropbear SSH version typo
+ - parse pre-banner header
+ - better errors handling
+
+### v1.0.20160803
+ - use OpenSSH 7.3 banner
+ - add new key-exchange algorithms
+
+### v1.0.20160207
+ - use OpenSSH 7.2 banner
+ - additional warnings for OpenSSH 7.2
+ - fix OpenSSH 7.0 failure messages
+ - add rijndael-cbc failure message from OpenSSH 6.7
+
+### v1.0.20160105
+ - multiple additional warnings
+ - support for none algorithm
+ - better compression handling
+ - ensure reading enough data (fixes few Linux SSH)
+
+### v1.0.20151230
+ - Dropbear SSH support
+
+### v1.0.20151223
+ - initial version
diff --git a/build_windows_executable.sh b/build_windows_executable.sh
new file mode 100755
index 0000000..93d794e
--- /dev/null
+++ b/build_windows_executable.sh
@@ -0,0 +1,130 @@
+#!/bin/bash
+
+#
+# The MIT License (MIT)
+#
+# Copyright (C) 2021 Joe Testa (jtesta@positronsecurity.com)
+#
+# Permission is hereby granted, free of charge, to any person obtaining a copy
+# of this software and associated documentation files (the "Software"), to deal
+# in the Software without restriction, including without limitation the rights
+# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+# copies of the Software, and to permit persons to whom the Software is
+# furnished to do so, subject to the following conditions:
+#
+# The above copyright notice and this permission notice shall be included in
+# all copies or substantial portions of the Software.
+#
+# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+# THE SOFTWARE.
+#
+
+################################################################################
+# build_windows_executable.sh
+#
+# Builds a Windows executable using PyInstaller.
+################################################################################
+
+
+PLATFORM="$(uname -s)"
+
+# This script is intended for use on Cygwin only.
+case "$PLATFORM" in
+ CYGWIN*) ;;
+ *)
+ echo "Platform not supported ($PLATFORM). This must be run in Cygwin only."
+ exit 1
+ ;;
+esac
+
+# Ensure that Python 3.x is installed.
+if [[ "$(python -V)" != "Python 3."* ]]; then
+ echo "Python v3.x not found. Install the latest stable version from: https://www.python.org/"
+ exit 1
+fi
+
+# Ensure that pyinstaller is installed.
+command -v pyinstaller >/dev/null 2>&1 || { echo >&2 "pyinstaller not found. Install with: 'pip install pyinstaller'"; exit 1; }
+
+# Ensure that the colorama module is installed.
+X=`pip show colorama` 2> /dev/null
+if [[ $? != 0 ]]; then
+ echo "Colorama module not found. Install with: 'pip install colorama'"
+ exit 1
+fi
+
+# Prompt for the version to release.
+echo -n "Enter the version to release, using format 'vX.X.X': "
+read -r version
+
+# Ensure that entered version fits required format.
+if [[ ! $version =~ ^v[0-9]\.[0-9]\.[0-9]$ ]]; then
+ echo "Error: version string does not match format vX.X.X!"
+ exit 1
+fi
+
+# Verify that version is correct.
+echo -n "Version will be set to '${version}'. Is this correct? (y/n): "
+read -r yn
+echo
+
+if [[ $yn != "y" ]]; then
+ echo "Build cancelled."
+ exit 1
+fi
+
+# Reset any local changes made to globals.py from a previous run.
+git checkout src/ssh_audit/globals.py 2> /dev/null
+
+# Update the man page.
+./update_windows_man_page.sh
+if [[ $? != 0 ]]; then
+ echo "Failed to run ./update_windows_man_page.sh"
+ exit 1
+fi
+
+# Do all operations from this point from the main source directory.
+pushd src/ssh_audit > /dev/null
+
+# Delete the existing VERSION variable and add the value that the user entered, above.
+sed -i '/^VERSION/d' globals.py
+echo "VERSION = '$version'" >> globals.py
+
+# Delete cached files if they exist from a prior run.
+rm -rf dist/ build/ ssh-audit.spec
+
+# Create a hard link from ssh_audit.py to ssh-audit.py.
+if [[ ! -f ssh-audit.py ]]; then
+ ln ssh_audit.py ssh-audit.py
+fi
+
+echo -e "\nRunning pyinstaller...\n"
+pyinstaller -F --icon ../../windows_icon.ico ssh-audit.py
+
+if [[ -f dist/ssh-audit.exe ]]; then
+ echo -e "\nExecutable created in $(pwd)/dist/ssh-audit.exe\n"
+else
+ echo -e "\nFAILED to create $(pwd)/dist/ssh-audit.exe!\n"
+ exit 1
+fi
+
+# Ensure that the version string doesn't have '-dev' in it.
+X=`dist/ssh-audit.exe | grep -E 'ssh-audit.exe v.+\-dev'` > /dev/null
+if [[ $? == 0 ]]; then
+ echo -e "\nError: executable's version number includes '-dev'."
+ exit 1
+fi
+
+# Remove the cache files created during the build process, along with the link we created, above.
+rm -rf build/ ssh-audit.spec ssh-audit.py
+
+# Reset the changes we made to globals.py.
+git checkout globals.py 2> /dev/null
+
+popd > /dev/null
+exit 0
diff --git a/docker_test.sh b/docker_test.sh
new file mode 100755
index 0000000..0afdd9a
--- /dev/null
+++ b/docker_test.sh
@@ -0,0 +1,716 @@
+#!/bin/bash
+
+#
+# This script will set up a docker image with multiple versions of OpenSSH, then
+# use it to run tests.
+#
+# For debugging purposes, here is a cheat sheet for manually running the docker image:
+#
+# docker run -p 2222:22 -it ssh-audit-test:X /bin/bash
+# docker run -p 2222:22 --security-opt seccomp:unconfined -it ssh-audit-test /debug.sh
+# docker run -d -p 2222:22 ssh-audit-test:X /openssh/sshd-5.6p1 -D -f /etc/ssh/sshd_config-5.6p1_test1
+# docker run -d -p 2222:22 ssh-audit-test:X /openssh/sshd-8.0p1 -D -f /etc/ssh/sshd_config-8.0p1_test1
+#
+
+
+# This is the docker tag for the image. If this tag doesn't exist, then we assume the
+# image is out of date, and generate a new one with this tag.
+IMAGE_VERSION=3
+
+# This is the name of our docker image.
+IMAGE_NAME=positronsecurity/ssh-audit-test-framework
+
+
+# Terminal colors.
+CLR="\033[0m"
+RED="\033[0;31m"
+YELLOW="\033[0;33m"
+GREEN="\033[0;32m"
+REDB="\033[1;31m" # Red + bold
+GREENB="\033[1;32m" # Green + bold
+
+# Program return values.
+PROGRAM_RETVAL_FAILURE=3
+PROGRAM_RETVAL_WARNING=2
+PROGRAM_RETVAL_CONNECTION_ERROR=1
+PROGRAM_RETVAL_GOOD=0
+
+
+# Counts the number of test failures.
+num_failures=0
+
+
+# Returns 0 if current docker image exists.
+function check_if_docker_image_exists {
+ images=`docker image ls | egrep "$IMAGE_NAME[[:space:]]+$IMAGE_VERSION"`
+}
+
+
+# Uncompresses and compiles the specified version of Dropbear.
+function compile_dropbear {
+ version=$1
+ compile 'Dropbear' $version
+}
+
+
+# Uncompresses and compiles the specified version of OpenSSH.
+function compile_openssh {
+ version=$1
+ compile 'OpenSSH' $version
+}
+
+
+# Uncompresses and compiles the specified version of TinySSH.
+function compile_tinyssh {
+ version=$1
+ compile 'TinySSH' $version
+}
+
+
+function compile {
+ project=$1
+ version=$2
+
+ tarball=
+ uncompress_options=
+ source_dir=
+ server_executable=
+ if [[ $project == 'OpenSSH' ]]; then
+ tarball="openssh-${version}.tar.gz"
+ uncompress_options="xzf"
+ source_dir="openssh-${version}"
+ server_executable=sshd
+ elif [[ $project == 'Dropbear' ]]; then
+ tarball="dropbear-${version}.tar.bz2"
+ uncompress_options="xjf"
+ source_dir="dropbear-${version}"
+ server_executable=dropbear
+ elif [[ $project == 'TinySSH' ]]; then
+ tarball="${version}.tar.gz"
+ uncompress_options="xzf"
+ source_dir="tinyssh-${version}"
+ server_executable='build/bin/tinysshd'
+ fi
+
+ echo "Uncompressing ${project} ${version}..."
+ tar $uncompress_options $tarball
+
+ echo "Compiling ${project} ${version}..."
+ pushd $source_dir > /dev/null
+
+ # TinySSH has no configure script... only a Makefile.
+ if [[ $project == 'TinySSH' ]]; then
+ make -j 10
+ else
+ ./configure && make -j 10
+ fi
+
+ if [[ ! -f $server_executable ]]; then
+ echo -e "${REDB}Error: ${server_executable} not built!${CLR}"
+ exit 1
+ fi
+
+ echo -e "\n${GREEN}Successfully built ${project} ${version}${CLR}\n"
+ popd > /dev/null
+}
+
+
+# Creates a new docker image.
+function create_docker_image {
+ # Create a new temporary directory.
+ TMP_DIR=`mktemp -d /tmp/sshaudit-docker-XXXXXXXXXX`
+
+ # Copy the Dockerfile and all files in the test/docker/ dir to our new temp directory.
+ find test/docker/ -maxdepth 1 -type f | xargs cp -t $TMP_DIR
+
+ # Make the temp directory our working directory for the duration of the build
+ # process.
+ pushd $TMP_DIR > /dev/null
+
+ # Get the release keys.
+ get_dropbear_release_key
+ get_openssh_release_key
+ get_tinyssh_release_key
+
+ # Aside from checking the GPG signatures, we also compare against this known-good
+ # SHA-256 hash just in case.
+ get_openssh '4.0p1' '5adb9b2c2002650e15216bf94ed9db9541d9a17c96fcd876784861a8890bc92b'
+ get_openssh '5.6p1' '538af53b2b8162c21a293bb004ae2bdb141abd250f61b4cea55244749f3c6c2b'
+ get_openssh '8.0p1' 'bd943879e69498e8031eb6b7f44d08cdc37d59a7ab689aa0b437320c3481fd68'
+ get_dropbear '2019.78' '525965971272270995364a0eb01f35180d793182e63dd0b0c3eb0292291644a4'
+ get_tinyssh '20190101' '554a9a94e53b370f0cd0c5fbbd322c34d1f695cbcea6a6a32dcb8c9f595b3fea'
+
+ # Compile the versions of OpenSSH.
+ compile_openssh '4.0p1'
+ compile_openssh '5.6p1'
+ compile_openssh '8.0p1'
+
+ # Compile the versions of Dropbear.
+ compile_dropbear '2019.78'
+
+ # Compile the versions of TinySSH.
+ compile_tinyssh '20190101'
+
+
+ # Rename the default config files so we know they are our originals.
+ mv openssh-4.0p1/sshd_config sshd_config-4.0p1_orig
+ mv openssh-5.6p1/sshd_config sshd_config-5.6p1_orig
+ mv openssh-8.0p1/sshd_config sshd_config-8.0p1_orig
+
+
+ # Create the configurations for each test.
+
+
+ #
+ # OpenSSH v4.0p1
+ #
+
+ # Test 1: Basic test.
+ create_openssh_config '4.0p1' 'test1' "HostKey /etc/ssh/ssh1_host_key\nHostKey /etc/ssh/ssh_host_rsa_key_1024\nHostKey /etc/ssh/ssh_host_dsa_key"
+
+
+ #
+ # OpenSSH v5.6p1
+ #
+
+ # Test 1: Basic test.
+ create_openssh_config '5.6p1' 'test1' "HostKey /etc/ssh/ssh_host_rsa_key_1024\nHostKey /etc/ssh/ssh_host_dsa_key"
+
+ # Test 2: RSA 1024 host key with RSA 1024 certificate.
+ create_openssh_config '5.6p1' 'test2' "HostKey /etc/ssh/ssh_host_rsa_key_1024\nHostCertificate /etc/ssh/ssh_host_rsa_key_1024-cert_1024.pub"
+
+ # Test 3: RSA 1024 host key with RSA 3072 certificate.
+ create_openssh_config '5.6p1' 'test3' "HostKey /etc/ssh/ssh_host_rsa_key_1024\nHostCertificate /etc/ssh/ssh_host_rsa_key_1024-cert_3072.pub"
+
+ # Test 4: RSA 3072 host key with RSA 1024 certificate.
+ create_openssh_config '5.6p1' 'test4' "HostKey /etc/ssh/ssh_host_rsa_key_3072\nHostCertificate /etc/ssh/ssh_host_rsa_key_3072-cert_1024.pub"
+
+ # Test 5: RSA 3072 host key with RSA 3072 certificate.
+ create_openssh_config '5.6p1' 'test5' "HostKey /etc/ssh/ssh_host_rsa_key_3072\nHostCertificate /etc/ssh/ssh_host_rsa_key_3072-cert_3072.pub"
+
+
+ #
+ # OpenSSH v8.0p1
+ #
+
+ # Test 1: Basic test.
+ create_openssh_config '8.0p1' 'test1' "HostKey /etc/ssh/ssh_host_rsa_key_3072\nHostKey /etc/ssh/ssh_host_ecdsa_key\nHostKey /etc/ssh/ssh_host_ed25519_key"
+
+ # Test 2: ED25519 certificate test.
+ create_openssh_config '8.0p1' 'test2' "HostKey /etc/ssh/ssh_host_ed25519_key\nHostCertificate /etc/ssh/ssh_host_ed25519_key-cert.pub"
+
+ # Test 3: Hardened installation test.
+ create_openssh_config '8.0p1' 'test3' "HostKey /etc/ssh/ssh_host_ed25519_key\nKexAlgorithms curve25519-sha256,curve25519-sha256@libssh.org,diffie-hellman-group-exchange-sha256\nCiphers chacha20-poly1305@openssh.com,aes256-gcm@openssh.com,aes128-gcm@openssh.com,aes256-ctr,aes192-ctr,aes128-ctr\nMACs hmac-sha2-256-etm@openssh.com,hmac-sha2-512-etm@openssh.com,umac-128-etm@openssh.com"
+
+
+ # Now build the docker image!
+ docker build --tag $IMAGE_NAME:$IMAGE_VERSION .
+
+ popd > /dev/null
+ rm -rf $TMP_DIR
+}
+
+
+# Creates an OpenSSH configuration file for a specific test.
+function create_openssh_config {
+ openssh_version=$1
+ test_number=$2
+ config_text=$3
+
+ cp sshd_config-${openssh_version}_orig sshd_config-${openssh_version}_${test_number}
+ echo -e "${config_text}" >> sshd_config-${openssh_version}_${test_number}
+}
+
+
+# Downloads the Dropbear release key and adds it to the local keyring.
+function get_dropbear_release_key {
+ get_release_key 'Dropbear' 'https://matt.ucc.asn.au/dropbear/releases/dropbear-key-2015.asc' 'F29C6773' 'F734 7EF2 EE2E 07A2 6762 8CA9 4493 1494 F29C 6773'
+}
+
+
+# Downloads the OpenSSH release key and adds it to the local keyring.
+function get_openssh_release_key {
+ get_release_key 'OpenSSH' 'https://ftp.openbsd.org/pub/OpenBSD/OpenSSH/RELEASE_KEY.asc' '6D920D30' '59C2 118E D206 D927 E667 EBE3 D3E5 F56B 6D92 0D30'
+}
+
+
+# Downloads the TinySSH release key and adds it to the local keyring.
+function get_tinyssh_release_key {
+ get_release_key 'TinySSH' '' '96939FF9' 'AADF 2EDF 5529 F170 2772 C8A2 DEC4 D246 931E F49B'
+}
+
+
+function get_release_key {
+ project=$1
+ key_url=$2
+ key_id=$3
+ release_key_fingerprint_expected=$4
+
+ # The TinySSH release key isn't on any website, apparently.
+ if [[ $project == 'TinySSH' ]]; then
+ gpg --keyserver keys.gnupg.net --recv-key $key_id
+ else
+ echo -e "\nGetting ${project} release key...\n"
+ wget -O key.asc $2
+
+ echo -e "\nImporting ${project} release key...\n"
+ gpg --import key.asc
+
+ rm key.asc
+ fi
+
+ local release_key_fingerprint_actual=`gpg --fingerprint ${key_id}`
+ if [[ $release_key_fingerprint_actual != *"$release_key_fingerprint_expected"* ]]; then
+ echo -e "\n${REDB}Error: ${project} release key fingerprint does not match expected value!\n\tExpected: $release_key_fingerprint_expected\n\tActual: $release_key_fingerprint_actual\n\nTerminating.${CLR}"
+ exit -1
+ fi
+ echo -e "\n\n${GREEN}${project} release key matches expected value.${CLR}\n"
+}
+
+
+# Downloads the specified version of Dropbear.
+function get_dropbear {
+ version=$1
+ tarball_checksum_expected=$2
+ get_source 'Dropbear' $version $tarball_checksum_expected
+}
+
+
+# Downloads the specified version of OpenSSH.
+function get_openssh {
+ version=$1
+ tarball_checksum_expected=$2
+ get_source 'OpenSSH' $version $tarball_checksum_expected
+}
+
+
+# Downloads the specified version of TinySSH.
+function get_tinyssh {
+ version=$1
+ tarball_checksum_expected=$2
+ get_source 'TinySSH' $version $tarball_checksum_expected
+}
+
+
+function get_source {
+ project=$1
+ version=$2
+ tarball_checksum_expected=$3
+
+ base_url_source=
+ base_url_sig=
+ tarball=
+ sig=
+ signer=
+ if [[ $project == 'OpenSSH' ]]; then
+ base_url_source='https://cdn.openbsd.org/pub/OpenBSD/OpenSSH/portable/'
+ base_url_sig=$base_url_source
+ tarball="openssh-${version}.tar.gz"
+ sig="${tarball}.asc"
+ signer="Damien Miller "
+ elif [[ $project == 'Dropbear' ]]; then
+ base_url_source='https://matt.ucc.asn.au/dropbear/releases/'
+ base_url_sig=$base_url_source
+ tarball="dropbear-${version}.tar.bz2"
+ sig="${tarball}.asc"
+ signer="Dropbear SSH Release Signing <matt@ucc.asn.au>"
+ elif [[ $project == 'TinySSH' ]]; then
+ base_url_source='https://github.com/janmojzis/tinyssh/archive/'
+ base_url_sig="https://github.com/janmojzis/tinyssh/releases/download/${version}/"
+ tarball="${version}.tar.gz"
+ sig="${tarball}.asc"
+ signer="Jan Mojžíš <jan.mojzis@gmail.com>"
+ fi
+
+ echo -e "\nGetting ${project} ${version} sources...\n"
+ wget "${base_url_source}${tarball}"
+
+ echo -e "\nGetting ${project} ${version} signature...\n"
+ wget "${base_url_sig}${sig}"
+
+
+ # Older OpenSSH releases were .sigs.
+ if [[ ($project == 'OpenSSH') && (! -f $sig) ]]; then
+ wget ${base_url_sig}openssh-${version}.tar.gz.sig
+ sig=openssh-${version}.tar.gz.sig
+ fi
+
+ local gpg_verify=`gpg --verify ${sig} ${tarball} 2>&1`
+ if [[ $gpg_verify != *"Good signature from \"${signer}"* ]]; then
+ echo -e "\n\n${REDB}Error: ${project} signature invalid!\n$gpg_verify\n\nTerminating.${CLR}"
+ exit -1
+ fi
+
+ # Check GPG's return value. 0 denotes a valid signature, and 1 is returned
+ # on invalid signatures.
+ if [[ $? != 0 ]]; then
+ echo -e "\n\n${REDB}Error: ${project} signature invalid! Verification returned code: $?\n\nTerminating.${CLR}"
+ exit -1
+ fi
+
+ echo -e "${GREEN}Signature on ${project} sources verified.${CLR}\n"
+
+ local checksum_actual=`sha256sum ${tarball} | cut -f1 -d" "`
+ if [[ $checksum_actual != $tarball_checksum_expected ]]; then
+ echo -e "${REDB}Error: ${project} checksum is invalid!\n Expected: ${tarball_checksum_expected}\n Actual: ${checksum_actual}\n\n Terminating.${CLR}"
+ exit -1
+ fi
+}
+
+
+# Pulls the defined image from Dockerhub.
+function pull_docker_image {
+ docker pull $IMAGE_NAME:$IMAGE_VERSION
+ if [[ $? == 0 ]]; then
+ echo -e "${GREEN}Successfully downloaded image ${IMAGE_NAME}:${IMAGE_VERSION} from Dockerhub.${CLR}\n"
+ else
+ echo -e "${REDB}Failed to pull image ${IMAGE_NAME}:${IMAGE_VERSION} from Dockerhub! Error code: $?${CLR}\n"
+ exit -1
+ fi
+}
+
+
+# Runs a Dropbear test. Upon failure, a diff between the expected and actual results
+# is shown, then the script immediately terminates.
+function run_dropbear_test {
+ dropbear_version=$1
+ test_number=$2
+ options=$3
+ expected_retval=$4
+
+ run_test 'Dropbear' $dropbear_version $test_number "$options" $expected_retval
+}
+
+
+# Runs an OpenSSH test. Upon failure, a diff between the expected and actual results
+# is shown, then the script immediately terminates.
+function run_openssh_test {
+ openssh_version=$1
+ test_number=$2
+ expected_retval=$3
+
+ run_test 'OpenSSH' $openssh_version $test_number '' $expected_retval
+}
+
+
+# Runs a TinySSH test. Upon failure, a diff between the expected and actual results
+# is shown, then the script immediately terminates.
+function run_tinyssh_test {
+ tinyssh_version=$1
+ test_number=$2
+ expected_retval=$3
+
+ run_test 'TinySSH' $tinyssh_version $test_number '' $expected_retval
+}
+
+
+function run_test {
+ server_type=$1
+ version=$2
+ test_number=$3
+ options=$4
+ expected_retval=$5
+
+ failed=0 # Set to 1 if this test fails.
+ server_exec=
+ test_result_stdout=
+ test_result_json=
+ expected_result_stdout=
+ expected_result_json=
+ test_name=
+ if [[ $server_type == 'OpenSSH' ]]; then
+ server_exec="/openssh/sshd-${version} -D -f /etc/ssh/sshd_config-${version}_${test_number}"
+ test_result_stdout="${TEST_RESULT_DIR}/openssh_${version}_${test_number}.txt"
+ test_result_json="${TEST_RESULT_DIR}/openssh_${version}_${test_number}.json"
+ expected_result_stdout="test/docker/expected_results/openssh_${version}_${test_number}.txt"
+ expected_result_json="test/docker/expected_results/openssh_${version}_${test_number}.json"
+ test_name="OpenSSH ${version} ${test_number}"
+ options=
+ elif [[ $server_type == 'Dropbear' ]]; then
+ server_exec="/dropbear/dropbear-${version} -F ${options}"
+ test_result_stdout="${TEST_RESULT_DIR}/dropbear_${version}_${test_number}.txt"
+ test_result_json="${TEST_RESULT_DIR}/dropbear_${version}_${test_number}.json"
+ expected_result_stdout="test/docker/expected_results/dropbear_${version}_${test_number}.txt"
+ expected_result_json="test/docker/expected_results/dropbear_${version}_${test_number}.json"
+ test_name="Dropbear ${version} ${test_number}"
+ elif [[ $server_type == 'TinySSH' ]]; then
+ server_exec="/usr/bin/tcpserver -HRDl0 0.0.0.0 22 /tinysshd/tinyssh-20190101 -v /etc/tinyssh/"
+ test_result_stdout="${TEST_RESULT_DIR}/tinyssh_${version}_${test_number}.txt"
+ test_result_json="${TEST_RESULT_DIR}/tinyssh_${version}_${test_number}.json"
+ expected_result_stdout="test/docker/expected_results/tinyssh_${version}_${test_number}.txt"
+ expected_result_json="test/docker/expected_results/tinyssh_${version}_${test_number}.json"
+ test_name="TinySSH ${version} ${test_number}"
+ fi
+
+ cid=`docker run -d -p 2222:22 ${IMAGE_NAME}:${IMAGE_VERSION} ${server_exec}`
+ #echo "Running: docker run -d -p 2222:22 ${IMAGE_NAME}:${IMAGE_VERSION} ${server_exec}"
+ if [[ $? != 0 ]]; then
+ echo -e "${REDB}Failed to run docker image! (exit code: $?)${CLR}"
+ exit 1
+ fi
+
+ ./ssh-audit.py localhost:2222 > $test_result_stdout
+ actual_retval=$?
+ if [[ $actual_retval != $expected_retval ]]; then
+ echo -e "${REDB}Unexpected return value. Expected: ${expected_retval}; Actual: ${actual_retval}${CLR}"
+ docker container stop -t 0 $cid > /dev/null
+ exit 1
+ fi
+
+ ./ssh-audit.py -j localhost:2222 > $test_result_json
+ actual_retval=$?
+ if [[ $actual_retval != $expected_retval ]]; then
+ echo -e "${REDB}Unexpected return value. Expected: ${expected_retval}; Actual: ${actual_retval}${CLR}"
+ docker container stop -t 0 $cid > /dev/null
+ exit 1
+ fi
+
+ docker container stop -t 0 $cid > /dev/null
+ if [[ $? != 0 ]]; then
+ echo -e "${REDB}Failed to stop docker container ${cid}! (exit code: $?)${CLR}"
+ exit 1
+ fi
+
+ # TinySSH outputs a random string in each banner, which breaks our test. So
+ # we need to filter out the banner part of the output so we get stable, repeatable
+ # results.
+ if [[ $server_type == 'TinySSH' ]]; then
+ grep -v "(gen) banner: " ${test_result_stdout} > "${test_result_stdout}.tmp"
+ mv "${test_result_stdout}.tmp" ${test_result_stdout}
+ cat "${test_result_json}" | perl -pe 's/"comments": ".*?"/"comments": ""/' | perl -pe 's/"raw": ".+?"/"raw": ""/' > "${test_result_json}.tmp"
+ mv "${test_result_json}.tmp" ${test_result_json}
+ fi
+
+ diff=`diff -u ${expected_result_stdout} ${test_result_stdout}`
+ if [[ $? != 0 ]]; then
+ echo -e "${test_name} ${REDB}FAILED${CLR}.\n\n${diff}\n"
+ failed=1
+ num_failures=$((num_failures+1))
+ fi
+
+ diff=`diff -u ${expected_result_json} ${test_result_json}`
+ if [[ $? != 0 ]]; then
+ echo -e "${test_name} ${REDB}FAILED${CLR}.\n\n${diff}\n"
+ failed=1
+ num_failures=$((num_failures+1))
+ fi
+
+ if [[ $failed == 0 ]]; then
+ echo -e "${test_name} ${GREEN}passed${CLR}."
+ fi
+}
+
+function run_builtin_policy_test {
+ policy_name=$1 # The built-in policy name to use.
+ version=$2 # Version of OpenSSH to test with.
+ test_number=$3 # The test number to run.
+ server_options=$4 # The options to start the server with (i.e.: "-o option1,options2,...")
+ expected_exit_code=$5 # The expected exit code of ssh-audit.py.
+
+ server_exec="/openssh/sshd-${version} -D -f /etc/ssh/sshd_config-8.0p1_test1 ${server_options}"
+ test_result_stdout="${TEST_RESULT_DIR}/openssh_${version}_builtin_policy_${test_number}.txt"
+ test_result_json="${TEST_RESULT_DIR}/openssh_${version}_builtin_policy_${test_number}.json"
+ expected_result_stdout="test/docker/expected_results/openssh_${version}_builtin_policy_${test_number}.txt"
+ expected_result_json="test/docker/expected_results/openssh_${version}_builtin_policy_${test_number}.json"
+ test_name="OpenSSH ${version} built-in policy ${test_number}"
+
+ run_policy_test "${test_name}" "${server_exec}" "${policy_name}" "${test_result_stdout}" "${test_result_json}" "${expected_exit_code}"
+}
+
+
+function run_custom_policy_test {
+ config_number=$1 # The configuration number to use.
+ test_number=$2 # The policy test number to run.
+ expected_exit_code=$3 # The expected exit code of ssh-audit.py.
+
+ version=
+ config=
+ if [[ ${config_number} == 'config1' ]]; then
+ version='5.6p1'
+ config='sshd_config-5.6p1_test1'
+ elif [[ ${config_number} == 'config2' ]]; then
+ version='8.0p1'
+ config='sshd_config-8.0p1_test1'
+ elif [[ ${config_number} == 'config3' ]]; then
+ version='5.6p1'
+ config='sshd_config-5.6p1_test4'
+ fi
+
+ server_exec="/openssh/sshd-${version} -D -f /etc/ssh/${config}"
+ policy_path="test/docker/policies/policy_${test_number}.txt"
+ test_result_stdout="${TEST_RESULT_DIR}/openssh_${version}_custom_policy_${test_number}.txt"
+ test_result_json="${TEST_RESULT_DIR}/openssh_${version}_custom_policy_${test_number}.json"
+ expected_result_stdout="test/docker/expected_results/openssh_${version}_custom_policy_${test_number}.txt"
+ expected_result_json="test/docker/expected_results/openssh_${version}_custom_policy_${test_number}.json"
+ test_name="OpenSSH ${version} custom policy ${test_number}"
+
+ run_policy_test "${test_name}" "${server_exec}" "${policy_path}" "${test_result_stdout}" "${test_result_json}" "${expected_exit_code}"
+}
+
+
+function run_policy_test {
+ test_name=$1
+ server_exec=$2
+ policy_path=$3
+ test_result_stdout=$4
+ test_result_json=$5
+ expected_exit_code=$6
+
+
+ #echo "Running: docker run -d -p 2222:22 ${IMAGE_NAME}:${IMAGE_VERSION} ${server_exec}"
+ cid=`docker run -d -p 2222:22 ${IMAGE_NAME}:${IMAGE_VERSION} ${server_exec}`
+ if [[ $? != 0 ]]; then
+ echo -e "${REDB}Failed to run docker image! (exit code: $?)${CLR}"
+ exit 1
+ fi
+
+ #echo "Running: ./ssh-audit.py -P \"${policy_path}\" localhost:2222 > ${test_result_stdout}"
+ ./ssh-audit.py -P "${policy_path}" localhost:2222 > ${test_result_stdout}
+ actual_exit_code=$?
+ if [[ ${actual_exit_code} != ${expected_exit_code} ]]; then
+ echo -e "${test_name} ${REDB}FAILED${CLR} (expected exit code: ${expected_exit_code}; actual exit code: ${actual_exit_code}\n"
+ cat ${test_result_stdout}
+ docker container stop -t 0 $cid > /dev/null
+ exit 1
+ fi
+
+ #echo "Running: ./ssh-audit.py -P \"${policy_path}\" -j localhost:2222 > ${test_result_json}"
+ ./ssh-audit.py -P "${policy_path}" -j localhost:2222 > ${test_result_json}
+ actual_exit_code=$?
+ if [[ ${actual_exit_code} != ${expected_exit_code} ]]; then
+ echo -e "${test_name} ${REDB}FAILED${CLR} (expected exit code: ${expected_exit_code}; actual exit code: ${actual_exit_code}\n"
+ cat ${test_result_json}
+ docker container stop -t 0 $cid > /dev/null
+ exit 1
+ fi
+
+ docker container stop -t 0 $cid > /dev/null
+ if [[ $? != 0 ]]; then
+ echo -e "${REDB}Failed to stop docker container ${cid}! (exit code: $?)${CLR}"
+ exit 1
+ fi
+
+ diff=`diff -u ${expected_result_stdout} ${test_result_stdout}`
+ if [[ $? != 0 ]]; then
+ echo -e "${test_name} ${REDB}FAILED${CLR}.\n\n${diff}\n"
+ exit 1
+ fi
+
+ diff=`diff -u ${expected_result_json} ${test_result_json}`
+ if [[ $? != 0 ]]; then
+ echo -e "${test_name} ${REDB}FAILED${CLR}.\n\n${diff}\n"
+ exit 1
+ fi
+
+ echo -e "${test_name} ${GREEN}passed${CLR}."
+}
+
+
+# First check if docker is functional.
+docker version > /dev/null
+if [[ $? != 0 ]]; then
+ echo -e "${REDB}Error: 'docker version' command failed (error code: $?). Is docker installed and functioning?${CLR}"
+ exit 1
+fi
+
+
+# Check if the docker image is the most up-to-date version.
+docker_image_exists=0
+check_if_docker_image_exists
+if [[ $? == 0 ]]; then
+ docker_image_exists=1
+fi
+
+
+# Check if the user specified --create to build a new image.
+if [[ ($# == 1) && ($1 == "--create") ]]; then
+ # Ensure that the image name doesn't already exist before building.
+ if [[ $docker_image_exists == 1 ]]; then
+ echo -e "${REDB}Error: --create specified, but $IMAGE_NAME:$IMAGE_VERSION already exists!${CLR}"
+ exit 1
+ else
+ echo -e "\nCreating docker image $IMAGE_NAME:$IMAGE_VERSION..."
+ create_docker_image
+ echo -e "\n${GREEN}Done creating docker image!${CLR}"
+ exit 0
+ fi
+fi
+
+
+# If we weren't explicitly told to create a new image, and it doesn't exist, then pull it from Dockerhub.
+if [[ $docker_image_exists == 0 ]]; then
+ echo -e "\nPulling docker image $IMAGE_NAME:$IMAGE_VERSION..."
+ pull_docker_image
+fi
+
+
+echo -e "\n${GREEN}Starting tests...${CLR}"
+
+# Create a temporary directory to write test results to.
+TEST_RESULT_DIR=`mktemp -d /tmp/ssh-audit_test-results_XXXXXXXXXX`
+
+# Now run all the tests.
+echo -e "\nRunning tests..."
+run_openssh_test '4.0p1' 'test1' $PROGRAM_RETVAL_FAILURE
+echo
+run_openssh_test '5.6p1' 'test1' $PROGRAM_RETVAL_FAILURE
+run_openssh_test '5.6p1' 'test2' $PROGRAM_RETVAL_FAILURE
+run_openssh_test '5.6p1' 'test3' $PROGRAM_RETVAL_FAILURE
+run_openssh_test '5.6p1' 'test4' $PROGRAM_RETVAL_FAILURE
+run_openssh_test '5.6p1' 'test5' $PROGRAM_RETVAL_FAILURE
+echo
+run_openssh_test '8.0p1' 'test1' $PROGRAM_RETVAL_FAILURE
+run_openssh_test '8.0p1' 'test2' $PROGRAM_RETVAL_FAILURE
+run_openssh_test '8.0p1' 'test3' $PROGRAM_RETVAL_GOOD
+echo
+run_dropbear_test '2019.78' 'test1' '-r /etc/dropbear/dropbear_rsa_host_key_1024 -r /etc/dropbear/dropbear_dss_host_key -r /etc/dropbear/dropbear_ecdsa_host_key' 3
+echo
+run_tinyssh_test '20190101' 'test1' $PROGRAM_RETVAL_WARNING
+echo
+echo
+run_custom_policy_test 'config1' 'test1' $PROGRAM_RETVAL_GOOD
+run_custom_policy_test 'config1' 'test2' $PROGRAM_RETVAL_FAILURE
+run_custom_policy_test 'config1' 'test3' $PROGRAM_RETVAL_FAILURE
+run_custom_policy_test 'config1' 'test4' $PROGRAM_RETVAL_FAILURE
+run_custom_policy_test 'config1' 'test5' $PROGRAM_RETVAL_FAILURE
+run_custom_policy_test 'config2' 'test6' $PROGRAM_RETVAL_GOOD
+
+# Passing test with host key certificate and CA key certificates.
+run_custom_policy_test 'config3' 'test7' $PROGRAM_RETVAL_GOOD
+
+# Failing test with host key certificate and non-compliant CA key length.
+run_custom_policy_test 'config3' 'test8' $PROGRAM_RETVAL_FAILURE
+
+# Failing test with non-compliant host key certificate and CA key certificate.
+run_custom_policy_test 'config3' 'test9' $PROGRAM_RETVAL_FAILURE
+
+# Failing test with non-compliant host key certificate and non-compliant CA key certificate.
+run_custom_policy_test 'config3' 'test10' $PROGRAM_RETVAL_FAILURE
+
+# Passing test with host key size check.
+run_custom_policy_test 'config2' 'test11' $PROGRAM_RETVAL_GOOD
+
+# Failing test with non-compliant host key size check.
+run_custom_policy_test 'config2' 'test12' $PROGRAM_RETVAL_FAILURE
+
+# Passing test with DH modulus test.
+run_custom_policy_test 'config2' 'test13' $PROGRAM_RETVAL_GOOD
+
+# Failing test with DH modulus test.
+run_custom_policy_test 'config2' 'test14' $PROGRAM_RETVAL_FAILURE
+
+# Passing test for built-in OpenSSH 8.0p1 server policy.
+run_builtin_policy_test "Hardened OpenSSH Server v8.0 (version 1)" "8.0p1" "test1" "-o HostKeyAlgorithms=rsa-sha2-512,rsa-sha2-256,ssh-ed25519 -o KexAlgorithms=curve25519-sha256,curve25519-sha256@libssh.org,diffie-hellman-group16-sha512,diffie-hellman-group18-sha512,diffie-hellman-group-exchange-sha256 -o Ciphers=chacha20-poly1305@openssh.com,aes256-gcm@openssh.com,aes128-gcm@openssh.com,aes256-ctr,aes192-ctr,aes128-ctr -o MACs=hmac-sha2-256-etm@openssh.com,hmac-sha2-512-etm@openssh.com,umac-128-etm@openssh.com" $PROGRAM_RETVAL_GOOD
+
+# Failing test for built-in OpenSSH 8.0p1 server policy (MACs not hardened).
+run_builtin_policy_test "Hardened OpenSSH Server v8.0 (version 1)" "8.0p1" "test2" "-o HostKeyAlgorithms=rsa-sha2-512,rsa-sha2-256,ssh-ed25519 -o KexAlgorithms=curve25519-sha256,curve25519-sha256@libssh.org,diffie-hellman-group16-sha512,diffie-hellman-group18-sha512,diffie-hellman-group-exchange-sha256 -o Ciphers=chacha20-poly1305@openssh.com,aes256-gcm@openssh.com,aes128-gcm@openssh.com,aes256-ctr,aes192-ctr,aes128-ctr" $PROGRAM_RETVAL_FAILURE
+
+
+if [[ $num_failures == 0 ]]; then
+ echo -e "\n${GREENB}ALL TESTS PASS!${CLR}\n"
+ rm -rf $TEST_RESULT_DIR
+else
+ echo -e "\n${REDB}${num_failures} TESTS FAILED!${CLR}\n"
+fi
+
+exit 0
diff --git a/pyproject.toml b/pyproject.toml
new file mode 100644
index 0000000..8dc9882
--- /dev/null
+++ b/pyproject.toml
@@ -0,0 +1,7 @@
+[build-system]
+# https://pip.pypa.io/en/stable/reference/pip/#pep-517-and-518-support
+requires = [
+ "setuptools>=40.8.0",
+ "wheel"
+]
+build-backend = "setuptools.build_meta"
diff --git a/setup.cfg b/setup.cfg
new file mode 100644
index 0000000..5d44bb6
--- /dev/null
+++ b/setup.cfg
@@ -0,0 +1,41 @@
+[metadata]
+name = ssh-audit
+version = attr: ssh_audit.globals.VERSION
+author = Joe Testa
+author_email = jtesta@positronsecurity.com
+description = An SSH server & client configuration security auditing tool
+long_description = file: README.md
+long_description_content_type = text/markdown
+license_file = LICENSE
+url = https://github.com/jtesta/ssh-audit
+project_urls =
+ Source Code = https://github.com/jtesta/ssh-audit
+ Bug Tracker = https://github.com/jtesta/ssh-audit/issues
+classifiers =
+ Development Status :: 5 - Production/Stable
+ Intended Audience :: Information Technology
+ Intended Audience :: System Administrators
+ License :: OSI Approved :: MIT License
+ Operating System :: OS Independent
+ Programming Language :: Python :: 3
+ Programming Language :: Python :: 3.6
+ Programming Language :: Python :: 3.7
+ Programming Language :: Python :: 3.8
+ Programming Language :: Python :: 3.9
+ Programming Language :: Python :: Implementation :: CPython
+ Programming Language :: Python :: Implementation :: PyPy
+ Topic :: Security
+ Topic :: Security :: Cryptography
+
+[options]
+packages = find:
+package_dir =
+ = src
+python_requires = >=3.6,<4
+
+[options.packages.find]
+where = src
+
+[options.entry_points]
+console_scripts =
+ ssh-audit = ssh_audit.ssh_audit:main
diff --git a/setup.py b/setup.py
new file mode 100644
index 0000000..f9d087a
--- /dev/null
+++ b/setup.py
@@ -0,0 +1,23 @@
+import re
+import sys
+from setuptools import setup
+
+print_warning = False
+m = re.search(r'^VERSION\s*=\s*\'v(\d\.\d\.\d)\'', open('src/ssh_audit/globals.py').read(), re.M)
+if m is None:
+ # If we failed to parse the stable version, see if this is the development version.
+ m = re.search(r'^VERSION\s*=\s*\'v(\d\.\d\.\d-dev)\'', open('src/ssh_audit/globals.py').read(), re.M)
+ if m is None:
+ print("Error: could not parse VERSION variable from ssh_audit.py.")
+ sys.exit(1)
+ else: # Continue with the development version, but print a warning later.
+ print_warning = True
+
+version = m.group(1)
+print("\n\nPackaging ssh-audit v%s...\n\n" % version)
+
+# see setup.cfg
+setup()
+
+if print_warning:
+ print("\n\n !!! WARNING: development version detected (%s). Are you sure you want to package this version? Probably not...\n" % version)
diff --git a/snapcraft.yaml b/snapcraft.yaml
new file mode 100644
index 0000000..3eb39a6
--- /dev/null
+++ b/snapcraft.yaml
@@ -0,0 +1,21 @@
+name: ssh-audit
+version: '2.4.0-1'
+license: 'MIT'
+summary: ssh-audit
+description: |
+ SSH server and client security configuration auditor. Official repository: <https://github.com/jtesta/ssh-audit>
+
+base: core20
+grade: stable
+confinement: strict
+
+apps:
+ ssh-audit:
+ command: bin/ssh-audit
+ plugs: [network,network-bind]
+
+parts:
+ ssh-audit:
+ plugin: python
+ # python-version: python3
+ source: .
diff --git a/src/ssh_audit/__init__.py b/src/ssh_audit/__init__.py
new file mode 100644
index 0000000..e69de29
--- /dev/null
+++ b/src/ssh_audit/__init__.py
diff --git a/src/ssh_audit/__main__.py b/src/ssh_audit/__main__.py
new file mode 100644
index 0000000..9ae64fa
--- /dev/null
+++ b/src/ssh_audit/__main__.py
@@ -0,0 +1,16 @@
+import sys
+import traceback
+
+from ssh_audit.ssh_audit import main
+from ssh_audit import exitcodes
+
+
+exit_code = exitcodes.GOOD
+
+try:
+ exit_code = main()
+except Exception:
+ exit_code = exitcodes.UNKNOWN_ERROR
+ print(traceback.format_exc())
+
+sys.exit(exit_code)
diff --git a/src/ssh_audit/algorithm.py b/src/ssh_audit/algorithm.py
new file mode 100644
index 0000000..e213264
--- /dev/null
+++ b/src/ssh_audit/algorithm.py
@@ -0,0 +1,61 @@
+"""
+ The MIT License (MIT)
+
+ Copyright (C) 2017 Andris Raugulis (moo@arthepsy.eu)
+
+ Permission is hereby granted, free of charge, to any person obtaining a copy
+ of this software and associated documentation files (the "Software"), to deal
+ in the Software without restriction, including without limitation the rights
+ to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+ copies of the Software, and to permit persons to whom the Software is
+ furnished to do so, subject to the following conditions:
+
+ The above copyright notice and this permission notice shall be included in
+ all copies or substantial portions of the Software.
+
+ THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+ IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+ FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+ AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+ LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+ OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+ THE SOFTWARE.
+"""
+# pylint: disable=unused-import
+from typing import Dict, List, Set, Sequence, Tuple, Iterable # noqa: F401
+from typing import Callable, Optional, Union, Any # noqa: F401
+
+from ssh_audit.product import Product
+
+
+class Algorithm:
+
+ @staticmethod
+ def get_ssh_version(version_desc: str) -> Tuple[str, str, bool]:
+ is_client = version_desc.endswith('C')
+ if is_client:
+ version_desc = version_desc[:-1]
+ if version_desc.startswith('d'):
+ return Product.DropbearSSH, version_desc[1:], is_client
+ elif version_desc.startswith('l1'):
+ return Product.LibSSH, version_desc[2:], is_client
+ else:
+ return Product.OpenSSH, version_desc, is_client
+
+ @classmethod
+ def get_since_text(cls, versions: List[Optional[str]]) -> Optional[str]:
+ tv = []
+ if len(versions) == 0 or versions[0] is None:
+ return None
+ for v in versions[0].split(','):
+ ssh_prod, ssh_ver, is_cli = cls.get_ssh_version(v)
+ if not ssh_ver:
+ continue
+ if ssh_prod in [Product.LibSSH]:
+ continue
+ if is_cli:
+ ssh_ver = '{} (client only)'.format(ssh_ver)
+ tv.append('{} {}'.format(ssh_prod, ssh_ver))
+ if len(tv) == 0:
+ return None
+ return 'available since ' + ', '.join(tv).rstrip(', ')
diff --git a/src/ssh_audit/algorithms.py b/src/ssh_audit/algorithms.py
new file mode 100644
index 0000000..7c9a850
--- /dev/null
+++ b/src/ssh_audit/algorithms.py
@@ -0,0 +1,223 @@
+"""
+ The MIT License (MIT)
+
+ Copyright (C) 2017-2021 Joe Testa (jtesta@positronsecurity.com)
+ Copyright (C) 2017 Andris Raugulis (moo@arthepsy.eu)
+
+ Permission is hereby granted, free of charge, to any person obtaining a copy
+ of this software and associated documentation files (the "Software"), to deal
+ in the Software without restriction, including without limitation the rights
+ to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+ copies of the Software, and to permit persons to whom the Software is
+ furnished to do so, subject to the following conditions:
+
+ The above copyright notice and this permission notice shall be included in
+ all copies or substantial portions of the Software.
+
+ THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+ IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+ FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+ AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+ LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+ OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+ THE SOFTWARE.
+"""
+# pylint: disable=unused-import
+from typing import Dict, List, Set, Sequence, Tuple, Iterable # noqa: F401
+from typing import Callable, Optional, Union, Any # noqa: F401
+
+from ssh_audit.algorithm import Algorithm
+from ssh_audit.product import Product
+from ssh_audit.software import Software
+from ssh_audit.ssh1_kexdb import SSH1_KexDB
+from ssh_audit.ssh1_publickeymessage import SSH1_PublicKeyMessage
+from ssh_audit.ssh2_kex import SSH2_Kex
+from ssh_audit.ssh2_kexdb import SSH2_KexDB
+from ssh_audit.timeframe import Timeframe
+from ssh_audit.utils import Utils
+
+
+class Algorithms:
+ def __init__(self, pkm: Optional[SSH1_PublicKeyMessage], kex: Optional[SSH2_Kex]) -> None:
+ self.__ssh1kex = pkm
+ self.__ssh2kex = kex
+
+ @property
+ def ssh1kex(self) -> Optional[SSH1_PublicKeyMessage]:
+ return self.__ssh1kex
+
+ @property
+ def ssh2kex(self) -> Optional[SSH2_Kex]:
+ return self.__ssh2kex
+
+ @property
+ def ssh1(self) -> Optional['Algorithms.Item']:
+ if self.ssh1kex is None:
+ return None
+ item = Algorithms.Item(1, SSH1_KexDB.ALGORITHMS)
+ item.add('key', ['ssh-rsa1'])
+ item.add('enc', self.ssh1kex.supported_ciphers)
+ item.add('aut', self.ssh1kex.supported_authentications)
+ return item
+
+ @property
+ def ssh2(self) -> Optional['Algorithms.Item']:
+ if self.ssh2kex is None:
+ return None
+ item = Algorithms.Item(2, SSH2_KexDB.ALGORITHMS)
+ item.add('kex', self.ssh2kex.kex_algorithms)
+ item.add('key', self.ssh2kex.key_algorithms)
+ item.add('enc', self.ssh2kex.server.encryption)
+ item.add('mac', self.ssh2kex.server.mac)
+ return item
+
+ @property
+ def values(self) -> Iterable['Algorithms.Item']:
+ for item in [self.ssh1, self.ssh2]:
+ if item is not None:
+ yield item
+
+ @property
+ def maxlen(self) -> int:
+ def _ml(items: Sequence[str]) -> int:
+ return max(len(i) for i in items)
+ maxlen = 0
+ if self.ssh1kex is not None:
+ maxlen = max(_ml(self.ssh1kex.supported_ciphers),
+ _ml(self.ssh1kex.supported_authentications),
+ maxlen)
+ if self.ssh2kex is not None:
+ maxlen = max(_ml(self.ssh2kex.kex_algorithms),
+ _ml(self.ssh2kex.key_algorithms),
+ _ml(self.ssh2kex.server.encryption),
+ _ml(self.ssh2kex.server.mac),
+ maxlen)
+ return maxlen
+
+ def get_ssh_timeframe(self, for_server: Optional[bool] = None) -> 'Timeframe':
+ timeframe = Timeframe()
+ for alg_pair in self.values:
+ alg_db = alg_pair.db
+ for alg_type, alg_list in alg_pair.items():
+ for alg_name in alg_list:
+ alg_name_native = Utils.to_text(alg_name)
+ alg_desc = alg_db[alg_type].get(alg_name_native)
+ if alg_desc is None:
+ continue
+ versions = alg_desc[0]
+ timeframe.update(versions, for_server)
+ return timeframe
+
+ def get_recommendations(self, software: Optional['Software'], for_server: bool = True) -> Tuple[Optional['Software'], Dict[int, Dict[str, Dict[str, Dict[str, int]]]]]:
+ # pylint: disable=too-many-locals,too-many-statements
+ vproducts = [Product.OpenSSH,
+ Product.DropbearSSH,
+ Product.LibSSH,
+ Product.TinySSH]
+ # Set to True if server is not one of vproducts, above.
+ unknown_software = False
+ if software is not None:
+ if software.product not in vproducts:
+ unknown_software = True
+
+ # The code below is commented out because it would try to guess what the server is,
+ # usually resulting in wild & incorrect recommendations.
+ # if software is None:
+ # ssh_timeframe = self.get_ssh_timeframe(for_server)
+ # for product in vproducts:
+ # if product not in ssh_timeframe:
+ # continue
+ # version = ssh_timeframe.get_from(product, for_server)
+ # if version is not None:
+ # software = SSH.Software(None, product, version, None, None)
+ # break
+ rec: Dict[int, Dict[str, Dict[str, Dict[str, int]]]] = {}
+ if software is None:
+ unknown_software = True
+ for alg_pair in self.values:
+ sshv, alg_db = alg_pair.sshv, alg_pair.db
+ rec[sshv] = {}
+ for alg_type, alg_list in alg_pair.items():
+ if alg_type == 'aut':
+ continue
+ rec[sshv][alg_type] = {'add': {}, 'del': {}, 'chg': {}}
+ for n, alg_desc in alg_db[alg_type].items():
+ versions = alg_desc[0]
+ empty_version = False
+ if len(versions) == 0 or versions[0] is None:
+ empty_version = True
+ else:
+ matches = False
+ if unknown_software:
+ matches = True
+ for v in versions[0].split(','):
+ ssh_prefix, ssh_version, is_cli = Algorithm.get_ssh_version(v)
+ if not ssh_version:
+ continue
+ if (software is not None) and (ssh_prefix != software.product):
+ continue
+ if is_cli and for_server:
+ continue
+ if (software is not None) and (software.compare_version(ssh_version) < 0):
+ continue
+ matches = True
+ break
+ if not matches:
+ continue
+ adl, faults = len(alg_desc), 0
+ for i in range(1, 3):
+ if not adl > i:
+ continue
+ fc = len(alg_desc[i])
+ if fc > 0:
+ faults += pow(10, 2 - i) * fc
+ if n not in alg_list:
+ # Don't recommend certificate or token types; these will only appear in the server's list if they are fully configured & functional on the server.
+ if faults > 0 or (alg_type == 'key' and (('-cert-' in n) or (n.startswith('sk-')))) or empty_version:
+ continue
+ rec[sshv][alg_type]['add'][n] = 0
+ else:
+ if faults == 0:
+ continue
+ if n in ['diffie-hellman-group-exchange-sha256', 'rsa-sha2-256', 'rsa-sha2-512', 'rsa-sha2-256-cert-v01@openssh.com', 'rsa-sha2-512-cert-v01@openssh.com']:
+ rec[sshv][alg_type]['chg'][n] = faults
+ else:
+ rec[sshv][alg_type]['del'][n] = faults
+ # If we are working with unknown software, drop all add recommendations, because we don't know if they're valid.
+ if unknown_software:
+ rec[sshv][alg_type]['add'] = {}
+ add_count = len(rec[sshv][alg_type]['add'])
+ del_count = len(rec[sshv][alg_type]['del'])
+ chg_count = len(rec[sshv][alg_type]['chg'])
+
+ if add_count == 0:
+ del rec[sshv][alg_type]['add']
+ if del_count == 0:
+ del rec[sshv][alg_type]['del']
+ if chg_count == 0:
+ del rec[sshv][alg_type]['chg']
+ if len(rec[sshv][alg_type]) == 0:
+ del rec[sshv][alg_type]
+ if len(rec[sshv]) == 0:
+ del rec[sshv]
+ return software, rec
+
+ class Item:
+ def __init__(self, sshv: int, db: Dict[str, Dict[str, List[List[Optional[str]]]]]) -> None:
+ self.__sshv = sshv
+ self.__db = db
+ self.__storage: Dict[str, List[str]] = {}
+
+ @property
+ def sshv(self) -> int:
+ return self.__sshv
+
+ @property
+ def db(self) -> Dict[str, Dict[str, List[List[Optional[str]]]]]:
+ return self.__db
+
+ def add(self, key: str, value: List[str]) -> None:
+ self.__storage[key] = value
+
+ def items(self) -> Iterable[Tuple[str, List[str]]]:
+ return self.__storage.items()
diff --git a/src/ssh_audit/auditconf.py b/src/ssh_audit/auditconf.py
new file mode 100644
index 0000000..1d51240
--- /dev/null
+++ b/src/ssh_audit/auditconf.py
@@ -0,0 +1,98 @@
+"""
+ The MIT License (MIT)
+
+ Copyright (C) 2017-2021 Joe Testa (jtesta@positronsecurity.com)
+ Copyright (C) 2017 Andris Raugulis (moo@arthepsy.eu)
+
+ Permission is hereby granted, free of charge, to any person obtaining a copy
+ of this software and associated documentation files (the "Software"), to deal
+ in the Software without restriction, including without limitation the rights
+ to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+ copies of the Software, and to permit persons to whom the Software is
+ furnished to do so, subject to the following conditions:
+
+ The above copyright notice and this permission notice shall be included in
+ all copies or substantial portions of the Software.
+
+ THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+ IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+ FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+ AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+ LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+ OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+ THE SOFTWARE.
+"""
+# pylint: disable=unused-import
+from typing import Dict, List, Set, Sequence, Tuple, Iterable # noqa: F401
+from typing import Callable, Optional, Union, Any # noqa: F401
+
+from ssh_audit.policy import Policy
+from ssh_audit.utils import Utils
+
+
+class AuditConf:
+ # pylint: disable=too-many-instance-attributes
+ def __init__(self, host: str = '', port: int = 22) -> None:
+ self.host = host
+ self.port = port
+ self.ssh1 = True
+ self.ssh2 = True
+ self.batch = False
+ self.client_audit = False
+ self.colors = True
+ self.json = False
+ self.json_print_indent = False
+ self.verbose = False
+ self.level = 'info'
+ self.ip_version_preference: List[int] = [] # Holds only 5 possible values: [] (no preference), [4] (use IPv4 only), [6] (use IPv6 only), [46] (use both IPv4 and IPv6, but prioritize v4), and [64] (use both IPv4 and IPv6, but prioritize v6).
+ self.ipv4 = False
+ self.ipv6 = False
+ self.make_policy = False # When True, creates a policy file from an audit scan.
+ self.policy_file: Optional[str] = None # File system path to a policy
+ self.policy: Optional[Policy] = None # Policy object
+ self.timeout = 5.0
+ self.timeout_set = False # Set to True when the user explicitly sets it.
+ self.target_file: Optional[str] = None
+ self.target_list: List[str] = []
+ self.threads = 32
+ self.list_policies = False
+ self.lookup = ''
+ self.manual = False
+ self.debug = False
+
+ def __setattr__(self, name: str, value: Union[str, int, float, bool, Sequence[int]]) -> None:
+ valid = False
+ if name in ['batch', 'client_audit', 'colors', 'json', 'json_print_indent', 'list_policies', 'manual', 'make_policy', 'ssh1', 'ssh2', 'timeout_set', 'verbose', 'debug']:
+ valid, value = True, bool(value)
+ elif name in ['ipv4', 'ipv6']:
+ valid, value = True, bool(value)
+ if len(self.ip_version_preference) == 2: # Being called more than twice is not valid.
+ valid = False
+ elif value:
+ self.ip_version_preference.append(4 if name == 'ipv4' else 6)
+ elif name == 'port':
+ valid, port = True, Utils.parse_int(value)
+ if port < 1 or port > 65535:
+ raise ValueError('invalid port: {}'.format(value))
+ value = port
+ elif name in ['level']:
+ if value not in ('info', 'warn', 'fail'):
+ raise ValueError('invalid level: {}'.format(value))
+ valid = True
+ elif name == 'host':
+ valid = True
+ elif name == 'timeout':
+ value = Utils.parse_float(value)
+ if value == -1.0:
+ raise ValueError('invalid timeout: {}'.format(value))
+ valid = True
+ elif name in ['ip_version_preference', 'lookup', 'policy_file', 'policy', 'target_file', 'target_list']:
+ valid = True
+ elif name == "threads":
+ valid, num_threads = True, Utils.parse_int(value)
+ if num_threads < 1:
+ raise ValueError('invalid number of threads: {}'.format(value))
+ value = num_threads
+
+ if valid:
+ object.__setattr__(self, name, value)
diff --git a/src/ssh_audit/banner.py b/src/ssh_audit/banner.py
new file mode 100644
index 0000000..da6ee17
--- /dev/null
+++ b/src/ssh_audit/banner.py
@@ -0,0 +1,92 @@
+"""
+ The MIT License (MIT)
+
+ Copyright (C) 2017 Andris Raugulis (moo@arthepsy.eu)
+
+ Permission is hereby granted, free of charge, to any person obtaining a copy
+ of this software and associated documentation files (the "Software"), to deal
+ in the Software without restriction, including without limitation the rights
+ to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+ copies of the Software, and to permit persons to whom the Software is
+ furnished to do so, subject to the following conditions:
+
+ The above copyright notice and this permission notice shall be included in
+ all copies or substantial portions of the Software.
+
+ THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+ IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+ FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+ AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+ LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+ OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+ THE SOFTWARE.
+"""
+import re
+
+# pylint: disable=unused-import
+from typing import Dict, List, Set, Sequence, Tuple, Iterable # noqa: F401
+from typing import Callable, Optional, Union, Any # noqa: F401
+
+from ssh_audit.utils import Utils
+
+
+class Banner:
+ _RXP, _RXR = r'SSH-\d\.\s*?\d+', r'(-\s*([^\s]*)(?:\s+(.*))?)?'
+ RX_PROTOCOL = re.compile(re.sub(r'\\d(\+?)', r'(\\d\g<1>)', _RXP))
+ RX_BANNER = re.compile(r'^({0}(?:(?:-{0})*)){1}$'.format(_RXP, _RXR))
+
+ def __init__(self, protocol: Tuple[int, int], software: Optional[str], comments: Optional[str], valid_ascii: bool) -> None:
+ self.__protocol = protocol
+ self.__software = software
+ self.__comments = comments
+ self.__valid_ascii = valid_ascii
+
+ @property
+ def protocol(self) -> Tuple[int, int]:
+ return self.__protocol
+
+ @property
+ def software(self) -> Optional[str]:
+ return self.__software
+
+ @property
+ def comments(self) -> Optional[str]:
+ return self.__comments
+
+ @property
+ def valid_ascii(self) -> bool:
+ return self.__valid_ascii
+
+ def __str__(self) -> str:
+ r = 'SSH-{}.{}'.format(self.protocol[0], self.protocol[1])
+ if self.software is not None:
+ r += '-{}'.format(self.software)
+ if bool(self.comments):
+ r += ' {}'.format(self.comments)
+ return r
+
+ def __repr__(self) -> str:
+ p = '{}.{}'.format(self.protocol[0], self.protocol[1])
+ r = 'protocol={}'.format(p)
+ if self.software is not None:
+ r += ', software={}'.format(self.software)
+ if bool(self.comments):
+ r += ', comments={}'.format(self.comments)
+ return '<{}({})>'.format(self.__class__.__name__, r)
+
+ @classmethod
+ def parse(cls, banner: str) -> Optional['Banner']:
+ valid_ascii = Utils.is_print_ascii(banner)
+ ascii_banner = Utils.to_print_ascii(banner)
+ mx = cls.RX_BANNER.match(ascii_banner)
+ if mx is None:
+ return None
+ protocol = min(re.findall(cls.RX_PROTOCOL, mx.group(1)))
+ protocol = (int(protocol[0]), int(protocol[1]))
+ software = (mx.group(3) or '').strip() or None
+ if software is None and (mx.group(2) or '').startswith('-'):
+ software = ''
+ comments = (mx.group(4) or '').strip() or None
+ if comments is not None:
+ comments = re.sub(r'\s+', ' ', comments)
+ return cls(protocol, software, comments, valid_ascii)
diff --git a/src/ssh_audit/exitcodes.py b/src/ssh_audit/exitcodes.py
new file mode 100644
index 0000000..5392f1d
--- /dev/null
+++ b/src/ssh_audit/exitcodes.py
@@ -0,0 +1,6 @@
+# The program return values corresponding to failure(s) encountered, warning(s) encountered, connection errors, and no problems found, respectively.
+FAILURE = 3
+WARNING = 2
+CONNECTION_ERROR = 1
+GOOD = 0
+UNKNOWN_ERROR = -1
diff --git a/src/ssh_audit/fingerprint.py b/src/ssh_audit/fingerprint.py
new file mode 100644
index 0000000..a9d2caf
--- /dev/null
+++ b/src/ssh_audit/fingerprint.py
@@ -0,0 +1,43 @@
+"""
+ The MIT License (MIT)
+
+ Copyright (C) 2017-2021 Joe Testa (jtesta@positronsecurity.com)
+ Copyright (C) 2017 Andris Raugulis (moo@arthepsy.eu)
+
+ Permission is hereby granted, free of charge, to any person obtaining a copy
+ of this software and associated documentation files (the "Software"), to deal
+ in the Software without restriction, including without limitation the rights
+ to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+ copies of the Software, and to permit persons to whom the Software is
+ furnished to do so, subject to the following conditions:
+
+ The above copyright notice and this permission notice shall be included in
+ all copies or substantial portions of the Software.
+
+ THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+ IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+ FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+ AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+ LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+ OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+ THE SOFTWARE.
+"""
+import base64
+import hashlib
+
+
+class Fingerprint:
+ def __init__(self, fpd: bytes) -> None:
+ self.__fpd = fpd
+
+ @property
+ def md5(self) -> str:
+ h = hashlib.md5(self.__fpd).hexdigest()
+ r = ':'.join(h[i:i + 2] for i in range(0, len(h), 2))
+ return 'MD5:{}'.format(r)
+
+ @property
+ def sha256(self) -> str:
+ h = base64.b64encode(hashlib.sha256(self.__fpd).digest())
+ r = h.decode('ascii').rstrip('=')
+ return 'SHA256:{}'.format(r)
diff --git a/src/ssh_audit/gextest.py b/src/ssh_audit/gextest.py
new file mode 100644
index 0000000..f82a856
--- /dev/null
+++ b/src/ssh_audit/gextest.py
@@ -0,0 +1,162 @@
+"""
+ The MIT License (MIT)
+
+ Copyright (C) 2017-2021 Joe Testa (jtesta@positronsecurity.com)
+
+ Permission is hereby granted, free of charge, to any person obtaining a copy
+ of this software and associated documentation files (the "Software"), to deal
+ in the Software without restriction, including without limitation the rights
+ to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+ copies of the Software, and to permit persons to whom the Software is
+ furnished to do so, subject to the following conditions:
+
+ The above copyright notice and this permission notice shall be included in
+ all copies or substantial portions of the Software.
+
+ THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+ IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+ FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+ AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+ LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+ OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+ THE SOFTWARE.
+"""
+
+# pylint: disable=unused-import
+from typing import Dict, List, Set, Sequence, Tuple, Iterable # noqa: F401
+from typing import Callable, Optional, Union, Any # noqa: F401
+
+import traceback
+
+from ssh_audit.kexdh import KexGroupExchange_SHA1, KexGroupExchange_SHA256
+from ssh_audit.ssh2_kexdb import SSH2_KexDB
+from ssh_audit.ssh2_kex import SSH2_Kex
+from ssh_audit.ssh_socket import SSH_Socket
+from ssh_audit.outputbuffer import OutputBuffer
+
+
+# Performs DH group exchanges to find what moduli are supported, and checks
+# their size.
+class GEXTest:
+
+ # Creates a new connection to the server. Returns True on success, or False.
+ @staticmethod
+ def reconnect(out: 'OutputBuffer', s: 'SSH_Socket', kex: 'SSH2_Kex', gex_alg: str) -> bool:
+ if s.is_connected():
+ return True
+
+ err = s.connect()
+ if err is not None:
+ out.v(err, write_now=True)
+ return False
+
+ _, _, err = s.get_banner()
+ if err is not None:
+ out.v(err, write_now=True)
+ s.close()
+ return False
+
+ # Send our KEX using the specified group-exchange and most of the
+ # server's own values.
+ s.send_kexinit(key_exchanges=[gex_alg], hostkeys=kex.key_algorithms, ciphers=kex.server.encryption, macs=kex.server.mac, compressions=kex.server.compression, languages=kex.server.languages)
+
+ try:
+ # Parse the server's KEX.
+ _, payload = s.read_packet(2)
+ SSH2_Kex.parse(payload)
+ except Exception:
+ out.v("Failed to parse server's kex. Stack trace:\n%s" % str(traceback.format_exc()), write_now=True)
+ return False
+
+ return True
+
+ # Runs the DH moduli test against the specified target.
+ @staticmethod
+ def run(out: 'OutputBuffer', s: 'SSH_Socket', kex: 'SSH2_Kex') -> None:
+ GEX_ALGS = {
+ 'diffie-hellman-group-exchange-sha1': KexGroupExchange_SHA1,
+ 'diffie-hellman-group-exchange-sha256': KexGroupExchange_SHA256,
+ }
+
+ # The previous RSA tests put the server in a state we can't
+ # test. So we need a new connection to start with a clean
+ # slate.
+ if s.is_connected():
+ s.close()
+
+ # Check if the server supports any of the group-exchange
+ # algorithms. If so, test each one.
+ for gex_alg, kex_group_class in GEX_ALGS.items():
+ if gex_alg in kex.kex_algorithms:
+ out.d('Preparing to perform DH group exchange using ' + gex_alg + '...', write_now=True)
+
+ if GEXTest.reconnect(out, s, kex, gex_alg) is False:
+ break
+
+ kex_group = kex_group_class()
+ smallest_modulus = -1
+
+ # First try a range of weak sizes.
+ try:
+ kex_group.send_init_gex(s, 512, 1024, 1536)
+ kex_group.recv_reply(s, False)
+
+ # Its been observed that servers will return a group
+ # larger than the requested max. So just because we
+ # got here, doesn't mean the server is vulnerable...
+ smallest_modulus = kex_group.get_dh_modulus_size()
+
+ except Exception:
+ pass
+ finally:
+ s.close()
+
+ # Try an array of specific modulus sizes... one at a time.
+ reconnect_failed = False
+ for bits in [512, 768, 1024, 1536, 2048, 3072, 4096]:
+
+ # If we found one modulus size already, but we're about
+ # to test a larger one, don't bother.
+ if bits >= smallest_modulus > 0:
+ break
+
+ out.d('Preparing to perform DH group exchange using ' + gex_alg + ' with modulus size ' + str(bits) + '...', write_now=True)
+
+ if GEXTest.reconnect(out, s, kex, gex_alg) is False:
+ reconnect_failed = True
+ break
+
+ try:
+ kex_group.send_init_gex(s, bits, bits, bits)
+ kex_group.recv_reply(s, False)
+ smallest_modulus = kex_group.get_dh_modulus_size()
+ except Exception:
+ # import traceback
+ # print(traceback.format_exc())
+ pass
+ finally:
+ # The server is in a state that is not re-testable,
+ # so there's nothing else to do with this open
+ # connection.
+ s.close()
+
+ if smallest_modulus > 0:
+ kex.set_dh_modulus_size(gex_alg, smallest_modulus)
+
+ # We flag moduli smaller than 2048 as a failure.
+ if smallest_modulus < 2048:
+ text = 'using small %d-bit modulus' % smallest_modulus
+ lst = SSH2_KexDB.ALGORITHMS['kex'][gex_alg]
+ # For 'diffie-hellman-group-exchange-sha256', add
+ # a failure reason.
+ if len(lst) == 1:
+ lst.append([text])
+ # For 'diffie-hellman-group-exchange-sha1', delete
+ # the existing failure reason (which is vague), and
+ # insert our own.
+ else:
+ del lst[1]
+ lst.insert(1, [text])
+
+ if reconnect_failed:
+ break
diff --git a/src/ssh_audit/globals.py b/src/ssh_audit/globals.py
new file mode 100644
index 0000000..c15359d
--- /dev/null
+++ b/src/ssh_audit/globals.py
@@ -0,0 +1,27 @@
+"""
+ The MIT License (MIT)
+
+ Copyright (C) 2017-2021 Joe Testa (jtesta@positronsecurity.com)
+
+ Permission is hereby granted, free of charge, to any person obtaining a copy
+ of this software and associated documentation files (the "Software"), to deal
+ in the Software without restriction, including without limitation the rights
+ to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+ copies of the Software, and to permit persons to whom the Software is
+ furnished to do so, subject to the following conditions:
+
+ The above copyright notice and this permission notice shall be included in
+ all copies or substantial portions of the Software.
+
+ THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+ IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+ FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+ AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+ LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+ OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+ THE SOFTWARE.
+"""
+VERSION = 'v2.5.0'
+SSH_HEADER = 'SSH-{0}-OpenSSH_8.2' # SSH software to impersonate
+GITHUB_ISSUES_URL = 'https://github.com/jtesta/ssh-audit/issues' # The URL to the Github issues tracker.
+WINDOWS_MAN_PAGE = ''
diff --git a/src/ssh_audit/hostkeytest.py b/src/ssh_audit/hostkeytest.py
new file mode 100644
index 0000000..6f87ac5
--- /dev/null
+++ b/src/ssh_audit/hostkeytest.py
@@ -0,0 +1,188 @@
+"""
+ The MIT License (MIT)
+
+ Copyright (C) 2017-2021 Joe Testa (jtesta@positronsecurity.com)
+
+ Permission is hereby granted, free of charge, to any person obtaining a copy
+ of this software and associated documentation files (the "Software"), to deal
+ in the Software without restriction, including without limitation the rights
+ to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+ copies of the Software, and to permit persons to whom the Software is
+ furnished to do so, subject to the following conditions:
+
+ The above copyright notice and this permission notice shall be included in
+ all copies or substantial portions of the Software.
+
+ THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+ IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+ FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+ AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+ LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+ OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+ THE SOFTWARE.
+"""
+
+# pylint: disable=unused-import
+from typing import Dict, List, Set, Sequence, Tuple, Iterable # noqa: F401
+from typing import Callable, Optional, Union, Any # noqa: F401
+
+import traceback
+
+from ssh_audit.kexdh import KexDH, KexGroup1, KexGroup14_SHA1, KexGroup14_SHA256, KexCurve25519_SHA256, KexGroup16_SHA512, KexGroup18_SHA512, KexGroupExchange_SHA1, KexGroupExchange_SHA256, KexNISTP256, KexNISTP384, KexNISTP521
+from ssh_audit.ssh2_kex import SSH2_Kex
+from ssh_audit.ssh2_kexdb import SSH2_KexDB
+from ssh_audit.ssh_socket import SSH_Socket
+from ssh_audit.outputbuffer import OutputBuffer
+
+
+# Obtains host keys, checks their size, and derives their fingerprints.
+class HostKeyTest:
+ # Tracks the RSA host key types. As of this writing, testing one in this family yields valid results for the rest.
+ RSA_FAMILY = ['ssh-rsa', 'rsa-sha2-256', 'rsa-sha2-512']
+
+ # Dict holding the host key types we should extract & parse. 'cert' is True to denote that a host key type handles certificates (thus requires additional parsing). 'variable_key_len' is True for host key types that can have variable sizes (True only for RSA types, as the rest are of fixed-size). After the host key type is fully parsed, the key 'parsed' is added with a value of True.
+ HOST_KEY_TYPES = {
+ 'ssh-rsa': {'cert': False, 'variable_key_len': True},
+ 'rsa-sha2-256': {'cert': False, 'variable_key_len': True},
+ 'rsa-sha2-512': {'cert': False, 'variable_key_len': True},
+
+ 'ssh-rsa-cert-v01@openssh.com': {'cert': True, 'variable_key_len': True},
+ 'rsa-sha2-256-cert-v01@openssh.com': {'cert': True, 'variable_key_len': True},
+ 'rsa-sha2-512-cert-v01@openssh.com': {'cert': True, 'variable_key_len': True},
+
+ 'ssh-ed25519': {'cert': False, 'variable_key_len': False},
+ 'ssh-ed25519-cert-v01@openssh.com': {'cert': True, 'variable_key_len': False},
+ }
+
+ @staticmethod
+ def run(out: 'OutputBuffer', s: 'SSH_Socket', server_kex: 'SSH2_Kex') -> None:
+ KEX_TO_DHGROUP = {
+ 'diffie-hellman-group1-sha1': KexGroup1,
+ 'diffie-hellman-group14-sha1': KexGroup14_SHA1,
+ 'diffie-hellman-group14-sha256': KexGroup14_SHA256,
+ 'curve25519-sha256': KexCurve25519_SHA256,
+ 'curve25519-sha256@libssh.org': KexCurve25519_SHA256,
+ 'diffie-hellman-group16-sha512': KexGroup16_SHA512,
+ 'diffie-hellman-group18-sha512': KexGroup18_SHA512,
+ 'diffie-hellman-group-exchange-sha1': KexGroupExchange_SHA1,
+ 'diffie-hellman-group-exchange-sha256': KexGroupExchange_SHA256,
+ 'ecdh-sha2-nistp256': KexNISTP256,
+ 'ecdh-sha2-nistp384': KexNISTP384,
+ 'ecdh-sha2-nistp521': KexNISTP521,
+ # 'kexguess2@matt.ucc.asn.au': ???
+ }
+
+ # Pick the first kex algorithm that the server supports, which we
+ # happen to support as well.
+ kex_str = None
+ kex_group = None
+ for server_kex_alg in server_kex.kex_algorithms:
+ if server_kex_alg in KEX_TO_DHGROUP:
+ kex_str = server_kex_alg
+ kex_group = KEX_TO_DHGROUP[kex_str]()
+ break
+
+ if kex_str is not None and kex_group is not None:
+ HostKeyTest.perform_test(out, s, server_kex, kex_str, kex_group, HostKeyTest.HOST_KEY_TYPES)
+
+ @staticmethod
+ def perform_test(out: 'OutputBuffer', s: 'SSH_Socket', server_kex: 'SSH2_Kex', kex_str: str, kex_group: 'KexDH', host_key_types: Dict[str, Dict[str, bool]]) -> None:
+ hostkey_modulus_size = 0
+ ca_modulus_size = 0
+
+ # If the connection still exists, close it so we can test
+ # using a clean slate (otherwise it may exist in a non-testable
+ # state).
+ if s.is_connected():
+ s.close()
+
+ # For each host key type...
+ for host_key_type in host_key_types:
+ # Skip those already handled (i.e.: those in the RSA family, as testing one tests them all).
+ if 'parsed' in host_key_types[host_key_type] and host_key_types[host_key_type]['parsed']:
+ continue
+
+ # If this host key type is supported by the server, we test it.
+ if host_key_type in server_kex.key_algorithms:
+ out.d('Preparing to obtain ' + host_key_type + ' host key...', write_now=True)
+
+ cert = host_key_types[host_key_type]['cert']
+ variable_key_len = host_key_types[host_key_type]['variable_key_len']
+
+ # If the connection is closed, re-open it and get the kex again.
+ if not s.is_connected():
+ err = s.connect()
+ if err is not None:
+ out.v(err, write_now=True)
+ return
+
+ _, _, err = s.get_banner()
+ if err is not None:
+ out.v(err, write_now=True)
+ s.close()
+ return
+
+ # Send our KEX using the specified group-exchange and most of the server's own values.
+ s.send_kexinit(key_exchanges=[kex_str], hostkeys=[host_key_type], ciphers=server_kex.server.encryption, macs=server_kex.server.mac, compressions=server_kex.server.compression, languages=server_kex.server.languages)
+
+ try:
+ # Parse the server's KEX.
+ _, payload = s.read_packet()
+ SSH2_Kex.parse(payload)
+ except Exception:
+ out.v("Failed to parse server's kex. Stack trace:\n%s" % str(traceback.format_exc()), write_now=True)
+ return
+
+ # Do the initial DH exchange. The server responds back
+ # with the host key and its length. Bingo. We also get back the host key fingerprint.
+ kex_group.send_init(s)
+ try:
+ host_key = kex_group.recv_reply(s, variable_key_len)
+ if host_key is not None:
+ server_kex.set_host_key(host_key_type, host_key)
+ except Exception:
+ pass
+
+ hostkey_modulus_size = kex_group.get_hostkey_size()
+ ca_modulus_size = kex_group.get_ca_size()
+
+ # Close the socket, as the connection has
+ # been put in a state that later tests can't use.
+ s.close()
+
+ # If the host key modulus or CA modulus was successfully parsed, check to see that its a safe size.
+ if hostkey_modulus_size > 0 or ca_modulus_size > 0:
+ # Set the hostkey size for all RSA key types since 'ssh-rsa',
+ # 'rsa-sha2-256', etc. are all using the same host key.
+ # Note, however, that this may change in the future.
+ if cert is False and host_key_type in HostKeyTest.RSA_FAMILY:
+ for rsa_type in HostKeyTest.RSA_FAMILY:
+ server_kex.set_rsa_key_size(rsa_type, hostkey_modulus_size)
+ elif cert is True:
+ server_kex.set_rsa_key_size(host_key_type, hostkey_modulus_size, ca_modulus_size)
+
+ # Keys smaller than 2048 result in a failure. Update the database accordingly.
+ if (cert is False) and (hostkey_modulus_size < 2048):
+ for rsa_type in HostKeyTest.RSA_FAMILY:
+ alg_list = SSH2_KexDB.ALGORITHMS['key'][rsa_type]
+
+ # If no failure list exists, add an empty failure list.
+ if len(alg_list) < 2:
+ alg_list.append([])
+ alg_list[1].append('using small %d-bit modulus' % hostkey_modulus_size)
+ elif (cert is True) and ((hostkey_modulus_size < 2048) or (ca_modulus_size > 0 and ca_modulus_size < 2048)): # pylint: disable=chained-comparison
+ alg_list = SSH2_KexDB.ALGORITHMS['key'][host_key_type]
+ min_modulus = min(hostkey_modulus_size, ca_modulus_size)
+ min_modulus = min_modulus if min_modulus > 0 else max(hostkey_modulus_size, ca_modulus_size)
+
+ # If no failure list exists, add an empty failure list.
+ if len(alg_list) < 2:
+ alg_list.append([])
+ alg_list[1].append('using small %d-bit modulus' % min_modulus)
+
+ # If this host key type is in the RSA family, then mark them all as parsed (since results in one are valid for them all).
+ if host_key_type in HostKeyTest.RSA_FAMILY:
+ for rsa_type in HostKeyTest.RSA_FAMILY:
+ host_key_types[rsa_type]['parsed'] = True
+ else:
+ host_key_types[host_key_type]['parsed'] = True
diff --git a/src/ssh_audit/kexdh.py b/src/ssh_audit/kexdh.py
new file mode 100644
index 0000000..536f0a8
--- /dev/null
+++ b/src/ssh_audit/kexdh.py
@@ -0,0 +1,365 @@
+"""
+ The MIT License (MIT)
+
+ Copyright (C) 2017-2021 Joe Testa (jtesta@positronsecurity.com)
+ Copyright (C) 2017 Andris Raugulis (moo@arthepsy.eu)
+
+ Permission is hereby granted, free of charge, to any person obtaining a copy
+ of this software and associated documentation files (the "Software"), to deal
+ in the Software without restriction, including without limitation the rights
+ to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+ copies of the Software, and to permit persons to whom the Software is
+ furnished to do so, subject to the following conditions:
+
+ The above copyright notice and this permission notice shall be included in
+ all copies or substantial portions of the Software.
+
+ THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+ IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+ FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+ AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+ LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+ OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+ THE SOFTWARE.
+"""
+import binascii
+import os
+import random
+import struct
+
+# pylint: disable=unused-import
+from typing import Dict, List, Set, Sequence, Tuple, Iterable # noqa: F401
+from typing import Callable, Optional, Union, Any # noqa: F401
+
+from ssh_audit.protocol import Protocol
+from ssh_audit.ssh_socket import SSH_Socket
+
+
+class KexDH: # pragma: nocover
+ def __init__(self, kex_name: str, hash_alg: str, g: int, p: int) -> None:
+ self.__kex_name = kex_name # pylint: disable=unused-private-member
+ self.__hash_alg = hash_alg # pylint: disable=unused-private-member
+ self.__g = 0
+ self.__p = 0
+ self.__q = 0
+ self.__x = 0
+ self.__e = 0
+ self.set_params(g, p)
+
+ self.__ed25519_pubkey: Optional[bytes] = None # pylint: disable=unused-private-member
+ self.__hostkey_type: Optional[bytes] = None
+ self.__hostkey_e = 0 # pylint: disable=unused-private-member
+ self.__hostkey_n = 0 # pylint: disable=unused-private-member
+ self.__hostkey_n_len = 0 # Length of the host key modulus.
+ self.__ca_n_len = 0 # Length of the CA key modulus (if hostkey is a cert).
+
+ def set_params(self, g: int, p: int) -> None:
+ self.__g = g
+ self.__p = p
+ self.__q = (self.__p - 1) // 2
+ self.__x = 0
+ self.__e = 0
+
+ def send_init(self, s: SSH_Socket, init_msg: int = Protocol.MSG_KEXDH_INIT) -> None:
+ r = random.SystemRandom()
+ self.__x = r.randrange(2, self.__q)
+ self.__e = pow(self.__g, self.__x, self.__p)
+ s.write_byte(init_msg)
+ s.write_mpint2(self.__e)
+ s.send_packet()
+
+ # Parse a KEXDH_REPLY or KEXDH_GEX_REPLY message from the server. This
+ # contains the host key, among other things. Function returns the host
+ # key blob (from which the fingerprint can be calculated).
+ def recv_reply(self, s: 'SSH_Socket', parse_host_key_size: bool = True) -> Optional[bytes]:
+ packet_type, payload = s.read_packet(2)
+
+ # Skip any & all MSG_DEBUG messages.
+ while packet_type == Protocol.MSG_DEBUG:
+ packet_type, payload = s.read_packet(2)
+
+ if packet_type != -1 and packet_type not in [Protocol.MSG_KEXDH_REPLY, Protocol.MSG_KEXDH_GEX_REPLY]: # pylint: disable=no-else-raise
+ # TODO: change Exception to something more specific.
+ raise Exception('Expected MSG_KEXDH_REPLY (%d) or MSG_KEXDH_GEX_REPLY (%d), but got %d instead.' % (Protocol.MSG_KEXDH_REPLY, Protocol.MSG_KEXDH_GEX_REPLY, packet_type))
+ elif packet_type == -1:
+ # A connection error occurred. We can't parse anything, so just
+ # return. The host key modulus (and perhaps certificate modulus)
+ # will remain at length 0.
+ return None
+
+ hostkey_len = 0 # pylint: disable=unused-variable
+ hostkey_type_len = hostkey_e_len = 0 # pylint: disable=unused-variable
+ key_id_len = principles_len = 0 # pylint: disable=unused-variable
+ critical_options_len = extensions_len = 0 # pylint: disable=unused-variable
+ nonce_len = ca_key_len = ca_key_type_len = 0 # pylint: disable=unused-variable
+ ca_key_len = ca_key_type_len = ca_key_e_len = 0 # pylint: disable=unused-variable
+
+ key_id = principles = None # pylint: disable=unused-variable
+ critical_options = extensions = None # pylint: disable=unused-variable
+ nonce = ca_key = ca_key_type = None # pylint: disable=unused-variable
+ ca_key_e = ca_key_n = None # pylint: disable=unused-variable
+
+ # Get the host key blob, F, and signature.
+ ptr = 0
+ hostkey, hostkey_len, ptr = KexDH.__get_bytes(payload, ptr)
+
+ # If we are not supposed to parse the host key size (i.e.: it is a type that is of fixed size such as ed25519), then stop here.
+ if not parse_host_key_size:
+ return hostkey
+
+ _, _, ptr = KexDH.__get_bytes(payload, ptr)
+ _, _, ptr = KexDH.__get_bytes(payload, ptr)
+
+ # Now pick apart the host key blob.
+ # Get the host key type (i.e.: 'ssh-rsa', 'ssh-ed25519', etc).
+ ptr = 0
+ self.__hostkey_type, hostkey_type_len, ptr = KexDH.__get_bytes(hostkey, ptr)
+
+ # If this is an RSA certificate, skip over the nonce.
+ if self.__hostkey_type.startswith(b'ssh-rsa-cert-v0'):
+ nonce, nonce_len, ptr = KexDH.__get_bytes(hostkey, ptr)
+
+ # The public key exponent.
+ hostkey_e, hostkey_e_len, ptr = KexDH.__get_bytes(hostkey, ptr)
+ self.__hostkey_e = int(binascii.hexlify(hostkey_e), 16) # pylint: disable=unused-private-member
+
+ # Here is the modulus size & actual modulus of the host key public key.
+ hostkey_n, self.__hostkey_n_len, ptr = KexDH.__get_bytes(hostkey, ptr)
+ self.__hostkey_n = int(binascii.hexlify(hostkey_n), 16) # pylint: disable=unused-private-member
+
+ # If this is an RSA certificate, continue parsing to extract the CA
+ # key.
+ if self.__hostkey_type.startswith(b'ssh-rsa-cert-v0'):
+ # Skip over the serial number.
+ ptr += 8
+
+ # Get the certificate type.
+ cert_type = int(binascii.hexlify(hostkey[ptr:ptr + 4]), 16)
+ ptr += 4
+
+ # Only SSH2_CERT_TYPE_HOST (2) makes sense in this context.
+ if cert_type == 2:
+
+ # Skip the key ID (this is the serial number of the
+ # certificate).
+ key_id, key_id_len, ptr = KexDH.__get_bytes(hostkey, ptr)
+
+ # The principles, which are... I don't know what.
+ principles, principles_len, ptr = KexDH.__get_bytes(hostkey, ptr)
+
+ # Skip over the timestamp that this certificate is valid after.
+ ptr += 8
+
+ # Skip over the timestamp that this certificate is valid before.
+ ptr += 8
+
+ # TODO: validate the principles, and time range.
+
+ # The critical options.
+ critical_options, critical_options_len, ptr = KexDH.__get_bytes(hostkey, ptr)
+
+ # Certificate extensions.
+ extensions, extensions_len, ptr = KexDH.__get_bytes(hostkey, ptr)
+
+ # Another nonce.
+ nonce, nonce_len, ptr = KexDH.__get_bytes(hostkey, ptr)
+
+ # Finally, we get to the CA key.
+ ca_key, ca_key_len, ptr = KexDH.__get_bytes(hostkey, ptr)
+
+ # Last in the host key blob is the CA signature. It isn't
+ # interesting to us, so we won't bother parsing any further.
+ # The CA key has the modulus, however...
+ ptr = 0
+
+ # 'ssh-rsa', 'rsa-sha2-256', etc.
+ ca_key_type, ca_key_type_len, ptr = KexDH.__get_bytes(ca_key, ptr)
+
+ # CA's public key exponent.
+ ca_key_e, ca_key_e_len, ptr = KexDH.__get_bytes(ca_key, ptr)
+
+ # CA's modulus. Bingo.
+ ca_key_n, self.__ca_n_len, ptr = KexDH.__get_bytes(ca_key, ptr)
+
+ return hostkey
+
+ @staticmethod
+ def __get_bytes(buf: bytes, ptr: int) -> Tuple[bytes, int, int]:
+ num_bytes = struct.unpack('>I', buf[ptr:ptr + 4])[0]
+ ptr += 4
+ return buf[ptr:ptr + num_bytes], num_bytes, ptr + num_bytes
+
+ # Converts a modulus length in bytes to its size in bits, after some
+ # possible adjustments.
+ @staticmethod
+ def __adjust_key_size(size: int) -> int:
+ size = size * 8
+ # Actual keys are observed to be about 8 bits bigger than expected
+ # (i.e.: 1024-bit keys have a 1032-bit modulus). Check if this is
+ # the case, and subtract 8 if so. This simply improves readability
+ # in the UI.
+ if (size >> 3) % 2 != 0:
+ size = size - 8
+ return size
+
+ # Returns the size of the hostkey, in bits.
+ def get_hostkey_size(self) -> int:
+ return KexDH.__adjust_key_size(self.__hostkey_n_len)
+
+ # Returns the size of the CA key, in bits.
+ def get_ca_size(self) -> int:
+ return KexDH.__adjust_key_size(self.__ca_n_len)
+
+ # Returns the size of the DH modulus, in bits.
+ def get_dh_modulus_size(self) -> int:
+ # -2 to account for the '0b' prefix in the string.
+ return len(bin(self.__p)) - 2
+
+
+class KexGroup1(KexDH): # pragma: nocover
+ def __init__(self) -> None:
+ # rfc2409: second oakley group
+ p = int('ffffffffffffffffc90fdaa22168c234c4c6628b80dc1cd129024e088a67cc74020bbea63b139b22514a08798e3404ddef9519b3cd3a431b302b0a6df25f14374fe1356d6d51c245e485b576625e7ec6f44c42e9a637ed6b0bff5cb6f406b7edee386bfb5a899fa5ae9f24117c4b1fe649286651ece65381ffffffffffffffff', 16)
+ super(KexGroup1, self).__init__('KexGroup1', 'sha1', 2, p)
+
+
+class KexGroup14(KexDH): # pragma: nocover
+ def __init__(self, hash_alg: str) -> None:
+ # rfc3526: 2048-bit modp group
+ p = int('ffffffffffffffffc90fdaa22168c234c4c6628b80dc1cd129024e088a67cc74020bbea63b139b22514a08798e3404ddef9519b3cd3a431b302b0a6df25f14374fe1356d6d51c245e485b576625e7ec6f44c42e9a637ed6b0bff5cb6f406b7edee386bfb5a899fa5ae9f24117c4b1fe649286651ece45b3dc2007cb8a163bf0598da48361c55d39a69163fa8fd24cf5f83655d23dca3ad961c62f356208552bb9ed529077096966d670c354e4abc9804f1746c08ca18217c32905e462e36ce3be39e772c180e86039b2783a2ec07a28fb5c55df06f4c52c9de2bcbf6955817183995497cea956ae515d2261898fa051015728e5a8aacaa68ffffffffffffffff', 16)
+ super(KexGroup14, self).__init__('KexGroup14', hash_alg, 2, p)
+
+
+class KexGroup14_SHA1(KexGroup14):
+ def __init__(self) -> None:
+ super(KexGroup14_SHA1, self).__init__('sha1')
+
+
+class KexGroup14_SHA256(KexGroup14):
+ def __init__(self) -> None:
+ super(KexGroup14_SHA256, self).__init__('sha256')
+
+
+class KexGroup16_SHA512(KexDH):
+ def __init__(self) -> None:
+ # rfc3526: 4096-bit modp group
+ p = int('ffffffffffffffffc90fdaa22168c234c4c6628b80dc1cd129024e088a67cc74020bbea63b139b22514a08798e3404ddef9519b3cd3a431b302b0a6df25f14374fe1356d6d51c245e485b576625e7ec6f44c42e9a637ed6b0bff5cb6f406b7edee386bfb5a899fa5ae9f24117c4b1fe649286651ece45b3dc2007cb8a163bf0598da48361c55d39a69163fa8fd24cf5f83655d23dca3ad961c62f356208552bb9ed529077096966d670c354e4abc9804f1746c08ca18217c32905e462e36ce3be39e772c180e86039b2783a2ec07a28fb5c55df06f4c52c9de2bcbf6955817183995497cea956ae515d2261898fa051015728e5a8aaac42dad33170d04507a33a85521abdf1cba64ecfb850458dbef0a8aea71575d060c7db3970f85a6e1e4c7abf5ae8cdb0933d71e8c94e04a25619dcee3d2261ad2ee6bf12ffa06d98a0864d87602733ec86a64521f2b18177b200cbbe117577a615d6c770988c0bad946e208e24fa074e5ab3143db5bfce0fd108e4b82d120a92108011a723c12a787e6d788719a10bdba5b2699c327186af4e23c1a946834b6150bda2583e9ca2ad44ce8dbbbc2db04de8ef92e8efc141fbecaa6287c59474e6bc05d99b2964fa090c3a2233ba186515be7ed1f612970cee2d7afb81bdd762170481cd0069127d5b05aa993b4ea988d8fddc186ffb7dc90a6c08f4df435c934063199ffffffffffffffff', 16)
+ super(KexGroup16_SHA512, self).__init__('KexGroup16_SHA512', 'sha512', 2, p)
+
+
+class KexGroup18_SHA512(KexDH):
+ def __init__(self) -> None:
+ # rfc3526: 8192-bit modp group
+ p = int('ffffffffffffffffc90fdaa22168c234c4c6628b80dc1cd129024e088a67cc74020bbea63b139b22514a08798e3404ddef9519b3cd3a431b302b0a6df25f14374fe1356d6d51c245e485b576625e7ec6f44c42e9a637ed6b0bff5cb6f406b7edee386bfb5a899fa5ae9f24117c4b1fe649286651ece45b3dc2007cb8a163bf0598da48361c55d39a69163fa8fd24cf5f83655d23dca3ad961c62f356208552bb9ed529077096966d670c354e4abc9804f1746c08ca18217c32905e462e36ce3be39e772c180e86039b2783a2ec07a28fb5c55df06f4c52c9de2bcbf6955817183995497cea956ae515d2261898fa051015728e5a8aaac42dad33170d04507a33a85521abdf1cba64ecfb850458dbef0a8aea71575d060c7db3970f85a6e1e4c7abf5ae8cdb0933d71e8c94e04a25619dcee3d2261ad2ee6bf12ffa06d98a0864d87602733ec86a64521f2b18177b200cbbe117577a615d6c770988c0bad946e208e24fa074e5ab3143db5bfce0fd108e4b82d120a92108011a723c12a787e6d788719a10bdba5b2699c327186af4e23c1a946834b6150bda2583e9ca2ad44ce8dbbbc2db04de8ef92e8efc141fbecaa6287c59474e6bc05d99b2964fa090c3a2233ba186515be7ed1f612970cee2d7afb81bdd762170481cd0069127d5b05aa993b4ea988d8fddc186ffb7dc90a6c08f4df435c93402849236c3fab4d27c7026c1d4dcb2602646dec9751e763dba37bdf8ff9406ad9e530ee5db382f413001aeb06a53ed9027d831179727b0865a8918da3edbebcf9b14ed44ce6cbaced4bb1bdb7f1447e6cc254b332051512bd7af426fb8f401378cd2bf5983ca01c64b92ecf032ea15d1721d03f482d7ce6e74fef6d55e702f46980c82b5a84031900b1c9e59e7c97fbec7e8f323a97a7e36cc88be0f1d45b7ff585ac54bd407b22b4154aacc8f6d7ebf48e1d814cc5ed20f8037e0a79715eef29be32806a1d58bb7c5da76f550aa3d8a1fbff0eb19ccb1a313d55cda56c9ec2ef29632387fe8d76e3c0468043e8f663f4860ee12bf2d5b0b7474d6e694f91e6dbe115974a3926f12fee5e438777cb6a932df8cd8bec4d073b931ba3bc832b68d9dd300741fa7bf8afc47ed2576f6936ba424663aab639c5ae4f5683423b4742bf1c978238f16cbe39d652de3fdb8befc848ad922222e04a4037c0713eb57a81a23f0c73473fc646cea306b4bcbc8862f8385ddfa9d4b7fa2c087e879683303ed5bdd3a062b3cf5b3a278a66d2a13f83f44f82ddf310ee074ab6a364597e899a0255dc164f31cc50846851df9ab48195ded7ea1b1d510bd7ee74d73faf36bc31ecfa268359046f4eb879f924009438b481c6cd7889a002ed5ee382bc9190da6fc026e479558e4475677e9aa9e3050e2765694dfc81f56e880b96e7160c980dd98edd3dfffffffffffffffff', 16)
+ super(KexGroup18_SHA512, self).__init__('KexGroup18_SHA512', 'sha512', 2, p)
+
+
+class KexCurve25519_SHA256(KexDH):
+ def __init__(self) -> None:
+ super(KexCurve25519_SHA256, self).__init__('KexCurve25519_SHA256', 'sha256', 0, 0)
+
+ # To start an ED25519 kex, we simply send a random 256-bit number as the
+ # public key.
+ def send_init(self, s: 'SSH_Socket', init_msg: int = Protocol.MSG_KEXDH_INIT) -> None:
+ self.__ed25519_pubkey = os.urandom(32)
+ s.write_byte(init_msg)
+ s.write_string(self.__ed25519_pubkey)
+ s.send_packet()
+
+
+class KexNISTP256(KexDH):
+ def __init__(self) -> None:
+ super(KexNISTP256, self).__init__('KexNISTP256', 'sha256', 0, 0)
+
+ # Because the server checks that the value sent here is valid (i.e.: it lies
+ # on the curve, among other things), we would have to write a lot of code
+ # or import an elliptic curve library in order to randomly generate a
+ # valid elliptic point each time. Hence, we will simply send a static
+ # value, which is enough for us to extract the server's host key.
+ def send_init(self, s: 'SSH_Socket', init_msg: int = Protocol.MSG_KEXDH_INIT) -> None:
+ s.write_byte(init_msg)
+ s.write_string(b'\x04\x0b\x60\x44\x9f\x8a\x11\x9e\xc7\x81\x0c\xa9\x98\xfc\xb7\x90\xaa\x6b\x26\x8c\x12\x4a\xc0\x09\xbb\xdf\xc4\x2c\x4c\x2c\x99\xb6\xe1\x71\xa0\xd4\xb3\x62\x47\x74\xb3\x39\x0c\xf2\x88\x4a\x84\x6b\x3b\x15\x77\xa5\x77\xd2\xa9\xc9\x94\xf9\xd5\x66\x19\xcd\x02\x34\xd1')
+ s.send_packet()
+
+
+class KexNISTP384(KexDH):
+ def __init__(self) -> None:
+ super(KexNISTP384, self).__init__('KexNISTP384', 'sha256', 0, 0)
+
+ # See comment for KexNISTP256.send_init().
+ def send_init(self, s: 'SSH_Socket', init_msg: int = Protocol.MSG_KEXDH_INIT) -> None:
+ s.write_byte(init_msg)
+ s.write_string(b'\x04\xe2\x9b\x84\xce\xa1\x39\x50\xfe\x1e\xa3\x18\x70\x1c\xe2\x7a\xe4\xb5\x6f\xdf\x93\x9f\xd4\xf4\x08\xcc\x9b\x02\x10\xa4\xca\x77\x9c\x2e\x51\x44\x1d\x50\x7a\x65\x4e\x7e\x2f\x10\x2d\x2d\x4a\x32\xc9\x8e\x18\x75\x90\x6c\x19\x10\xda\xcc\xa8\xe9\xf4\xc4\x3a\x53\x80\x35\xf4\x97\x9c\x04\x16\xf9\x5a\xdc\xcc\x05\x94\x29\xfa\xc4\xd6\x87\x4e\x13\x21\xdb\x3d\x12\xac\xbd\x20\x3b\x60\xff\xe6\x58\x42')
+ s.send_packet()
+
+
+class KexNISTP521(KexDH):
+ def __init__(self) -> None:
+ super(KexNISTP521, self).__init__('KexNISTP521', 'sha256', 0, 0)
+
+ # See comment for KexNISTP256.send_init().
+ def send_init(self, s: 'SSH_Socket', init_msg: int = Protocol.MSG_KEXDH_INIT) -> None:
+ s.write_byte(init_msg)
+ s.write_string(b'\x04\x01\x02\x90\x29\xe9\x8f\xa8\x04\xaf\x1c\x00\xf9\xc6\x29\xc0\x39\x74\x8e\xea\x47\x7e\x7c\xf7\x15\x6e\x43\x3b\x59\x13\x53\x43\xb0\xae\x0b\xe7\xe6\x7c\x55\x73\x52\xa5\x2a\xc1\x42\xde\xfc\xf4\x1f\x8b\x5a\x8d\xfa\xcd\x0a\x65\x77\xa8\xce\x68\xd2\xc6\x26\xb5\x3f\xee\x4b\x01\x7b\xd2\x96\x23\x69\x53\xc7\x01\xe1\x0d\x39\xe9\x87\x49\x3b\xc8\xec\xda\x0c\xf9\xca\xad\x89\x42\x36\x6f\x93\x78\x78\x31\x55\x51\x09\x51\xc0\x96\xd7\xea\x61\xbf\xc2\x44\x08\x80\x43\xed\xc6\xbb\xfb\x94\xbd\xf8\xdf\x2b\xd8\x0b\x2e\x29\x1b\x8c\xc4\x8a\x04\x2d\x3a')
+ s.send_packet()
+
+
+class KexGroupExchange(KexDH):
+ def __init__(self, classname: str, hash_alg: str) -> None:
+ super(KexGroupExchange, self).__init__(classname, hash_alg, 0, 0)
+
+ def send_init(self, s: 'SSH_Socket', init_msg: int = Protocol.MSG_KEXDH_GEX_REQUEST) -> None:
+ self.send_init_gex(s)
+
+ # The group exchange starts with sending a message to the server with
+ # the minimum, maximum, and preferred number of bits are for the DH group.
+ # The server responds with a generator and prime modulus that matches that,
+ # then the handshake continues on like a normal DH handshake (except the
+ # SSH message types differ).
+ def send_init_gex(self, s: 'SSH_Socket', minbits: int = 1024, prefbits: int = 2048, maxbits: int = 8192) -> None:
+
+ # Send the initial group exchange request. Tell the server what range
+ # of modulus sizes we will accept, along with our preference.
+ s.write_byte(Protocol.MSG_KEXDH_GEX_REQUEST)
+ s.write_int(minbits)
+ s.write_int(prefbits)
+ s.write_int(maxbits)
+ s.send_packet()
+
+ packet_type, payload = s.read_packet(2)
+ if packet_type not in [Protocol.MSG_KEXDH_GEX_GROUP, Protocol.MSG_DEBUG]:
+ # TODO: replace with a better exception type.
+ raise Exception('Expected MSG_KEXDH_GEX_REPLY (%d), but got %d instead.' % (Protocol.MSG_KEXDH_GEX_REPLY, packet_type))
+
+ # Skip any & all MSG_DEBUG messages.
+ while packet_type == Protocol.MSG_DEBUG:
+ packet_type, payload = s.read_packet(2)
+
+ # Parse the modulus (p) and generator (g) values from the server.
+ ptr = 0
+ p_len = struct.unpack('>I', payload[ptr:ptr + 4])[0]
+ ptr += 4
+
+ p = int(binascii.hexlify(payload[ptr:ptr + p_len]), 16)
+ ptr += p_len
+
+ g_len = struct.unpack('>I', payload[ptr:ptr + 4])[0]
+ ptr += 4
+
+ g = int(binascii.hexlify(payload[ptr:ptr + g_len]), 16)
+ ptr += g_len
+
+ # Now that we got the generator and modulus, perform the DH exchange
+ # like usual.
+ super(KexGroupExchange, self).set_params(g, p)
+ super(KexGroupExchange, self).send_init(s, Protocol.MSG_KEXDH_GEX_INIT)
+
+
+class KexGroupExchange_SHA1(KexGroupExchange):
+ def __init__(self) -> None:
+ super(KexGroupExchange_SHA1, self).__init__('KexGroupExchange_SHA1', 'sha1')
+
+
+class KexGroupExchange_SHA256(KexGroupExchange):
+ def __init__(self) -> None:
+ super(KexGroupExchange_SHA256, self).__init__('KexGroupExchange_SHA256', 'sha256')
diff --git a/src/ssh_audit/outputbuffer.py b/src/ssh_audit/outputbuffer.py
new file mode 100644
index 0000000..38723e3
--- /dev/null
+++ b/src/ssh_audit/outputbuffer.py
@@ -0,0 +1,185 @@
+"""
+ The MIT License (MIT)
+
+ Copyright (C) 2021 Joe Testa (jtesta@positronsecurity.com)
+ Copyright (C) 2017 Andris Raugulis (moo@arthepsy.eu)
+
+ Permission is hereby granted, free of charge, to any person obtaining a copy
+ of this software and associated documentation files (the "Software"), to deal
+ in the Software without restriction, including without limitation the rights
+ to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+ copies of the Software, and to permit persons to whom the Software is
+ furnished to do so, subject to the following conditions:
+
+ The above copyright notice and this permission notice shall be included in
+ all copies or substantial portions of the Software.
+
+ THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+ IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+ FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+ AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+ LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+ OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+ THE SOFTWARE.
+"""
+import os
+import sys
+
+# pylint: disable=unused-import
+from typing import Dict, List, Set, Sequence, Tuple, Iterable # noqa: F401
+from typing import Callable, Optional, Union, Any # noqa: F401
+
+from ssh_audit.utils import Utils
+
+
+class OutputBuffer:
+ LEVELS: Sequence[str] = ('info', 'warn', 'fail')
+ COLORS = {'head': 36, 'good': 32, 'warn': 33, 'fail': 31}
+
+ # Use brighter colors on Windows for better readability.
+ if Utils.is_windows():
+ COLORS = {'head': 96, 'good': 92, 'warn': 93, 'fail': 91}
+
+ def __init__(self, buffer_output: bool = True) -> None:
+ self.buffer_output = buffer_output
+ self.buffer: List[str] = []
+ self.in_section = False
+ self.section: List[str] = []
+ self.batch = False
+ self.verbose = False
+ self.debug = False
+ self.use_colors = True
+ self.json = False
+ self.__level = 0
+ self.__is_color_supported = ('colorama' in sys.modules) or (os.name == 'posix')
+ self.line_ended = True
+
+ def _print(self, level: str, s: str = '', line_ended: bool = True) -> None:
+ '''Saves output to buffer (if in buffered mode), or immediately prints to stdout otherwise.'''
+
+ # If we're logging only 'warn' or above, and this is an 'info', ignore message.
+ if self.get_level(level) < self.__level:
+ return
+
+ if self.use_colors and self.colors_supported and len(s) > 0 and level != 'info':
+ s = "\033[0;%dm%s\033[0m" % (self.COLORS[level], s)
+
+ if self.buffer_output:
+ # Select which list to add to. If we are in a 'with' statement, then this goes in the section buffer, otherwise the general buffer.
+ buf = self.section if self.in_section else self.buffer
+
+ # Determine if a new line should be added, or if the last line should be appended.
+ if not self.line_ended:
+ last_entry = -1 if len(buf) > 0 else 0
+ buf[last_entry] = buf[last_entry] + s
+ else:
+ buf.append(s)
+
+ # When False, this tells the next call to append to the last line we just added.
+ self.line_ended = line_ended
+ else:
+ print(s)
+
+ def get_buffer(self) -> str:
+ '''Returns all buffered output, then clears the buffer.'''
+ self.flush_section()
+
+ buffer_str = "\n".join(self.buffer)
+ self.buffer = []
+ return buffer_str
+
+ def write(self) -> None:
+ '''Writes the output to stdout.'''
+ self.flush_section()
+ print(self.get_buffer(), flush=True)
+
+ def reset(self) -> None:
+ self.flush_section()
+ self.get_buffer()
+
+ @property
+ def level(self) -> str:
+ '''Returns the minimum level for output.'''
+ if self.__level < len(self.LEVELS):
+ return self.LEVELS[self.__level]
+ return 'unknown'
+
+ @level.setter
+ def level(self, name: str) -> None:
+ '''Sets the minimum level for output (one of: 'info', 'warn', 'fail').'''
+ self.__level = self.get_level(name)
+
+ def get_level(self, name: str) -> int:
+ cname = 'info' if name == 'good' else name
+ if cname not in self.LEVELS:
+ return sys.maxsize
+ return self.LEVELS.index(cname)
+
+ @property
+ def colors_supported(self) -> bool:
+ '''Returns True if the system supports color output.'''
+ return self.__is_color_supported
+
+ # When used in a 'with' block, the output to goes into a section; this can be sorted separately when add_section_to_buffer() is later called.
+ def __enter__(self) -> 'OutputBuffer':
+ self.in_section = True
+ return self
+
+ def __exit__(self, *args: Any) -> None:
+ self.in_section = False
+
+ def flush_section(self, sort_section: bool = False) -> None:
+ '''Appends section output (optionally sorting it first) to the end of the buffer, then clears the section output.'''
+ if sort_section:
+ self.section.sort()
+
+ self.buffer.extend(self.section)
+ self.section = []
+
+ def is_section_empty(self) -> bool:
+ '''Returns True if the section buffer is empty, otherwise False.'''
+ return len(self.section) == 0
+
+ def head(self, s: str, line_ended: bool = True) -> 'OutputBuffer':
+ if not self.batch:
+ self._print('head', s, line_ended)
+ return self
+
+ def fail(self, s: str, line_ended: bool = True) -> 'OutputBuffer':
+ self._print('fail', s, line_ended)
+ return self
+
+ def warn(self, s: str, line_ended: bool = True) -> 'OutputBuffer':
+ self._print('warn', s, line_ended)
+ return self
+
+ def info(self, s: str, line_ended: bool = True) -> 'OutputBuffer':
+ self._print('info', s, line_ended)
+ return self
+
+ def good(self, s: str, line_ended: bool = True) -> 'OutputBuffer':
+ self._print('good', s, line_ended)
+ return self
+
+ def sep(self) -> 'OutputBuffer':
+ if not self.batch:
+ self._print('info')
+ return self
+
+ def v(self, s: str, write_now: bool = False) -> 'OutputBuffer':
+ '''Prints a message if verbose output is enabled.'''
+ if self.verbose or self.debug:
+ self.info(s)
+ if write_now:
+ self.write()
+
+ return self
+
+ def d(self, s: str, write_now: bool = False) -> 'OutputBuffer':
+ '''Prints a message if verbose output is enabled.'''
+ if self.debug:
+ self.info(s)
+ if write_now:
+ self.write()
+
+ return self
diff --git a/src/ssh_audit/policy.py b/src/ssh_audit/policy.py
new file mode 100644
index 0000000..fd9d2c7
--- /dev/null
+++ b/src/ssh_audit/policy.py
@@ -0,0 +1,521 @@
+"""
+ The MIT License (MIT)
+
+ Copyright (C) 2020-2021 Joe Testa (jtesta@positronsecurity.com)
+
+ Permission is hereby granted, free of charge, to any person obtaining a copy
+ of this software and associated documentation files (the "Software"), to deal
+ in the Software without restriction, including without limitation the rights
+ to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+ copies of the Software, and to permit persons to whom the Software is
+ furnished to do so, subject to the following conditions:
+
+ The above copyright notice and this permission notice shall be included in
+ all copies or substantial portions of the Software.
+
+ THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+ IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+ FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+ AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+ LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+ OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+ THE SOFTWARE.
+"""
+import sys
+
+from typing import Dict, List, Tuple
+from typing import Optional, Any, Union, cast
+from datetime import date
+
+from ssh_audit import exitcodes
+from ssh_audit.ssh2_kex import SSH2_Kex # pylint: disable=unused-import
+from ssh_audit.banner import Banner # pylint: disable=unused-import
+
+
+# Validates policy files and performs policy testing
+class Policy:
+
+ # Each field maps directly to a private member variable of the Policy class.
+ BUILTIN_POLICIES: Dict[str, Dict[str, Union[Optional[str], Optional[List[str]], bool, Dict[str, int]]]] = {
+
+ # Ubuntu Server policies
+
+ 'Hardened Ubuntu Server 16.04 LTS (version 1)': {'version': '1', 'banner': None, 'compressions': None, 'host_keys': ['ssh-ed25519'], 'optional_host_keys': ['ssh-ed25519-cert-v01@openssh.com', 'rsa-sha2-256-cert-v01@openssh.com', 'rsa-sha2-512-cert-v01@openssh.com'], 'kex': ['curve25519-sha256@libssh.org', 'diffie-hellman-group-exchange-sha256'], 'ciphers': ['chacha20-poly1305@openssh.com', 'aes256-gcm@openssh.com', 'aes128-gcm@openssh.com', 'aes256-ctr', 'aes192-ctr', 'aes128-ctr'], 'macs': ['hmac-sha2-256-etm@openssh.com', 'hmac-sha2-512-etm@openssh.com', 'umac-128-etm@openssh.com'], 'hostkey_sizes': None, 'cakey_sizes': None, 'dh_modulus_sizes': {'diffie-hellman-group-exchange-sha256': 2048}, 'server_policy': True},
+
+ 'Hardened Ubuntu Server 18.04 LTS (version 1)': {'version': '1', 'banner': None, 'compressions': None, 'host_keys': ['ssh-ed25519'], 'optional_host_keys': ['ssh-ed25519-cert-v01@openssh.com', 'rsa-sha2-256-cert-v01@openssh.com', 'rsa-sha2-512-cert-v01@openssh.com'], 'kex': ['curve25519-sha256', 'curve25519-sha256@libssh.org', 'diffie-hellman-group16-sha512', 'diffie-hellman-group18-sha512', 'diffie-hellman-group-exchange-sha256'], 'ciphers': ['chacha20-poly1305@openssh.com', 'aes256-gcm@openssh.com', 'aes128-gcm@openssh.com', 'aes256-ctr', 'aes192-ctr', 'aes128-ctr'], 'macs': ['hmac-sha2-256-etm@openssh.com', 'hmac-sha2-512-etm@openssh.com', 'umac-128-etm@openssh.com'], 'hostkey_sizes': None, 'cakey_sizes': None, 'dh_modulus_sizes': {'diffie-hellman-group-exchange-sha256': 2048}, 'server_policy': True},
+
+ 'Hardened Ubuntu Server 20.04 LTS (version 1)': {'version': '1', 'banner': None, 'compressions': None, 'host_keys': ['rsa-sha2-512', 'rsa-sha2-256', 'ssh-ed25519'], 'optional_host_keys': ['sk-ssh-ed25519@openssh.com', 'ssh-ed25519-cert-v01@openssh.com', 'sk-ssh-ed25519-cert-v01@openssh.com', 'rsa-sha2-256-cert-v01@openssh.com', 'rsa-sha2-512-cert-v01@openssh.com'], 'kex': ['curve25519-sha256', 'curve25519-sha256@libssh.org', 'diffie-hellman-group16-sha512', 'diffie-hellman-group18-sha512', 'diffie-hellman-group-exchange-sha256'], 'ciphers': ['chacha20-poly1305@openssh.com', 'aes256-gcm@openssh.com', 'aes128-gcm@openssh.com', 'aes256-ctr', 'aes192-ctr', 'aes128-ctr'], 'macs': ['hmac-sha2-256-etm@openssh.com', 'hmac-sha2-512-etm@openssh.com', 'umac-128-etm@openssh.com'], 'hostkey_sizes': {'rsa-sha2-256': 4096, 'rsa-sha2-512': 4096}, 'cakey_sizes': {'rsa-sha2-256-cert-v01@openssh.com': 4096, 'rsa-sha2-512-cert-v01@openssh.com': 4096}, 'dh_modulus_sizes': {'diffie-hellman-group-exchange-sha256': 2048}, 'server_policy': True},
+
+
+ # Generic OpenSSH Server policies
+
+ 'Hardened OpenSSH Server v7.7 (version 1)': {'version': '1', 'banner': None, 'compressions': None, 'host_keys': ['rsa-sha2-512', 'rsa-sha2-256', 'ssh-ed25519'], 'optional_host_keys': ['ssh-ed25519-cert-v01@openssh.com', 'rsa-sha2-256-cert-v01@openssh.com', 'rsa-sha2-512-cert-v01@openssh.com'], 'kex': ['curve25519-sha256', 'curve25519-sha256@libssh.org', 'diffie-hellman-group16-sha512', 'diffie-hellman-group18-sha512', 'diffie-hellman-group-exchange-sha256'], 'ciphers': ['chacha20-poly1305@openssh.com', 'aes256-gcm@openssh.com', 'aes128-gcm@openssh.com', 'aes256-ctr', 'aes192-ctr', 'aes128-ctr'], 'macs': ['hmac-sha2-256-etm@openssh.com', 'hmac-sha2-512-etm@openssh.com', 'umac-128-etm@openssh.com'], 'hostkey_sizes': None, 'cakey_sizes': None, 'dh_modulus_sizes': {'diffie-hellman-group-exchange-sha256': 2048}, 'server_policy': True},
+
+ 'Hardened OpenSSH Server v7.8 (version 1)': {'version': '1', 'banner': None, 'compressions': None, 'host_keys': ['rsa-sha2-512', 'rsa-sha2-256', 'ssh-ed25519'], 'optional_host_keys': ['ssh-ed25519-cert-v01@openssh.com', 'rsa-sha2-256-cert-v01@openssh.com', 'rsa-sha2-512-cert-v01@openssh.com'], 'kex': ['curve25519-sha256', 'curve25519-sha256@libssh.org', 'diffie-hellman-group16-sha512', 'diffie-hellman-group18-sha512', 'diffie-hellman-group-exchange-sha256'], 'ciphers': ['chacha20-poly1305@openssh.com', 'aes256-gcm@openssh.com', 'aes128-gcm@openssh.com', 'aes256-ctr', 'aes192-ctr', 'aes128-ctr'], 'macs': ['hmac-sha2-256-etm@openssh.com', 'hmac-sha2-512-etm@openssh.com', 'umac-128-etm@openssh.com'], 'hostkey_sizes': None, 'cakey_sizes': None, 'dh_modulus_sizes': {'diffie-hellman-group-exchange-sha256': 2048}, 'server_policy': True},
+
+ 'Hardened OpenSSH Server v7.9 (version 1)': {'version': '1', 'banner': None, 'compressions': None, 'host_keys': ['rsa-sha2-512', 'rsa-sha2-256', 'ssh-ed25519'], 'optional_host_keys': ['ssh-ed25519-cert-v01@openssh.com', 'rsa-sha2-256-cert-v01@openssh.com', 'rsa-sha2-512-cert-v01@openssh.com'], 'kex': ['curve25519-sha256', 'curve25519-sha256@libssh.org', 'diffie-hellman-group16-sha512', 'diffie-hellman-group18-sha512', 'diffie-hellman-group-exchange-sha256'], 'ciphers': ['chacha20-poly1305@openssh.com', 'aes256-gcm@openssh.com', 'aes128-gcm@openssh.com', 'aes256-ctr', 'aes192-ctr', 'aes128-ctr'], 'macs': ['hmac-sha2-256-etm@openssh.com', 'hmac-sha2-512-etm@openssh.com', 'umac-128-etm@openssh.com'], 'hostkey_sizes': None, 'cakey_sizes': None, 'dh_modulus_sizes': {'diffie-hellman-group-exchange-sha256': 2048}, 'server_policy': True},
+
+ 'Hardened OpenSSH Server v8.0 (version 1)': {'version': '1', 'banner': None, 'compressions': None, 'host_keys': ['rsa-sha2-512', 'rsa-sha2-256', 'ssh-ed25519'], 'optional_host_keys': ['ssh-ed25519-cert-v01@openssh.com', 'rsa-sha2-256-cert-v01@openssh.com', 'rsa-sha2-512-cert-v01@openssh.com'], 'kex': ['curve25519-sha256', 'curve25519-sha256@libssh.org', 'diffie-hellman-group16-sha512', 'diffie-hellman-group18-sha512', 'diffie-hellman-group-exchange-sha256'], 'ciphers': ['chacha20-poly1305@openssh.com', 'aes256-gcm@openssh.com', 'aes128-gcm@openssh.com', 'aes256-ctr', 'aes192-ctr', 'aes128-ctr'], 'macs': ['hmac-sha2-256-etm@openssh.com', 'hmac-sha2-512-etm@openssh.com', 'umac-128-etm@openssh.com'], 'hostkey_sizes': None, 'cakey_sizes': None, 'dh_modulus_sizes': {'diffie-hellman-group-exchange-sha256': 2048}, 'server_policy': True},
+
+ 'Hardened OpenSSH Server v8.1 (version 1)': {'version': '1', 'banner': None, 'compressions': None, 'host_keys': ['rsa-sha2-512', 'rsa-sha2-256', 'ssh-ed25519'], 'optional_host_keys': ['sk-ssh-ed25519@openssh.com', 'ssh-ed25519-cert-v01@openssh.com', 'sk-ssh-ed25519-cert-v01@openssh.com', 'rsa-sha2-256-cert-v01@openssh.com', 'rsa-sha2-512-cert-v01@openssh.com'], 'kex': ['curve25519-sha256', 'curve25519-sha256@libssh.org', 'diffie-hellman-group16-sha512', 'diffie-hellman-group18-sha512', 'diffie-hellman-group-exchange-sha256'], 'ciphers': ['chacha20-poly1305@openssh.com', 'aes256-gcm@openssh.com', 'aes128-gcm@openssh.com', 'aes256-ctr', 'aes192-ctr', 'aes128-ctr'], 'macs': ['hmac-sha2-256-etm@openssh.com', 'hmac-sha2-512-etm@openssh.com', 'umac-128-etm@openssh.com'], 'hostkey_sizes': None, 'cakey_sizes': None, 'dh_modulus_sizes': {'diffie-hellman-group-exchange-sha256': 2048}, 'server_policy': True},
+
+ 'Hardened OpenSSH Server v8.2 (version 1)': {'version': '1', 'banner': None, 'compressions': None, 'host_keys': ['rsa-sha2-512', 'rsa-sha2-256', 'ssh-ed25519'], 'optional_host_keys': ['sk-ssh-ed25519@openssh.com', 'ssh-ed25519-cert-v01@openssh.com', 'sk-ssh-ed25519-cert-v01@openssh.com', 'rsa-sha2-256-cert-v01@openssh.com', 'rsa-sha2-512-cert-v01@openssh.com'], 'kex': ['curve25519-sha256', 'curve25519-sha256@libssh.org', 'diffie-hellman-group16-sha512', 'diffie-hellman-group18-sha512', 'diffie-hellman-group-exchange-sha256'], 'ciphers': ['chacha20-poly1305@openssh.com', 'aes256-gcm@openssh.com', 'aes128-gcm@openssh.com', 'aes256-ctr', 'aes192-ctr', 'aes128-ctr'], 'macs': ['hmac-sha2-256-etm@openssh.com', 'hmac-sha2-512-etm@openssh.com', 'umac-128-etm@openssh.com'], 'hostkey_sizes': {'rsa-sha2-256': 4096, 'rsa-sha2-512': 4096}, 'cakey_sizes': {'rsa-sha2-256-cert-v01@openssh.com': 4096, 'rsa-sha2-512-cert-v01@openssh.com': 4096}, 'dh_modulus_sizes': {'diffie-hellman-group-exchange-sha256': 2048}, 'server_policy': True},
+
+ 'Hardened OpenSSH Server v8.3 (version 1)': {'version': '1', 'banner': None, 'compressions': None, 'host_keys': ['rsa-sha2-512', 'rsa-sha2-256', 'ssh-ed25519'], 'optional_host_keys': ['sk-ssh-ed25519@openssh.com', 'ssh-ed25519-cert-v01@openssh.com', 'sk-ssh-ed25519-cert-v01@openssh.com', 'rsa-sha2-256-cert-v01@openssh.com', 'rsa-sha2-512-cert-v01@openssh.com'], 'kex': ['curve25519-sha256', 'curve25519-sha256@libssh.org', 'diffie-hellman-group16-sha512', 'diffie-hellman-group18-sha512', 'diffie-hellman-group-exchange-sha256'], 'ciphers': ['chacha20-poly1305@openssh.com', 'aes256-gcm@openssh.com', 'aes128-gcm@openssh.com', 'aes256-ctr', 'aes192-ctr', 'aes128-ctr'], 'macs': ['hmac-sha2-256-etm@openssh.com', 'hmac-sha2-512-etm@openssh.com', 'umac-128-etm@openssh.com'], 'hostkey_sizes': {'rsa-sha2-256': 4096, 'rsa-sha2-512': 4096}, 'cakey_sizes': {'rsa-sha2-256-cert-v01@openssh.com': 4096, 'rsa-sha2-512-cert-v01@openssh.com': 4096}, 'dh_modulus_sizes': {'diffie-hellman-group-exchange-sha256': 2048}, 'server_policy': True},
+
+ 'Hardened OpenSSH Server v8.4 (version 1)': {'version': '1', 'banner': None, 'compressions': None, 'host_keys': ['rsa-sha2-512', 'rsa-sha2-256', 'ssh-ed25519'], 'optional_host_keys': ['sk-ssh-ed25519@openssh.com', 'ssh-ed25519-cert-v01@openssh.com', 'sk-ssh-ed25519-cert-v01@openssh.com', 'rsa-sha2-256-cert-v01@openssh.com', 'rsa-sha2-512-cert-v01@openssh.com'], 'kex': ['curve25519-sha256', 'curve25519-sha256@libssh.org', 'diffie-hellman-group16-sha512', 'diffie-hellman-group18-sha512', 'diffie-hellman-group-exchange-sha256'], 'ciphers': ['chacha20-poly1305@openssh.com', 'aes256-gcm@openssh.com', 'aes128-gcm@openssh.com', 'aes256-ctr', 'aes192-ctr', 'aes128-ctr'], 'macs': ['hmac-sha2-256-etm@openssh.com', 'hmac-sha2-512-etm@openssh.com', 'umac-128-etm@openssh.com'], 'hostkey_sizes': {'rsa-sha2-256': 4096, 'rsa-sha2-512': 4096}, 'cakey_sizes': {'rsa-sha2-256-cert-v01@openssh.com': 4096, 'rsa-sha2-512-cert-v01@openssh.com': 4096}, 'dh_modulus_sizes': {'diffie-hellman-group-exchange-sha256': 2048}, 'server_policy': True},
+
+ 'Hardened OpenSSH Server v8.5 (version 1)': {'version': '1', 'banner': None, 'compressions': None, 'host_keys': ['rsa-sha2-512', 'rsa-sha2-256', 'ssh-ed25519'], 'optional_host_keys': ['sk-ssh-ed25519@openssh.com', 'ssh-ed25519-cert-v01@openssh.com', 'sk-ssh-ed25519-cert-v01@openssh.com', 'rsa-sha2-256-cert-v01@openssh.com', 'rsa-sha2-512-cert-v01@openssh.com'], 'kex': ['curve25519-sha256', 'curve25519-sha256@libssh.org', 'diffie-hellman-group16-sha512', 'diffie-hellman-group18-sha512', 'diffie-hellman-group-exchange-sha256'], 'ciphers': ['chacha20-poly1305@openssh.com', 'aes256-gcm@openssh.com', 'aes128-gcm@openssh.com', 'aes256-ctr', 'aes192-ctr', 'aes128-ctr'], 'macs': ['hmac-sha2-256-etm@openssh.com', 'hmac-sha2-512-etm@openssh.com', 'umac-128-etm@openssh.com'], 'hostkey_sizes': {'rsa-sha2-256': 4096, 'rsa-sha2-512': 4096}, 'cakey_sizes': {'rsa-sha2-256-cert-v01@openssh.com': 4096, 'rsa-sha2-512-cert-v01@openssh.com': 4096}, 'dh_modulus_sizes': {'diffie-hellman-group-exchange-sha256': 2048}, 'server_policy': True},
+
+ 'Hardened OpenSSH Server v8.6 (version 1)': {'version': '1', 'banner': None, 'compressions': None, 'host_keys': ['rsa-sha2-512', 'rsa-sha2-256', 'ssh-ed25519'], 'optional_host_keys': ['sk-ssh-ed25519@openssh.com', 'ssh-ed25519-cert-v01@openssh.com', 'sk-ssh-ed25519-cert-v01@openssh.com', 'rsa-sha2-256-cert-v01@openssh.com', 'rsa-sha2-512-cert-v01@openssh.com'], 'kex': ['curve25519-sha256', 'curve25519-sha256@libssh.org', 'diffie-hellman-group16-sha512', 'diffie-hellman-group18-sha512', 'diffie-hellman-group-exchange-sha256'], 'ciphers': ['chacha20-poly1305@openssh.com', 'aes256-gcm@openssh.com', 'aes128-gcm@openssh.com', 'aes256-ctr', 'aes192-ctr', 'aes128-ctr'], 'macs': ['hmac-sha2-256-etm@openssh.com', 'hmac-sha2-512-etm@openssh.com', 'umac-128-etm@openssh.com'], 'hostkey_sizes': {'rsa-sha2-256': 4096, 'rsa-sha2-512': 4096}, 'cakey_sizes': {'rsa-sha2-256-cert-v01@openssh.com': 4096, 'rsa-sha2-512-cert-v01@openssh.com': 4096}, 'dh_modulus_sizes': {'diffie-hellman-group-exchange-sha256': 2048}, 'server_policy': True},
+
+ 'Hardened OpenSSH Server v8.7 (version 1)': {'version': '1', 'banner': None, 'compressions': None, 'host_keys': ['rsa-sha2-512', 'rsa-sha2-256', 'ssh-ed25519'], 'optional_host_keys': ['sk-ssh-ed25519@openssh.com', 'ssh-ed25519-cert-v01@openssh.com', 'sk-ssh-ed25519-cert-v01@openssh.com', 'rsa-sha2-256-cert-v01@openssh.com', 'rsa-sha2-512-cert-v01@openssh.com'], 'kex': ['curve25519-sha256', 'curve25519-sha256@libssh.org', 'diffie-hellman-group16-sha512', 'diffie-hellman-group18-sha512', 'diffie-hellman-group-exchange-sha256'], 'ciphers': ['chacha20-poly1305@openssh.com', 'aes256-gcm@openssh.com', 'aes128-gcm@openssh.com', 'aes256-ctr', 'aes192-ctr', 'aes128-ctr'], 'macs': ['hmac-sha2-256-etm@openssh.com', 'hmac-sha2-512-etm@openssh.com', 'umac-128-etm@openssh.com'], 'hostkey_sizes': {'rsa-sha2-256': 4096, 'rsa-sha2-512': 4096}, 'cakey_sizes': {'rsa-sha2-256-cert-v01@openssh.com': 4096, 'rsa-sha2-512-cert-v01@openssh.com': 4096}, 'dh_modulus_sizes': {'diffie-hellman-group-exchange-sha256': 2048}, 'server_policy': True},
+
+
+ # Ubuntu Client policies
+
+ 'Hardened Ubuntu Client 16.04 LTS (version 2)': {'version': '2', 'banner': None, 'compressions': None, 'host_keys': ['ssh-ed25519', 'ssh-ed25519-cert-v01@openssh.com', 'rsa-sha2-256', 'rsa-sha2-512'], 'optional_host_keys': None, 'kex': ['curve25519-sha256@libssh.org', 'diffie-hellman-group-exchange-sha256', 'ext-info-c'], 'ciphers': ['chacha20-poly1305@openssh.com', 'aes256-gcm@openssh.com', 'aes128-gcm@openssh.com', 'aes256-ctr', 'aes192-ctr', 'aes128-ctr'], 'macs': ['hmac-sha2-256-etm@openssh.com', 'hmac-sha2-512-etm@openssh.com', 'umac-128-etm@openssh.com'], 'hostkey_sizes': None, 'cakey_sizes': None, 'dh_modulus_sizes': None, 'server_policy': False},
+
+ 'Hardened Ubuntu Client 18.04 LTS (version 2)': {'version': '2', 'banner': None, 'compressions': None, 'host_keys': ['ssh-ed25519', 'ssh-ed25519-cert-v01@openssh.com', 'rsa-sha2-256', 'rsa-sha2-512'], 'optional_host_keys': None, 'kex': ['curve25519-sha256', 'curve25519-sha256@libssh.org', 'diffie-hellman-group16-sha512', 'diffie-hellman-group18-sha512', 'diffie-hellman-group-exchange-sha256', 'ext-info-c'], 'ciphers': ['chacha20-poly1305@openssh.com', 'aes256-gcm@openssh.com', 'aes128-gcm@openssh.com', 'aes256-ctr', 'aes192-ctr', 'aes128-ctr'], 'macs': ['hmac-sha2-256-etm@openssh.com', 'hmac-sha2-512-etm@openssh.com', 'umac-128-etm@openssh.com'], 'hostkey_sizes': None, 'cakey_sizes': None, 'dh_modulus_sizes': None, 'server_policy': False},
+
+ 'Hardened Ubuntu Client 20.04 LTS (version 2)': {'version': '2', 'banner': None, 'compressions': None, 'host_keys': ['ssh-ed25519', 'ssh-ed25519-cert-v01@openssh.com', 'sk-ssh-ed25519@openssh.com', 'sk-ssh-ed25519-cert-v01@openssh.com', 'rsa-sha2-256', 'rsa-sha2-256-cert-v01@openssh.com', 'rsa-sha2-512', 'rsa-sha2-512-cert-v01@openssh.com'], 'optional_host_keys': None, 'kex': ['curve25519-sha256', 'curve25519-sha256@libssh.org', 'diffie-hellman-group16-sha512', 'diffie-hellman-group18-sha512', 'diffie-hellman-group-exchange-sha256', 'ext-info-c'], 'ciphers': ['chacha20-poly1305@openssh.com', 'aes256-gcm@openssh.com', 'aes128-gcm@openssh.com', 'aes256-ctr', 'aes192-ctr', 'aes128-ctr'], 'macs': ['hmac-sha2-256-etm@openssh.com', 'hmac-sha2-512-etm@openssh.com', 'umac-128-etm@openssh.com'], 'hostkey_sizes': None, 'cakey_sizes': None, 'dh_modulus_sizes': None, 'server_policy': False},
+
+ }
+
+
+ def __init__(self, policy_file: Optional[str] = None, policy_data: Optional[str] = None, manual_load: bool = False) -> None:
+ self._name: Optional[str] = None
+ self._version: Optional[str] = None
+ self._banner: Optional[str] = None
+ self._compressions: Optional[List[str]] = None
+ self._host_keys: Optional[List[str]] = None
+ self._optional_host_keys: Optional[List[str]] = None
+ self._kex: Optional[List[str]] = None
+ self._ciphers: Optional[List[str]] = None
+ self._macs: Optional[List[str]] = None
+ self._hostkey_sizes: Optional[Dict[str, int]] = None
+ self._cakey_sizes: Optional[Dict[str, int]] = None
+ self._dh_modulus_sizes: Optional[Dict[str, int]] = None
+ self._server_policy = True
+
+ self._name_and_version: str = ''
+
+ # Ensure that only one mode was specified.
+ num_modes = 0
+ if policy_file is not None:
+ num_modes += 1
+ if policy_data is not None:
+ num_modes += 1
+ if manual_load is True:
+ num_modes += 1
+
+ if num_modes != 1:
+ raise RuntimeError('Exactly one of the following can be specified only: policy_file, policy_data, or manual_load')
+
+ if manual_load:
+ return
+
+ if policy_file is not None:
+ try:
+ with open(policy_file, "r", encoding='utf-8') as f:
+ policy_data = f.read()
+ except FileNotFoundError:
+ print("Error: policy file not found: %s" % policy_file)
+ sys.exit(exitcodes.UNKNOWN_ERROR)
+
+ lines = []
+ if policy_data is not None:
+ lines = policy_data.split("\n")
+
+ for line in lines:
+ line = line.strip()
+ if (len(line) == 0) or line.startswith('#'):
+ continue
+
+ key = None
+ val = None
+ try:
+ key, val = line.split('=')
+ except ValueError as ve:
+ raise ValueError("could not parse line: %s" % line) from ve
+
+ key = key.strip()
+ val = val.strip()
+
+ if key not in ['name', 'version', 'banner', 'compressions', 'host keys', 'optional host keys', 'key exchanges', 'ciphers', 'macs', 'client policy'] and not key.startswith('hostkey_size_') and not key.startswith('cakey_size_') and not key.startswith('dh_modulus_size_'):
+ raise ValueError("invalid field found in policy: %s" % line)
+
+ if key in ['name', 'banner']:
+
+ # If the banner value is blank, set it to "" so that the code below handles it.
+ if len(val) < 2:
+ val = "\"\""
+
+ if (val[0] != '"') or (val[-1] != '"'):
+ raise ValueError('the value for the %s field must be enclosed in quotes: %s' % (key, val))
+
+ # Remove the surrounding quotes, and unescape quotes & newlines.
+ val = val[1:-1]. replace("\\\"", "\"").replace("\\n", "\n")
+
+ if key == 'name':
+ self._name = val
+ elif key == 'banner':
+ self._banner = val
+ elif key == 'version':
+ self._version = val
+ elif key in ['compressions', 'host keys', 'optional host keys', 'key exchanges', 'ciphers', 'macs']:
+ try:
+ algs = val.split(',')
+ except ValueError:
+ # If the value has no commas, then set the algorithm list to just the value.
+ algs = [val]
+
+ # Strip whitespace in each algorithm name.
+ algs = [alg.strip() for alg in algs]
+
+ if key == 'compressions':
+ self._compressions = algs
+ elif key == 'host keys':
+ self._host_keys = algs
+ elif key == 'optional host keys':
+ self._optional_host_keys = algs
+ elif key == 'key exchanges':
+ self._kex = algs
+ elif key == 'ciphers':
+ self._ciphers = algs
+ elif key == 'macs':
+ self._macs = algs
+ elif key.startswith('hostkey_size_'):
+ hostkey_type = key[13:]
+ if self._hostkey_sizes is None:
+ self._hostkey_sizes = {}
+ self._hostkey_sizes[hostkey_type] = int(val)
+ elif key.startswith('cakey_size_'):
+ cakey_type = key[11:]
+ if self._cakey_sizes is None:
+ self._cakey_sizes = {}
+ self._cakey_sizes[cakey_type] = int(val)
+ elif key.startswith('dh_modulus_size_'):
+ dh_modulus_type = key[16:]
+ if self._dh_modulus_sizes is None:
+ self._dh_modulus_sizes = {}
+ self._dh_modulus_sizes[dh_modulus_type] = int(val)
+ elif key.startswith('client policy') and val.lower() == 'true':
+ self._server_policy = False
+
+
+ if self._name is None:
+ raise ValueError('The policy does not have a name field.')
+ if self._version is None:
+ raise ValueError('The policy does not have a version field.')
+
+ self._name_and_version = "%s (version %s)" % (self._name, self._version)
+
+
+ @staticmethod
+ def _append_error(errors: List[Any], mismatched_field: str, expected_required: Optional[List[str]], expected_optional: Optional[List[str]], actual: List[str]) -> None:
+ if expected_required is None:
+ expected_required = ['']
+ if expected_optional is None:
+ expected_optional = ['']
+ errors.append({'mismatched_field': mismatched_field, 'expected_required': expected_required, 'expected_optional': expected_optional, 'actual': actual})
+
+
+ @staticmethod
+ def create(source: Optional[str], banner: Optional['Banner'], kex: Optional['SSH2_Kex'], client_audit: bool) -> str:
+ '''Creates a policy based on a server configuration. Returns a string.'''
+
+ today = date.today().strftime('%Y/%m/%d')
+ compressions = None
+ host_keys = None
+ kex_algs = None
+ ciphers = None
+ macs = None
+ rsa_hostkey_sizes_str = ''
+ rsa_cakey_sizes_str = ''
+ dh_modulus_sizes_str = ''
+ client_policy_str = ''
+
+ if client_audit:
+ client_policy_str = "\n# Set to true to signify this is a policy for clients, not servers.\nclient policy = true\n"
+
+ if kex is not None:
+ if kex.server.compression is not None:
+ compressions = ', '.join(kex.server.compression)
+ if kex.key_algorithms is not None:
+ host_keys = ', '.join(kex.key_algorithms)
+ if kex.kex_algorithms is not None:
+ kex_algs = ', '.join(kex.kex_algorithms)
+ if kex.server.encryption is not None:
+ ciphers = ', '.join(kex.server.encryption)
+ if kex.server.mac is not None:
+ macs = ', '.join(kex.server.mac)
+ if kex.rsa_key_sizes():
+ rsa_key_sizes_dict = kex.rsa_key_sizes()
+ for host_key_type in sorted(rsa_key_sizes_dict):
+ hostkey_size, cakey_size = rsa_key_sizes_dict[host_key_type]
+
+ rsa_hostkey_sizes_str = "%shostkey_size_%s = %d\n" % (rsa_hostkey_sizes_str, host_key_type, hostkey_size)
+ if cakey_size != -1:
+ rsa_cakey_sizes_str = "%scakey_size_%s = %d\n" % (rsa_cakey_sizes_str, host_key_type, cakey_size)
+
+ if len(rsa_hostkey_sizes_str) > 0:
+ rsa_hostkey_sizes_str = "\n# RSA host key sizes.\n%s" % rsa_hostkey_sizes_str
+ if len(rsa_cakey_sizes_str) > 0:
+ rsa_cakey_sizes_str = "\n# RSA CA key sizes.\n%s" % rsa_cakey_sizes_str
+ if kex.dh_modulus_sizes():
+ dh_modulus_sizes_dict = kex.dh_modulus_sizes()
+ for gex_type in sorted(dh_modulus_sizes_dict):
+ modulus_size, _ = dh_modulus_sizes_dict[gex_type]
+ dh_modulus_sizes_str = "%sdh_modulus_size_%s = %d\n" % (dh_modulus_sizes_str, gex_type, modulus_size)
+ if len(dh_modulus_sizes_str) > 0:
+ dh_modulus_sizes_str = "\n# Group exchange DH modulus sizes.\n%s" % dh_modulus_sizes_str
+
+
+ policy_data = '''#
+# Custom policy based on %s (created on %s)
+#
+%s
+# The name of this policy (displayed in the output during scans). Must be in quotes.
+name = "Custom Policy (based on %s on %s)"
+
+# The version of this policy (displayed in the output during scans). Not parsed, and may be any value, including strings.
+version = 1
+
+# The banner that must match exactly. Commented out to ignore banners, since minor variability in the banner is sometimes normal.
+# banner = "%s"
+
+# The compression options that must match exactly (order matters). Commented out to ignore by default.
+# compressions = %s
+%s%s%s
+# The host key types that must match exactly (order matters).
+host keys = %s
+
+# Host key types that may optionally appear.
+#optional host keys = ssh-ed25519-cert-v01@openssh.com,sk-ssh-ed25519@openssh.com,sk-ssh-ed25519-cert-v01@openssh.com,rsa-sha2-256-cert-v01@openssh.com,rsa-sha2-512-cert-v01@openssh.com
+
+# The key exchange algorithms that must match exactly (order matters).
+key exchanges = %s
+
+# The ciphers that must match exactly (order matters).
+ciphers = %s
+
+# The MACs that must match exactly (order matters).
+macs = %s
+''' % (source, today, client_policy_str, source, today, banner, compressions, rsa_hostkey_sizes_str, rsa_cakey_sizes_str, dh_modulus_sizes_str, host_keys, kex_algs, ciphers, macs)
+
+ return policy_data
+
+
+ def evaluate(self, banner: Optional['Banner'], kex: Optional['SSH2_Kex']) -> Tuple[bool, List[Dict[str, str]], str]:
+ '''Evaluates a server configuration against this policy. Returns a tuple of a boolean (True if server adheres to policy) and an array of strings that holds error messages.'''
+
+ ret = True
+ errors: List[Any] = []
+
+ banner_str = str(banner)
+ if (self._banner is not None) and (banner_str != self._banner):
+ ret = False
+ self._append_error(errors, 'Banner', [self._banner], None, [banner_str])
+
+ # All subsequent tests require a valid kex, so end here if we don't have one.
+ if kex is None:
+ return ret, errors, self._get_error_str(errors)
+
+ if (self._compressions is not None) and (kex.server.compression != self._compressions):
+ ret = False
+ self._append_error(errors, 'Compression', self._compressions, None, kex.server.compression)
+
+ # If a list of optional host keys was given in the policy, remove any of its entries from the list retrieved from the server. This allows us to do an exact comparison with the expected list below.
+ pruned_host_keys = kex.key_algorithms
+ if self._optional_host_keys is not None:
+ pruned_host_keys = [x for x in kex.key_algorithms if x not in self._optional_host_keys]
+
+ if (self._host_keys is not None) and (pruned_host_keys != self._host_keys):
+ ret = False
+ self._append_error(errors, 'Host keys', self._host_keys, self._optional_host_keys, kex.key_algorithms)
+
+ if self._hostkey_sizes is not None:
+ hostkey_types = list(self._hostkey_sizes.keys())
+ hostkey_types.sort() # Sorted to make testing output repeatable.
+ for hostkey_type in hostkey_types:
+ expected_hostkey_size = self._hostkey_sizes[hostkey_type]
+ if hostkey_type in kex.rsa_key_sizes():
+ actual_hostkey_size, actual_cakey_size = kex.rsa_key_sizes()[hostkey_type]
+ if actual_hostkey_size != expected_hostkey_size:
+ ret = False
+ self._append_error(errors, 'RSA host key (%s) sizes' % hostkey_type, [str(expected_hostkey_size)], None, [str(actual_hostkey_size)])
+
+ if self._cakey_sizes is not None:
+ hostkey_types = list(self._cakey_sizes.keys())
+ hostkey_types.sort() # Sorted to make testing output repeatable.
+ for hostkey_type in hostkey_types:
+ expected_cakey_size = self._cakey_sizes[hostkey_type]
+ if hostkey_type in kex.rsa_key_sizes():
+ actual_hostkey_size, actual_cakey_size = kex.rsa_key_sizes()[hostkey_type]
+ if actual_cakey_size != expected_cakey_size:
+ ret = False
+ self._append_error(errors, 'RSA CA key (%s) sizes' % hostkey_type, [str(expected_cakey_size)], None, [str(actual_cakey_size)])
+
+ if kex.kex_algorithms != self._kex:
+ ret = False
+ self._append_error(errors, 'Key exchanges', self._kex, None, kex.kex_algorithms)
+
+ if (self._ciphers is not None) and (kex.server.encryption != self._ciphers):
+ ret = False
+ self._append_error(errors, 'Ciphers', self._ciphers, None, kex.server.encryption)
+
+ if (self._macs is not None) and (kex.server.mac != self._macs):
+ ret = False
+ self._append_error(errors, 'MACs', self._macs, None, kex.server.mac)
+
+ if self._dh_modulus_sizes is not None:
+ dh_modulus_types = list(self._dh_modulus_sizes.keys())
+ dh_modulus_types.sort() # Sorted to make testing output repeatable.
+ for dh_modulus_type in dh_modulus_types:
+ expected_dh_modulus_size = self._dh_modulus_sizes[dh_modulus_type]
+ if dh_modulus_type in kex.dh_modulus_sizes():
+ actual_dh_modulus_size, _ = kex.dh_modulus_sizes()[dh_modulus_type]
+ if expected_dh_modulus_size != actual_dh_modulus_size:
+ ret = False
+ self._append_error(errors, 'Group exchange (%s) modulus sizes' % dh_modulus_type, [str(expected_dh_modulus_size)], None, [str(actual_dh_modulus_size)])
+
+ return ret, errors, self._get_error_str(errors)
+
+
+ @staticmethod
+ def _get_error_str(errors: List[Any]) -> str:
+ '''Transforms an error struct to a flat string of error messages.'''
+
+ error_list = []
+ spacer = ''
+ for e in errors:
+ e_str = " * %s did not match.\n" % e['mismatched_field']
+ if ('expected_optional' in e) and (e['expected_optional'] != ['']):
+ e_str += " - Expected (required): %s\n - Expected (optional): %s\n" % (Policy._normalize_error_field(e['expected_required']), Policy._normalize_error_field(e['expected_optional']))
+ spacer = ' '
+ else:
+ e_str += " - Expected: %s\n" % Policy._normalize_error_field(e['expected_required'])
+ spacer = ' '
+ e_str += " - Actual:%s%s\n" % (spacer, Policy._normalize_error_field(e['actual']))
+ error_list.append(e_str)
+
+ error_list.sort() # To ensure repeatable results for testing.
+
+ error_str = ''
+ if len(error_list) > 0:
+ error_str = "\n".join(error_list)
+
+ return error_str
+
+
+ def get_name_and_version(self) -> str:
+ '''Returns a string of this Policy's name and version.'''
+ return self._name_and_version
+
+
+ def is_server_policy(self) -> bool:
+ '''Returns True if this is a server policy, or False if this is a client policy.'''
+ return self._server_policy
+
+
+ @staticmethod
+ def list_builtin_policies() -> Tuple[List[str], List[str]]:
+ '''Returns two lists: a list of names of built-in server policies, and a list of names of built-in client policies, respectively.'''
+ server_policy_names = []
+ client_policy_names = []
+
+ for policy_name, policy in Policy.BUILTIN_POLICIES.items():
+ if policy['server_policy']:
+ server_policy_names.append(policy_name)
+ else:
+ client_policy_names.append(policy_name)
+
+ server_policy_names.sort()
+ client_policy_names.sort()
+ return server_policy_names, client_policy_names
+
+
+ @staticmethod
+ def load_builtin_policy(policy_name: str) -> Optional['Policy']:
+ '''Returns a Policy with the specified built-in policy name loaded, or None if no policy of that name exists.'''
+ p = None
+ if policy_name in Policy.BUILTIN_POLICIES:
+ policy_struct = Policy.BUILTIN_POLICIES[policy_name]
+ p = Policy(manual_load=True)
+ policy_name_without_version = policy_name[0:policy_name.rfind(' (')]
+ p._name = policy_name_without_version # pylint: disable=protected-access
+ p._version = cast(str, policy_struct['version']) # pylint: disable=protected-access
+ p._banner = cast(Optional[str], policy_struct['banner']) # pylint: disable=protected-access
+ p._compressions = cast(Optional[List[str]], policy_struct['compressions']) # pylint: disable=protected-access
+ p._host_keys = cast(Optional[List[str]], policy_struct['host_keys']) # pylint: disable=protected-access
+ p._optional_host_keys = cast(Optional[List[str]], policy_struct['optional_host_keys']) # pylint: disable=protected-access
+ p._kex = cast(Optional[List[str]], policy_struct['kex']) # pylint: disable=protected-access
+ p._ciphers = cast(Optional[List[str]], policy_struct['ciphers']) # pylint: disable=protected-access
+ p._macs = cast(Optional[List[str]], policy_struct['macs']) # pylint: disable=protected-access
+ p._hostkey_sizes = cast(Optional[Dict[str, int]], policy_struct['hostkey_sizes']) # pylint: disable=protected-access
+ p._cakey_sizes = cast(Optional[Dict[str, int]], policy_struct['cakey_sizes']) # pylint: disable=protected-access
+ p._dh_modulus_sizes = cast(Optional[Dict[str, int]], policy_struct['dh_modulus_sizes']) # pylint: disable=protected-access
+ p._server_policy = cast(bool, policy_struct['server_policy']) # pylint: disable=protected-access
+
+ p._name_and_version = "%s (version %s)" % (p._name, p._version) # pylint: disable=protected-access
+
+ return p
+
+
+ @staticmethod
+ def _normalize_error_field(field: List[str]) -> Any:
+ '''If field is an array with a string parsable as an integer, return that integer. Otherwise, return the field joined with commas.'''
+ if len(field) == 1:
+ try:
+ return int(field[0])
+ except ValueError:
+ return field[0]
+ else:
+ return ', '.join(field)
+
+
+ def __str__(self) -> str:
+ undefined = '{undefined}'
+
+ name = undefined
+ version = undefined
+ banner = undefined
+ compressions_str = undefined
+ host_keys_str = undefined
+ optional_host_keys_str = undefined
+ kex_str = undefined
+ ciphers_str = undefined
+ macs_str = undefined
+ hostkey_sizes_str = undefined
+ cakey_sizes_str = undefined
+ dh_modulus_sizes_str = undefined
+
+
+ if self._name is not None:
+ name = '[%s]' % self._name
+ if self._version is not None:
+ version = '[%s]' % self._version
+ if self._banner is not None:
+ banner = '[%s]' % self._banner
+
+ if self._compressions is not None:
+ compressions_str = ', '.join(self._compressions)
+ if self._host_keys is not None:
+ host_keys_str = ', '.join(self._host_keys)
+ if self._optional_host_keys is not None:
+ optional_host_keys_str = ', '.join(self._optional_host_keys)
+ if self._kex is not None:
+ kex_str = ', '.join(self._kex)
+ if self._ciphers is not None:
+ ciphers_str = ', '.join(self._ciphers)
+ if self._macs is not None:
+ macs_str = ', '.join(self._macs)
+ if self._hostkey_sizes is not None:
+ hostkey_sizes_str = str(self._hostkey_sizes)
+ if self._cakey_sizes is not None:
+ cakey_sizes_str = str(self._cakey_sizes)
+ if self._dh_modulus_sizes is not None:
+ dh_modulus_sizes_str = str(self._dh_modulus_sizes)
+
+ return "Name: %s\nVersion: %s\nBanner: %s\nCompressions: %s\nHost Keys: %s\nOptional Host Keys: %s\nKey Exchanges: %s\nCiphers: %s\nMACs: %s\nHost Key Sizes: %s\nCA Key Sizes: %s\nDH Modulus Sizes: %s\nServer Policy: %r" % (name, version, banner, compressions_str, host_keys_str, optional_host_keys_str, kex_str, ciphers_str, macs_str, hostkey_sizes_str, cakey_sizes_str, dh_modulus_sizes_str, self._server_policy)
diff --git a/src/ssh_audit/product.py b/src/ssh_audit/product.py
new file mode 100644
index 0000000..2d8ae06
--- /dev/null
+++ b/src/ssh_audit/product.py
@@ -0,0 +1,31 @@
+"""
+ The MIT License (MIT)
+
+ Copyright (C) 2017 Andris Raugulis (moo@arthepsy.eu)
+
+ Permission is hereby granted, free of charge, to any person obtaining a copy
+ of this software and associated documentation files (the "Software"), to deal
+ in the Software without restriction, including without limitation the rights
+ to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+ copies of the Software, and to permit persons to whom the Software is
+ furnished to do so, subject to the following conditions:
+
+ The above copyright notice and this permission notice shall be included in
+ all copies or substantial portions of the Software.
+
+ THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+ IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+ FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+ AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+ LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+ OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+ THE SOFTWARE.
+"""
+
+
+class Product: # pylint: disable=too-few-public-methods
+ OpenSSH = 'OpenSSH'
+ DropbearSSH = 'Dropbear SSH'
+ LibSSH = 'libssh'
+ TinySSH = 'TinySSH'
+ PuTTY = 'PuTTY'
diff --git a/src/ssh_audit/protocol.py b/src/ssh_audit/protocol.py
new file mode 100644
index 0000000..1b1b216
--- /dev/null
+++ b/src/ssh_audit/protocol.py
@@ -0,0 +1,36 @@
+"""
+ The MIT License (MIT)
+
+ Copyright (C) 2017 Andris Raugulis (moo@arthepsy.eu)
+
+ Permission is hereby granted, free of charge, to any person obtaining a copy
+ of this software and associated documentation files (the "Software"), to deal
+ in the Software without restriction, including without limitation the rights
+ to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+ copies of the Software, and to permit persons to whom the Software is
+ furnished to do so, subject to the following conditions:
+
+ The above copyright notice and this permission notice shall be included in
+ all copies or substantial portions of the Software.
+
+ THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+ IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+ FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+ AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+ LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+ OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+ THE SOFTWARE.
+"""
+
+
+class Protocol: # pylint: disable=too-few-public-methods
+ SMSG_PUBLIC_KEY = 2
+ MSG_DEBUG = 4
+ MSG_KEXINIT = 20
+ MSG_NEWKEYS = 21
+ MSG_KEXDH_INIT = 30
+ MSG_KEXDH_REPLY = 31
+ MSG_KEXDH_GEX_REQUEST = 34
+ MSG_KEXDH_GEX_GROUP = 31
+ MSG_KEXDH_GEX_INIT = 32
+ MSG_KEXDH_GEX_REPLY = 33
diff --git a/src/ssh_audit/readbuf.py b/src/ssh_audit/readbuf.py
new file mode 100644
index 0000000..c9405c4
--- /dev/null
+++ b/src/ssh_audit/readbuf.py
@@ -0,0 +1,92 @@
+"""
+ The MIT License (MIT)
+
+ Copyright (C) 2017 Andris Raugulis (moo@arthepsy.eu)
+
+ Permission is hereby granted, free of charge, to any person obtaining a copy
+ of this software and associated documentation files (the "Software"), to deal
+ in the Software without restriction, including without limitation the rights
+ to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+ copies of the Software, and to permit persons to whom the Software is
+ furnished to do so, subject to the following conditions:
+
+ The above copyright notice and this permission notice shall be included in
+ all copies or substantial portions of the Software.
+
+ THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+ IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+ FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+ AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+ LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+ OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+ THE SOFTWARE.
+"""
+import io
+import struct
+
+# pylint: disable=unused-import
+from typing import Dict, List, Set, Sequence, Tuple, Iterable # noqa: F401
+from typing import Callable, Optional, Union, Any # noqa: F401
+
+
+class ReadBuf:
+ def __init__(self, data: Optional[bytes] = None) -> None:
+ super(ReadBuf, self).__init__()
+ self._buf = io.BytesIO(data) if data is not None else io.BytesIO()
+ self._len = len(data) if data is not None else 0
+
+ @property
+ def unread_len(self) -> int:
+ return self._len - self._buf.tell()
+
+ def read(self, size: int) -> bytes:
+ return self._buf.read(size)
+
+ def read_byte(self) -> int:
+ v: int = struct.unpack('B', self.read(1))[0]
+ return v
+
+ def read_bool(self) -> bool:
+ return self.read_byte() != 0
+
+ def read_int(self) -> int:
+ v: int = struct.unpack('>I', self.read(4))[0]
+ return v
+
+ def read_list(self) -> List[str]:
+ list_size = self.read_int()
+ return self.read(list_size).decode('utf-8', 'replace').split(',')
+
+ def read_string(self) -> bytes:
+ n = self.read_int()
+ return self.read(n)
+
+ @classmethod
+ def _parse_mpint(cls, v: bytes, pad: bytes, f: str) -> int:
+ r = 0
+ if len(v) % 4 != 0:
+ v = pad * (4 - (len(v) % 4)) + v
+ for i in range(0, len(v), 4):
+ r = (r << 32) | struct.unpack(f, v[i:i + 4])[0]
+ return r
+
+ def read_mpint1(self) -> int:
+ # NOTE: Data Type Enc @ http://www.snailbook.com/docs/protocol-1.5.txt
+ bits = struct.unpack('>H', self.read(2))[0]
+ n = (bits + 7) // 8
+ return self._parse_mpint(self.read(n), b'\x00', '>I')
+
+ def read_mpint2(self) -> int:
+ # NOTE: Section 5 @ https://www.ietf.org/rfc/rfc4251.txt
+ v = self.read_string()
+ if len(v) == 0:
+ return 0
+ pad, f = (b'\xff', '>i') if ord(v[0:1]) & 0x80 != 0 else (b'\x00', '>I')
+ return self._parse_mpint(v, pad, f)
+
+ def read_line(self) -> str:
+ return self._buf.readline().rstrip().decode('utf-8', 'replace')
+
+ def reset(self) -> None:
+ self._buf = io.BytesIO()
+ self._len = 0
diff --git a/src/ssh_audit/software.py b/src/ssh_audit/software.py
new file mode 100644
index 0000000..f13ebeb
--- /dev/null
+++ b/src/ssh_audit/software.py
@@ -0,0 +1,227 @@
+"""
+ The MIT License (MIT)
+
+ Copyright (C) 2017 Andris Raugulis (moo@arthepsy.eu)
+
+ Permission is hereby granted, free of charge, to any person obtaining a copy
+ of this software and associated documentation files (the "Software"), to deal
+ in the Software without restriction, including without limitation the rights
+ to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+ copies of the Software, and to permit persons to whom the Software is
+ furnished to do so, subject to the following conditions:
+
+ The above copyright notice and this permission notice shall be included in
+ all copies or substantial portions of the Software.
+
+ THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+ IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+ FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+ AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+ LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+ OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+ THE SOFTWARE.
+"""
+import re
+
+# pylint: disable=unused-import
+from typing import Dict, List, Set, Sequence, Tuple, Iterable # noqa: F401
+from typing import Callable, Optional, Union, Any # noqa: F401
+
+from ssh_audit.banner import Banner
+from ssh_audit.product import Product
+
+
+class Software:
+ def __init__(self, vendor: Optional[str], product: str, version: str, patch: Optional[str], os_version: Optional[str]) -> None:
+ self.__vendor = vendor
+ self.__product = product
+ self.__version = version
+ self.__patch = patch
+ self.__os = os_version
+
+ @property
+ def vendor(self) -> Optional[str]:
+ return self.__vendor
+
+ @property
+ def product(self) -> str:
+ return self.__product
+
+ @property
+ def version(self) -> str:
+ return self.__version
+
+ @property
+ def patch(self) -> Optional[str]:
+ return self.__patch
+
+ @property
+ def os(self) -> Optional[str]:
+ return self.__os
+
+ def compare_version(self, other: Union[None, 'Software', str]) -> int:
+ # pylint: disable=too-many-branches,too-many-return-statements
+ if other is None:
+ return 1
+ if isinstance(other, Software):
+ other = '{}{}'.format(other.version, other.patch or '')
+ else:
+ other = str(other)
+ mx = re.match(r'^([\d\.]+\d+)(.*)$', other)
+ if mx is not None:
+ oversion, opatch = mx.group(1), mx.group(2).strip()
+ else:
+ oversion, opatch = other, ''
+ if self.version < oversion:
+ return -1
+ elif self.version > oversion:
+ return 1
+ spatch = self.patch or ''
+ if self.product == Product.DropbearSSH:
+ if not re.match(r'^test\d.*$', opatch):
+ opatch = 'z{}'.format(opatch)
+ if not re.match(r'^test\d.*$', spatch):
+ spatch = 'z{}'.format(spatch)
+ elif self.product == Product.OpenSSH:
+ mx1 = re.match(r'^p(\d).*', opatch)
+ mx2 = re.match(r'^p(\d).*', spatch)
+ if not (bool(mx1) and bool(mx2)):
+ if mx1 is not None:
+ opatch = mx1.group(1)
+ if mx2 is not None:
+ spatch = mx2.group(1)
+ # OpenBSD version and p1 versions are considered the same.
+ if ((spatch == '') and (opatch == '1')) or ((spatch == '1') and (opatch == '')):
+ return 0
+ if spatch < opatch:
+ return -1
+ elif spatch > opatch:
+ return 1
+ return 0
+
+ def between_versions(self, vfrom: str, vtill: str) -> bool:
+ if bool(vfrom) and self.compare_version(vfrom) < 0:
+ return False
+ if bool(vtill) and self.compare_version(vtill) > 0:
+ return False
+ return True
+
+ def display(self, full: bool = True) -> str:
+ r = '{} '.format(self.vendor) if bool(self.vendor) else ''
+ r += self.product
+ if bool(self.version):
+ r += ' {}'.format(self.version)
+ if full:
+ patch = self.patch or ''
+ if self.product == Product.OpenSSH:
+ mx = re.match(r'^(p\d)(.*)$', patch)
+ if mx is not None:
+ r += mx.group(1)
+ patch = mx.group(2).strip()
+ if bool(patch):
+ r += ' ({})'.format(patch)
+ if bool(self.os):
+ r += ' running on {}'.format(self.os)
+ return r
+
+ def __str__(self) -> str:
+ return self.display()
+
+ def __repr__(self) -> str:
+ r = 'vendor={}, '.format(self.vendor) if bool(self.vendor) else ''
+ r += 'product={}'.format(self.product)
+ if bool(self.version):
+ r += ', version={}'.format(self.version)
+ if bool(self.patch):
+ r += ', patch={}'.format(self.patch)
+ if bool(self.os):
+ r += ', os={}'.format(self.os)
+ return '<{}({})>'.format(self.__class__.__name__, r)
+
+ @staticmethod
+ def _fix_patch(patch: str) -> Optional[str]:
+ return re.sub(r'^[-_\.]+', '', patch) or None
+
+ @staticmethod
+ def _fix_date(d: Optional[str]) -> Optional[str]:
+ if d is not None and len(d) == 8:
+ return '{}-{}-{}'.format(d[:4], d[4:6], d[6:8])
+ else:
+ return None
+
+ @classmethod
+ def _extract_os_version(cls, c: Optional[str]) -> Optional[str]:
+ if c is None:
+ return None
+ mx = re.match(r'^NetBSD(?:_Secure_Shell)?(?:[\s-]+(\d{8})(.*))?$', c)
+ if mx is not None:
+ d = cls._fix_date(mx.group(1))
+ return 'NetBSD' if d is None else 'NetBSD ({})'.format(d)
+ mx = re.match(r'^FreeBSD(?:\slocalisations)?[\s-]+(\d{8})(.*)$', c)
+ if not bool(mx):
+ mx = re.match(r'^[^@]+@FreeBSD\.org[\s-]+(\d{8})(.*)$', c)
+ if mx is not None:
+ d = cls._fix_date(mx.group(1))
+ return 'FreeBSD' if d is None else 'FreeBSD ({})'.format(d)
+ w = ['RemotelyAnywhere', 'DesktopAuthority', 'RemoteSupportManager']
+ for win_soft in w:
+ mx = re.match(r'^in ' + win_soft + r' ([\d\.]+\d)$', c)
+ if mx is not None:
+ ver = mx.group(1)
+ return 'Microsoft Windows ({} {})'.format(win_soft, ver)
+ generic = ['NetBSD', 'FreeBSD']
+ for g in generic:
+ if c.startswith(g) or c.endswith(g):
+ return g
+ return None
+
+ @classmethod
+ def parse(cls, banner: 'Banner') -> Optional['Software']:
+ # pylint: disable=too-many-return-statements
+ software = str(banner.software)
+ mx = re.match(r'^dropbear_([\d\.]+\d+)(.*)', software)
+ v: Optional[str] = None
+ if mx is not None:
+ patch = cls._fix_patch(mx.group(2))
+ v, p = 'Matt Johnston', Product.DropbearSSH
+ v = None
+ return cls(v, p, mx.group(1), patch, None)
+ mx = re.match(r'^OpenSSH[_\.-]+([\d\.]+\d+)(.*)', software)
+ if mx is not None:
+ patch = cls._fix_patch(mx.group(2))
+ v, p = 'OpenBSD', Product.OpenSSH
+ v = None
+ os_version = cls._extract_os_version(banner.comments)
+ return cls(v, p, mx.group(1), patch, os_version)
+ mx = re.match(r'^libssh-([\d\.]+\d+)(.*)', software)
+ if mx is not None:
+ patch = cls._fix_patch(mx.group(2))
+ v, p = None, Product.LibSSH
+ os_version = cls._extract_os_version(banner.comments)
+ return cls(v, p, mx.group(1), patch, os_version)
+ mx = re.match(r'^libssh_([\d\.]+\d+)(.*)', software)
+ if mx is not None:
+ patch = cls._fix_patch(mx.group(2))
+ v, p = None, Product.LibSSH
+ os_version = cls._extract_os_version(banner.comments)
+ return cls(v, p, mx.group(1), patch, os_version)
+ mx = re.match(r'^RomSShell_([\d\.]+\d+)(.*)', software)
+ if mx is not None:
+ patch = cls._fix_patch(mx.group(2))
+ v, p = 'Allegro Software', 'RomSShell'
+ return cls(v, p, mx.group(1), patch, None)
+ mx = re.match(r'^mpSSH_([\d\.]+\d+)', software)
+ if mx is not None:
+ v, p = 'HP', 'iLO (Integrated Lights-Out) sshd'
+ return cls(v, p, mx.group(1), None, None)
+ mx = re.match(r'^Cisco-([\d\.]+\d+)', software)
+ if mx is not None:
+ v, p = 'Cisco', 'IOS/PIX sshd'
+ return cls(v, p, mx.group(1), None, None)
+ mx = re.match(r'^tinyssh_(.*)', software)
+ if mx is not None:
+ return cls(None, Product.TinySSH, mx.group(1), None, None)
+ mx = re.match(r'^PuTTY_Release_(.*)', software)
+ if mx:
+ return cls(None, Product.PuTTY, mx.group(1), None, None)
+ return None
diff --git a/src/ssh_audit/ssh1.py b/src/ssh_audit/ssh1.py
new file mode 100644
index 0000000..f966b74
--- /dev/null
+++ b/src/ssh_audit/ssh1.py
@@ -0,0 +1,40 @@
+"""
+ The MIT License (MIT)
+
+ Copyright (C) 2017 Andris Raugulis (moo@arthepsy.eu)
+
+ Permission is hereby granted, free of charge, to any person obtaining a copy
+ of this software and associated documentation files (the "Software"), to deal
+ in the Software without restriction, including without limitation the rights
+ to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+ copies of the Software, and to permit persons to whom the Software is
+ furnished to do so, subject to the following conditions:
+
+ The above copyright notice and this permission notice shall be included in
+ all copies or substantial portions of the Software.
+
+ THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+ IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+ FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+ AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+ LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+ OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+ THE SOFTWARE.
+"""
+# pylint: disable=unused-import
+from typing import Dict, List, Set, Sequence, Tuple, Iterable # noqa: F401
+from typing import Callable, Optional, Union, Any # noqa: F401
+
+from ssh_audit.ssh1_crc32 import SSH1_CRC32
+
+
+class SSH1:
+ _crc32: Optional[SSH1_CRC32] = None
+ CIPHERS = ['none', 'idea', 'des', '3des', 'tss', 'rc4', 'blowfish']
+ AUTHS = ['none', 'rhosts', 'rsa', 'password', 'rhosts_rsa', 'tis', 'kerberos']
+
+ @classmethod
+ def crc32(cls, v: bytes) -> int:
+ if cls._crc32 is None:
+ cls._crc32 = SSH1_CRC32()
+ return cls._crc32.calc(v)
diff --git a/src/ssh_audit/ssh1_crc32.py b/src/ssh_audit/ssh1_crc32.py
new file mode 100644
index 0000000..55089ed
--- /dev/null
+++ b/src/ssh_audit/ssh1_crc32.py
@@ -0,0 +1,47 @@
+"""
+ The MIT License (MIT)
+
+ Copyright (C) 2017 Andris Raugulis (moo@arthepsy.eu)
+
+ Permission is hereby granted, free of charge, to any person obtaining a copy
+ of this software and associated documentation files (the "Software"), to deal
+ in the Software without restriction, including without limitation the rights
+ to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+ copies of the Software, and to permit persons to whom the Software is
+ furnished to do so, subject to the following conditions:
+
+ The above copyright notice and this permission notice shall be included in
+ all copies or substantial portions of the Software.
+
+ THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+ IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+ FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+ AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+ LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+ OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+ THE SOFTWARE.
+"""
+# pylint: disable=unused-import
+from typing import Dict, List, Set, Sequence, Tuple, Iterable # noqa: F401
+from typing import Callable, Optional, Union, Any # noqa: F401
+
+
+class SSH1_CRC32:
+ def __init__(self) -> None:
+ self._table = [0] * 256
+ for i in range(256):
+ crc = 0
+ n = i
+ for _ in range(8):
+ x = (crc ^ n) & 1
+ crc = (crc >> 1) ^ (x * 0xedb88320)
+ n = n >> 1
+ self._table[i] = crc
+
+ def calc(self, v: bytes) -> int:
+ crc, length = 0, len(v)
+ for i in range(length):
+ n = ord(v[i:i + 1])
+ n = n ^ (crc & 0xff)
+ crc = (crc >> 8) ^ self._table[n]
+ return crc
diff --git a/src/ssh_audit/ssh1_kexdb.py b/src/ssh_audit/ssh1_kexdb.py
new file mode 100644
index 0000000..c9810ce
--- /dev/null
+++ b/src/ssh_audit/ssh1_kexdb.py
@@ -0,0 +1,58 @@
+"""
+ The MIT License (MIT)
+
+ Copyright (C) 2017 Andris Raugulis (moo@arthepsy.eu)
+
+ Permission is hereby granted, free of charge, to any person obtaining a copy
+ of this software and associated documentation files (the "Software"), to deal
+ in the Software without restriction, including without limitation the rights
+ to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+ copies of the Software, and to permit persons to whom the Software is
+ furnished to do so, subject to the following conditions:
+
+ The above copyright notice and this permission notice shall be included in
+ all copies or substantial portions of the Software.
+
+ THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+ IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+ FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+ AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+ LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+ OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+ THE SOFTWARE.
+"""
+# pylint: disable=unused-import
+from typing import Dict, List, Set, Sequence, Tuple, Iterable # noqa: F401
+from typing import Callable, Optional, Union, Any # noqa: F401
+
+
+class SSH1_KexDB: # pylint: disable=too-few-public-methods
+
+ FAIL_PLAINTEXT = 'no encryption/integrity'
+ FAIL_OPENSSH37_REMOVE = 'removed since OpenSSH 3.7'
+ FAIL_NA_BROKEN = 'not implemented in OpenSSH, broken algorithm'
+ FAIL_NA_UNSAFE = 'not implemented in OpenSSH (server), unsafe algorithm'
+ TEXT_CIPHER_IDEA = 'cipher used by commercial SSH'
+
+ ALGORITHMS: Dict[str, Dict[str, List[List[Optional[str]]]]] = {
+ 'key': {
+ 'ssh-rsa1': [['1.2.2']],
+ },
+ 'enc': {
+ 'none': [['1.2.2'], [FAIL_PLAINTEXT]],
+ 'idea': [[None], [], [], [TEXT_CIPHER_IDEA]],
+ 'des': [['2.3.0C'], [FAIL_NA_UNSAFE]],
+ '3des': [['1.2.2']],
+ 'tss': [[''], [FAIL_NA_BROKEN]],
+ 'rc4': [[], [FAIL_NA_BROKEN]],
+ 'blowfish': [['1.2.2']],
+ },
+ 'aut': {
+ 'rhosts': [['1.2.2', '3.6'], [FAIL_OPENSSH37_REMOVE]],
+ 'rsa': [['1.2.2']],
+ 'password': [['1.2.2']],
+ 'rhosts_rsa': [['1.2.2']],
+ 'tis': [['1.2.2']],
+ 'kerberos': [['1.2.2', '3.6'], [FAIL_OPENSSH37_REMOVE]],
+ }
+ }
diff --git a/src/ssh_audit/ssh1_publickeymessage.py b/src/ssh_audit/ssh1_publickeymessage.py
new file mode 100644
index 0000000..67c4dec
--- /dev/null
+++ b/src/ssh_audit/ssh1_publickeymessage.py
@@ -0,0 +1,144 @@
+"""
+ The MIT License (MIT)
+
+ Copyright (C) 2017 Andris Raugulis (moo@arthepsy.eu)
+
+ Permission is hereby granted, free of charge, to any person obtaining a copy
+ of this software and associated documentation files (the "Software"), to deal
+ in the Software without restriction, including without limitation the rights
+ to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+ copies of the Software, and to permit persons to whom the Software is
+ furnished to do so, subject to the following conditions:
+
+ The above copyright notice and this permission notice shall be included in
+ all copies or substantial portions of the Software.
+
+ THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+ IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+ FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+ AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+ LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+ OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+ THE SOFTWARE.
+"""
+# pylint: disable=unused-import
+from typing import Dict, List, Set, Sequence, Tuple, Iterable # noqa: F401
+from typing import Callable, Optional, Union, Any # noqa: F401
+
+from ssh_audit.ssh1 import SSH1
+from ssh_audit.readbuf import ReadBuf
+from ssh_audit.utils import Utils
+from ssh_audit.writebuf import WriteBuf
+
+
+class SSH1_PublicKeyMessage:
+ def __init__(self, cookie: bytes, skey: Tuple[int, int, int], hkey: Tuple[int, int, int], pflags: int, cmask: int, amask: int) -> None:
+ if len(skey) != 3:
+ raise ValueError('invalid server key pair: {}'.format(skey))
+ if len(hkey) != 3:
+ raise ValueError('invalid host key pair: {}'.format(hkey))
+ self.__cookie = cookie
+ self.__server_key = skey
+ self.__host_key = hkey
+ self.__protocol_flags = pflags
+ self.__supported_ciphers_mask = cmask
+ self.__supported_authentications_mask = amask
+
+ @property
+ def cookie(self) -> bytes:
+ return self.__cookie
+
+ @property
+ def server_key_bits(self) -> int:
+ return self.__server_key[0]
+
+ @property
+ def server_key_public_exponent(self) -> int:
+ return self.__server_key[1]
+
+ @property
+ def server_key_public_modulus(self) -> int:
+ return self.__server_key[2]
+
+ @property
+ def host_key_bits(self) -> int:
+ return self.__host_key[0]
+
+ @property
+ def host_key_public_exponent(self) -> int:
+ return self.__host_key[1]
+
+ @property
+ def host_key_public_modulus(self) -> int:
+ return self.__host_key[2]
+
+ @property
+ def host_key_fingerprint_data(self) -> bytes:
+ # pylint: disable=protected-access
+ mod = WriteBuf._create_mpint(self.host_key_public_modulus, False)
+ e = WriteBuf._create_mpint(self.host_key_public_exponent, False)
+ return mod + e
+
+ @property
+ def protocol_flags(self) -> int:
+ return self.__protocol_flags
+
+ @property
+ def supported_ciphers_mask(self) -> int:
+ return self.__supported_ciphers_mask
+
+ @property
+ def supported_ciphers(self) -> List[str]:
+ ciphers = []
+ for i in range(len(SSH1.CIPHERS)): # pylint: disable=consider-using-enumerate
+ if self.__supported_ciphers_mask & (1 << i) != 0:
+ ciphers.append(Utils.to_text(SSH1.CIPHERS[i]))
+ return ciphers
+
+ @property
+ def supported_authentications_mask(self) -> int:
+ return self.__supported_authentications_mask
+
+ @property
+ def supported_authentications(self) -> List[str]:
+ auths = []
+ for i in range(1, len(SSH1.AUTHS)):
+ if self.__supported_authentications_mask & (1 << i) != 0:
+ auths.append(Utils.to_text(SSH1.AUTHS[i]))
+ return auths
+
+ def write(self, wbuf: 'WriteBuf') -> None:
+ wbuf.write(self.cookie)
+ wbuf.write_int(self.server_key_bits)
+ wbuf.write_mpint1(self.server_key_public_exponent)
+ wbuf.write_mpint1(self.server_key_public_modulus)
+ wbuf.write_int(self.host_key_bits)
+ wbuf.write_mpint1(self.host_key_public_exponent)
+ wbuf.write_mpint1(self.host_key_public_modulus)
+ wbuf.write_int(self.protocol_flags)
+ wbuf.write_int(self.supported_ciphers_mask)
+ wbuf.write_int(self.supported_authentications_mask)
+
+ @property
+ def payload(self) -> bytes:
+ wbuf = WriteBuf()
+ self.write(wbuf)
+ return wbuf.write_flush()
+
+ @classmethod
+ def parse(cls, payload: bytes) -> 'SSH1_PublicKeyMessage':
+ buf = ReadBuf(payload)
+ cookie = buf.read(8)
+ server_key_bits = buf.read_int()
+ server_key_exponent = buf.read_mpint1()
+ server_key_modulus = buf.read_mpint1()
+ skey = (server_key_bits, server_key_exponent, server_key_modulus)
+ host_key_bits = buf.read_int()
+ host_key_exponent = buf.read_mpint1()
+ host_key_modulus = buf.read_mpint1()
+ hkey = (host_key_bits, host_key_exponent, host_key_modulus)
+ pflags = buf.read_int()
+ cmask = buf.read_int()
+ amask = buf.read_int()
+ pkm = cls(cookie, skey, hkey, pflags, cmask, amask)
+ return pkm
diff --git a/src/ssh_audit/ssh2_kex.py b/src/ssh_audit/ssh2_kex.py
new file mode 100644
index 0000000..0c31f3d
--- /dev/null
+++ b/src/ssh_audit/ssh2_kex.py
@@ -0,0 +1,136 @@
+"""
+ The MIT License (MIT)
+
+ Copyright (C) 2017-2020 Joe Testa (jtesta@positronsecurity.com)
+ Copyright (C) 2017 Andris Raugulis (moo@arthepsy.eu)
+
+ Permission is hereby granted, free of charge, to any person obtaining a copy
+ of this software and associated documentation files (the "Software"), to deal
+ in the Software without restriction, including without limitation the rights
+ to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+ copies of the Software, and to permit persons to whom the Software is
+ furnished to do so, subject to the following conditions:
+
+ The above copyright notice and this permission notice shall be included in
+ all copies or substantial portions of the Software.
+
+ THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+ IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+ FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+ AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+ LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+ OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+ THE SOFTWARE.
+"""
+# pylint: disable=unused-import
+from typing import Dict, List, Set, Sequence, Tuple, Iterable # noqa: F401
+from typing import Callable, Optional, Union, Any # noqa: F401
+
+from ssh_audit.ssh2_kexparty import SSH2_KexParty
+from ssh_audit.readbuf import ReadBuf
+from ssh_audit.writebuf import WriteBuf
+
+
+class SSH2_Kex:
+ def __init__(self, cookie: bytes, kex_algs: List[str], key_algs: List[str], cli: 'SSH2_KexParty', srv: 'SSH2_KexParty', follows: bool, unused: int = 0) -> None:
+ self.__cookie = cookie
+ self.__kex_algs = kex_algs
+ self.__key_algs = key_algs
+ self.__client = cli
+ self.__server = srv
+ self.__follows = follows
+ self.__unused = unused
+
+ self.__rsa_key_sizes: Dict[str, Tuple[int, int]] = {}
+ self.__dh_modulus_sizes: Dict[str, Tuple[int, int]] = {}
+ self.__host_keys: Dict[str, bytes] = {}
+
+ @property
+ def cookie(self) -> bytes:
+ return self.__cookie
+
+ @property
+ def kex_algorithms(self) -> List[str]:
+ return self.__kex_algs
+
+ @property
+ def key_algorithms(self) -> List[str]:
+ return self.__key_algs
+
+ # client_to_server
+ @property
+ def client(self) -> 'SSH2_KexParty':
+ return self.__client
+
+ # server_to_client
+ @property
+ def server(self) -> 'SSH2_KexParty':
+ return self.__server
+
+ @property
+ def follows(self) -> bool:
+ return self.__follows
+
+ @property
+ def unused(self) -> int:
+ return self.__unused
+
+ def set_rsa_key_size(self, rsa_type: str, hostkey_size: int, ca_size: int = -1) -> None:
+ self.__rsa_key_sizes[rsa_type] = (hostkey_size, ca_size)
+
+ def rsa_key_sizes(self) -> Dict[str, Tuple[int, int]]:
+ return self.__rsa_key_sizes
+
+ def set_dh_modulus_size(self, gex_alg: str, modulus_size: int) -> None:
+ self.__dh_modulus_sizes[gex_alg] = (modulus_size, -1)
+
+ def dh_modulus_sizes(self) -> Dict[str, Tuple[int, int]]:
+ return self.__dh_modulus_sizes
+
+ def set_host_key(self, key_type: str, hostkey: bytes) -> None:
+ self.__host_keys[key_type] = hostkey
+
+ def host_keys(self) -> Dict[str, bytes]:
+ return self.__host_keys
+
+ def write(self, wbuf: 'WriteBuf') -> None:
+ wbuf.write(self.cookie)
+ wbuf.write_list(self.kex_algorithms)
+ wbuf.write_list(self.key_algorithms)
+ wbuf.write_list(self.client.encryption)
+ wbuf.write_list(self.server.encryption)
+ wbuf.write_list(self.client.mac)
+ wbuf.write_list(self.server.mac)
+ wbuf.write_list(self.client.compression)
+ wbuf.write_list(self.server.compression)
+ wbuf.write_list(self.client.languages)
+ wbuf.write_list(self.server.languages)
+ wbuf.write_bool(self.follows)
+ wbuf.write_int(self.__unused)
+
+ @property
+ def payload(self) -> bytes:
+ wbuf = WriteBuf()
+ self.write(wbuf)
+ return wbuf.write_flush()
+
+ @classmethod
+ def parse(cls, payload: bytes) -> 'SSH2_Kex':
+ buf = ReadBuf(payload)
+ cookie = buf.read(16)
+ kex_algs = buf.read_list()
+ key_algs = buf.read_list()
+ cli_enc = buf.read_list()
+ srv_enc = buf.read_list()
+ cli_mac = buf.read_list()
+ srv_mac = buf.read_list()
+ cli_compression = buf.read_list()
+ srv_compression = buf.read_list()
+ cli_languages = buf.read_list()
+ srv_languages = buf.read_list()
+ follows = buf.read_bool()
+ unused = buf.read_int()
+ cli = SSH2_KexParty(cli_enc, cli_mac, cli_compression, cli_languages)
+ srv = SSH2_KexParty(srv_enc, srv_mac, srv_compression, srv_languages)
+ kex = cls(cookie, kex_algs, key_algs, cli, srv, follows, unused)
+ return kex
diff --git a/src/ssh_audit/ssh2_kexdb.py b/src/ssh_audit/ssh2_kexdb.py
new file mode 100644
index 0000000..643d9c6
--- /dev/null
+++ b/src/ssh_audit/ssh2_kexdb.py
@@ -0,0 +1,280 @@
+"""
+ The MIT License (MIT)
+
+ Copyright (C) 2017-2021 Joe Testa (jtesta@positronsecurity.com)
+ Copyright (C) 2017 Andris Raugulis (moo@arthepsy.eu)
+
+ Permission is hereby granted, free of charge, to any person obtaining a copy
+ of this software and associated documentation files (the "Software"), to deal
+ in the Software without restriction, including without limitation the rights
+ to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+ copies of the Software, and to permit persons to whom the Software is
+ furnished to do so, subject to the following conditions:
+
+ The above copyright notice and this permission notice shall be included in
+ all copies or substantial portions of the Software.
+
+ THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+ IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+ FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+ AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+ LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+ OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+ THE SOFTWARE.
+"""
+# pylint: disable=unused-import
+from typing import Dict, List, Set, Sequence, Tuple, Iterable # noqa: F401
+from typing import Callable, Optional, Union, Any # noqa: F401
+
+
+class SSH2_KexDB: # pylint: disable=too-few-public-methods
+ WARN_OPENSSH74_UNSAFE = 'disabled (in client) since OpenSSH 7.4, unsafe algorithm'
+ WARN_OPENSSH72_LEGACY = 'disabled (in client) since OpenSSH 7.2, legacy algorithm'
+ FAIL_OPENSSH70_LEGACY = 'removed since OpenSSH 7.0, legacy algorithm'
+ FAIL_OPENSSH70_WEAK = 'removed (in server) and disabled (in client) since OpenSSH 7.0, weak algorithm'
+ FAIL_OPENSSH70_LOGJAM = 'disabled (in client) since OpenSSH 7.0, logjam attack'
+ INFO_OPENSSH69_CHACHA = 'default cipher since OpenSSH 6.9.'
+ FAIL_OPENSSH67_UNSAFE = 'removed (in server) since OpenSSH 6.7, unsafe algorithm'
+ FAIL_OPENSSH61_REMOVE = 'removed since OpenSSH 6.1, removed from specification'
+ FAIL_OPENSSH31_REMOVE = 'removed since OpenSSH 3.1'
+ INFO_OPENSSH82_FUTURE_DEPRECATION = 'a future deprecation notice has been issued in OpenSSH 8.2: https://www.openssh.com/txt/release-8.2'
+ FAIL_DBEAR67_DISABLED = 'disabled since Dropbear SSH 2015.67'
+ FAIL_DBEAR53_DISABLED = 'disabled since Dropbear SSH 0.53'
+ FAIL_DEPRECATED_CIPHER = 'deprecated cipher'
+ FAIL_WEAK_CIPHER = 'using weak cipher'
+ FAIL_WEAK_ALGORITHM = 'using weak/obsolete algorithm'
+ FAIL_PLAINTEXT = 'no encryption/integrity'
+ FAIL_DEPRECATED_MAC = 'deprecated MAC'
+ FAIL_1024BIT_MODULUS = 'using small 1024-bit modulus'
+ FAIL_UNPROVEN = 'using unproven algorithm'
+ FAIL_HASH_WEAK = 'using weak hashing algorithm'
+ WARN_CURVES_WEAK = 'using weak elliptic curves'
+ WARN_RNDSIG_KEY = 'using weak random number generator could reveal the key'
+ WARN_HASH_WEAK = 'using weak hashing algorithm'
+ WARN_CIPHER_MODE = 'using weak cipher mode'
+ WARN_BLOCK_SIZE = 'using small 64-bit block size'
+ WARN_CIPHER_WEAK = 'using weak cipher'
+ WARN_ENCRYPT_AND_MAC = 'using encrypt-and-MAC mode'
+ WARN_TAG_SIZE = 'using small 64-bit tag size'
+ WARN_TAG_SIZE_96 = 'using small 96-bit tag size'
+ WARN_EXPERIMENTAL = 'using experimental algorithm'
+ WARN_OBSOLETE = 'using obsolete algorithm'
+ WARN_UNTRUSTED = 'using untrusted algorithm'
+
+ ALGORITHMS: Dict[str, Dict[str, List[List[Optional[str]]]]] = {
+ # Format: 'algorithm_name': [['version_first_appeared_in'], [reason_for_failure1, reason_for_failure2, ...], [warning1, warning2, ...]]
+ 'kex': {
+ 'diffie-hellman-group1-sha1': [['2.3.0,d0.28,l10.2', '6.6', '6.9'], [FAIL_1024BIT_MODULUS, FAIL_OPENSSH67_UNSAFE, FAIL_OPENSSH70_LOGJAM], [WARN_HASH_WEAK]],
+ 'gss-group1-sha1-toWM5Slw5Ew8Mqkay+al2g==': [[], [FAIL_1024BIT_MODULUS, FAIL_OPENSSH67_UNSAFE, FAIL_OPENSSH70_LOGJAM], [WARN_HASH_WEAK]],
+ 'gss-gex-sha1-eipGX3TCiQSrx573bT1o1Q==': [[], [], [WARN_HASH_WEAK]],
+ 'gss-gex-sha1-toWM5Slw5Ew8Mqkay+al2g==': [[], [], [WARN_HASH_WEAK]],
+ 'gss-gex-sha1-': [[], [], [WARN_HASH_WEAK]],
+ 'gss-group1-sha1-eipGX3TCiQSrx573bT1o1Q==': [[], [FAIL_1024BIT_MODULUS], [WARN_HASH_WEAK]],
+ 'gss-group1-sha1-': [[], [FAIL_1024BIT_MODULUS], [WARN_HASH_WEAK]],
+ 'gss-group14-sha1-': [[], [], [WARN_HASH_WEAK]],
+ 'gss-group14-sha1-eipGX3TCiQSrx573bT1o1Q==': [[], [], [WARN_HASH_WEAK]],
+ 'gss-group14-sha1-toWM5Slw5Ew8Mqkay+al2g==': [[], [], [WARN_HASH_WEAK]],
+ 'gss-group14-sha256-': [[]],
+ 'gss-group14-sha256-toWM5Slw5Ew8Mqkay+al2g==': [[]],
+ 'gss-group15-sha512-': [[]],
+ 'gss-group15-sha512-toWM5Slw5Ew8Mqkay+al2g==': [[]],
+ 'gss-group16-sha512-': [[]],
+ 'gss-nistp256-sha256-': [[], [WARN_CURVES_WEAK]],
+ 'gss-curve25519-sha256-': [[]],
+ 'diffie-hellman-group1-sha256': [[], [FAIL_1024BIT_MODULUS]],
+ 'diffie-hellman-group14-sha1': [['3.9,d0.53,l10.6.0'], [], [WARN_HASH_WEAK]],
+ 'diffie-hellman-group14-sha256': [['7.3,d2016.73']],
+ 'diffie-hellman-group14-sha256@ssh.com': [[]],
+ 'diffie-hellman-group15-sha256': [[]],
+ 'diffie-hellman-group15-sha256@ssh.com': [[]],
+ 'diffie-hellman-group15-sha384@ssh.com': [[]],
+ 'diffie-hellman-group15-sha512': [[]],
+ 'diffie-hellman-group16-sha256': [[]],
+ 'diffie-hellman-group16-sha384@ssh.com': [[]],
+ 'diffie-hellman-group16-sha512': [['7.3,d2016.73']],
+ 'diffie-hellman-group16-sha512@ssh.com': [[]],
+ 'diffie-hellman-group17-sha512': [[]],
+ 'diffie-hellman-group18-sha512': [['7.3']],
+ 'diffie-hellman-group18-sha512@ssh.com': [[]],
+ 'diffie-hellman-group-exchange-sha1': [['2.3.0', '6.6', None], [FAIL_OPENSSH67_UNSAFE], [WARN_HASH_WEAK]],
+ 'diffie-hellman-group-exchange-sha256': [['4.4']],
+ 'diffie-hellman-group-exchange-sha256@ssh.com': [[]],
+ 'diffie-hellman-group-exchange-sha512@ssh.com': [[]],
+ 'ecdh-sha2-curve25519': [[], []],
+ 'ecdh-sha2-nistb233': [[], [WARN_CURVES_WEAK]],
+ 'ecdh-sha2-nistb409': [[], [WARN_CURVES_WEAK]],
+ 'ecdh-sha2-nistk163': [[], [WARN_CURVES_WEAK]],
+ 'ecdh-sha2-nistk233': [[], [WARN_CURVES_WEAK]],
+ 'ecdh-sha2-nistk283': [[], [WARN_CURVES_WEAK]],
+ 'ecdh-sha2-nistk409': [[], [WARN_CURVES_WEAK]],
+ 'ecdh-sha2-nistp192': [[], [WARN_CURVES_WEAK]],
+ 'ecdh-sha2-nistp224': [[], [WARN_CURVES_WEAK]],
+ 'ecdh-sha2-nistp256': [['5.7,d2013.62,l10.6.0'], [WARN_CURVES_WEAK]],
+ 'ecdh-sha2-nistp384': [['5.7,d2013.62'], [WARN_CURVES_WEAK]],
+ 'ecdh-sha2-nistp521': [['5.7,d2013.62'], [WARN_CURVES_WEAK]],
+ 'ecdh-sha2-nistt571': [[], [WARN_CURVES_WEAK]],
+ 'ecdh-sha2-1.3.132.0.10': [[]], # ECDH over secp256k1 (i.e.: the Bitcoin curve)
+ 'curve25519-sha256@libssh.org': [['6.5,d2013.62,l10.6.0']],
+ 'curve25519-sha256': [['7.4,d2018.76']],
+ 'curve448-sha512': [[]],
+ 'kexguess2@matt.ucc.asn.au': [['d2013.57']],
+ 'rsa1024-sha1': [[], [FAIL_1024BIT_MODULUS], [WARN_HASH_WEAK]],
+ 'rsa2048-sha256': [[]],
+ 'sntrup4591761x25519-sha512@tinyssh.org': [['8.0', '8.4'], [], [WARN_EXPERIMENTAL]],
+ 'sntrup761x25519-sha512@openssh.com': [['8.5'], [], [WARN_EXPERIMENTAL]],
+ 'kexAlgoCurve25519SHA256': [[]],
+ 'Curve25519SHA256': [[]],
+ 'ext-info-c': [[]], # Extension negotiation (RFC 8308)
+ 'ext-info-s': [[]], # Extension negotiation (RFC 8308)
+ },
+ 'key': {
+ 'ssh-rsa1': [[], [FAIL_WEAK_ALGORITHM]],
+ 'rsa-sha2-256': [['7.2']],
+ 'rsa-sha2-512': [['7.2']],
+ 'ssh-ed25519': [['6.5,l10.7.0']],
+ 'ssh-ed25519-cert-v01@openssh.com': [['6.5']],
+ 'ssh-rsa': [['2.5.0,d0.28,l10.2'], [FAIL_HASH_WEAK], [], [INFO_OPENSSH82_FUTURE_DEPRECATION]],
+ 'ssh-dss': [['2.1.0,d0.28,l10.2', '6.9'], [FAIL_1024BIT_MODULUS, FAIL_OPENSSH70_WEAK], [WARN_RNDSIG_KEY]],
+ 'ecdsa-sha2-nistp256': [['5.7,d2013.62,l10.6.4'], [WARN_CURVES_WEAK], [WARN_RNDSIG_KEY]],
+ 'ecdsa-sha2-nistp384': [['5.7,d2013.62,l10.6.4'], [WARN_CURVES_WEAK], [WARN_RNDSIG_KEY]],
+ 'ecdsa-sha2-nistp521': [['5.7,d2013.62,l10.6.4'], [WARN_CURVES_WEAK], [WARN_RNDSIG_KEY]],
+ 'ecdsa-sha2-1.3.132.0.10': [[], [], [WARN_RNDSIG_KEY]], # ECDSA over secp256k1 (i.e.: the Bitcoin curve)
+ 'x509v3-sign-dss': [[], [FAIL_1024BIT_MODULUS, FAIL_OPENSSH70_WEAK], [WARN_RNDSIG_KEY]],
+ 'x509v3-sign-rsa': [[], [FAIL_HASH_WEAK], [], [INFO_OPENSSH82_FUTURE_DEPRECATION]],
+ 'x509v3-sign-rsa-sha256@ssh.com': [[]],
+ 'x509v3-ssh-dss': [[], [FAIL_1024BIT_MODULUS, FAIL_OPENSSH70_WEAK], [WARN_RNDSIG_KEY]],
+ 'x509v3-ssh-rsa': [[], [FAIL_HASH_WEAK], [], [INFO_OPENSSH82_FUTURE_DEPRECATION]],
+ 'ssh-rsa-cert-v00@openssh.com': [['5.4', '6.9'], [FAIL_OPENSSH70_LEGACY, FAIL_HASH_WEAK], [], [INFO_OPENSSH82_FUTURE_DEPRECATION]],
+ 'ssh-dss-cert-v00@openssh.com': [['5.4', '6.9'], [FAIL_1024BIT_MODULUS, FAIL_OPENSSH70_LEGACY], [WARN_RNDSIG_KEY]],
+ 'ssh-rsa-cert-v01@openssh.com': [['5.6'], [FAIL_HASH_WEAK], [], [INFO_OPENSSH82_FUTURE_DEPRECATION]],
+ 'ssh-dss-cert-v01@openssh.com': [['5.6', '6.9'], [FAIL_1024BIT_MODULUS, FAIL_OPENSSH70_WEAK], [WARN_RNDSIG_KEY]],
+ 'ecdsa-sha2-nistp256-cert-v01@openssh.com': [['5.7'], [WARN_CURVES_WEAK], [WARN_RNDSIG_KEY]],
+ 'ecdsa-sha2-nistp384-cert-v01@openssh.com': [['5.7'], [WARN_CURVES_WEAK], [WARN_RNDSIG_KEY]],
+ 'ecdsa-sha2-nistp521-cert-v01@openssh.com': [['5.7'], [WARN_CURVES_WEAK], [WARN_RNDSIG_KEY]],
+ 'rsa-sha2-256-cert-v01@openssh.com': [['7.8']],
+ 'rsa-sha2-512-cert-v01@openssh.com': [['7.8']],
+ 'ssh-rsa-sha256@ssh.com': [[]],
+ 'ssh-dss-sha256@ssh.com': [[], [FAIL_1024BIT_MODULUS]],
+ 'sk-ecdsa-sha2-nistp256-cert-v01@openssh.com': [['8.2'], [WARN_CURVES_WEAK], [WARN_RNDSIG_KEY]],
+ 'sk-ecdsa-sha2-nistp256@openssh.com': [['8.2'], [WARN_CURVES_WEAK], [WARN_RNDSIG_KEY]],
+ 'sk-ssh-ed25519-cert-v01@openssh.com': [['8.2']],
+ 'sk-ssh-ed25519@openssh.com': [['8.2']],
+ 'ssh-gost2001': [[], [], [WARN_UNTRUSTED]],
+ 'ssh-gost2012-256': [[], [], [WARN_UNTRUSTED]],
+ 'ssh-gost2012-512': [[], [], [WARN_UNTRUSTED]],
+ 'spi-sign-rsa': [[]],
+ 'ssh-ed448': [[]],
+ 'x509v3-ecdsa-sha2-nistp256': [[], [WARN_CURVES_WEAK]],
+ 'x509v3-ecdsa-sha2-nistp384': [[], [WARN_CURVES_WEAK]],
+ 'x509v3-ecdsa-sha2-nistp521': [[], [WARN_CURVES_WEAK]],
+ 'x509v3-rsa2048-sha256': [[]],
+ },
+ 'enc': {
+ 'none': [['1.2.2,d2013.56,l10.2'], [FAIL_PLAINTEXT]],
+ 'des': [[], [FAIL_WEAK_CIPHER], [WARN_CIPHER_MODE, WARN_BLOCK_SIZE]],
+ 'des-cbc': [[], [FAIL_WEAK_CIPHER], [WARN_CIPHER_MODE, WARN_BLOCK_SIZE]],
+ 'des-cbc@ssh.com': [[], [FAIL_WEAK_CIPHER], [WARN_CIPHER_MODE, WARN_BLOCK_SIZE]],
+ 'des-cbc-ssh1': [[], [FAIL_WEAK_CIPHER], [WARN_CIPHER_MODE, WARN_BLOCK_SIZE]],
+ '3des': [[], [FAIL_OPENSSH67_UNSAFE], [WARN_OPENSSH74_UNSAFE, WARN_CIPHER_WEAK, WARN_CIPHER_MODE, WARN_BLOCK_SIZE]],
+ '3des-cbc': [['1.2.2,d0.28,l10.2', '6.6', None], [FAIL_OPENSSH67_UNSAFE], [WARN_OPENSSH74_UNSAFE, WARN_CIPHER_WEAK, WARN_CIPHER_MODE, WARN_BLOCK_SIZE]],
+ '3des-ctr': [['d0.52'], [FAIL_WEAK_CIPHER]],
+ 'blowfish': [[], [FAIL_WEAK_ALGORITHM], [WARN_BLOCK_SIZE]],
+ 'blowfish-cbc': [['1.2.2,d0.28,l10.2', '6.6,d0.52', '7.1,d0.52'], [FAIL_OPENSSH67_UNSAFE, FAIL_DBEAR53_DISABLED], [WARN_OPENSSH72_LEGACY, WARN_CIPHER_MODE, WARN_BLOCK_SIZE]],
+ 'blowfish-ctr': [[], [FAIL_OPENSSH67_UNSAFE, FAIL_DBEAR53_DISABLED], [WARN_OPENSSH72_LEGACY, WARN_CIPHER_MODE, WARN_BLOCK_SIZE]],
+ 'twofish-cbc': [['d0.28', 'd2014.66'], [FAIL_DBEAR67_DISABLED], [WARN_CIPHER_MODE]],
+ 'twofish128-cbc': [['d0.47', 'd2014.66'], [FAIL_DBEAR67_DISABLED], [WARN_CIPHER_MODE]],
+ 'twofish192-cbc': [[], [], [WARN_CIPHER_MODE]],
+ 'twofish256-cbc': [['d0.47', 'd2014.66'], [FAIL_DBEAR67_DISABLED], [WARN_CIPHER_MODE]],
+ 'twofish-ctr': [[]],
+ 'twofish128-ctr': [['d2015.68']],
+ 'twofish192-ctr': [[]],
+ 'twofish256-ctr': [['d2015.68']],
+ 'serpent128-cbc': [[], [FAIL_DEPRECATED_CIPHER], [WARN_CIPHER_MODE]],
+ 'serpent192-cbc': [[], [FAIL_DEPRECATED_CIPHER], [WARN_CIPHER_MODE]],
+ 'serpent256-cbc': [[], [FAIL_DEPRECATED_CIPHER], [WARN_CIPHER_MODE]],
+ 'serpent128-ctr': [[], [FAIL_DEPRECATED_CIPHER]],
+ 'serpent192-ctr': [[], [FAIL_DEPRECATED_CIPHER]],
+ 'serpent256-ctr': [[], [FAIL_DEPRECATED_CIPHER]],
+ 'idea-cbc': [[], [FAIL_DEPRECATED_CIPHER], [WARN_CIPHER_MODE]],
+ 'idea-ctr': [[], [FAIL_DEPRECATED_CIPHER]],
+ 'cast128-ctr': [[], [FAIL_DEPRECATED_CIPHER]],
+ 'cast128-cbc': [['2.1.0', '6.6', '7.1'], [FAIL_OPENSSH67_UNSAFE], [WARN_OPENSSH72_LEGACY, WARN_CIPHER_MODE, WARN_BLOCK_SIZE]],
+ 'arcfour': [['2.1.0', '6.6', '7.1'], [FAIL_OPENSSH67_UNSAFE], [WARN_OPENSSH72_LEGACY, WARN_CIPHER_WEAK]],
+ 'arcfour128': [['4.2', '6.6', '7.1'], [FAIL_OPENSSH67_UNSAFE], [WARN_OPENSSH72_LEGACY, WARN_CIPHER_WEAK]],
+ 'arcfour256': [['4.2', '6.6', '7.1'], [FAIL_OPENSSH67_UNSAFE], [WARN_OPENSSH72_LEGACY, WARN_CIPHER_WEAK]],
+ 'aes128-cbc': [['2.3.0,d0.28,l10.2', '6.6', None], [FAIL_OPENSSH67_UNSAFE], [WARN_CIPHER_MODE]],
+ 'aes192-cbc': [['2.3.0,l10.2', '6.6', None], [FAIL_OPENSSH67_UNSAFE], [WARN_CIPHER_MODE]],
+ 'aes256-cbc': [['2.3.0,d0.47,l10.2', '6.6', None], [FAIL_OPENSSH67_UNSAFE], [WARN_CIPHER_MODE]],
+ 'rijndael128-cbc': [['2.3.0', '3.0.2'], [FAIL_OPENSSH31_REMOVE], [WARN_CIPHER_MODE]],
+ 'rijndael192-cbc': [['2.3.0', '3.0.2'], [FAIL_OPENSSH31_REMOVE], [WARN_CIPHER_MODE]],
+ 'rijndael256-cbc': [['2.3.0', '3.0.2'], [FAIL_OPENSSH31_REMOVE], [WARN_CIPHER_MODE]],
+ 'rijndael-cbc@lysator.liu.se': [['2.3.0', '6.6', '7.1'], [FAIL_OPENSSH67_UNSAFE], [WARN_OPENSSH72_LEGACY, WARN_CIPHER_MODE]],
+ 'aes128-ctr': [['3.7,d0.52,l10.4.1']],
+ 'aes192-ctr': [['3.7,l10.4.1']],
+ 'aes256-ctr': [['3.7,d0.52,l10.4.1']],
+ 'aes128-gcm': [[]],
+ 'aes256-gcm': [[]],
+ 'AEAD_AES_128_GCM': [[]],
+ 'AEAD_AES_256_GCM': [[]],
+ 'aes128-gcm@openssh.com': [['6.2']],
+ 'aes256-gcm@openssh.com': [['6.2']],
+ 'chacha20-poly1305': [[], [], [], [INFO_OPENSSH69_CHACHA]],
+ 'chacha20-poly1305@openssh.com': [['6.5'], [], [], [INFO_OPENSSH69_CHACHA]],
+ 'camellia128-cbc': [[], [], [WARN_CIPHER_MODE]],
+ 'camellia128-ctr': [[]],
+ 'camellia192-cbc': [[], [], [WARN_CIPHER_MODE]],
+ 'camellia192-ctr': [[]],
+ 'camellia256-cbc': [[], [], [WARN_CIPHER_MODE]],
+ 'camellia256-ctr': [[]],
+ 'crypticore128@ssh.com': [[], [FAIL_UNPROVEN]],
+ 'seed-cbc@ssh.com': [[], [], [WARN_OBSOLETE, WARN_CIPHER_MODE]],
+ },
+ 'mac': {
+ 'none': [['d2013.56'], [FAIL_PLAINTEXT]],
+ 'hmac-sha1': [['2.1.0,d0.28,l10.2'], [], [WARN_ENCRYPT_AND_MAC, WARN_HASH_WEAK]],
+ 'hmac-sha1-96': [['2.5.0,d0.47', '6.6', '7.1'], [FAIL_OPENSSH67_UNSAFE], [WARN_OPENSSH72_LEGACY, WARN_ENCRYPT_AND_MAC, WARN_HASH_WEAK]],
+ 'hmac-sha2-56': [[], [], [WARN_TAG_SIZE, WARN_ENCRYPT_AND_MAC]],
+ 'hmac-sha2-224': [[], [], [WARN_TAG_SIZE, WARN_ENCRYPT_AND_MAC]],
+ 'hmac-sha2-256': [['5.9,d2013.56,l10.7.0'], [], [WARN_ENCRYPT_AND_MAC]],
+ 'hmac-sha2-256-96': [['5.9', '6.0'], [FAIL_OPENSSH61_REMOVE], [WARN_ENCRYPT_AND_MAC]],
+ 'hmac-sha2-384': [[], [], [WARN_ENCRYPT_AND_MAC]],
+ 'hmac-sha2-512': [['5.9,d2013.56,l10.7.0'], [], [WARN_ENCRYPT_AND_MAC]],
+ 'hmac-sha2-512-96': [['5.9', '6.0'], [FAIL_OPENSSH61_REMOVE], [WARN_ENCRYPT_AND_MAC]],
+ 'hmac-sha3-224': [[], [], [WARN_ENCRYPT_AND_MAC]],
+ 'hmac-sha3-256': [[], [], [WARN_ENCRYPT_AND_MAC]],
+ 'hmac-sha3-384': [[], [], [WARN_ENCRYPT_AND_MAC]],
+ 'hmac-sha3-512': [[], [], [WARN_ENCRYPT_AND_MAC]],
+ 'hmac-sha256': [[], [], [WARN_ENCRYPT_AND_MAC]],
+ 'hmac-sha256-96@ssh.com': [[], [], [WARN_ENCRYPT_AND_MAC, WARN_TAG_SIZE]],
+ 'hmac-sha256@ssh.com': [[], [], [WARN_ENCRYPT_AND_MAC]],
+ 'hmac-sha512': [[], [], [WARN_ENCRYPT_AND_MAC]],
+ 'hmac-sha512@ssh.com': [[], [], [WARN_ENCRYPT_AND_MAC]],
+ 'hmac-md5': [['2.1.0,d0.28', '6.6', '7.1'], [FAIL_OPENSSH67_UNSAFE], [WARN_OPENSSH72_LEGACY, WARN_ENCRYPT_AND_MAC, WARN_HASH_WEAK]],
+ 'hmac-md5-96': [['2.5.0', '6.6', '7.1'], [FAIL_OPENSSH67_UNSAFE], [WARN_OPENSSH72_LEGACY, WARN_ENCRYPT_AND_MAC, WARN_HASH_WEAK]],
+ 'hmac-ripemd': [[], [FAIL_DEPRECATED_MAC], [WARN_OPENSSH72_LEGACY, WARN_ENCRYPT_AND_MAC]],
+ 'hmac-ripemd160': [['2.5.0', '6.6', '7.1'], [FAIL_OPENSSH67_UNSAFE], [WARN_OPENSSH72_LEGACY, WARN_ENCRYPT_AND_MAC]],
+ 'hmac-ripemd160-96': [[], [FAIL_DEPRECATED_MAC], [WARN_ENCRYPT_AND_MAC, WARN_TAG_SIZE]],
+ 'hmac-ripemd160@openssh.com': [['2.1.0', '6.6', '7.1'], [FAIL_OPENSSH67_UNSAFE], [WARN_OPENSSH72_LEGACY, WARN_ENCRYPT_AND_MAC]],
+ 'umac-64@openssh.com': [['4.7'], [], [WARN_ENCRYPT_AND_MAC, WARN_TAG_SIZE]],
+ 'umac-128@openssh.com': [['6.2'], [], [WARN_ENCRYPT_AND_MAC]],
+ 'hmac-sha1-etm@openssh.com': [['6.2'], [], [WARN_HASH_WEAK]],
+ 'hmac-sha1-96-etm@openssh.com': [['6.2', '6.6', None], [FAIL_OPENSSH67_UNSAFE], [WARN_HASH_WEAK]],
+ 'hmac-sha2-256-96-etm@openssh.com': [[], [], [WARN_TAG_SIZE_96]], # Despite the @openssh.com tag, it doesn't appear that this was ever shipped with OpenSSH; it is only implemented in AsyncSSH (?).
+ 'hmac-sha2-512-96-etm@openssh.com': [[], [], [WARN_TAG_SIZE_96]], # Despite the @openssh.com tag, it doesn't appear that this was ever shipped with OpenSSH; it is only implemented in AsyncSSH (?).
+ 'hmac-sha2-256-etm@openssh.com': [['6.2']],
+ 'hmac-sha2-512-etm@openssh.com': [['6.2']],
+ 'hmac-md5-etm@openssh.com': [['6.2', '6.6', '7.1'], [FAIL_OPENSSH67_UNSAFE], [WARN_OPENSSH72_LEGACY, WARN_HASH_WEAK]],
+ 'hmac-md5-96-etm@openssh.com': [['6.2', '6.6', '7.1'], [FAIL_OPENSSH67_UNSAFE], [WARN_OPENSSH72_LEGACY, WARN_HASH_WEAK]],
+ 'hmac-ripemd160-etm@openssh.com': [['6.2', '6.6', '7.1'], [FAIL_OPENSSH67_UNSAFE], [WARN_OPENSSH72_LEGACY]],
+ 'umac-32@openssh.com': [[], [], [WARN_ENCRYPT_AND_MAC, WARN_TAG_SIZE]], # Despite having the @openssh.com suffix, this may never have shipped with OpenSSH (!).
+ 'umac-64-etm@openssh.com': [['6.2'], [], [WARN_TAG_SIZE]],
+ 'umac-96@openssh.com': [[], [], [WARN_ENCRYPT_AND_MAC]], # Despite having the @openssh.com suffix, this may never have shipped with OpenSSH (!).
+ 'umac-128-etm@openssh.com': [['6.2']],
+ 'aes128-gcm': [[]],
+ 'aes256-gcm': [[]],
+ 'chacha20-poly1305@openssh.com': [[]], # Despite the @openssh.com tag, this was never shipped as a MAC in OpenSSH (only as a cipher); it is only implemented as a MAC in Syncplify.
+ 'crypticore-mac@ssh.com': [[], [FAIL_UNPROVEN]],
+ 'AEAD_AES_128_GCM': [[]],
+ 'AEAD_AES_256_GCM': [[]],
+ }
+ }
diff --git a/src/ssh_audit/ssh2_kexparty.py b/src/ssh_audit/ssh2_kexparty.py
new file mode 100644
index 0000000..52b5ac5
--- /dev/null
+++ b/src/ssh_audit/ssh2_kexparty.py
@@ -0,0 +1,50 @@
+"""
+ The MIT License (MIT)
+
+ Copyright (C) 2017 Andris Raugulis (moo@arthepsy.eu)
+
+ Permission is hereby granted, free of charge, to any person obtaining a copy
+ of this software and associated documentation files (the "Software"), to deal
+ in the Software without restriction, including without limitation the rights
+ to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+ copies of the Software, and to permit persons to whom the Software is
+ furnished to do so, subject to the following conditions:
+
+ The above copyright notice and this permission notice shall be included in
+ all copies or substantial portions of the Software.
+
+ THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+ IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+ FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+ AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+ LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+ OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+ THE SOFTWARE.
+"""
+# pylint: disable=unused-import
+from typing import Dict, List, Set, Sequence, Tuple, Iterable # noqa: F401
+from typing import Callable, Optional, Union, Any # noqa: F401
+
+
+class SSH2_KexParty:
+ def __init__(self, enc: List[str], mac: List[str], compression: List[str], languages: List[str]) -> None:
+ self.__enc = enc
+ self.__mac = mac
+ self.__compression = compression
+ self.__languages = languages
+
+ @property
+ def encryption(self) -> List[str]:
+ return self.__enc
+
+ @property
+ def mac(self) -> List[str]:
+ return self.__mac
+
+ @property
+ def compression(self) -> List[str]:
+ return self.__compression
+
+ @property
+ def languages(self) -> List[str]:
+ return self.__languages
diff --git a/src/ssh_audit/ssh_audit.py b/src/ssh_audit/ssh_audit.py
new file mode 100755
index 0000000..38b5095
--- /dev/null
+++ b/src/ssh_audit/ssh_audit.py
@@ -0,0 +1,1132 @@
+#!/usr/bin/env python3
+"""
+ The MIT License (MIT)
+
+ Copyright (C) 2017-2021 Joe Testa (jtesta@positronsecurity.com)
+ Copyright (C) 2017 Andris Raugulis (moo@arthepsy.eu)
+
+ Permission is hereby granted, free of charge, to any person obtaining a copy
+ of this software and associated documentation files (the "Software"), to deal
+ in the Software without restriction, including without limitation the rights
+ to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+ copies of the Software, and to permit persons to whom the Software is
+ furnished to do so, subject to the following conditions:
+
+ The above copyright notice and this permission notice shall be included in
+ all copies or substantial portions of the Software.
+
+ THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+ IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+ FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+ AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+ LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+ OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+ THE SOFTWARE.
+"""
+import concurrent.futures
+import copy
+import getopt
+import json
+import os
+import re
+import sys
+import traceback
+
+# pylint: disable=unused-import
+from typing import Dict, List, Set, Sequence, Tuple, Iterable # noqa: F401
+from typing import Callable, Optional, Union, Any # noqa: F401
+
+from ssh_audit.globals import VERSION
+from ssh_audit.globals import WINDOWS_MAN_PAGE
+from ssh_audit.algorithm import Algorithm
+from ssh_audit.algorithms import Algorithms
+from ssh_audit.auditconf import AuditConf
+from ssh_audit.banner import Banner
+from ssh_audit import exitcodes
+from ssh_audit.fingerprint import Fingerprint
+from ssh_audit.gextest import GEXTest
+from ssh_audit.hostkeytest import HostKeyTest
+from ssh_audit.outputbuffer import OutputBuffer
+from ssh_audit.policy import Policy
+from ssh_audit.product import Product
+from ssh_audit.protocol import Protocol
+from ssh_audit.software import Software
+from ssh_audit.ssh1_kexdb import SSH1_KexDB
+from ssh_audit.ssh1_publickeymessage import SSH1_PublicKeyMessage
+from ssh_audit.ssh2_kex import SSH2_Kex
+from ssh_audit.ssh2_kexdb import SSH2_KexDB
+from ssh_audit.ssh_socket import SSH_Socket
+from ssh_audit.utils import Utils
+from ssh_audit.versionvulnerabilitydb import VersionVulnerabilityDB
+
+# Only import colorama under Windows. Other OSes can natively handle terminal colors.
+if sys.platform == 'win32':
+ try:
+ from colorama import init as colorama_init
+ colorama_init()
+ except ImportError:
+ pass
+
+
+def usage(err: Optional[str] = None) -> None:
+ retval = exitcodes.GOOD
+ uout = OutputBuffer()
+ p = os.path.basename(sys.argv[0])
+ uout.head('# {} {}, https://github.com/jtesta/ssh-audit\n'.format(p, VERSION))
+ if err is not None and len(err) > 0:
+ uout.fail('\n' + err)
+ retval = exitcodes.UNKNOWN_ERROR
+ uout.info('usage: {0} [options] <host>\n'.format(p))
+ uout.info(' -h, --help print this help')
+ uout.info(' -1, --ssh1 force ssh version 1 only')
+ uout.info(' -2, --ssh2 force ssh version 2 only')
+ uout.info(' -4, --ipv4 enable IPv4 (order of precedence)')
+ uout.info(' -6, --ipv6 enable IPv6 (order of precedence)')
+ uout.info(' -b, --batch batch output')
+ uout.info(' -c, --client-audit starts a server on port 2222 to audit client\n software config (use -p to change port;\n use -t to change timeout)')
+ uout.info(' -d, --debug debug output')
+ uout.info(' -j, --json JSON output (use -jj to enable indents)')
+ uout.info(' -l, --level=<level> minimum output level (info|warn|fail)')
+ uout.info(' -L, --list-policies list all the official, built-in policies')
+ uout.info(' --lookup=<alg1,alg2,...> looks up an algorithm(s) without\n connecting to a server')
+ uout.info(' -M, --make-policy=<policy.txt> creates a policy based on the target server\n (i.e.: the target server has the ideal\n configuration that other servers should\n adhere to)')
+ uout.info(' -m, --manual print the man page (Windows only)')
+ uout.info(' -n, --no-colors disable colors')
+ uout.info(' -p, --port=<port> port to connect')
+ uout.info(' -P, --policy=<policy.txt> run a policy test using the specified policy')
+ uout.info(' -t, --timeout=<secs> timeout (in seconds) for connection and reading\n (default: 5)')
+ uout.info(' -T, --targets=<hosts.txt> a file containing a list of target hosts (one\n per line, format HOST[:PORT]). Use --threads\n to control concurrent scans.')
+ uout.info(' --threads=<threads> number of threads to use when scanning multiple\n targets (-T/--targets) (default: 32)')
+ uout.info(' -v, --verbose verbose output')
+ uout.sep()
+ uout.write()
+ sys.exit(retval)
+
+
+def output_algorithms(out: OutputBuffer, title: str, alg_db: Dict[str, Dict[str, List[List[Optional[str]]]]], alg_type: str, algorithms: List[str], unknown_algs: List[str], is_json_output: bool, program_retval: int, maxlen: int = 0, alg_sizes: Optional[Dict[str, Tuple[int, int]]] = None) -> int: # pylint: disable=too-many-arguments
+ with out:
+ for algorithm in algorithms:
+ program_retval = output_algorithm(out, alg_db, alg_type, algorithm, unknown_algs, program_retval, maxlen, alg_sizes)
+ if not out.is_section_empty() and not is_json_output:
+ out.head('# ' + title)
+ out.flush_section()
+ out.sep()
+
+ return program_retval
+
+
+def output_algorithm(out: OutputBuffer, alg_db: Dict[str, Dict[str, List[List[Optional[str]]]]], alg_type: str, alg_name: str, unknown_algs: List[str], program_retval: int, alg_max_len: int = 0, alg_sizes: Optional[Dict[str, Tuple[int, int]]] = None) -> int:
+ prefix = '(' + alg_type + ') '
+ if alg_max_len == 0:
+ alg_max_len = len(alg_name)
+ padding = '' if out.batch else ' ' * (alg_max_len - len(alg_name))
+
+ # If this is an RSA host key or DH GEX, append the size to its name and fix
+ # the padding.
+ alg_name_with_size = None
+ if (alg_sizes is not None) and (alg_name in alg_sizes):
+ hostkey_size, ca_size = alg_sizes[alg_name]
+ if ca_size > 0:
+ alg_name_with_size = '%s (%d-bit cert/%d-bit CA)' % (alg_name, hostkey_size, ca_size)
+ padding = padding[0:-15]
+ else:
+ alg_name_with_size = '%s (%d-bit)' % (alg_name, hostkey_size)
+ padding = padding[0:-11]
+
+ texts = []
+ if len(alg_name.strip()) == 0:
+ return program_retval
+ alg_name_native = Utils.to_text(alg_name)
+ if alg_name_native in alg_db[alg_type]:
+ alg_desc = alg_db[alg_type][alg_name_native]
+ ldesc = len(alg_desc)
+ for idx, level in enumerate(['fail', 'warn', 'info']):
+ if level == 'info':
+ versions = alg_desc[0]
+ since_text = Algorithm.get_since_text(versions)
+ if since_text is not None and len(since_text) > 0:
+ texts.append((level, since_text))
+ idx = idx + 1
+ if ldesc > idx:
+ for t in alg_desc[idx]:
+ if t is None:
+ continue
+ texts.append((level, t))
+ if len(texts) == 0:
+ texts.append(('info', ''))
+ else:
+ texts.append(('warn', 'unknown algorithm'))
+ unknown_algs.append(alg_name)
+
+ alg_name = alg_name_with_size if alg_name_with_size is not None else alg_name
+ first = True
+ for level, text in texts:
+ if level == 'fail':
+ program_retval = exitcodes.FAILURE
+ elif level == 'warn' and program_retval != exitcodes.FAILURE: # If a failure was found previously, don't downgrade to warning.
+ program_retval = exitcodes.WARNING
+
+ f = getattr(out, level)
+ comment = (padding + ' -- [' + level + '] ' + text) if text != '' else ''
+ if first:
+ if first and level == 'info':
+ f = out.good
+ f(prefix + alg_name + comment)
+ first = False
+ else: # pylint: disable=else-if-used
+ if out.verbose:
+ f(prefix + alg_name + comment)
+ elif text != '':
+ comment = (padding + ' `- [' + level + '] ' + text)
+ f(' ' * len(prefix + alg_name) + comment)
+
+ return program_retval
+
+
+def output_compatibility(out: OutputBuffer, algs: Algorithms, client_audit: bool, for_server: bool = True) -> None:
+
+ # Don't output any compatibility info if we're doing a client audit.
+ if client_audit:
+ return
+
+ ssh_timeframe = algs.get_ssh_timeframe(for_server)
+ comp_text = []
+ for ssh_prod in [Product.OpenSSH, Product.DropbearSSH]:
+ if ssh_prod not in ssh_timeframe:
+ continue
+ v_from = ssh_timeframe.get_from(ssh_prod, for_server)
+ v_till = ssh_timeframe.get_till(ssh_prod, for_server)
+ if v_from is None:
+ continue
+ if v_till is None:
+ comp_text.append('{} {}+'.format(ssh_prod, v_from))
+ elif v_from == v_till:
+ comp_text.append('{} {}'.format(ssh_prod, v_from))
+ else:
+ software = Software(None, ssh_prod, v_from, None, None)
+ if software.compare_version(v_till) > 0:
+ tfmt = '{0} {1}+ (some functionality from {2})'
+ else:
+ tfmt = '{0} {1}-{2}'
+ comp_text.append(tfmt.format(ssh_prod, v_from, v_till))
+ if len(comp_text) > 0:
+ out.good('(gen) compatibility: ' + ', '.join(comp_text))
+
+
+def output_security_sub(out: OutputBuffer, sub: str, software: Optional[Software], client_audit: bool, padlen: int) -> None:
+ secdb = VersionVulnerabilityDB.CVE if sub == 'cve' else VersionVulnerabilityDB.TXT
+ if software is None or software.product not in secdb:
+ return
+ for line in secdb[software.product]:
+ vfrom: str = ''
+ vtill: str = ''
+ vfrom, vtill = line[0:2]
+ if not software.between_versions(vfrom, vtill):
+ continue
+ target: int = 0
+ name: str = ''
+ target, name = line[2:4]
+ is_server = target & 1 == 1
+ is_client = target & 2 == 2
+ # is_local = target & 4 == 4
+
+ # If this security entry applies only to servers, but we're testing a client, then skip it. Similarly, skip entries that apply only to clients, but we're testing a server.
+ if (is_server and not is_client and client_audit) or (is_client and not is_server and not client_audit):
+ continue
+ p = '' if out.batch else ' ' * (padlen - len(name))
+ if sub == 'cve':
+ cvss: float = 0.0
+ descr: str = ''
+ cvss, descr = line[4:6]
+
+ # Critical CVSS scores (>= 8.0) are printed as a fail, otherwise they are printed as a warning.
+ out_func = out.warn
+ if cvss >= 8.0:
+ out_func = out.fail
+ out_func('(cve) {}{} -- (CVSSv2: {}) {}'.format(name, p, cvss, descr))
+ else:
+ descr = line[4]
+ out.fail('(sec) {}{} -- {}'.format(name, p, descr))
+
+
+def output_security(out: OutputBuffer, banner: Optional[Banner], client_audit: bool, padlen: int, is_json_output: bool) -> None:
+ with out:
+ if banner is not None:
+ software = Software.parse(banner)
+ output_security_sub(out, 'cve', software, client_audit, padlen)
+ output_security_sub(out, 'txt', software, client_audit, padlen)
+ if banner.protocol[0] == 1:
+ p = '' if out.batch else ' ' * (padlen - 14)
+ out.fail('(sec) SSH v1 enabled{} -- SSH v1 can be exploited to recover plaintext passwords'.format(p))
+ if not out.is_section_empty() and not is_json_output:
+ out.head('# security')
+ out.flush_section()
+ out.sep()
+
+
+def output_fingerprints(out: OutputBuffer, algs: Algorithms, is_json_output: bool) -> None:
+ with out:
+ fps = []
+ if algs.ssh1kex is not None:
+ name = 'ssh-rsa1'
+ fp = Fingerprint(algs.ssh1kex.host_key_fingerprint_data)
+ # bits = algs.ssh1kex.host_key_bits
+ fps.append((name, fp))
+ if algs.ssh2kex is not None:
+ host_keys = algs.ssh2kex.host_keys()
+ for host_key_type in algs.ssh2kex.host_keys():
+ if host_keys[host_key_type] is None:
+ continue
+
+ fp = Fingerprint(host_keys[host_key_type])
+
+ # Workaround for Python's order-indifference in dicts. We might get a random RSA type (ssh-rsa, rsa-sha2-256, or rsa-sha2-512), so running the tool against the same server three times may give three different host key types here. So if we have any RSA type, we will simply hard-code it to 'ssh-rsa'.
+ if host_key_type in HostKeyTest.RSA_FAMILY:
+ host_key_type = 'ssh-rsa'
+
+ # Skip over certificate host types (or we would return invalid fingerprints).
+ if '-cert-' not in host_key_type:
+ fps.append((host_key_type, fp))
+ # Similarly, the host keys can be processed in random order due to Python's order-indifference in dicts. So we sort this list before printing; this makes automated testing possible.
+ fps = sorted(fps)
+ for fpp in fps:
+ name, fp = fpp
+ out.good('(fin) {}: {}'.format(name, fp.sha256))
+
+ # Output the MD5 hash too if verbose mode is enabled.
+ if out.verbose:
+ out.info('(fin) {}: {} -- [info] do not rely on MD5 fingerprints for server identification; it is insecure for this use case'.format(name, fp.md5))
+
+ if not out.is_section_empty() and not is_json_output:
+ out.head('# fingerprints')
+ out.flush_section()
+ out.sep()
+
+
+# Returns True if no warnings or failures encountered in configuration.
+def output_recommendations(out: OutputBuffer, algs: Algorithms, software: Optional[Software], is_json_output: bool, padlen: int = 0) -> bool:
+
+ ret = True
+ # PuTTY's algorithms cannot be modified, so there's no point in issuing recommendations.
+ if (software is not None) and (software.product == Product.PuTTY):
+ max_vuln_version = 0.0
+ max_cvssv2_severity = 0.0
+ # Search the CVE database for the most recent vulnerable version and the max CVSSv2 score.
+ for cve_list in VersionVulnerabilityDB.CVE['PuTTY']:
+ vuln_version = float(cve_list[1])
+ cvssv2_severity = cve_list[4]
+
+ if vuln_version > max_vuln_version:
+ max_vuln_version = vuln_version
+ if cvssv2_severity > max_cvssv2_severity:
+ max_cvssv2_severity = cvssv2_severity
+
+ fn = out.warn
+ if max_cvssv2_severity > 8.0:
+ fn = out.fail
+
+ # Assuming that PuTTY versions will always increment by 0.01, we can calculate the first safe version by adding 0.01 to the latest vulnerable version.
+ current_version = float(software.version)
+ upgrade_to_version = max_vuln_version + 0.01
+ if current_version < upgrade_to_version:
+ out.head('# recommendations')
+ fn('(rec) Upgrade to PuTTY v%.2f' % upgrade_to_version)
+ out.sep()
+ ret = False
+ return ret
+
+ for_server = True
+ with out:
+ software, alg_rec = algs.get_recommendations(software, for_server)
+ for sshv in range(2, 0, -1):
+ if sshv not in alg_rec:
+ continue
+ for alg_type in ['kex', 'key', 'enc', 'mac']:
+ if alg_type not in alg_rec[sshv]:
+ continue
+ for action in ['del', 'add', 'chg']:
+ if action not in alg_rec[sshv][alg_type]:
+ continue
+ for name in alg_rec[sshv][alg_type][action]:
+ p = '' if out.batch else ' ' * (padlen - len(name))
+ chg_additional_info = ''
+ if action == 'del':
+ an, sg, fn = 'remove', '-', out.warn
+ ret = False
+ if alg_rec[sshv][alg_type][action][name] >= 10:
+ fn = out.fail
+ elif action == 'add':
+ an, sg, fn = 'append', '+', out.good
+ elif action == 'chg':
+ an, sg, fn = 'change', '!', out.fail
+ ret = False
+ chg_additional_info = ' (increase modulus size to 2048 bits or larger)'
+ b = '(SSH{})'.format(sshv) if sshv == 1 else ''
+ fm = '(rec) {0}{1}{2}-- {3} algorithm to {4}{5} {6}'
+ fn(fm.format(sg, name, p, alg_type, an, chg_additional_info, b))
+ if not out.is_section_empty() and not is_json_output:
+ if software is not None:
+ title = '(for {})'.format(software.display(False))
+ else:
+ title = ''
+ out.head('# algorithm recommendations {}'.format(title))
+ out.flush_section(sort_section=True) # Sort the output so that it is always stable (needed for repeatable testing).
+ out.sep()
+ return ret
+
+
+# Output additional information & notes.
+def output_info(out: OutputBuffer, software: Optional['Software'], client_audit: bool, any_problems: bool, is_json_output: bool) -> None:
+ with out:
+ # Tell user that PuTTY cannot be hardened at the protocol-level.
+ if client_audit and (software is not None) and (software.product == Product.PuTTY):
+ out.warn('(nfo) PuTTY does not have the option of restricting any algorithms during the SSH handshake.')
+
+ # If any warnings or failures were given, print a link to the hardening guides.
+ if any_problems:
+ out.warn('(nfo) For hardening guides on common OSes, please see: <https://www.ssh-audit.com/hardening_guides.html>')
+
+ if not out.is_section_empty() and not is_json_output:
+ out.head('# additional info')
+ out.flush_section()
+ out.sep()
+
+
+# Returns a exitcodes.* flag to denote if any failures or warnings were encountered.
+def output(out: OutputBuffer, aconf: AuditConf, banner: Optional[Banner], header: List[str], client_host: Optional[str] = None, kex: Optional[SSH2_Kex] = None, pkm: Optional[SSH1_PublicKeyMessage] = None, print_target: bool = False) -> int:
+
+ program_retval = exitcodes.GOOD
+ client_audit = client_host is not None # If set, this is a client audit.
+ sshv = 1 if pkm is not None else 2
+ algs = Algorithms(pkm, kex)
+ with out:
+ if print_target:
+ host = aconf.host
+
+ # Print the port if it's not the default of 22.
+ if aconf.port != 22:
+
+ # Check if this is an IPv6 address, as that is printed in a different format.
+ if Utils.is_ipv6_address(aconf.host):
+ host = '[%s]:%d' % (aconf.host, aconf.port)
+ else:
+ host = '%s:%d' % (aconf.host, aconf.port)
+
+ out.good('(gen) target: {}'. format(host))
+ if client_audit:
+ out.good('(gen) client IP: {}'.format(client_host))
+ if len(header) > 0:
+ out.info('(gen) header: ' + '\n'.join(header))
+ if banner is not None:
+ banner_line = '(gen) banner: {}'.format(banner)
+ if sshv == 1 or banner.protocol[0] == 1:
+ out.fail(banner_line)
+ out.fail('(gen) protocol SSH1 enabled')
+ else:
+ out.good(banner_line)
+
+ if not banner.valid_ascii:
+ # NOTE: RFC 4253, Section 4.2
+ out.warn('(gen) banner contains non-printable ASCII')
+
+ software = Software.parse(banner)
+ if software is not None:
+ out.good('(gen) software: {}'.format(software))
+ else:
+ software = None
+ output_compatibility(out, algs, client_audit)
+ if kex is not None:
+ compressions = [x for x in kex.server.compression if x != 'none']
+ if len(compressions) > 0:
+ cmptxt = 'enabled ({})'.format(', '.join(compressions))
+ else:
+ cmptxt = 'disabled'
+ out.good('(gen) compression: {}'.format(cmptxt))
+ if not out.is_section_empty() and not aconf.json: # Print output when it exists and JSON output isn't requested.
+ out.head('# general')
+ out.flush_section()
+ out.sep()
+ maxlen = algs.maxlen + 1
+ output_security(out, banner, client_audit, maxlen, aconf.json)
+ # Filled in by output_algorithms() with unidentified algs.
+ unknown_algorithms: List[str] = []
+ if pkm is not None:
+ adb = SSH1_KexDB.ALGORITHMS
+ ciphers = pkm.supported_ciphers
+ auths = pkm.supported_authentications
+ title, atype = 'SSH1 host-key algorithms', 'key'
+ program_retval = output_algorithms(out, title, adb, atype, ['ssh-rsa1'], unknown_algorithms, aconf.json, program_retval, maxlen)
+ title, atype = 'SSH1 encryption algorithms (ciphers)', 'enc'
+ program_retval = output_algorithms(out, title, adb, atype, ciphers, unknown_algorithms, aconf.json, program_retval, maxlen)
+ title, atype = 'SSH1 authentication types', 'aut'
+ program_retval = output_algorithms(out, title, adb, atype, auths, unknown_algorithms, aconf.json, program_retval, maxlen)
+ if kex is not None:
+ adb = SSH2_KexDB.ALGORITHMS
+ title, atype = 'key exchange algorithms', 'kex'
+ program_retval = output_algorithms(out, title, adb, atype, kex.kex_algorithms, unknown_algorithms, aconf.json, program_retval, maxlen, kex.dh_modulus_sizes())
+ title, atype = 'host-key algorithms', 'key'
+ program_retval = output_algorithms(out, title, adb, atype, kex.key_algorithms, unknown_algorithms, aconf.json, program_retval, maxlen, kex.rsa_key_sizes())
+ title, atype = 'encryption algorithms (ciphers)', 'enc'
+ program_retval = output_algorithms(out, title, adb, atype, kex.server.encryption, unknown_algorithms, aconf.json, program_retval, maxlen)
+ title, atype = 'message authentication code algorithms', 'mac'
+ program_retval = output_algorithms(out, title, adb, atype, kex.server.mac, unknown_algorithms, aconf.json, program_retval, maxlen)
+ output_fingerprints(out, algs, aconf.json)
+ perfect_config = output_recommendations(out, algs, software, aconf.json, maxlen)
+ output_info(out, software, client_audit, not perfect_config, aconf.json)
+
+ if aconf.json:
+ out.reset()
+ # Build & write the JSON struct.
+ out.info(json.dumps(build_struct(aconf.host, banner, kex=kex, client_host=client_host), indent=4 if aconf.json_print_indent else None, sort_keys=True))
+ elif len(unknown_algorithms) > 0: # If we encountered any unknown algorithms, ask the user to report them.
+ out.warn("\n\n!!! WARNING: unknown algorithm(s) found!: %s. Please email the full output above to the maintainer (jtesta@positronsecurity.com), or create a Github issue at <https://github.com/jtesta/ssh-audit/issues>.\n" % ','.join(unknown_algorithms))
+
+ return program_retval
+
+
+def evaluate_policy(out: OutputBuffer, aconf: AuditConf, banner: Optional['Banner'], client_host: Optional[str], kex: Optional['SSH2_Kex'] = None) -> bool:
+
+ if aconf.policy is None:
+ raise RuntimeError('Internal error: cannot evaluate against null Policy!')
+
+ passed, error_struct, error_str = aconf.policy.evaluate(banner, kex)
+ if aconf.json:
+ json_struct = {'host': aconf.host, 'policy': aconf.policy.get_name_and_version(), 'passed': passed, 'errors': error_struct}
+ out.info(json.dumps(json_struct, indent=4 if aconf.json_print_indent else None, sort_keys=True))
+ else:
+ spacing = ''
+ if aconf.client_audit:
+ out.info("Client IP: %s" % client_host)
+ spacing = " " # So the fields below line up with 'Client IP: '.
+ else:
+ host = aconf.host
+ if aconf.port != 22:
+ # Check if this is an IPv6 address, as that is printed in a different format.
+ if Utils.is_ipv6_address(aconf.host):
+ host = '[%s]:%d' % (aconf.host, aconf.port)
+ else:
+ host = '%s:%d' % (aconf.host, aconf.port)
+
+ out.info("Host: %s" % host)
+ out.info("Policy: %s%s" % (spacing, aconf.policy.get_name_and_version()))
+ out.info("Result: %s" % spacing, line_ended=False)
+
+ # Use these nice unicode characters in the result message, unless we're on Windows (the cmd.exe terminal doesn't display them properly).
+ icon_good = "✔ "
+ icon_fail = "⌠"
+ if Utils.is_windows():
+ icon_good = ""
+ icon_fail = ""
+
+ if passed:
+ out.good("%sPassed" % icon_good)
+ else:
+ out.fail("%sFailed!" % icon_fail)
+ out.warn("\nErrors:\n%s" % error_str)
+
+ return passed
+
+
+def list_policies(out: OutputBuffer) -> None:
+ '''Prints a list of server & client policies.'''
+
+ server_policy_names, client_policy_names = Policy.list_builtin_policies()
+
+ if len(server_policy_names) > 0:
+ out.head('\nServer policies:\n')
+ out.info(" * \"%s\"" % "\"\n * \"".join(server_policy_names))
+
+ if len(client_policy_names) > 0:
+ out.head('\nClient policies:\n')
+ out.info(" * \"%s\"" % "\"\n * \"".join(client_policy_names))
+
+ out.sep()
+ if len(server_policy_names) == 0 and len(client_policy_names) == 0:
+ out.fail("Error: no built-in policies found!")
+ else:
+ out.info("\nHint: Use -P and provide the full name of a policy to run a policy scan with.\n")
+ out.write()
+
+
+def make_policy(aconf: AuditConf, banner: Optional['Banner'], kex: Optional['SSH2_Kex'], client_host: Optional[str]) -> None:
+
+ # Set the source of this policy to the server host if this is a server audit, otherwise set it to the client address.
+ source: Optional[str] = aconf.host
+ if aconf.client_audit:
+ source = client_host
+
+ policy_data = Policy.create(source, banner, kex, aconf.client_audit)
+
+ if aconf.policy_file is None:
+ raise RuntimeError('Internal error: cannot write policy file since filename is None!')
+
+ # Open with mode 'x' (creates the file, or fails if it already exist).
+ succeeded = True
+ try:
+ with open(aconf.policy_file, 'x', encoding='utf-8') as f:
+ f.write(policy_data)
+ except FileExistsError:
+ succeeded = False
+
+ if succeeded:
+ print("Wrote policy to %s. Customize as necessary, then run a policy scan with -P option." % aconf.policy_file)
+ else:
+ print("Error: file already exists: %s" % aconf.policy_file)
+
+
+def process_commandline(out: OutputBuffer, args: List[str], usage_cb: Callable[..., None]) -> 'AuditConf': # pylint: disable=too-many-statements
+ # pylint: disable=too-many-branches
+ aconf = AuditConf()
+ try:
+ sopts = 'h1246M:p:P:jbcnvl:t:T:Lmd'
+ lopts = ['help', 'ssh1', 'ssh2', 'ipv4', 'ipv6', 'make-policy=', 'port=', 'policy=', 'json', 'batch', 'client-audit', 'no-colors', 'verbose', 'level=', 'timeout=', 'targets=', 'list-policies', 'lookup=', 'threads=', 'manual', 'debug']
+ opts, args = getopt.gnu_getopt(args, sopts, lopts)
+ except getopt.GetoptError as err:
+ usage_cb(str(err))
+ aconf.ssh1, aconf.ssh2 = False, False
+ host: str = ''
+ oport: Optional[str] = None
+ port: int = 0
+ for o, a in opts:
+ if o in ('-h', '--help'):
+ usage_cb()
+ elif o in ('-1', '--ssh1'):
+ aconf.ssh1 = True
+ elif o in ('-2', '--ssh2'):
+ aconf.ssh2 = True
+ elif o in ('-4', '--ipv4'):
+ aconf.ipv4 = True
+ elif o in ('-6', '--ipv6'):
+ aconf.ipv6 = True
+ elif o in ('-p', '--port'):
+ oport = a
+ elif o in ('-b', '--batch'):
+ aconf.batch = True
+ aconf.verbose = True
+ elif o in ('-c', '--client-audit'):
+ aconf.client_audit = True
+ elif o in ('-n', '--no-colors'):
+ aconf.colors = False
+ out.use_colors = False
+ elif o in ('-j', '--json'):
+ if aconf.json: # If specified twice, enable indent printing.
+ aconf.json_print_indent = True
+ else:
+ aconf.json = True
+ elif o in ('-v', '--verbose'):
+ aconf.verbose = True
+ out.verbose = True
+ elif o in ('-l', '--level'):
+ if a not in ('info', 'warn', 'fail'):
+ usage_cb('level {} is not valid'.format(a))
+ aconf.level = a
+ elif o in ('-t', '--timeout'):
+ aconf.timeout = float(a)
+ aconf.timeout_set = True
+ elif o in ('-M', '--make-policy'):
+ aconf.make_policy = True
+ aconf.policy_file = a
+ elif o in ('-P', '--policy'):
+ aconf.policy_file = a
+ elif o in ('-T', '--targets'):
+ aconf.target_file = a
+ elif o == '--threads':
+ aconf.threads = int(a)
+ elif o in ('-L', '--list-policies'):
+ aconf.list_policies = True
+ elif o == '--lookup':
+ aconf.lookup = a
+ elif o in ('-m', '--manual'):
+ aconf.manual = True
+ elif o in ('-d', '--debug'):
+ aconf.debug = True
+ out.debug = True
+
+ if len(args) == 0 and aconf.client_audit is False and aconf.target_file is None and aconf.list_policies is False and aconf.lookup == '' and aconf.manual is False:
+ usage_cb()
+
+ if aconf.manual:
+ return aconf
+
+ if aconf.lookup != '':
+ return aconf
+
+ if aconf.list_policies:
+ list_policies(out)
+ sys.exit(exitcodes.GOOD)
+
+ if aconf.client_audit is False and aconf.target_file is None:
+ if oport is not None:
+ host = args[0]
+ else:
+ host, port = Utils.parse_host_and_port(args[0])
+ if not host and aconf.target_file is None:
+ usage_cb('host is empty')
+
+ if port == 0 and oport is None:
+ if aconf.client_audit: # The default port to listen on during a client audit is 2222.
+ port = 2222
+ else:
+ port = 22
+
+ if oport is not None:
+ port = Utils.parse_int(oport)
+ if port <= 0 or port > 65535:
+ usage_cb('port {} is not valid'.format(oport))
+
+ aconf.host = host
+ aconf.port = port
+ if not (aconf.ssh1 or aconf.ssh2):
+ aconf.ssh1, aconf.ssh2 = True, True
+
+ # If a file containing a list of targets was given, read it.
+ if aconf.target_file is not None:
+ with open(aconf.target_file, 'r', encoding='utf-8') as f:
+ aconf.target_list = f.readlines()
+
+ # Strip out whitespace from each line in target file, and skip empty lines.
+ aconf.target_list = [target.strip() for target in aconf.target_list if target not in ("", "\n")]
+
+ # If a policy file was provided, validate it.
+ if (aconf.policy_file is not None) and (aconf.make_policy is False):
+
+ # First, see if this is a built-in policy name. If not, assume a file path was provided, and try to load it from disk.
+ aconf.policy = Policy.load_builtin_policy(aconf.policy_file)
+ if aconf.policy is None:
+ try:
+ aconf.policy = Policy(policy_file=aconf.policy_file)
+ except Exception as e:
+ out.fail("Error while loading policy file: %s: %s" % (str(e), traceback.format_exc()))
+ out.write()
+ sys.exit(exitcodes.UNKNOWN_ERROR)
+
+ # If the user wants to do a client audit, but provided a server policy, terminate.
+ if aconf.client_audit and aconf.policy.is_server_policy():
+ out.fail("Error: client audit selected, but server policy provided.")
+ out.write()
+ sys.exit(exitcodes.UNKNOWN_ERROR)
+
+ # If the user wants to do a server audit, but provided a client policy, terminate.
+ if aconf.client_audit is False and aconf.policy.is_server_policy() is False:
+ out.fail("Error: server audit selected, but client policy provided.")
+ out.write()
+ sys.exit(exitcodes.UNKNOWN_ERROR)
+
+ return aconf
+
+
+def build_struct(target_host: str, banner: Optional['Banner'], kex: Optional['SSH2_Kex'] = None, pkm: Optional['SSH1_PublicKeyMessage'] = None, client_host: Optional[str] = None) -> Any:
+
+ banner_str = ''
+ banner_protocol = None
+ banner_software = None
+ banner_comments = None
+ if banner is not None:
+ banner_str = str(banner)
+ banner_protocol = banner.protocol
+ banner_software = banner.software
+ banner_comments = banner.comments
+
+ res: Any = {
+ "banner": {
+ "raw": banner_str,
+ "protocol": banner_protocol,
+ "software": banner_software,
+ "comments": banner_comments,
+ },
+ }
+
+ # If we're scanning a client host, put the client's IP into the results. Otherwise, include the target host.
+ if client_host is not None:
+ res['client_ip'] = client_host
+ else:
+ res['target'] = target_host
+
+ if kex is not None:
+ res['compression'] = kex.server.compression
+
+ res['kex'] = []
+ alg_sizes = kex.dh_modulus_sizes()
+ for algorithm in kex.kex_algorithms:
+ entry: Any = {
+ 'algorithm': algorithm,
+ }
+ if algorithm in alg_sizes:
+ hostkey_size, ca_size = alg_sizes[algorithm]
+ entry['keysize'] = hostkey_size
+ if ca_size > 0:
+ entry['casize'] = ca_size
+ res['kex'].append(entry)
+
+ res['key'] = []
+ alg_sizes = kex.rsa_key_sizes()
+ for algorithm in kex.key_algorithms:
+ entry = {
+ 'algorithm': algorithm,
+ }
+ if algorithm in alg_sizes:
+ hostkey_size, ca_size = alg_sizes[algorithm]
+ entry['keysize'] = hostkey_size
+ if ca_size > 0:
+ entry['casize'] = ca_size
+ res['key'].append(entry)
+
+ res['enc'] = kex.server.encryption
+ res['mac'] = kex.server.mac
+ res['fingerprints'] = []
+ host_keys = kex.host_keys()
+
+ # Normalize all RSA key types to 'ssh-rsa'. Otherwise, due to Python's order-indifference dictionary types, we would iterate key types in unpredictable orders, which interferes with the docker testing framework (i.e.: tests would fail because elements are reported out of order, even though the output is semantically the same).
+ for host_key_type in list(host_keys.keys())[:]:
+ if host_key_type in HostKeyTest.RSA_FAMILY:
+ val = host_keys[host_key_type]
+ del host_keys[host_key_type]
+ host_keys['ssh-rsa'] = val
+
+ for host_key_type in sorted(host_keys):
+ if host_keys[host_key_type] is None:
+ continue
+
+ fp = Fingerprint(host_keys[host_key_type])
+
+ # Skip over certificate host types (or we would return invalid fingerprints).
+ if '-cert-' in host_key_type:
+ continue
+
+ # Add the SHA256 and MD5 fingerprints.
+ res['fingerprints'].append({
+ 'hostkey': host_key_type,
+ 'hash_alg': 'SHA256',
+ 'hash': fp.sha256[7:]
+ })
+ res['fingerprints'].append({
+ 'hostkey': host_key_type,
+ 'hash_alg': 'MD5',
+ 'hash': fp.md5[4:]
+ })
+ else:
+ pkm_supported_ciphers = None
+ pkm_supported_authentications = None
+ pkm_fp = None
+ if pkm is not None:
+ pkm_supported_ciphers = pkm.supported_ciphers
+ pkm_supported_authentications = pkm.supported_authentications
+ pkm_fp = Fingerprint(pkm.host_key_fingerprint_data).sha256
+
+ res['key'] = ['ssh-rsa1']
+ res['enc'] = pkm_supported_ciphers
+ res['aut'] = pkm_supported_authentications
+ res['fingerprints'] = [{
+ 'type': 'ssh-rsa1',
+ 'fp': pkm_fp,
+ }]
+
+ return res
+
+
+# Returns one of the exitcodes.* flags.
+def audit(out: OutputBuffer, aconf: AuditConf, sshv: Optional[int] = None, print_target: bool = False) -> int:
+ program_retval = exitcodes.GOOD
+ out.batch = aconf.batch
+ out.verbose = aconf.verbose
+ out.debug = aconf.debug
+ out.level = aconf.level
+ out.use_colors = aconf.colors
+ s = SSH_Socket(out, aconf.host, aconf.port, aconf.ip_version_preference, aconf.timeout, aconf.timeout_set)
+
+ if aconf.client_audit:
+ out.v("Listening for client connection on port %d..." % aconf.port, write_now=True)
+ s.listen_and_accept()
+ else:
+ out.v("Starting audit of %s:%d..." % ('[%s]' % aconf.host if Utils.is_ipv6_address(aconf.host) else aconf.host, aconf.port), write_now=True)
+ err = s.connect()
+
+ if err is not None:
+ out.fail(err)
+
+ # If we're running against multiple targets, return a connection error to the calling worker thread. Otherwise, write the error message to the console and exit.
+ if len(aconf.target_list) > 0:
+ return exitcodes.CONNECTION_ERROR
+ else:
+ out.write()
+ sys.exit(exitcodes.CONNECTION_ERROR)
+
+ if sshv is None:
+ sshv = 2 if aconf.ssh2 else 1
+ err = None
+ banner, header, err = s.get_banner(sshv)
+ if banner is None:
+ if err is None:
+ err = '[exception] did not receive banner.'
+ else:
+ err = '[exception] did not receive banner: {}'.format(err)
+ if err is None:
+ s.send_kexinit() # Send the algorithms we support (except we don't since this isn't a real SSH connection).
+
+ packet_type, payload = s.read_packet(sshv)
+ if packet_type < 0:
+ try:
+ if len(payload) > 0:
+ payload_txt = payload.decode('utf-8')
+ else:
+ payload_txt = 'empty'
+ except UnicodeDecodeError:
+ payload_txt = '"{}"'.format(repr(payload).lstrip('b')[1:-1])
+ if payload_txt == 'Protocol major versions differ.':
+ if sshv == 2 and aconf.ssh1:
+ ret = audit(out, aconf, 1)
+ out.write()
+ return ret
+ err = '[exception] error reading packet ({})'.format(payload_txt)
+ else:
+ err_pair = None
+ if sshv == 1 and packet_type != Protocol.SMSG_PUBLIC_KEY:
+ err_pair = ('SMSG_PUBLIC_KEY', Protocol.SMSG_PUBLIC_KEY)
+ elif sshv == 2 and packet_type != Protocol.MSG_KEXINIT:
+ err_pair = ('MSG_KEXINIT', Protocol.MSG_KEXINIT)
+ if err_pair is not None:
+ fmt = '[exception] did not receive {0} ({1}), ' + \
+ 'instead received unknown message ({2})'
+ err = fmt.format(err_pair[0], err_pair[1], packet_type)
+ if err is not None:
+ output(out, aconf, banner, header)
+ out.fail(err)
+ return exitcodes.CONNECTION_ERROR
+ if sshv == 1:
+ program_retval = output(out, aconf, banner, header, pkm=SSH1_PublicKeyMessage.parse(payload))
+ elif sshv == 2:
+ try:
+ kex = SSH2_Kex.parse(payload)
+ except Exception:
+ out.fail("Failed to parse server's kex. Stack trace:\n%s" % str(traceback.format_exc()))
+ return exitcodes.CONNECTION_ERROR
+
+ if aconf.client_audit is False:
+ HostKeyTest.run(out, s, kex)
+ GEXTest.run(out, s, kex)
+
+ # This is a standard audit scan.
+ if (aconf.policy is None) and (aconf.make_policy is False):
+ program_retval = output(out, aconf, banner, header, client_host=s.client_host, kex=kex, print_target=print_target)
+
+ # This is a policy test.
+ elif (aconf.policy is not None) and (aconf.make_policy is False):
+ program_retval = exitcodes.GOOD if evaluate_policy(out, aconf, banner, s.client_host, kex=kex) else exitcodes.FAILURE
+
+ # A new policy should be made from this scan.
+ elif (aconf.policy is None) and (aconf.make_policy is True):
+ make_policy(aconf, banner, kex, s.client_host)
+
+ else:
+ raise RuntimeError('Internal error while handling output: %r %r' % (aconf.policy is None, aconf.make_policy))
+
+ return program_retval
+
+
+def algorithm_lookup(out: OutputBuffer, alg_names: str) -> int:
+ '''Looks up a comma-separated list of algorithms and outputs their security properties. Returns an exitcodes.* flag.'''
+ retval = exitcodes.GOOD
+ alg_types = {
+ 'kex': 'key exchange algorithms',
+ 'key': 'host-key algorithms',
+ 'mac': 'message authentication code algorithms',
+ 'enc': 'encryption algorithms (ciphers)'
+ }
+
+ algorithm_names = alg_names.split(",")
+ adb = SSH2_KexDB.ALGORITHMS
+
+ # Use nested dictionary comprehension to iterate an outer dictionary where
+ # each key is an alg type that consists of a value (which is itself a
+ # dictionary) of alg names. Filter the alg names against the user supplied
+ # list of names.
+ algorithms_dict = {
+ outer_k: {
+ inner_k
+ for (inner_k, inner_v) in outer_v.items()
+ if inner_k in algorithm_names
+ }
+ for (outer_k, outer_v) in adb.items()
+ }
+
+ unknown_algorithms: List[str] = []
+ padding = len(max(algorithm_names, key=len))
+
+ for alg_type in alg_types:
+ if len(algorithms_dict[alg_type]) > 0:
+ title = str(alg_types.get(alg_type))
+ retval = output_algorithms(out, title, adb, alg_type, list(algorithms_dict[alg_type]), unknown_algorithms, False, retval, padding)
+
+ algorithms_dict_flattened = [
+ alg_name
+ for val in algorithms_dict.values()
+ for alg_name in val
+ ]
+
+ algorithms_not_found = [
+ alg_name
+ for alg_name in algorithm_names
+ if alg_name not in algorithms_dict_flattened
+ ]
+
+ similar_algorithms = [
+ alg_unknown + " --> (" + alg_type + ") " + alg_name
+ for alg_unknown in algorithms_not_found
+ for alg_type, alg_names in adb.items()
+ for alg_name in alg_names
+ # Perform a case-insensitive comparison using 'casefold'
+ # and match substrings using the 'in' operator.
+ if alg_unknown.casefold() in alg_name.casefold()
+ ]
+
+ if len(algorithms_not_found) > 0:
+ retval = exitcodes.FAILURE
+ out.head('# unknown algorithms')
+ for algorithm_not_found in algorithms_not_found:
+ out.fail(algorithm_not_found)
+
+ out.sep()
+
+ if len(similar_algorithms) > 0:
+ retval = exitcodes.FAILURE
+ out.head('# suggested similar algorithms')
+ for similar_algorithm in similar_algorithms:
+ out.warn(similar_algorithm)
+
+ return retval
+
+
+# Worker thread for scanning multiple targets concurrently.
+def target_worker_thread(host: str, port: int, shared_aconf: AuditConf) -> Tuple[int, str]:
+ ret = -1
+ string_output = ''
+
+ out = OutputBuffer()
+ out.verbose = shared_aconf.verbose
+ my_aconf = copy.deepcopy(shared_aconf)
+ my_aconf.host = host
+ my_aconf.port = port
+
+ # If we're outputting JSON, turn off colors and ensure 'info' level messages go through.
+ if my_aconf.json:
+ out.json = True
+ out.use_colors = False
+
+ out.v("Running against: %s:%d..." % (my_aconf.host, my_aconf.port), write_now=True)
+ try:
+ ret = audit(out, my_aconf, print_target=True)
+ string_output = out.get_buffer()
+ except Exception:
+ ret = -1
+ string_output = "An exception occurred while scanning %s:%d:\n%s" % (host, port, str(traceback.format_exc()))
+
+ return ret, string_output
+
+
+def windows_manual(out: OutputBuffer) -> int:
+ '''Prints the man page on Windows. Returns an exitcodes.* flag.'''
+
+ retval = exitcodes.GOOD
+
+ if sys.platform != 'win32':
+ out.fail("The '-m' and '--manual' parameters are reserved for use on Windows only.\nUsers of other operating systems should read the man page.")
+ retval = exitcodes.FAILURE
+ return retval
+
+ # If colors are disabled, strip the ANSI color codes from the man page.
+ windows_man_page = WINDOWS_MAN_PAGE
+ if not out.use_colors:
+ windows_man_page = re.sub(r'\x1b\[\d+?m', '', windows_man_page)
+
+ out.info(windows_man_page)
+ return retval
+
+
+def main() -> int:
+ out = OutputBuffer()
+ aconf = process_commandline(out, sys.argv[1:], usage)
+
+ # If we're on Windows, but the colorama module could not be imported, print a warning if we're in verbose mode.
+ if (sys.platform == 'win32') and ('colorama' not in sys.modules):
+ out.v("WARNING: colorama module not found. Colorized output will be disabled.", write_now=True)
+
+ # If we're outputting JSON, turn off colors and ensure 'info' level messages go through.
+ if aconf.json:
+ out.json = True
+ out.use_colors = False
+
+ if aconf.manual:
+ # If the colorama module was not be imported, turn off colors in order
+ # to output a plain text version of the man page.
+ if (sys.platform == 'win32') and ('colorama' not in sys.modules):
+ out.use_colors = False
+ retval = windows_manual(out)
+ out.write()
+ sys.exit(retval)
+
+ if aconf.lookup != '':
+ retval = algorithm_lookup(out, aconf.lookup)
+ out.write()
+ sys.exit(retval)
+
+ # If multiple targets were specified...
+ if len(aconf.target_list) > 0:
+ ret = exitcodes.GOOD
+
+ # If JSON output is desired, each target's results will be reported in its own list entry.
+ if aconf.json:
+ print('[', end='')
+
+ # Loop through each target in the list.
+ target_servers = []
+ for _, target in enumerate(aconf.target_list):
+ host, port = Utils.parse_host_and_port(target, default_port=22)
+ target_servers.append((host, port))
+
+ # A ranked list of return codes. Those with higher indices will take precendence over lower ones. For example, if three servers are scanned, yielding WARNING, GOOD, and UNKNOWN_ERROR, the overall result will be UNKNOWN_ERROR, since its index is the highest. Errors have highest priority, followed by failures, then warnings.
+ ranked_return_codes = [exitcodes.GOOD, exitcodes.WARNING, exitcodes.FAILURE, exitcodes.CONNECTION_ERROR, exitcodes.UNKNOWN_ERROR]
+
+ # Queue all worker threads.
+ num_target_servers = len(target_servers)
+ num_processed = 0
+ out.v("Scanning %u targets with %s%u threads..." % (num_target_servers, '(at most) ' if aconf.threads > num_target_servers else '', aconf.threads), write_now=True)
+ with concurrent.futures.ThreadPoolExecutor(max_workers=aconf.threads) as executor:
+ future_to_server = {executor.submit(target_worker_thread, target_server[0], target_server[1], aconf): target_server for target_server in target_servers}
+ for future in concurrent.futures.as_completed(future_to_server):
+ worker_ret, worker_output = future.result()
+
+ # If this worker's return code is ranked higher that what we've cached so far, update our cache.
+ if ranked_return_codes.index(worker_ret) > ranked_return_codes.index(ret):
+ ret = worker_ret
+
+ # print("Worker for %s:%d returned %d: [%s]" % (target_server[0], target_server[1], worker_ret, worker_output))
+ print(worker_output, end='' if aconf.json else "\n")
+
+ # Don't print a delimiter after the last target was handled.
+ num_processed += 1
+ if num_processed < num_target_servers:
+ if aconf.json:
+ print(", ", end='')
+ else:
+ print(("-" * 80) + "\n")
+
+ if aconf.json:
+ print(']')
+
+ else: # Just a scan against a single target.
+ ret = audit(out, aconf)
+ out.write()
+
+ return ret
+
+
+if __name__ == '__main__': # pragma: nocover
+ exit_code = exitcodes.GOOD
+
+ try:
+ exit_code = main()
+ except Exception:
+ exit_code = exitcodes.UNKNOWN_ERROR
+ print(traceback.format_exc())
+
+ sys.exit(exit_code)
diff --git a/src/ssh_audit/ssh_socket.py b/src/ssh_audit/ssh_socket.py
new file mode 100644
index 0000000..6853c1c
--- /dev/null
+++ b/src/ssh_audit/ssh_socket.py
@@ -0,0 +1,343 @@
+"""
+ The MIT License (MIT)
+
+ Copyright (C) 2017-2021 Joe Testa (jtesta@positronsecurity.com)
+ Copyright (C) 2017 Andris Raugulis (moo@arthepsy.eu)
+
+ Permission is hereby granted, free of charge, to any person obtaining a copy
+ of this software and associated documentation files (the "Software"), to deal
+ in the Software without restriction, including without limitation the rights
+ to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+ copies of the Software, and to permit persons to whom the Software is
+ furnished to do so, subject to the following conditions:
+
+ The above copyright notice and this permission notice shall be included in
+ all copies or substantial portions of the Software.
+
+ THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+ IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+ FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+ AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+ LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+ OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+ THE SOFTWARE.
+"""
+import errno
+import os
+import select
+import socket
+import struct
+import sys
+
+# pylint: disable=unused-import
+from typing import Dict, List, Set, Sequence, Tuple, Iterable # noqa: F401
+from typing import Callable, Optional, Union, Any # noqa: F401
+
+from ssh_audit import exitcodes
+from ssh_audit.banner import Banner
+from ssh_audit.globals import SSH_HEADER
+from ssh_audit.outputbuffer import OutputBuffer
+from ssh_audit.protocol import Protocol
+from ssh_audit.readbuf import ReadBuf
+from ssh_audit.ssh1 import SSH1
+from ssh_audit.ssh2_kex import SSH2_Kex
+from ssh_audit.ssh2_kexparty import SSH2_KexParty
+from ssh_audit.utils import Utils
+from ssh_audit.writebuf import WriteBuf
+
+
+class SSH_Socket(ReadBuf, WriteBuf):
+ class InsufficientReadException(Exception):
+ pass
+
+ SM_BANNER_SENT = 1
+
+ def __init__(self, outputbuffer: 'OutputBuffer', host: Optional[str], port: int, ip_version_preference: List[int] = [], timeout: Union[int, float] = 5, timeout_set: bool = False) -> None: # pylint: disable=dangerous-default-value
+ super(SSH_Socket, self).__init__()
+ self.__outputbuffer = outputbuffer
+ self.__sock: Optional[socket.socket] = None
+ self.__sock_map: Dict[int, socket.socket] = {}
+ self.__block_size = 8
+ self.__state = 0
+ self.__header: List[str] = []
+ self.__banner: Optional[Banner] = None
+ if host is None:
+ raise ValueError('undefined host')
+ nport = Utils.parse_int(port)
+ if nport < 1 or nport > 65535:
+ raise ValueError('invalid port: {}'.format(port))
+ self.__host = host
+ self.__port = nport
+ self.__ip_version_preference = ip_version_preference # Holds only 5 possible values: [] (no preference), [4] (use IPv4 only), [6] (use IPv6 only), [46] (use both IPv4 and IPv6, but prioritize v4), and [64] (use both IPv4 and IPv6, but prioritize v6).
+ self.__timeout = timeout
+ self.__timeout_set = timeout_set
+ self.client_host: Optional[str] = None
+ self.client_port = None
+
+ def _resolve(self) -> Iterable[Tuple[int, Tuple[Any, ...]]]:
+ # If __ip_version_preference has only one entry, then it means that ONLY that IP version should be used.
+ if len(self.__ip_version_preference) == 1:
+ family = socket.AF_INET if self.__ip_version_preference[0] == 4 else socket.AF_INET6
+ else:
+ family = socket.AF_UNSPEC
+ try:
+ stype = socket.SOCK_STREAM
+ r = socket.getaddrinfo(self.__host, self.__port, family, stype)
+
+ # If the user has a preference for using IPv4 over IPv6 (or vice-versa), then sort the list returned by getaddrinfo() so that the preferred address type comes first.
+ if len(self.__ip_version_preference) == 2:
+ r = sorted(r, key=lambda x: x[0], reverse=(self.__ip_version_preference[0] == 6))
+ for af, socktype, _proto, _canonname, addr in r:
+ if socktype == socket.SOCK_STREAM:
+ yield af, addr
+ except socket.error as e:
+ self.__outputbuffer.fail('[exception] {}'.format(e)).write()
+ sys.exit(exitcodes.CONNECTION_ERROR)
+
+ # Listens on a server socket and accepts one connection (used for
+ # auditing client connections).
+ def listen_and_accept(self) -> None:
+
+ try:
+ # Socket to listen on all IPv4 addresses.
+ s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
+ s.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
+ s.bind(('0.0.0.0', self.__port))
+ s.listen()
+ self.__sock_map[s.fileno()] = s
+ except Exception:
+ print("Warning: failed to listen on any IPv4 interfaces.")
+
+ try:
+ # Socket to listen on all IPv6 addresses.
+ s = socket.socket(socket.AF_INET6, socket.SOCK_STREAM)
+ s.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
+ s.setsockopt(socket.IPPROTO_IPV6, socket.IPV6_V6ONLY, 1)
+ s.bind(('::', self.__port))
+ s.listen()
+ self.__sock_map[s.fileno()] = s
+ except Exception:
+ print("Warning: failed to listen on any IPv6 interfaces.")
+
+ # If we failed to listen on any interfaces, terminate.
+ if len(self.__sock_map.keys()) == 0:
+ print("Error: failed to listen on any IPv4 and IPv6 interfaces!")
+ sys.exit(exitcodes.CONNECTION_ERROR)
+
+ # Wait for an incoming connection. If a timeout was explicitly
+ # set by the user, terminate when it elapses.
+ fds = None
+ time_elapsed = 0.0
+ interval = 1.0
+ while True:
+ # Wait for a connection on either socket.
+ fds = select.select(self.__sock_map.keys(), [], [], interval)
+ time_elapsed += interval
+
+ # We have incoming data on at least one of the sockets.
+ if len(fds[0]) > 0:
+ break
+
+ if self.__timeout_set and time_elapsed >= self.__timeout:
+ print("Timeout elapsed. Terminating...")
+ sys.exit(exitcodes.CONNECTION_ERROR)
+
+ # Accept the connection.
+ c, addr = self.__sock_map[fds[0][0]].accept()
+ self.client_host = addr[0]
+ self.client_port = addr[1]
+ c.settimeout(self.__timeout)
+ self.__sock = c
+
+ def connect(self) -> Optional[str]:
+ '''Returns None on success, or an error string.'''
+ err = None
+ for af, addr in self._resolve():
+ s = None
+ try:
+ s = socket.socket(af, socket.SOCK_STREAM)
+ s.settimeout(self.__timeout)
+ self.__outputbuffer.d(("Connecting to %s:%d..." % ('[%s]' % addr[0] if Utils.is_ipv6_address(addr[0]) else addr[0], addr[1])), write_now=True)
+ s.connect(addr)
+ self.__sock = s
+ return None
+ except socket.error as e:
+ err = e
+ self._close_socket(s)
+ if err is None:
+ errm = 'host {} has no DNS records'.format(self.__host)
+ else:
+ errt = (self.__host, self.__port, err)
+ errm = 'cannot connect to {} port {}: {}'.format(*errt)
+ return '[exception] {}'.format(errm)
+
+ def get_banner(self, sshv: int = 2) -> Tuple[Optional['Banner'], List[str], Optional[str]]:
+ self.__outputbuffer.d('Getting banner...', write_now=True)
+
+ if self.__sock is None:
+ return self.__banner, self.__header, 'not connected'
+ if self.__banner is not None:
+ return self.__banner, self.__header, None
+
+ banner = SSH_HEADER.format('1.5' if sshv == 1 else '2.0')
+ if self.__state < self.SM_BANNER_SENT:
+ self.send_banner(banner)
+
+ s = 0
+ e = None
+ while s >= 0:
+ s, e = self.recv()
+ if s < 0:
+ continue
+ while self.unread_len > 0:
+ line = self.read_line()
+ if len(line.strip()) == 0:
+ continue
+ self.__banner = Banner.parse(line)
+ if self.__banner is not None:
+ return self.__banner, self.__header, None
+ self.__header.append(line)
+
+ return self.__banner, self.__header, e
+
+ def recv(self, size: int = 2048) -> Tuple[int, Optional[str]]:
+ if self.__sock is None:
+ return -1, 'not connected'
+ try:
+ data = self.__sock.recv(size)
+ except socket.timeout:
+ return -1, 'timed out'
+ except socket.error as e:
+ if e.args[0] in (errno.EAGAIN, errno.EWOULDBLOCK):
+ return 0, 'retry'
+ return -1, str(e.args[-1])
+ if len(data) == 0:
+ return -1, None
+ pos = self._buf.tell()
+ self._buf.seek(0, 2)
+ self._buf.write(data)
+ self._len += len(data)
+ self._buf.seek(pos, 0)
+ return len(data), None
+
+ def send(self, data: bytes) -> Tuple[int, Optional[str]]:
+ if self.__sock is None:
+ return -1, 'not connected'
+ try:
+ self.__sock.send(data)
+ return 0, None
+ except socket.error as e:
+ return -1, str(e.args[-1])
+
+ # Send a KEXINIT with the lists of key exchanges, hostkeys, ciphers, MACs, compressions, and languages that we "support".
+ def send_kexinit(self, key_exchanges: List[str] = ['curve25519-sha256', 'curve25519-sha256@libssh.org', 'ecdh-sha2-nistp256', 'ecdh-sha2-nistp384', 'ecdh-sha2-nistp521', 'diffie-hellman-group-exchange-sha256', 'diffie-hellman-group16-sha512', 'diffie-hellman-group18-sha512', 'diffie-hellman-group14-sha256'], hostkeys: List[str] = ['rsa-sha2-512', 'rsa-sha2-256', 'ssh-rsa', 'ecdsa-sha2-nistp256', 'ssh-ed25519'], ciphers: List[str] = ['chacha20-poly1305@openssh.com', 'aes128-ctr', 'aes192-ctr', 'aes256-ctr', 'aes128-gcm@openssh.com', 'aes256-gcm@openssh.com'], macs: List[str] = ['umac-64-etm@openssh.com', 'umac-128-etm@openssh.com', 'hmac-sha2-256-etm@openssh.com', 'hmac-sha2-512-etm@openssh.com', 'hmac-sha1-etm@openssh.com', 'umac-64@openssh.com', 'umac-128@openssh.com', 'hmac-sha2-256', 'hmac-sha2-512', 'hmac-sha1'], compressions: List[str] = ['none', 'zlib@openssh.com'], languages: List[str] = ['']) -> None: # pylint: disable=dangerous-default-value
+ '''Sends the list of supported host keys, key exchanges, ciphers, and MACs. Emulates OpenSSH v8.2.'''
+
+ self.__outputbuffer.d('KEX initialisation...', write_now=True)
+
+ kexparty = SSH2_KexParty(ciphers, macs, compressions, languages)
+ kex = SSH2_Kex(os.urandom(16), key_exchanges, hostkeys, kexparty, kexparty, False, 0)
+
+ self.write_byte(Protocol.MSG_KEXINIT)
+ kex.write(self)
+ self.send_packet()
+
+ def send_banner(self, banner: str) -> None:
+ self.send(banner.encode() + b'\r\n')
+ if self.__state < self.SM_BANNER_SENT:
+ self.__state = self.SM_BANNER_SENT
+
+ def ensure_read(self, size: int) -> None:
+ while self.unread_len < size:
+ s, e = self.recv()
+ if s < 0:
+ raise SSH_Socket.InsufficientReadException(e)
+
+ def read_packet(self, sshv: int = 2) -> Tuple[int, bytes]:
+ try:
+ header = WriteBuf()
+ self.ensure_read(4)
+ packet_length = self.read_int()
+ header.write_int(packet_length)
+ # XXX: validate length
+ if sshv == 1:
+ padding_length = 8 - packet_length % 8
+ self.ensure_read(padding_length)
+ padding = self.read(padding_length)
+ header.write(padding)
+ payload_length = packet_length
+ check_size = padding_length + payload_length
+ else:
+ self.ensure_read(1)
+ padding_length = self.read_byte()
+ header.write_byte(padding_length)
+ payload_length = packet_length - padding_length - 1
+ check_size = 4 + 1 + payload_length + padding_length
+ if check_size % self.__block_size != 0:
+ self.__outputbuffer.fail('[exception] invalid ssh packet (block size)').write()
+ sys.exit(exitcodes.CONNECTION_ERROR)
+ self.ensure_read(payload_length)
+ if sshv == 1:
+ payload = self.read(payload_length - 4)
+ header.write(payload)
+ crc = self.read_int()
+ header.write_int(crc)
+ else:
+ payload = self.read(payload_length)
+ header.write(payload)
+ packet_type = ord(payload[0:1])
+ if sshv == 1:
+ rcrc = SSH1.crc32(padding + payload)
+ if crc != rcrc:
+ self.__outputbuffer.fail('[exception] packet checksum CRC32 mismatch.').write()
+ sys.exit(exitcodes.CONNECTION_ERROR)
+ else:
+ self.ensure_read(padding_length)
+ padding = self.read(padding_length)
+ payload = payload[1:]
+ return packet_type, payload
+ except SSH_Socket.InsufficientReadException as ex:
+ if ex.args[0] is None:
+ header.write(self.read(self.unread_len))
+ e = header.write_flush().strip()
+ else:
+ e = ex.args[0].encode('utf-8')
+ return -1, e
+
+ def send_packet(self) -> Tuple[int, Optional[str]]:
+ payload = self.write_flush()
+ padding = -(len(payload) + 5) % 8
+ if padding < 4:
+ padding += 8
+ plen = len(payload) + padding + 1
+ pad_bytes = b'\x00' * padding
+ data = struct.pack('>Ib', plen, padding) + payload + pad_bytes
+ return self.send(data)
+
+ def is_connected(self) -> bool:
+ """Returns true if this Socket is connected, False otherwise."""
+ return self.__sock is not None
+
+ def close(self) -> None:
+ self.__cleanup()
+ self.reset()
+ self.__state = 0
+ self.__header = []
+ self.__banner = None
+
+ def _close_socket(self, s: Optional[socket.socket]) -> None: # pylint: disable=no-self-use
+ try:
+ if s is not None:
+ s.shutdown(socket.SHUT_RDWR)
+ s.close() # pragma: nocover
+ except Exception:
+ pass
+
+ def __del__(self) -> None:
+ self.__cleanup()
+
+ def __cleanup(self) -> None:
+ self._close_socket(self.__sock)
+ for sock in self.__sock_map.values():
+ self._close_socket(sock)
+ self.__sock = None
diff --git a/src/ssh_audit/timeframe.py b/src/ssh_audit/timeframe.py
new file mode 100644
index 0000000..cb98d9b
--- /dev/null
+++ b/src/ssh_audit/timeframe.py
@@ -0,0 +1,77 @@
+"""
+ The MIT License (MIT)
+
+ Copyright (C) 2017 Andris Raugulis (moo@arthepsy.eu)
+
+ Permission is hereby granted, free of charge, to any person obtaining a copy
+ of this software and associated documentation files (the "Software"), to deal
+ in the Software without restriction, including without limitation the rights
+ to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+ copies of the Software, and to permit persons to whom the Software is
+ furnished to do so, subject to the following conditions:
+
+ The above copyright notice and this permission notice shall be included in
+ all copies or substantial portions of the Software.
+
+ THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+ IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+ FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+ AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+ LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+ OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+ THE SOFTWARE.
+"""
+# pylint: disable=unused-import
+from typing import Dict, List, Set, Sequence, Tuple, Iterable # noqa: F401
+from typing import Callable, Optional, Union, Any # noqa: F401
+
+from ssh_audit.algorithm import Algorithm
+
+
+class Timeframe:
+ def __init__(self) -> None:
+ self.__storage: Dict[str, List[Optional[str]]] = {}
+
+ def __contains__(self, product: str) -> bool:
+ return product in self.__storage
+
+ def __getitem__(self, product: str) -> Sequence[Optional[str]]:
+ return tuple(self.__storage.get(product, [None] * 4))
+
+ def __str__(self) -> str:
+ return self.__storage.__str__()
+
+ def __repr__(self) -> str:
+ return self.__str__()
+
+ def get_from(self, product: str, for_server: bool = True) -> Optional[str]:
+ return self[product][0 if bool(for_server) else 2]
+
+ def get_till(self, product: str, for_server: bool = True) -> Optional[str]:
+ return self[product][1 if bool(for_server) else 3]
+
+ def _update(self, versions: Optional[str], pos: int) -> None:
+ ssh_versions: Dict[str, str] = {}
+ for_srv, for_cli = pos < 2, pos > 1
+ for v in (versions or '').split(','):
+ ssh_prod, ssh_ver, is_cli = Algorithm.get_ssh_version(v)
+ if not ssh_ver or (is_cli and for_srv) or (not is_cli and for_cli and ssh_prod in ssh_versions):
+ continue
+ ssh_versions[ssh_prod] = ssh_ver
+ for ssh_product, ssh_version in ssh_versions.items():
+ if ssh_product not in self.__storage:
+ self.__storage[ssh_product] = [None] * 4
+ prev = self[ssh_product][pos]
+ if (prev is None or (prev < ssh_version and pos % 2 == 0) or (prev > ssh_version and pos % 2 == 1)):
+ self.__storage[ssh_product][pos] = ssh_version
+
+ def update(self, versions: List[Optional[str]], for_server: Optional[bool] = None) -> 'Timeframe':
+ for_cli = for_server is None or for_server is False
+ for_srv = for_server is None or for_server is True
+ vlen = len(versions)
+ for i in range(min(3, vlen)):
+ if for_srv and i < 2:
+ self._update(versions[i], i)
+ if for_cli and (i % 2 == 0 or vlen == 2):
+ self._update(versions[i], 3 - 0**i)
+ return self
diff --git a/src/ssh_audit/utils.py b/src/ssh_audit/utils.py
new file mode 100644
index 0000000..a17ecb6
--- /dev/null
+++ b/src/ssh_audit/utils.py
@@ -0,0 +1,165 @@
+"""
+ The MIT License (MIT)
+
+ Copyright (C) 2017-2020 Joe Testa (jtesta@positronsecurity.com)
+ Copyright (C) 2017 Andris Raugulis (moo@arthepsy.eu)
+
+ Permission is hereby granted, free of charge, to any person obtaining a copy
+ of this software and associated documentation files (the "Software"), to deal
+ in the Software without restriction, including without limitation the rights
+ to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+ copies of the Software, and to permit persons to whom the Software is
+ furnished to do so, subject to the following conditions:
+
+ The above copyright notice and this permission notice shall be included in
+ all copies or substantial portions of the Software.
+
+ THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+ IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+ FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+ AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+ LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+ OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+ THE SOFTWARE.
+"""
+import ipaddress
+import re
+import sys
+
+# pylint: disable=unused-import
+from typing import Dict, List, Set, Sequence, Tuple, Iterable # noqa: F401
+from typing import Callable, Optional, Union, Any # noqa: F401
+
+
+class Utils:
+ @classmethod
+ def _type_err(cls, v: Any, target: str) -> TypeError:
+ return TypeError('cannot convert {} to {}'.format(type(v), target))
+
+ @classmethod
+ def to_bytes(cls, v: Union[bytes, str], enc: str = 'utf-8') -> bytes:
+ if isinstance(v, bytes):
+ return v
+ elif isinstance(v, str):
+ return v.encode(enc)
+ raise cls._type_err(v, 'bytes')
+
+ @classmethod
+ def to_text(cls, v: Union[str, bytes], enc: str = 'utf-8') -> str:
+ if isinstance(v, str):
+ return v
+ elif isinstance(v, bytes):
+ return v.decode(enc)
+ raise cls._type_err(v, 'unicode text')
+
+ @classmethod
+ def _is_ascii(cls, v: str, char_filter: Callable[[int], bool] = lambda x: x <= 127) -> bool:
+ r = False
+ if isinstance(v, str):
+ for c in v:
+ i = cls.ctoi(c)
+ if not char_filter(i):
+ return r
+ r = True
+ return r
+
+ @classmethod
+ def _to_ascii(cls, v: str, char_filter: Callable[[int], bool] = lambda x: x <= 127, errors: str = 'replace') -> str:
+ if isinstance(v, str):
+ r = bytearray()
+ for c in v:
+ i = cls.ctoi(c)
+ if char_filter(i):
+ r.append(i)
+ else:
+ if errors == 'ignore':
+ continue
+ r.append(63)
+ return cls.to_text(r.decode('ascii'))
+ raise cls._type_err(v, 'ascii')
+
+ @classmethod
+ def is_ascii(cls, v: str) -> bool:
+ return cls._is_ascii(v)
+
+ @classmethod
+ def to_ascii(cls, v: str, errors: str = 'replace') -> str:
+ return cls._to_ascii(v, errors=errors)
+
+ @classmethod
+ def is_print_ascii(cls, v: str) -> bool:
+ return cls._is_ascii(v, lambda x: 126 >= x >= 32)
+
+ @classmethod
+ def to_print_ascii(cls, v: str, errors: str = 'replace') -> str:
+ return cls._to_ascii(v, lambda x: 126 >= x >= 32, errors)
+
+ @classmethod
+ def unique_seq(cls, seq: Sequence[Any]) -> Sequence[Any]:
+ seen: Set[Any] = set()
+
+ def _seen_add(x: Any) -> bool:
+ seen.add(x)
+ return False
+
+ if isinstance(seq, tuple):
+ return tuple(x for x in seq if x not in seen and not _seen_add(x))
+ else:
+ return [x for x in seq if x not in seen and not _seen_add(x)]
+
+ @classmethod
+ def ctoi(cls, c: Union[str, int]) -> int:
+ if isinstance(c, str):
+ return ord(c[0])
+ else:
+ return c
+
+ @staticmethod
+ def parse_int(v: Any) -> int:
+ try:
+ return int(v)
+ except ValueError:
+ return 0
+
+ @staticmethod
+ def parse_float(v: Any) -> float:
+ try:
+ return float(v)
+ except ValueError:
+ return -1.0
+
+ @staticmethod
+ def parse_host_and_port(host_and_port: str, default_port: int = 0) -> Tuple[str, int]:
+ '''Parses a string into a tuple of its host and port. The port is 0 if not specified.'''
+ host = host_and_port
+ port = default_port
+
+ mx = re.match(r'^\[([^\]]+)\](?::(\d+))?$', host_and_port)
+ if mx is not None:
+ host = mx.group(1)
+ port_str = mx.group(2)
+ if port_str is not None:
+ port = int(port_str)
+ else:
+ s = host_and_port.split(':')
+ if len(s) == 2:
+ host = s[0]
+ if len(s[1]) > 0:
+ port = int(s[1])
+
+ return host, port
+
+ @staticmethod
+ def is_ipv6_address(address: str) -> bool:
+ '''Returns True if address is an IPv6 address, otherwise False.'''
+ is_ipv6 = True
+ try:
+ ipaddress.IPv6Address(address)
+ except ipaddress.AddressValueError:
+ is_ipv6 = False
+
+ return is_ipv6
+
+ @staticmethod
+ def is_windows() -> bool:
+ return sys.platform in ['win32', 'cygwin']
diff --git a/src/ssh_audit/versionvulnerabilitydb.py b/src/ssh_audit/versionvulnerabilitydb.py
new file mode 100644
index 0000000..7f02c71
--- /dev/null
+++ b/src/ssh_audit/versionvulnerabilitydb.py
@@ -0,0 +1,151 @@
+"""
+ The MIT License (MIT)
+
+ Copyright (C) 2017-2020 Joe Testa (jtesta@positronsecurity.com)
+ Copyright (C) 2017 Andris Raugulis (moo@arthepsy.eu)
+
+ Permission is hereby granted, free of charge, to any person obtaining a copy
+ of this software and associated documentation files (the "Software"), to deal
+ in the Software without restriction, including without limitation the rights
+ to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+ copies of the Software, and to permit persons to whom the Software is
+ furnished to do so, subject to the following conditions:
+
+ The above copyright notice and this permission notice shall be included in
+ all copies or substantial portions of the Software.
+
+ THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+ IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+ FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+ AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+ LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+ OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+ THE SOFTWARE.
+"""
+# pylint: disable=unused-import
+from typing import Dict, List, Set, Sequence, Tuple, Iterable # noqa: F401
+from typing import Callable, Optional, Union, Any # noqa: F401
+
+
+class VersionVulnerabilityDB: # pylint: disable=too-few-public-methods
+ # Format: [starting_vuln_version, last_vuln_version, affected, CVE_ID, CVSSv2, description]
+ # affected: 1 = server, 2 = client, 4 = local
+ # Example: if it affects servers, both remote & local, then affected
+ # = 1. If it affects servers, but is a local issue only,
+ # then affected = 1 + 4 = 5.
+ CVE: Dict[str, List[List[Any]]] = {
+ 'Dropbear SSH': [
+ ['0.0', '2018.76', 1, 'CVE-2018-15599', 5.0, 'remote users may enumerate users on the system'],
+ ['0.0', '2017.74', 5, 'CVE-2017-9079', 4.7, 'local users can read certain files as root'],
+ ['0.0', '2017.74', 5, 'CVE-2017-9078', 9.3, 'local users may elevate privileges to root under certain conditions'],
+ ['0.0', '2016.73', 5, 'CVE-2016-7409', 2.1, 'local users can read process memory under limited conditions'],
+ ['0.0', '2016.73', 1, 'CVE-2016-7408', 6.5, 'remote users can execute arbitrary code'],
+ ['0.0', '2016.73', 5, 'CVE-2016-7407', 10.0, 'local users can execute arbitrary code'],
+ ['0.0', '2016.73', 1, 'CVE-2016-7406', 10.0, 'remote users can execute arbitrary code'],
+ ['0.44', '2015.71', 1, 'CVE-2016-3116', 5.5, 'bypass command restrictions via xauth command injection'],
+ ['0.28', '2013.58', 1, 'CVE-2013-4434', 5.0, 'discover valid usernames through different time delays'],
+ ['0.28', '2013.58', 1, 'CVE-2013-4421', 5.0, 'cause DoS via a compressed packet (memory consumption)'],
+ ['0.52', '2011.54', 1, 'CVE-2012-0920', 7.1, 'execute arbitrary code or bypass command restrictions'],
+ ['0.40', '0.48.1', 1, 'CVE-2007-1099', 7.5, 'conduct a MitM attack (no warning for hostkey mismatch)'],
+ ['0.28', '0.47', 1, 'CVE-2006-1206', 7.5, 'cause DoS via large number of connections (slot exhaustion)'],
+ ['0.39', '0.47', 1, 'CVE-2006-0225', 4.6, 'execute arbitrary commands via scp with crafted filenames'],
+ ['0.28', '0.46', 1, 'CVE-2005-4178', 6.5, 'execute arbitrary code via buffer overflow vulnerability'],
+ ['0.28', '0.42', 1, 'CVE-2004-2486', 7.5, 'execute arbitrary code via DSS verification code']],
+ 'libssh': [
+ ['0.6.4', '0.6.4', 1, 'CVE-2018-10933', 6.4, 'authentication bypass'],
+ ['0.7.0', '0.7.5', 1, 'CVE-2018-10933', 6.4, 'authentication bypass'],
+ ['0.8.0', '0.8.3', 1, 'CVE-2018-10933', 6.4, 'authentication bypass'],
+ ['0.1', '0.7.2', 1, 'CVE-2016-0739', 4.3, 'conduct a MitM attack (weakness in DH key generation)'],
+ ['0.5.1', '0.6.4', 1, 'CVE-2015-3146', 5.0, 'cause DoS via kex packets (null pointer dereference)'],
+ ['0.5.1', '0.6.3', 1, 'CVE-2014-8132', 5.0, 'cause DoS via kex init packet (dangling pointer)'],
+ ['0.4.7', '0.6.2', 1, 'CVE-2014-0017', 1.9, 'leak data via PRNG state reuse on forking servers'],
+ ['0.4.7', '0.5.3', 1, 'CVE-2013-0176', 4.3, 'cause DoS via kex packet (null pointer dereference)'],
+ ['0.4.7', '0.5.2', 1, 'CVE-2012-6063', 7.5, 'cause DoS or execute arbitrary code via sftp (double free)'],
+ ['0.4.7', '0.5.2', 1, 'CVE-2012-4562', 7.5, 'cause DoS or execute arbitrary code (overflow check)'],
+ ['0.4.7', '0.5.2', 1, 'CVE-2012-4561', 5.0, 'cause DoS via unspecified vectors (invalid pointer)'],
+ ['0.4.7', '0.5.2', 1, 'CVE-2012-4560', 7.5, 'cause DoS or execute arbitrary code (buffer overflow)'],
+ ['0.4.7', '0.5.2', 1, 'CVE-2012-4559', 6.8, 'cause DoS or execute arbitrary code (double free)']],
+ 'OpenSSH': [
+ ['1.0', '7.7', 1, 'CVE-2018-15473', 5.3, 'enumerate usernames due to timing discrepencies'],
+ ['7.2', '7.2p2', 1, 'CVE-2016-6515', 7.8, 'cause DoS via long password string (crypt CPU consumption)'],
+ ['1.2.2', '7.2', 1, 'CVE-2016-3115', 5.5, 'bypass command restrictions via crafted X11 forwarding data'],
+ ['5.4', '7.1', 1, 'CVE-2016-1907', 5.0, 'cause DoS via crafted network traffic (out of bounds read)'],
+ ['5.4', '7.1p1', 2, 'CVE-2016-0778', 4.6, 'cause DoS via requesting many forwardings (heap based buffer overflow)'],
+ ['5.0', '7.1p1', 2, 'CVE-2016-0777', 4.0, 'leak data via allowing transfer of entire buffer'],
+ ['6.0', '7.2p2', 5, 'CVE-2015-8325', 7.2, 'privilege escalation via triggering crafted environment'],
+ ['6.8', '6.9', 5, 'CVE-2015-6565', 7.2, 'cause DoS via writing to a device (terminal disruption)'],
+ ['5.0', '6.9', 5, 'CVE-2015-6564', 6.9, 'privilege escalation via leveraging sshd uid'],
+ ['5.0', '6.9', 5, 'CVE-2015-6563', 1.9, 'conduct impersonation attack'],
+ ['6.9p1', '6.9p1', 1, 'CVE-2015-5600', 8.5, 'cause Dos or aid in conduct brute force attack (CPU consumption)'],
+ ['6.0', '6.6', 1, 'CVE-2015-5352', 4.3, 'bypass access restrictions via a specific connection'],
+ ['6.0', '6.6', 2, 'CVE-2014-2653', 5.8, 'bypass SSHFP DNS RR check via unacceptable host certificate'],
+ ['5.0', '6.5', 1, 'CVE-2014-2532', 5.8, 'bypass environment restrictions via specific string before wildcard'],
+ ['1.2', '6.4', 1, 'CVE-2014-1692', 7.5, 'cause DoS via triggering error condition (memory corruption)'],
+ ['6.2', '6.3', 1, 'CVE-2013-4548', 6.0, 'bypass command restrictions via crafted packet data'],
+ ['1.2', '5.6', 1, 'CVE-2012-0814', 3.5, 'leak data via debug messages'],
+ ['1.2', '5.8', 1, 'CVE-2011-5000', 3.5, 'cause DoS via large value in certain length field (memory consumption)'],
+ ['5.6', '5.7', 2, 'CVE-2011-0539', 5.0, 'leak data or conduct hash collision attack'],
+ ['1.2', '6.1', 1, 'CVE-2010-5107', 5.0, 'cause DoS via large number of connections (slot exhaustion)'],
+ ['1.2', '5.8', 1, 'CVE-2010-4755', 4.0, 'cause DoS via crafted glob expression (CPU and memory consumption)'],
+ ['1.2', '5.6', 1, 'CVE-2010-4478', 7.5, 'bypass authentication check via crafted values'],
+ ['4.3', '4.8', 1, 'CVE-2009-2904', 6.9, 'privilege escalation via hard links to setuid programs'],
+ ['4.0', '5.1', 1, 'CVE-2008-5161', 2.6, 'recover plaintext data from ciphertext'],
+ ['1.2', '4.6', 1, 'CVE-2008-4109', 5.0, 'cause DoS via multiple login attempts (slot exhaustion)'],
+ ['1.2', '4.8', 1, 'CVE-2008-1657', 6.5, 'bypass command restrictions via modifying session file'],
+ ['1.2.2', '4.9', 1, 'CVE-2008-1483', 6.9, 'hijack forwarded X11 connections'],
+ ['4.0', '4.6', 1, 'CVE-2007-4752', 7.5, 'privilege escalation via causing an X client to be trusted'],
+ ['4.3p2', '4.3p2', 1, 'CVE-2007-3102', 4.3, 'allow attacker to write random data to audit log'],
+ ['1.2', '4.6', 1, 'CVE-2007-2243', 5.0, 'discover valid usernames through different responses'],
+ ['4.4', '4.4', 1, 'CVE-2006-5794', 7.5, 'bypass authentication'],
+ ['4.1', '4.1p1', 1, 'CVE-2006-5229', 2.6, 'discover valid usernames through different time delays'],
+ ['1.2', '4.3p2', 1, 'CVE-2006-5052', 5.0, 'discover valid usernames through different responses'],
+ ['1.2', '4.3p2', 1, 'CVE-2006-5051', 9.3, 'cause DoS or execute arbitrary code (double free)'],
+ ['4.5', '4.5', 1, 'CVE-2006-4925', 5.0, 'cause DoS via invalid protocol sequence (crash)'],
+ ['1.2', '4.3p2', 1, 'CVE-2006-4924', 7.8, 'cause DoS via crafted packet (CPU consumption)'],
+ ['3.8.1p1', '3.8.1p1', 1, 'CVE-2006-0883', 5.0, 'cause DoS via connecting multiple times (client connection refusal)'],
+ ['3.0', '4.2p1', 1, 'CVE-2006-0225', 4.6, 'execute arbitrary code'],
+ ['2.1', '4.1p1', 1, 'CVE-2005-2798', 5.0, 'leak data about authentication credentials'],
+ ['3.5', '3.5p1', 1, 'CVE-2004-2760', 6.8, 'leak data through different connection states'],
+ ['2.3', '3.7.1p2', 1, 'CVE-2004-2069', 5.0, 'cause DoS via large number of connections (slot exhaustion)'],
+ ['3.0', '3.4p1', 1, 'CVE-2004-0175', 4.3, 'leak data through directoy traversal'],
+ ['1.2', '3.9p1', 1, 'CVE-2003-1562', 7.6, 'leak data about authentication credentials'],
+ ['3.1p1', '3.7.1p1', 1, 'CVE-2003-0787', 7.5, 'privilege escalation via modifying stack'],
+ ['3.1p1', '3.7.1p1', 1, 'CVE-2003-0786', 10.0, 'privilege escalation via bypassing authentication'],
+ ['1.0', '3.7.1', 1, 'CVE-2003-0695', 7.5, 'cause DoS or execute arbitrary code'],
+ ['1.0', '3.7', 1, 'CVE-2003-0693', 10.0, 'execute arbitrary code'],
+ ['3.0', '3.6.1p2', 1, 'CVE-2003-0386', 7.5, 'bypass address restrictions for connection'],
+ ['3.1p1', '3.6.1p1', 1, 'CVE-2003-0190', 5.0, 'discover valid usernames through different time delays'],
+ ['3.2.2', '3.2.2', 1, 'CVE-2002-0765', 7.5, 'bypass authentication'],
+ ['1.2.2', '3.3p1', 1, 'CVE-2002-0640', 10.0, 'execute arbitrary code'],
+ ['1.2.2', '3.3p1', 1, 'CVE-2002-0639', 10.0, 'execute arbitrary code'],
+ ['2.1', '3.2', 1, 'CVE-2002-0575', 7.5, 'privilege escalation'],
+ ['2.1', '3.0.2p1', 2, 'CVE-2002-0083', 10.0, 'privilege escalation'],
+ ['3.0', '3.0p1', 1, 'CVE-2001-1507', 7.5, 'bypass authentication'],
+ ['1.2.3', '3.0.1p1', 5, 'CVE-2001-0872', 7.2, 'privilege escalation via crafted environment variables'],
+ ['1.2.3', '2.1.1', 1, 'CVE-2001-0361', 4.0, 'recover plaintext from ciphertext'],
+ ['1.2', '2.1', 1, 'CVE-2000-0525', 10.0, 'execute arbitrary code (improper privileges)']],
+ 'PuTTY': [
+ ['0.54', '0.73', 2, 'CVE-2020-XXXX', 5.0, 'out of bounds memory read'],
+ ['0.0', '0.72', 2, 'CVE-2019-17069', 5.0, 'potential DOS by remote SSHv1 server'],
+ ['0.71', '0.72', 2, 'CVE-2019-17068', 5.0, 'xterm bracketed paste mode command injection'],
+ ['0.52', '0.72', 2, 'CVE-2019-17067', 7.5, 'port rebinding weakness in port forward tunnel handling'],
+ ['0.0', '0.71', 2, 'CVE-2019-XXXX', 5.0, 'undefined vulnerability in obsolete SSHv1 protocol handling'],
+ ['0.0', '0.71', 6, 'CVE-2019-XXXX', 5.0, 'local privilege escalation in Pageant'],
+ ['0.0', '0.70', 2, 'CVE-2019-9898', 7.5, 'potential recycling of random numbers'],
+ ['0.0', '0.70', 2, 'CVE-2019-9897', 5.0, 'multiple denial-of-service issues from writing to the terminal'],
+ ['0.0', '0.70', 6, 'CVE-2019-9896', 4.6, 'local application hijacking through malicious Windows help file'],
+ ['0.0', '0.70', 2, 'CVE-2019-9894', 6.4, 'buffer overflow in RSA key exchange'],
+ ['0.0', '0.69', 6, 'CVE-2016-6167', 4.4, 'local application hijacking through untrusted DLL loading'],
+ ['0.0', '0.67', 2, 'CVE-2017-6542', 7.5, 'buffer overflow in UNIX client that can result in privilege escalation or denial-of-service'],
+ ['0.0', '0.66', 2, 'CVE-2016-2563', 7.5, 'buffer overflow in SCP command-line utility'],
+ ['0.0', '0.65', 2, 'CVE-2015-5309', 4.3, 'integer overflow in terminal-handling code'],
+ ]
+ }
+ TXT: Dict[str, List[List[Any]]] = {
+ 'Dropbear SSH': [
+ ['0.28', '0.34', 1, 'remote root exploit', 'remote format string buffer overflow exploit (exploit-db#387)']],
+ 'libssh': [
+ ['0.3.3', '0.3.3', 1, 'null pointer check', 'missing null pointer check in "crypt_set_algorithms_server"'],
+ ['0.3.3', '0.3.3', 1, 'integer overflow', 'integer overflow in "buffer_get_data"'],
+ ['0.3.3', '0.3.3', 3, 'heap overflow', 'heap overflow in "packet_decrypt"']]
+ }
diff --git a/src/ssh_audit/writebuf.py b/src/ssh_audit/writebuf.py
new file mode 100644
index 0000000..5f25d8c
--- /dev/null
+++ b/src/ssh_audit/writebuf.py
@@ -0,0 +1,108 @@
+"""
+ The MIT License (MIT)
+
+ Copyright (C) 2017 Andris Raugulis (moo@arthepsy.eu)
+
+ Permission is hereby granted, free of charge, to any person obtaining a copy
+ of this software and associated documentation files (the "Software"), to deal
+ in the Software without restriction, including without limitation the rights
+ to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+ copies of the Software, and to permit persons to whom the Software is
+ furnished to do so, subject to the following conditions:
+
+ The above copyright notice and this permission notice shall be included in
+ all copies or substantial portions of the Software.
+
+ THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+ IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+ FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+ AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+ LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+ OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+ THE SOFTWARE.
+"""
+import io
+import struct
+
+# pylint: disable=unused-import
+from typing import Dict, List, Set, Sequence, Tuple, Iterable # noqa: F401
+from typing import Callable, Optional, Union, Any # noqa: F401
+
+
+class WriteBuf:
+ def __init__(self, data: Optional[bytes] = None) -> None:
+ super(WriteBuf, self).__init__()
+ self._wbuf = io.BytesIO(data) if data is not None else io.BytesIO()
+
+ def write(self, data: bytes) -> 'WriteBuf':
+ self._wbuf.write(data)
+ return self
+
+ def write_byte(self, v: int) -> 'WriteBuf':
+ return self.write(struct.pack('B', v))
+
+ def write_bool(self, v: bool) -> 'WriteBuf':
+ return self.write_byte(1 if v else 0)
+
+ def write_int(self, v: int) -> 'WriteBuf':
+ return self.write(struct.pack('>I', v))
+
+ def write_string(self, v: Union[bytes, str]) -> 'WriteBuf':
+ if not isinstance(v, bytes):
+ v = bytes(bytearray(v, 'utf-8'))
+ self.write_int(len(v))
+ return self.write(v)
+
+ def write_list(self, v: List[str]) -> 'WriteBuf':
+ return self.write_string(','.join(v))
+
+ @classmethod
+ def _bitlength(cls, n: int) -> int:
+ try:
+ return n.bit_length()
+ except AttributeError:
+ return len(bin(n)) - (2 if n > 0 else 3)
+
+ @classmethod
+ def _create_mpint(cls, n: int, signed: bool = True, bits: Optional[int] = None) -> bytes:
+ if bits is None:
+ bits = cls._bitlength(n)
+ length = bits // 8 + (1 if n != 0 else 0)
+ ql = (length + 7) // 8
+ fmt, v2 = '>{}Q'.format(ql), [0] * ql
+ for i in range(ql):
+ v2[ql - i - 1] = n & 0xffffffffffffffff
+ n >>= 64
+ data = bytes(struct.pack(fmt, *v2)[-length:])
+ if not signed:
+ data = data.lstrip(b'\x00')
+ elif data.startswith(b'\xff\x80'):
+ data = data[1:]
+ return data
+
+ def write_mpint1(self, n: int) -> 'WriteBuf':
+ # NOTE: Data Type Enc @ http://www.snailbook.com/docs/protocol-1.5.txt
+ bits = self._bitlength(n)
+ data = self._create_mpint(n, False, bits)
+ self.write(struct.pack('>H', bits))
+ return self.write(data)
+
+ def write_mpint2(self, n: int) -> 'WriteBuf':
+ # NOTE: Section 5 @ https://www.ietf.org/rfc/rfc4251.txt
+ data = self._create_mpint(n)
+ return self.write_string(data)
+
+ def write_line(self, v: Union[bytes, str]) -> 'WriteBuf':
+ if not isinstance(v, bytes):
+ v = bytes(bytearray(v, 'utf-8'))
+ v += b'\r\n'
+ return self.write(v)
+
+ def write_flush(self) -> bytes:
+ payload = self._wbuf.getvalue()
+ self._wbuf.truncate(0)
+ self._wbuf.seek(0)
+ return payload
+
+ def reset(self) -> None:
+ self._wbuf = io.BytesIO()
diff --git a/ssh-audit.1 b/ssh-audit.1
new file mode 100644
index 0000000..6939627
--- /dev/null
+++ b/ssh-audit.1
@@ -0,0 +1,245 @@
+.TH SSH-AUDIT 1 "March 2, 2021"
+.SH NAME
+\fBssh-audit\fP \- SSH server & client configuration auditor
+.SH SYNOPSIS
+.B ssh-audit
+.RI [ options ] " <target_host>"
+.SH DESCRIPTION
+.PP
+\fBssh-audit\fP analyzes the configuration of SSH servers & clients, then warns the user of weak, obsolete, and/or un-tested cryptographic primitives. It is very useful for hardening SSH tunnels, which by default tend to be optimized for compatibility, not security.
+.PP
+See <https://www.ssh\-audit.com/> for official hardening guides for common platforms.
+
+.SH OPTIONS
+.TP
+.B -h, \-\-help
+.br
+Print short summary of options.
+
+.TP
+.B -1, \-\-ssh1
+.br
+Only perform an audit using SSH protocol version 1.
+
+.TP
+.B -2, \-\-ssh2
+.br
+Only perform an audit using SSH protocol version 2.
+
+.TP
+.B -4, \-\-ipv4
+.br
+Prioritize the usage of IPv4.
+
+.TP
+.B -6, \-\-ipv6
+.br
+Prioritize the usage of IPv6.
+
+.TP
+.B -b, \-\-batch
+.br
+Enables grepable output.
+
+.TP
+.B -c, \-\-client\-audit
+.br
+Starts a server on port 2222 to audit client software configuration. Use -p/--port=<port> to change port and -t/--timeout=<secs> to change listen timeout.
+
+.TP
+.B -d, \-\-debug
+.br
+Enable debug output.
+
+.TP
+.B -j, \-\-json
+.br
+Output results in JSON format. Specify twice (-jj) to enable indent printing (useful for debugging).
+
+.TP
+.B -l, \-\-level=<info|warn|fail>
+.br
+Specify the minimum output level. Default is info.
+
+.TP
+.B -L, \-\-list-policies
+.br
+List all official, built-in policies for common systems. Their full names can then be passed to -P/--policy.
+
+.TP
+.B \-\-lookup=<alg1,alg2,...>
+.br
+Look up the security information of an algorithm(s) in the internal database. Does not connect to a server.
+
+.TP
+.B -m, \-\-manual
+.br
+Print the man page (Windows only).
+
+.TP
+.B -M, \-\-make-policy=<custom_policy.txt>
+.br
+Creates a policy based on the target server. Useful when other servers should be compared to the target server's custom configuration (i.e.: a cluster environment). Note that the resulting policy can be edited manually.
+
+.TP
+.B -n, \-\-no-colors
+.br
+Disable color output.
+
+.TP
+.B -p, \-\-port=<port>
+.br
+The TCP port to connect to when auditing a server, or the port to listen on when auditing a client.
+
+.TP
+.B -P, \-\-policy=<"built-in policy name" | path/to/custom_policy.txt>
+.br
+Runs a policy audit against a target using the specified policy (see \fBPOLICY AUDIT\fP section for detailed description of this mode of operation). Combine with -c/--client-audit to audit a client configuration instead of a server. Use -L/--list-policies to list all official, built-in policies for common systems.
+
+.TP
+.B -t, \-\-timeout=<secs>
+.br
+The timeout, in seconds, for creating connections and reading data from the socket. Default is 5.
+
+.TP
+.B -T, \-\-targets=<hosts.txt>
+.br
+A file containing a list of target hosts. Each line must have one host, in the format of HOST[:PORT]. Use --threads to control concurrent scans.
+
+.TP
+.B \-\-threads=<threads>
+.br
+The number of threads to use when scanning multiple targets (with -T/--targets). Default is 32.
+
+.TP
+.B -v, \-\-verbose
+.br
+Enable verbose output.
+
+
+.SH STANDARD AUDIT
+.PP
+By default, \fBssh-audit\fP performs a standard audit. That is, it enumerates all host key types, key exchanges, ciphers, MACs, and other information, then color-codes them in output to the user. Cryptographic primitives with potential issues are displayed in yellow; primitives with serious flaws are displayed in red.
+
+
+.SH POLICY AUDIT
+.PP
+When the -P/--policy option is used, \fBssh-audit\fP performs a policy audit. The target's host key types, key exchanges, ciphers, MACs, and other information is compared to a set of expected values defined in the specified policy file. If everything matches, only a short message stating a passing result is reported. Otherwise, the field(s) that did not match are reported.
+
+.PP
+Policy auditing is helpful for ensuring a group of related servers are properly hardened to an exact specification.
+
+.PP
+The set of official built-in policies can be viewed with -L/--list-policies. Multiple servers can be audited with -T/--targets=<servers.txt>. Custom policies can be made from an ideal target server with -M/--make-policy=<custom_policy.txt>.
+
+
+.SH EXAMPLES
+.LP
+Basic server auditing:
+.RS
+.nf
+ssh-audit localhost
+ssh-audit 127.0.0.1
+ssh-audit 127.0.0.1:222
+ssh-audit ::1
+ssh-audit [::1]:222
+.fi
+.RE
+
+.LP
+To run a standard audit against many servers (place targets into servers.txt, one on each line in the format of HOST[:PORT]):
+.RS
+.nf
+ssh-audit -T servers.txt
+.fi
+.RE
+
+.LP
+To audit a client configuration (listens on port 2222 by default; connect using "ssh -p 2222 anything@localhost"):
+.RS
+.nf
+ssh-audit -c
+.fi
+.RE
+
+.LP
+To audit a client configuration, with a listener on port 4567:
+.RS
+.nf
+ssh-audit -c -p 4567
+.fi
+.RE
+
+.LP
+To list all official built-in policies (hint: use their full names with -P/--policy):
+.RS
+.nf
+ssh-audit -L
+.fi
+.RE
+
+.LP
+To run a built-in policy audit against a server (hint: use -L to see list of built-in policies):
+.RS
+.nf
+ssh-audit -P "Hardened Ubuntu Server 20.04 LTS (version 1)" targetserver
+.fi
+.RE
+
+
+.LP
+To run a custom policy audit against a server (hint: use -M/--make-policy to create a custom policy file):
+.RS
+.nf
+ssh-audit -P path/to/server_policy.txt targetserver
+.fi
+.RE
+
+.LP
+To run a policy audit against a client:
+.RS
+.nf
+ssh-audit -c -P ["policy name" | path/to/client_policy.txt]
+.fi
+.RE
+
+.LP
+To run a policy audit against many servers:
+.RS
+.nf
+ssh-audit -T servers.txt -P ["policy name" | path/to/server_policy.txt]
+.fi
+.RE
+
+.LP
+To create a policy based on a target server (which can be manually edited; see official built-in policies for syntax examples):
+.RS
+.nf
+ssh-audit -M new_policy.txt targetserver
+.fi
+.RE
+
+.SH RETURN VALUES
+When a successful connection is made and all algorithms are rated as "good", \fBssh-audit\fP returns 0. Other possible return values are:
+
+.RS
+.nf
+1 = connection error
+2 = at least one algorithm warning was found
+3 = at least one algorithm failure was found
+<any other non-zero value> = unknown error
+.fi
+.RE
+
+.SH SSH HARDENING GUIDES
+Hardening guides for common platforms can be found at: <https://www.ssh\-audit.com/>
+
+.SH BUG REPORTS
+Please file bug reports as a Github Issue at: <https://github.com/jtesta/ssh\-audit/issues>
+
+.SH AUTHOR
+.LP
+\fBssh-audit\fP was originally written by Andris Raugulis <moo@arthepsy.eu>, and maintained from 2015 to 2017.
+.br
+.LP
+Maintainership was assumed and development was resumed in 2017 by Joe Testa <jtesta@positronsecurity.com>.
diff --git a/ssh-audit.py b/ssh-audit.py
new file mode 100755
index 0000000..f0cf144
--- /dev/null
+++ b/ssh-audit.py
@@ -0,0 +1,21 @@
+#!/usr/bin/env python3
+"""src/ssh_audit/ssh_audit.py wrapper for backwards compatibility"""
+
+import sys
+import traceback
+from pathlib import Path
+
+sys.path.insert(0, str(Path(__file__).resolve().parent / "src"))
+
+from ssh_audit.ssh_audit import main # noqa: E402
+from ssh_audit import exitcodes # noqa: E402
+
+exit_code = exitcodes.GOOD
+
+try:
+ exit_code = main()
+except Exception:
+ exit_code = exitcodes.UNKNOWN_ERROR
+ print(traceback.format_exc())
+
+sys.exit(exit_code)
diff --git a/test/conftest.py b/test/conftest.py
new file mode 100644
index 0000000..e3878b1
--- /dev/null
+++ b/test/conftest.py
@@ -0,0 +1,145 @@
+import io
+import sys
+import socket
+import pytest
+
+
+@pytest.fixture(scope='module')
+def ssh_audit():
+ import ssh_audit.ssh_audit
+ return ssh_audit.ssh_audit
+
+
+# pylint: disable=attribute-defined-outside-init
+class _OutputSpy(list):
+ def begin(self):
+ self.__out = io.StringIO()
+ self.__old_stdout = sys.stdout
+ sys.stdout = self.__out
+
+ def flush(self):
+ lines = self.__out.getvalue().splitlines()
+ sys.stdout = self.__old_stdout
+ self.__out = None
+ return lines
+
+
+@pytest.fixture(scope='module')
+def output_spy():
+ return _OutputSpy()
+
+
+class _VirtualGlobalSocket:
+ def __init__(self, vsocket):
+ self.vsocket = vsocket
+ self.addrinfodata = {}
+
+ # pylint: disable=unused-argument
+ def create_connection(self, address, timeout=0, source_address=None):
+ # pylint: disable=protected-access
+ return self.vsocket._connect(address, True)
+
+ # pylint: disable=unused-argument
+ def socket(self,
+ family=socket.AF_INET,
+ socktype=socket.SOCK_STREAM,
+ proto=0,
+ fileno=None):
+ return self.vsocket
+
+ def getaddrinfo(self, host, port, family=0, socktype=0, proto=0, flags=0):
+ key = '{}#{}'.format(host, port)
+ if key in self.addrinfodata:
+ data = self.addrinfodata[key]
+ if isinstance(data, Exception):
+ raise data
+ return data
+ if host == 'localhost':
+ r = []
+ if family in (0, socket.AF_INET):
+ r.append((socket.AF_INET, 1, 6, '', ('127.0.0.1', port)))
+ if family in (0, socket.AF_INET6):
+ r.append((socket.AF_INET6, 1, 6, '', ('::1', port)))
+ return r
+ return []
+
+
+class _VirtualSocket:
+ def __init__(self):
+ self.sock_address = ('127.0.0.1', 0)
+ self.peer_address = None
+ self._connected = False
+ self.timeout = -1.0
+ self.rdata = []
+ self.sdata = []
+ self.errors = {}
+ self.gsock = _VirtualGlobalSocket(self)
+
+ def _check_err(self, method):
+ method_error = self.errors.get(method)
+ if method_error:
+ raise method_error
+
+ def connect(self, address):
+ return self._connect(address, False)
+
+ def _connect(self, address, ret=True):
+ self.peer_address = address
+ self._connected = True
+ self._check_err('connect')
+ return self if ret else None
+
+ def settimeout(self, timeout):
+ self.timeout = timeout
+
+ def gettimeout(self):
+ return self.timeout
+
+ def getpeername(self):
+ if self.peer_address is None or not self._connected:
+ raise OSError(57, 'Socket is not connected')
+ return self.peer_address
+
+ def getsockname(self):
+ return self.sock_address
+
+ def bind(self, address):
+ self.sock_address = address
+
+ def listen(self, backlog):
+ pass
+
+ def accept(self):
+ # pylint: disable=protected-access
+ conn = _VirtualSocket()
+ conn.sock_address = self.sock_address
+ conn.peer_address = ('127.0.0.1', 0)
+ conn._connected = True
+ return conn, conn.peer_address
+
+ def recv(self, bufsize, flags=0):
+ # pylint: disable=unused-argument
+ if not self._connected:
+ raise OSError(54, 'Connection reset by peer')
+ if not len(self.rdata) > 0:
+ return b''
+ data = self.rdata.pop(0)
+ if isinstance(data, Exception):
+ raise data
+ return data
+
+ def send(self, data):
+ if self.peer_address is None or not self._connected:
+ raise OSError(32, 'Broken pipe')
+ self._check_err('send')
+ self.sdata.append(data)
+
+
+@pytest.fixture()
+def virtual_socket(monkeypatch):
+ vsocket = _VirtualSocket()
+ gsock = vsocket.gsock
+ monkeypatch.setattr(socket, 'create_connection', gsock.create_connection)
+ monkeypatch.setattr(socket, 'socket', gsock.socket)
+ monkeypatch.setattr(socket, 'getaddrinfo', gsock.getaddrinfo)
+ return vsocket
diff --git a/test/docker/.ed25519.sk b/test/docker/.ed25519.sk
new file mode 100644
index 0000000..58b097b
--- /dev/null
+++ b/test/docker/.ed25519.sk
@@ -0,0 +1 @@
+iÜ›ãÈÍíVÌè¿ÓZ/Dì<†î|S“zË=°:×1vu}¢ï„JòÝ·ŸŠ"à^Bb&U‰ìP« ’CJ? \ No newline at end of file
diff --git a/test/docker/Dockerfile b/test/docker/Dockerfile
new file mode 100644
index 0000000..eef0139
--- /dev/null
+++ b/test/docker/Dockerfile
@@ -0,0 +1,32 @@
+FROM ubuntu:16.04
+
+COPY openssh-4.0p1/sshd /openssh/sshd-4.0p1
+COPY openssh-5.6p1/sshd /openssh/sshd-5.6p1
+COPY openssh-8.0p1/sshd /openssh/sshd-8.0p1
+COPY dropbear-2019.78/dropbear /dropbear/dropbear-2019.78
+COPY tinyssh-20190101/build/bin/tinysshd /tinysshd/tinyssh-20190101
+
+# Dropbear host keys.
+COPY dropbear_*_host_key* /etc/dropbear/
+
+# OpenSSH configs.
+COPY sshd_config* /etc/ssh/
+
+# OpenSSH host keys & moduli file.
+COPY ssh_host_* /etc/ssh/
+COPY ssh1_host_* /etc/ssh/
+COPY moduli_1024 /usr/local/etc/moduli
+
+# TinySSH host keys.
+COPY ed25519.pk /etc/tinyssh/
+COPY .ed25519.sk /etc/tinyssh/
+
+COPY debug.sh /debug.sh
+
+RUN apt update 2> /dev/null
+RUN apt install -y libssl-dev strace rsyslog ucspi-tcp 2> /dev/null
+RUN apt clean 2> /dev/null
+RUN useradd -s /bin/false sshd
+RUN mkdir /var/empty
+
+EXPOSE 22
diff --git a/test/docker/debug.sh b/test/docker/debug.sh
new file mode 100755
index 0000000..c4be343
--- /dev/null
+++ b/test/docker/debug.sh
@@ -0,0 +1,9 @@
+#!/bin/bash
+
+# This script is run on in docker container. It will enable logging for sshd in
+# /var/log/auth.log.
+
+/etc/init.d/rsyslog start
+sleep 1
+/openssh/sshd-5.6p1 -o LogLevel=DEBUG3 -f /etc/ssh/sshd_config-5.6p1_test1
+/bin/bash
diff --git a/test/docker/dropbear_dss_host_key b/test/docker/dropbear_dss_host_key
new file mode 100644
index 0000000..3388632
--- /dev/null
+++ b/test/docker/dropbear_dss_host_key
Binary files differ
diff --git a/test/docker/dropbear_ecdsa_host_key b/test/docker/dropbear_ecdsa_host_key
new file mode 100644
index 0000000..318ebb0
--- /dev/null
+++ b/test/docker/dropbear_ecdsa_host_key
Binary files differ
diff --git a/test/docker/dropbear_rsa_host_key_1024 b/test/docker/dropbear_rsa_host_key_1024
new file mode 100644
index 0000000..d9ce331
--- /dev/null
+++ b/test/docker/dropbear_rsa_host_key_1024
Binary files differ
diff --git a/test/docker/dropbear_rsa_host_key_3072 b/test/docker/dropbear_rsa_host_key_3072
new file mode 100644
index 0000000..006249a
--- /dev/null
+++ b/test/docker/dropbear_rsa_host_key_3072
Binary files differ
diff --git a/test/docker/ed25519.pk b/test/docker/ed25519.pk
new file mode 100644
index 0000000..82cfb47
--- /dev/null
+++ b/test/docker/ed25519.pk
@@ -0,0 +1 @@
+1vu}¢ï„JòÝ·ŸŠ"à^Bb&U‰ìP« ’CJ? \ No newline at end of file
diff --git a/test/docker/expected_results/dropbear_2019.78_test1.json b/test/docker/expected_results/dropbear_2019.78_test1.json
new file mode 100644
index 0000000..186cb06
--- /dev/null
+++ b/test/docker/expected_results/dropbear_2019.78_test1.json
@@ -0,0 +1 @@
+{"banner": {"comments": null, "protocol": [2, 0], "raw": "SSH-2.0-dropbear_2019.78", "software": "dropbear_2019.78"}, "compression": ["zlib@openssh.com", "none"], "enc": ["aes128-ctr", "aes256-ctr", "aes128-cbc", "aes256-cbc", "3des-ctr", "3des-cbc"], "fingerprints": [{"hash": "CDfAU12pjQS7/91kg7gYacza0U/6PDbE04Ic3IpYxkM", "hash_alg": "SHA256", "hostkey": "ssh-rsa"}, {"hash": "63:7f:54:f7:0a:28:7f:75:0b:f4:07:0b:fc:66:51:a2", "hash_alg": "MD5", "hostkey": "ssh-rsa"}], "kex": [{"algorithm": "curve25519-sha256"}, {"algorithm": "curve25519-sha256@libssh.org"}, {"algorithm": "ecdh-sha2-nistp521"}, {"algorithm": "ecdh-sha2-nistp384"}, {"algorithm": "ecdh-sha2-nistp256"}, {"algorithm": "diffie-hellman-group14-sha256"}, {"algorithm": "diffie-hellman-group14-sha1"}, {"algorithm": "kexguess2@matt.ucc.asn.au"}], "key": [{"algorithm": "ecdsa-sha2-nistp256"}, {"algorithm": "ssh-rsa", "keysize": 1024}, {"algorithm": "ssh-dss"}], "mac": ["hmac-sha1-96", "hmac-sha1", "hmac-sha2-256"], "target": "localhost"}
diff --git a/test/docker/expected_results/dropbear_2019.78_test1.txt b/test/docker/expected_results/dropbear_2019.78_test1.txt
new file mode 100644
index 0000000..93205f4
--- /dev/null
+++ b/test/docker/expected_results/dropbear_2019.78_test1.txt
@@ -0,0 +1,88 @@
+# general
+(gen) banner: SSH-2.0-dropbear_2019.78
+(gen) software: Dropbear SSH 2019.78
+(gen) compatibility: OpenSSH 7.4+ (some functionality from 6.6), Dropbear SSH 2018.76+
+(gen) compression: enabled (zlib@openssh.com)
+
+# key exchange algorithms
+(kex) curve25519-sha256 -- [info] available since OpenSSH 7.4, Dropbear SSH 2018.76
+(kex) curve25519-sha256@libssh.org -- [info] available since OpenSSH 6.5, Dropbear SSH 2013.62
+(kex) ecdh-sha2-nistp521 -- [fail] using weak elliptic curves
+ `- [info] available since OpenSSH 5.7, Dropbear SSH 2013.62
+(kex) ecdh-sha2-nistp384 -- [fail] using weak elliptic curves
+ `- [info] available since OpenSSH 5.7, Dropbear SSH 2013.62
+(kex) ecdh-sha2-nistp256 -- [fail] using weak elliptic curves
+ `- [info] available since OpenSSH 5.7, Dropbear SSH 2013.62
+(kex) diffie-hellman-group14-sha256 -- [info] available since OpenSSH 7.3, Dropbear SSH 2016.73
+(kex) diffie-hellman-group14-sha1 -- [warn] using weak hashing algorithm
+ `- [info] available since OpenSSH 3.9, Dropbear SSH 0.53
+(kex) kexguess2@matt.ucc.asn.au -- [info] available since Dropbear SSH 2013.57
+
+# host-key algorithms
+(key) ecdsa-sha2-nistp256 -- [fail] using weak elliptic curves
+ `- [warn] using weak random number generator could reveal the key
+ `- [info] available since OpenSSH 5.7, Dropbear SSH 2013.62
+(key) ssh-rsa (1024-bit) -- [fail] using weak hashing algorithm
+ `- [fail] using small 1024-bit modulus
+ `- [info] available since OpenSSH 2.5.0, Dropbear SSH 0.28
+ `- [info] a future deprecation notice has been issued in OpenSSH 8.2: https://www.openssh.com/txt/release-8.2
+(key) ssh-dss -- [fail] using small 1024-bit modulus
+ `- [fail] removed (in server) and disabled (in client) since OpenSSH 7.0, weak algorithm
+ `- [warn] using weak random number generator could reveal the key
+ `- [info] available since OpenSSH 2.1.0, Dropbear SSH 0.28
+
+# encryption algorithms (ciphers)
+(enc) aes128-ctr -- [info] available since OpenSSH 3.7, Dropbear SSH 0.52
+(enc) aes256-ctr -- [info] available since OpenSSH 3.7, Dropbear SSH 0.52
+(enc) aes128-cbc -- [fail] removed (in server) since OpenSSH 6.7, unsafe algorithm
+ `- [warn] using weak cipher mode
+ `- [info] available since OpenSSH 2.3.0, Dropbear SSH 0.28
+(enc) aes256-cbc -- [fail] removed (in server) since OpenSSH 6.7, unsafe algorithm
+ `- [warn] using weak cipher mode
+ `- [info] available since OpenSSH 2.3.0, Dropbear SSH 0.47
+(enc) 3des-ctr -- [fail] using weak cipher
+ `- [info] available since Dropbear SSH 0.52
+(enc) 3des-cbc -- [fail] removed (in server) since OpenSSH 6.7, unsafe algorithm
+ `- [warn] disabled (in client) since OpenSSH 7.4, unsafe algorithm
+ `- [warn] using weak cipher
+ `- [warn] using weak cipher mode
+ `- [warn] using small 64-bit block size
+ `- [info] available since OpenSSH 1.2.2, Dropbear SSH 0.28
+
+# message authentication code algorithms
+(mac) hmac-sha1-96 -- [fail] removed (in server) since OpenSSH 6.7, unsafe algorithm
+ `- [warn] disabled (in client) since OpenSSH 7.2, legacy algorithm
+ `- [warn] using encrypt-and-MAC mode
+ `- [warn] using weak hashing algorithm
+ `- [info] available since OpenSSH 2.5.0, Dropbear SSH 0.47
+(mac) hmac-sha1 -- [warn] using encrypt-and-MAC mode
+ `- [warn] using weak hashing algorithm
+ `- [info] available since OpenSSH 2.1.0, Dropbear SSH 0.28
+(mac) hmac-sha2-256 -- [warn] using encrypt-and-MAC mode
+ `- [info] available since OpenSSH 5.9, Dropbear SSH 2013.56
+
+# fingerprints
+(fin) ssh-rsa: SHA256:CDfAU12pjQS7/91kg7gYacza0U/6PDbE04Ic3IpYxkM
+
+# algorithm recommendations (for Dropbear SSH 2019.78)
+(rec) -3des-cbc -- enc algorithm to remove 
+(rec) -3des-ctr -- enc algorithm to remove 
+(rec) -aes128-cbc -- enc algorithm to remove 
+(rec) -aes256-cbc -- enc algorithm to remove 
+(rec) -ecdh-sha2-nistp256 -- kex algorithm to remove 
+(rec) -ecdh-sha2-nistp384 -- kex algorithm to remove 
+(rec) -ecdh-sha2-nistp521 -- kex algorithm to remove 
+(rec) -ecdsa-sha2-nistp256 -- key algorithm to remove 
+(rec) -hmac-sha1-96 -- mac algorithm to remove 
+(rec) -ssh-dss -- key algorithm to remove 
+(rec) -ssh-rsa -- key algorithm to remove 
+(rec) +diffie-hellman-group16-sha512 -- kex algorithm to append 
+(rec) +twofish128-ctr -- enc algorithm to append 
+(rec) +twofish256-ctr -- enc algorithm to append 
+(rec) -diffie-hellman-group14-sha1 -- kex algorithm to remove 
+(rec) -hmac-sha1 -- mac algorithm to remove 
+(rec) -hmac-sha2-256 -- mac algorithm to remove 
+
+# additional info
+(nfo) For hardening guides on common OSes, please see: <https://www.ssh-audit.com/hardening_guides.html>
+
diff --git a/test/docker/expected_results/openssh_4.0p1_test1.json b/test/docker/expected_results/openssh_4.0p1_test1.json
new file mode 100644
index 0000000..33e63cf
--- /dev/null
+++ b/test/docker/expected_results/openssh_4.0p1_test1.json
@@ -0,0 +1 @@
+{"banner": {"comments": null, "protocol": [1, 99], "raw": "SSH-1.99-OpenSSH_4.0", "software": "OpenSSH_4.0"}, "compression": ["none", "zlib"], "enc": ["aes128-cbc", "3des-cbc", "blowfish-cbc", "cast128-cbc", "arcfour", "aes192-cbc", "aes256-cbc", "rijndael-cbc@lysator.liu.se", "aes128-ctr", "aes192-ctr", "aes256-ctr"], "fingerprints": [{"hash": "YZ457EBcJTSxRKI3yXRgtAj3PBf5B9/F36b1SVooml4", "hash_alg": "SHA256", "hostkey": "ssh-rsa"}, {"hash": "3c:c3:38:f8:55:39:c0:4a:5a:17:89:60:2c:a1:fc:6a", "hash_alg": "MD5", "hostkey": "ssh-rsa"}], "kex": [{"algorithm": "diffie-hellman-group-exchange-sha1", "keysize": 1024}, {"algorithm": "diffie-hellman-group14-sha1"}, {"algorithm": "diffie-hellman-group1-sha1"}], "key": [{"algorithm": "ssh-rsa", "keysize": 1024}, {"algorithm": "ssh-dss"}], "mac": ["hmac-md5", "hmac-sha1", "hmac-ripemd160", "hmac-ripemd160@openssh.com", "hmac-sha1-96", "hmac-md5-96"], "target": "localhost"}
diff --git a/test/docker/expected_results/openssh_4.0p1_test1.txt b/test/docker/expected_results/openssh_4.0p1_test1.txt
new file mode 100644
index 0000000..419105a
--- /dev/null
+++ b/test/docker/expected_results/openssh_4.0p1_test1.txt
@@ -0,0 +1,145 @@
+# general
+(gen) banner: SSH-1.99-OpenSSH_4.0
+(gen) protocol SSH1 enabled
+(gen) software: OpenSSH 4.0
+(gen) compatibility: OpenSSH 3.9-6.6, Dropbear SSH 0.53+ (some functionality from 0.52)
+(gen) compression: enabled (zlib)
+
+# security
+(cve) CVE-2018-15473 -- (CVSSv2: 5.3) enumerate usernames due to timing discrepencies
+(cve) CVE-2016-3115 -- (CVSSv2: 5.5) bypass command restrictions via crafted X11 forwarding data
+(cve) CVE-2014-1692 -- (CVSSv2: 7.5) cause DoS via triggering error condition (memory corruption)
+(cve) CVE-2012-0814 -- (CVSSv2: 3.5) leak data via debug messages
+(cve) CVE-2011-5000 -- (CVSSv2: 3.5) cause DoS via large value in certain length field (memory consumption)
+(cve) CVE-2010-5107 -- (CVSSv2: 5.0) cause DoS via large number of connections (slot exhaustion)
+(cve) CVE-2010-4755 -- (CVSSv2: 4.0) cause DoS via crafted glob expression (CPU and memory consumption)
+(cve) CVE-2010-4478 -- (CVSSv2: 7.5) bypass authentication check via crafted values
+(cve) CVE-2008-5161 -- (CVSSv2: 2.6) recover plaintext data from ciphertext
+(cve) CVE-2008-4109 -- (CVSSv2: 5.0) cause DoS via multiple login attempts (slot exhaustion)
+(cve) CVE-2008-1657 -- (CVSSv2: 6.5) bypass command restrictions via modifying session file
+(cve) CVE-2008-1483 -- (CVSSv2: 6.9) hijack forwarded X11 connections
+(cve) CVE-2007-4752 -- (CVSSv2: 7.5) privilege escalation via causing an X client to be trusted
+(cve) CVE-2007-2243 -- (CVSSv2: 5.0) discover valid usernames through different responses
+(cve) CVE-2006-5052 -- (CVSSv2: 5.0) discover valid usernames through different responses
+(cve) CVE-2006-5051 -- (CVSSv2: 9.3) cause DoS or execute arbitrary code (double free)
+(cve) CVE-2006-4924 -- (CVSSv2: 7.8) cause DoS via crafted packet (CPU consumption)
+(cve) CVE-2006-0225 -- (CVSSv2: 4.6) execute arbitrary code
+(cve) CVE-2005-2798 -- (CVSSv2: 5.0) leak data about authentication credentials
+(sec) SSH v1 enabled -- SSH v1 can be exploited to recover plaintext passwords
+
+# key exchange algorithms
+(kex) diffie-hellman-group-exchange-sha1 (1024-bit) -- [fail] using small 1024-bit modulus
+ `- [warn] using weak hashing algorithm
+ `- [info] available since OpenSSH 2.3.0
+(kex) diffie-hellman-group14-sha1 -- [warn] using weak hashing algorithm
+ `- [info] available since OpenSSH 3.9, Dropbear SSH 0.53
+(kex) diffie-hellman-group1-sha1 -- [fail] using small 1024-bit modulus
+ `- [fail] removed (in server) since OpenSSH 6.7, unsafe algorithm
+ `- [fail] disabled (in client) since OpenSSH 7.0, logjam attack
+ `- [warn] using weak hashing algorithm
+ `- [info] available since OpenSSH 2.3.0, Dropbear SSH 0.28
+
+# host-key algorithms
+(key) ssh-rsa (1024-bit) -- [fail] using weak hashing algorithm
+ `- [fail] using small 1024-bit modulus
+ `- [info] available since OpenSSH 2.5.0, Dropbear SSH 0.28
+ `- [info] a future deprecation notice has been issued in OpenSSH 8.2: https://www.openssh.com/txt/release-8.2
+(key) ssh-dss -- [fail] using small 1024-bit modulus
+ `- [fail] removed (in server) and disabled (in client) since OpenSSH 7.0, weak algorithm
+ `- [warn] using weak random number generator could reveal the key
+ `- [info] available since OpenSSH 2.1.0, Dropbear SSH 0.28
+
+# encryption algorithms (ciphers)
+(enc) aes128-cbc -- [fail] removed (in server) since OpenSSH 6.7, unsafe algorithm
+ `- [warn] using weak cipher mode
+ `- [info] available since OpenSSH 2.3.0, Dropbear SSH 0.28
+(enc) 3des-cbc -- [fail] removed (in server) since OpenSSH 6.7, unsafe algorithm
+ `- [warn] disabled (in client) since OpenSSH 7.4, unsafe algorithm
+ `- [warn] using weak cipher
+ `- [warn] using weak cipher mode
+ `- [warn] using small 64-bit block size
+ `- [info] available since OpenSSH 1.2.2, Dropbear SSH 0.28
+(enc) blowfish-cbc -- [fail] removed (in server) since OpenSSH 6.7, unsafe algorithm
+ `- [fail] disabled since Dropbear SSH 0.53
+ `- [warn] disabled (in client) since OpenSSH 7.2, legacy algorithm
+ `- [warn] using weak cipher mode
+ `- [warn] using small 64-bit block size
+ `- [info] available since OpenSSH 1.2.2, Dropbear SSH 0.28
+(enc) cast128-cbc -- [fail] removed (in server) since OpenSSH 6.7, unsafe algorithm
+ `- [warn] disabled (in client) since OpenSSH 7.2, legacy algorithm
+ `- [warn] using weak cipher mode
+ `- [warn] using small 64-bit block size
+ `- [info] available since OpenSSH 2.1.0
+(enc) arcfour -- [fail] removed (in server) since OpenSSH 6.7, unsafe algorithm
+ `- [warn] disabled (in client) since OpenSSH 7.2, legacy algorithm
+ `- [warn] using weak cipher
+ `- [info] available since OpenSSH 2.1.0
+(enc) aes192-cbc -- [fail] removed (in server) since OpenSSH 6.7, unsafe algorithm
+ `- [warn] using weak cipher mode
+ `- [info] available since OpenSSH 2.3.0
+(enc) aes256-cbc -- [fail] removed (in server) since OpenSSH 6.7, unsafe algorithm
+ `- [warn] using weak cipher mode
+ `- [info] available since OpenSSH 2.3.0, Dropbear SSH 0.47
+(enc) rijndael-cbc@lysator.liu.se -- [fail] removed (in server) since OpenSSH 6.7, unsafe algorithm
+ `- [warn] disabled (in client) since OpenSSH 7.2, legacy algorithm
+ `- [warn] using weak cipher mode
+ `- [info] available since OpenSSH 2.3.0
+(enc) aes128-ctr -- [info] available since OpenSSH 3.7, Dropbear SSH 0.52
+(enc) aes192-ctr -- [info] available since OpenSSH 3.7
+(enc) aes256-ctr -- [info] available since OpenSSH 3.7, Dropbear SSH 0.52
+
+# message authentication code algorithms
+(mac) hmac-md5 -- [fail] removed (in server) since OpenSSH 6.7, unsafe algorithm
+ `- [warn] disabled (in client) since OpenSSH 7.2, legacy algorithm
+ `- [warn] using encrypt-and-MAC mode
+ `- [warn] using weak hashing algorithm
+ `- [info] available since OpenSSH 2.1.0, Dropbear SSH 0.28
+(mac) hmac-sha1 -- [warn] using encrypt-and-MAC mode
+ `- [warn] using weak hashing algorithm
+ `- [info] available since OpenSSH 2.1.0, Dropbear SSH 0.28
+(mac) hmac-ripemd160 -- [fail] removed (in server) since OpenSSH 6.7, unsafe algorithm
+ `- [warn] disabled (in client) since OpenSSH 7.2, legacy algorithm
+ `- [warn] using encrypt-and-MAC mode
+ `- [info] available since OpenSSH 2.5.0
+(mac) hmac-ripemd160@openssh.com -- [fail] removed (in server) since OpenSSH 6.7, unsafe algorithm
+ `- [warn] disabled (in client) since OpenSSH 7.2, legacy algorithm
+ `- [warn] using encrypt-and-MAC mode
+ `- [info] available since OpenSSH 2.1.0
+(mac) hmac-sha1-96 -- [fail] removed (in server) since OpenSSH 6.7, unsafe algorithm
+ `- [warn] disabled (in client) since OpenSSH 7.2, legacy algorithm
+ `- [warn] using encrypt-and-MAC mode
+ `- [warn] using weak hashing algorithm
+ `- [info] available since OpenSSH 2.5.0, Dropbear SSH 0.47
+(mac) hmac-md5-96 -- [fail] removed (in server) since OpenSSH 6.7, unsafe algorithm
+ `- [warn] disabled (in client) since OpenSSH 7.2, legacy algorithm
+ `- [warn] using encrypt-and-MAC mode
+ `- [warn] using weak hashing algorithm
+ `- [info] available since OpenSSH 2.5.0
+
+# fingerprints
+(fin) ssh-rsa: SHA256:YZ457EBcJTSxRKI3yXRgtAj3PBf5B9/F36b1SVooml4
+
+# algorithm recommendations (for OpenSSH 4.0)
+(rec) -3des-cbc -- enc algorithm to remove 
+(rec) -aes128-cbc -- enc algorithm to remove 
+(rec) -aes192-cbc -- enc algorithm to remove 
+(rec) -aes256-cbc -- enc algorithm to remove 
+(rec) -arcfour -- enc algorithm to remove 
+(rec) -blowfish-cbc -- enc algorithm to remove 
+(rec) -cast128-cbc -- enc algorithm to remove 
+(rec) -diffie-hellman-group-exchange-sha1 -- kex algorithm to remove 
+(rec) -diffie-hellman-group1-sha1 -- kex algorithm to remove 
+(rec) -hmac-md5 -- mac algorithm to remove 
+(rec) -hmac-md5-96 -- mac algorithm to remove 
+(rec) -hmac-ripemd160 -- mac algorithm to remove 
+(rec) -hmac-ripemd160@openssh.com -- mac algorithm to remove 
+(rec) -hmac-sha1-96 -- mac algorithm to remove 
+(rec) -rijndael-cbc@lysator.liu.se -- enc algorithm to remove 
+(rec) -ssh-dss -- key algorithm to remove 
+(rec) -ssh-rsa -- key algorithm to remove 
+(rec) -diffie-hellman-group14-sha1 -- kex algorithm to remove 
+(rec) -hmac-sha1 -- mac algorithm to remove 
+
+# additional info
+(nfo) For hardening guides on common OSes, please see: <https://www.ssh-audit.com/hardening_guides.html>
+
diff --git a/test/docker/expected_results/openssh_5.6p1_custom_policy_test1.json b/test/docker/expected_results/openssh_5.6p1_custom_policy_test1.json
new file mode 100644
index 0000000..52d9e38
--- /dev/null
+++ b/test/docker/expected_results/openssh_5.6p1_custom_policy_test1.json
@@ -0,0 +1 @@
+{"errors": [], "host": "localhost", "passed": true, "policy": "Docker policy: test1 (version 1)"}
diff --git a/test/docker/expected_results/openssh_5.6p1_custom_policy_test1.txt b/test/docker/expected_results/openssh_5.6p1_custom_policy_test1.txt
new file mode 100644
index 0000000..01146f8
--- /dev/null
+++ b/test/docker/expected_results/openssh_5.6p1_custom_policy_test1.txt
@@ -0,0 +1,3 @@
+Host: localhost:2222
+Policy: Docker policy: test1 (version 1)
+Result: ✔ Passed
diff --git a/test/docker/expected_results/openssh_5.6p1_custom_policy_test10.json b/test/docker/expected_results/openssh_5.6p1_custom_policy_test10.json
new file mode 100644
index 0000000..7bf35ae
--- /dev/null
+++ b/test/docker/expected_results/openssh_5.6p1_custom_policy_test10.json
@@ -0,0 +1 @@
+{"errors": [{"actual": ["3072"], "expected_optional": [""], "expected_required": ["4096"], "mismatched_field": "RSA host key (ssh-rsa-cert-v01@openssh.com) sizes"}, {"actual": ["1024"], "expected_optional": [""], "expected_required": ["4096"], "mismatched_field": "RSA CA key (ssh-rsa-cert-v01@openssh.com) sizes"}], "host": "localhost", "passed": false, "policy": "Docker poliicy: test10 (version 1)"}
diff --git a/test/docker/expected_results/openssh_5.6p1_custom_policy_test10.txt b/test/docker/expected_results/openssh_5.6p1_custom_policy_test10.txt
new file mode 100644
index 0000000..7f8befc
--- /dev/null
+++ b/test/docker/expected_results/openssh_5.6p1_custom_policy_test10.txt
@@ -0,0 +1,13 @@
+Host: localhost:2222
+Policy: Docker poliicy: test10 (version 1)
+Result: ⌠Failed!
+
+Errors:
+ * RSA CA key (ssh-rsa-cert-v01@openssh.com) sizes did not match.
+ - Expected: 4096
+ - Actual: 1024
+
+ * RSA host key (ssh-rsa-cert-v01@openssh.com) sizes did not match.
+ - Expected: 4096
+ - Actual: 3072
+
diff --git a/test/docker/expected_results/openssh_5.6p1_custom_policy_test2.json b/test/docker/expected_results/openssh_5.6p1_custom_policy_test2.json
new file mode 100644
index 0000000..4d39f6a
--- /dev/null
+++ b/test/docker/expected_results/openssh_5.6p1_custom_policy_test2.json
@@ -0,0 +1 @@
+{"errors": [{"actual": ["diffie-hellman-group-exchange-sha256", "diffie-hellman-group-exchange-sha1", "diffie-hellman-group14-sha1", "diffie-hellman-group1-sha1"], "expected_optional": [""], "expected_required": ["kex_alg1", "kex_alg2"], "mismatched_field": "Key exchanges"}], "host": "localhost", "passed": false, "policy": "Docker policy: test2 (version 1)"}
diff --git a/test/docker/expected_results/openssh_5.6p1_custom_policy_test2.txt b/test/docker/expected_results/openssh_5.6p1_custom_policy_test2.txt
new file mode 100644
index 0000000..e88e44b
--- /dev/null
+++ b/test/docker/expected_results/openssh_5.6p1_custom_policy_test2.txt
@@ -0,0 +1,9 @@
+Host: localhost:2222
+Policy: Docker policy: test2 (version 1)
+Result: ⌠Failed!
+
+Errors:
+ * Key exchanges did not match.
+ - Expected: kex_alg1, kex_alg2
+ - Actual: diffie-hellman-group-exchange-sha256, diffie-hellman-group-exchange-sha1, diffie-hellman-group14-sha1, diffie-hellman-group1-sha1
+
diff --git a/test/docker/expected_results/openssh_5.6p1_custom_policy_test3.json b/test/docker/expected_results/openssh_5.6p1_custom_policy_test3.json
new file mode 100644
index 0000000..519fc40
--- /dev/null
+++ b/test/docker/expected_results/openssh_5.6p1_custom_policy_test3.json
@@ -0,0 +1 @@
+{"errors": [{"actual": ["ssh-rsa", "ssh-dss"], "expected_optional": [""], "expected_required": ["ssh-rsa", "ssh-dss", "key_alg1"], "mismatched_field": "Host keys"}], "host": "localhost", "passed": false, "policy": "Docker policy: test3 (version 1)"}
diff --git a/test/docker/expected_results/openssh_5.6p1_custom_policy_test3.txt b/test/docker/expected_results/openssh_5.6p1_custom_policy_test3.txt
new file mode 100644
index 0000000..cf7eefc
--- /dev/null
+++ b/test/docker/expected_results/openssh_5.6p1_custom_policy_test3.txt
@@ -0,0 +1,9 @@
+Host: localhost:2222
+Policy: Docker policy: test3 (version 1)
+Result: ⌠Failed!
+
+Errors:
+ * Host keys did not match.
+ - Expected: ssh-rsa, ssh-dss, key_alg1
+ - Actual: ssh-rsa, ssh-dss
+
diff --git a/test/docker/expected_results/openssh_5.6p1_custom_policy_test4.json b/test/docker/expected_results/openssh_5.6p1_custom_policy_test4.json
new file mode 100644
index 0000000..83db23b
--- /dev/null
+++ b/test/docker/expected_results/openssh_5.6p1_custom_policy_test4.json
@@ -0,0 +1 @@
+{"errors": [{"actual": ["aes128-ctr", "aes192-ctr", "aes256-ctr", "arcfour256", "arcfour128", "aes128-cbc", "3des-cbc", "blowfish-cbc", "cast128-cbc", "aes192-cbc", "aes256-cbc", "arcfour", "rijndael-cbc@lysator.liu.se"], "expected_optional": [""], "expected_required": ["cipher_alg1", "cipher_alg2"], "mismatched_field": "Ciphers"}], "host": "localhost", "passed": false, "policy": "Docker policy: test4 (version 1)"}
diff --git a/test/docker/expected_results/openssh_5.6p1_custom_policy_test4.txt b/test/docker/expected_results/openssh_5.6p1_custom_policy_test4.txt
new file mode 100644
index 0000000..514b715
--- /dev/null
+++ b/test/docker/expected_results/openssh_5.6p1_custom_policy_test4.txt
@@ -0,0 +1,9 @@
+Host: localhost:2222
+Policy: Docker policy: test4 (version 1)
+Result: ⌠Failed!
+
+Errors:
+ * Ciphers did not match.
+ - Expected: cipher_alg1, cipher_alg2
+ - Actual: aes128-ctr, aes192-ctr, aes256-ctr, arcfour256, arcfour128, aes128-cbc, 3des-cbc, blowfish-cbc, cast128-cbc, aes192-cbc, aes256-cbc, arcfour, rijndael-cbc@lysator.liu.se
+
diff --git a/test/docker/expected_results/openssh_5.6p1_custom_policy_test5.json b/test/docker/expected_results/openssh_5.6p1_custom_policy_test5.json
new file mode 100644
index 0000000..3fbfe09
--- /dev/null
+++ b/test/docker/expected_results/openssh_5.6p1_custom_policy_test5.json
@@ -0,0 +1 @@
+{"errors": [{"actual": ["hmac-md5", "hmac-sha1", "umac-64@openssh.com", "hmac-ripemd160", "hmac-ripemd160@openssh.com", "hmac-sha1-96", "hmac-md5-96"], "expected_optional": [""], "expected_required": ["hmac-md5", "hmac-sha1", "umac-64@openssh.com", "hmac-ripemd160", "hmac-ripemd160@openssh.com", "hmac_alg1", "hmac-md5-96"], "mismatched_field": "MACs"}], "host": "localhost", "passed": false, "policy": "Docker policy: test5 (version 1)"}
diff --git a/test/docker/expected_results/openssh_5.6p1_custom_policy_test5.txt b/test/docker/expected_results/openssh_5.6p1_custom_policy_test5.txt
new file mode 100644
index 0000000..746ca8c
--- /dev/null
+++ b/test/docker/expected_results/openssh_5.6p1_custom_policy_test5.txt
@@ -0,0 +1,9 @@
+Host: localhost:2222
+Policy: Docker policy: test5 (version 1)
+Result: ⌠Failed!
+
+Errors:
+ * MACs did not match.
+ - Expected: hmac-md5, hmac-sha1, umac-64@openssh.com, hmac-ripemd160, hmac-ripemd160@openssh.com, hmac_alg1, hmac-md5-96
+ - Actual: hmac-md5, hmac-sha1, umac-64@openssh.com, hmac-ripemd160, hmac-ripemd160@openssh.com, hmac-sha1-96, hmac-md5-96
+
diff --git a/test/docker/expected_results/openssh_5.6p1_custom_policy_test7.json b/test/docker/expected_results/openssh_5.6p1_custom_policy_test7.json
new file mode 100644
index 0000000..cb4b860
--- /dev/null
+++ b/test/docker/expected_results/openssh_5.6p1_custom_policy_test7.json
@@ -0,0 +1 @@
+{"errors": [], "host": "localhost", "passed": true, "policy": "Docker poliicy: test7 (version 1)"}
diff --git a/test/docker/expected_results/openssh_5.6p1_custom_policy_test7.txt b/test/docker/expected_results/openssh_5.6p1_custom_policy_test7.txt
new file mode 100644
index 0000000..4014b23
--- /dev/null
+++ b/test/docker/expected_results/openssh_5.6p1_custom_policy_test7.txt
@@ -0,0 +1,3 @@
+Host: localhost:2222
+Policy: Docker poliicy: test7 (version 1)
+Result: ✔ Passed
diff --git a/test/docker/expected_results/openssh_5.6p1_custom_policy_test8.json b/test/docker/expected_results/openssh_5.6p1_custom_policy_test8.json
new file mode 100644
index 0000000..a7fa650
--- /dev/null
+++ b/test/docker/expected_results/openssh_5.6p1_custom_policy_test8.json
@@ -0,0 +1 @@
+{"errors": [{"actual": ["1024"], "expected_optional": [""], "expected_required": ["2048"], "mismatched_field": "RSA CA key (ssh-rsa-cert-v01@openssh.com) sizes"}], "host": "localhost", "passed": false, "policy": "Docker poliicy: test8 (version 1)"}
diff --git a/test/docker/expected_results/openssh_5.6p1_custom_policy_test8.txt b/test/docker/expected_results/openssh_5.6p1_custom_policy_test8.txt
new file mode 100644
index 0000000..36dceba
--- /dev/null
+++ b/test/docker/expected_results/openssh_5.6p1_custom_policy_test8.txt
@@ -0,0 +1,9 @@
+Host: localhost:2222
+Policy: Docker poliicy: test8 (version 1)
+Result: ⌠Failed!
+
+Errors:
+ * RSA CA key (ssh-rsa-cert-v01@openssh.com) sizes did not match.
+ - Expected: 2048
+ - Actual: 1024
+
diff --git a/test/docker/expected_results/openssh_5.6p1_custom_policy_test9.json b/test/docker/expected_results/openssh_5.6p1_custom_policy_test9.json
new file mode 100644
index 0000000..6e9faec
--- /dev/null
+++ b/test/docker/expected_results/openssh_5.6p1_custom_policy_test9.json
@@ -0,0 +1 @@
+{"errors": [{"actual": ["3072"], "expected_optional": [""], "expected_required": ["4096"], "mismatched_field": "RSA host key (ssh-rsa-cert-v01@openssh.com) sizes"}], "host": "localhost", "passed": false, "policy": "Docker poliicy: test9 (version 1)"}
diff --git a/test/docker/expected_results/openssh_5.6p1_custom_policy_test9.txt b/test/docker/expected_results/openssh_5.6p1_custom_policy_test9.txt
new file mode 100644
index 0000000..fc91c9f
--- /dev/null
+++ b/test/docker/expected_results/openssh_5.6p1_custom_policy_test9.txt
@@ -0,0 +1,9 @@
+Host: localhost:2222
+Policy: Docker poliicy: test9 (version 1)
+Result: ⌠Failed!
+
+Errors:
+ * RSA host key (ssh-rsa-cert-v01@openssh.com) sizes did not match.
+ - Expected: 4096
+ - Actual: 3072
+
diff --git a/test/docker/expected_results/openssh_5.6p1_test1.json b/test/docker/expected_results/openssh_5.6p1_test1.json
new file mode 100644
index 0000000..f189bd8
--- /dev/null
+++ b/test/docker/expected_results/openssh_5.6p1_test1.json
@@ -0,0 +1 @@
+{"banner": {"comments": null, "protocol": [2, 0], "raw": "SSH-2.0-OpenSSH_5.6", "software": "OpenSSH_5.6"}, "compression": ["none", "zlib@openssh.com"], "enc": ["aes128-ctr", "aes192-ctr", "aes256-ctr", "arcfour256", "arcfour128", "aes128-cbc", "3des-cbc", "blowfish-cbc", "cast128-cbc", "aes192-cbc", "aes256-cbc", "arcfour", "rijndael-cbc@lysator.liu.se"], "fingerprints": [{"hash": "YZ457EBcJTSxRKI3yXRgtAj3PBf5B9/F36b1SVooml4", "hash_alg": "SHA256", "hostkey": "ssh-rsa"}, {"hash": "3c:c3:38:f8:55:39:c0:4a:5a:17:89:60:2c:a1:fc:6a", "hash_alg": "MD5", "hostkey": "ssh-rsa"}], "kex": [{"algorithm": "diffie-hellman-group-exchange-sha256", "keysize": 1024}, {"algorithm": "diffie-hellman-group-exchange-sha1", "keysize": 1024}, {"algorithm": "diffie-hellman-group14-sha1"}, {"algorithm": "diffie-hellman-group1-sha1"}], "key": [{"algorithm": "ssh-rsa", "keysize": 1024}, {"algorithm": "ssh-dss"}], "mac": ["hmac-md5", "hmac-sha1", "umac-64@openssh.com", "hmac-ripemd160", "hmac-ripemd160@openssh.com", "hmac-sha1-96", "hmac-md5-96"], "target": "localhost"}
diff --git a/test/docker/expected_results/openssh_5.6p1_test1.txt b/test/docker/expected_results/openssh_5.6p1_test1.txt
new file mode 100644
index 0000000..014d7e7
--- /dev/null
+++ b/test/docker/expected_results/openssh_5.6p1_test1.txt
@@ -0,0 +1,153 @@
+# general
+(gen) banner: SSH-2.0-OpenSSH_5.6
+(gen) software: OpenSSH 5.6
+(gen) compatibility: OpenSSH 4.7-6.6, Dropbear SSH 0.53+ (some functionality from 0.52)
+(gen) compression: enabled (zlib@openssh.com)
+
+# security
+(cve) CVE-2018-15473 -- (CVSSv2: 5.3) enumerate usernames due to timing discrepencies
+(cve) CVE-2016-3115 -- (CVSSv2: 5.5) bypass command restrictions via crafted X11 forwarding data
+(cve) CVE-2016-1907 -- (CVSSv2: 5.0) cause DoS via crafted network traffic (out of bounds read)
+(cve) CVE-2015-6564 -- (CVSSv2: 6.9) privilege escalation via leveraging sshd uid
+(cve) CVE-2015-6563 -- (CVSSv2: 1.9) conduct impersonation attack
+(cve) CVE-2014-2532 -- (CVSSv2: 5.8) bypass environment restrictions via specific string before wildcard
+(cve) CVE-2014-1692 -- (CVSSv2: 7.5) cause DoS via triggering error condition (memory corruption)
+(cve) CVE-2012-0814 -- (CVSSv2: 3.5) leak data via debug messages
+(cve) CVE-2011-5000 -- (CVSSv2: 3.5) cause DoS via large value in certain length field (memory consumption)
+(cve) CVE-2010-5107 -- (CVSSv2: 5.0) cause DoS via large number of connections (slot exhaustion)
+(cve) CVE-2010-4755 -- (CVSSv2: 4.0) cause DoS via crafted glob expression (CPU and memory consumption)
+(cve) CVE-2010-4478 -- (CVSSv2: 7.5) bypass authentication check via crafted values
+
+# key exchange algorithms
+(kex) diffie-hellman-group-exchange-sha256 (1024-bit) -- [fail] using small 1024-bit modulus
+ `- [info] available since OpenSSH 4.4
+(kex) diffie-hellman-group-exchange-sha1 (1024-bit) -- [fail] using small 1024-bit modulus
+ `- [warn] using weak hashing algorithm
+ `- [info] available since OpenSSH 2.3.0
+(kex) diffie-hellman-group14-sha1 -- [warn] using weak hashing algorithm
+ `- [info] available since OpenSSH 3.9, Dropbear SSH 0.53
+(kex) diffie-hellman-group1-sha1 -- [fail] using small 1024-bit modulus
+ `- [fail] removed (in server) since OpenSSH 6.7, unsafe algorithm
+ `- [fail] disabled (in client) since OpenSSH 7.0, logjam attack
+ `- [warn] using weak hashing algorithm
+ `- [info] available since OpenSSH 2.3.0, Dropbear SSH 0.28
+
+# host-key algorithms
+(key) ssh-rsa (1024-bit) -- [fail] using weak hashing algorithm
+ `- [fail] using small 1024-bit modulus
+ `- [info] available since OpenSSH 2.5.0, Dropbear SSH 0.28
+ `- [info] a future deprecation notice has been issued in OpenSSH 8.2: https://www.openssh.com/txt/release-8.2
+(key) ssh-dss -- [fail] using small 1024-bit modulus
+ `- [fail] removed (in server) and disabled (in client) since OpenSSH 7.0, weak algorithm
+ `- [warn] using weak random number generator could reveal the key
+ `- [info] available since OpenSSH 2.1.0, Dropbear SSH 0.28
+
+# encryption algorithms (ciphers)
+(enc) aes128-ctr -- [info] available since OpenSSH 3.7, Dropbear SSH 0.52
+(enc) aes192-ctr -- [info] available since OpenSSH 3.7
+(enc) aes256-ctr -- [info] available since OpenSSH 3.7, Dropbear SSH 0.52
+(enc) arcfour256 -- [fail] removed (in server) since OpenSSH 6.7, unsafe algorithm
+ `- [warn] disabled (in client) since OpenSSH 7.2, legacy algorithm
+ `- [warn] using weak cipher
+ `- [info] available since OpenSSH 4.2
+(enc) arcfour128 -- [fail] removed (in server) since OpenSSH 6.7, unsafe algorithm
+ `- [warn] disabled (in client) since OpenSSH 7.2, legacy algorithm
+ `- [warn] using weak cipher
+ `- [info] available since OpenSSH 4.2
+(enc) aes128-cbc -- [fail] removed (in server) since OpenSSH 6.7, unsafe algorithm
+ `- [warn] using weak cipher mode
+ `- [info] available since OpenSSH 2.3.0, Dropbear SSH 0.28
+(enc) 3des-cbc -- [fail] removed (in server) since OpenSSH 6.7, unsafe algorithm
+ `- [warn] disabled (in client) since OpenSSH 7.4, unsafe algorithm
+ `- [warn] using weak cipher
+ `- [warn] using weak cipher mode
+ `- [warn] using small 64-bit block size
+ `- [info] available since OpenSSH 1.2.2, Dropbear SSH 0.28
+(enc) blowfish-cbc -- [fail] removed (in server) since OpenSSH 6.7, unsafe algorithm
+ `- [fail] disabled since Dropbear SSH 0.53
+ `- [warn] disabled (in client) since OpenSSH 7.2, legacy algorithm
+ `- [warn] using weak cipher mode
+ `- [warn] using small 64-bit block size
+ `- [info] available since OpenSSH 1.2.2, Dropbear SSH 0.28
+(enc) cast128-cbc -- [fail] removed (in server) since OpenSSH 6.7, unsafe algorithm
+ `- [warn] disabled (in client) since OpenSSH 7.2, legacy algorithm
+ `- [warn] using weak cipher mode
+ `- [warn] using small 64-bit block size
+ `- [info] available since OpenSSH 2.1.0
+(enc) aes192-cbc -- [fail] removed (in server) since OpenSSH 6.7, unsafe algorithm
+ `- [warn] using weak cipher mode
+ `- [info] available since OpenSSH 2.3.0
+(enc) aes256-cbc -- [fail] removed (in server) since OpenSSH 6.7, unsafe algorithm
+ `- [warn] using weak cipher mode
+ `- [info] available since OpenSSH 2.3.0, Dropbear SSH 0.47
+(enc) arcfour -- [fail] removed (in server) since OpenSSH 6.7, unsafe algorithm
+ `- [warn] disabled (in client) since OpenSSH 7.2, legacy algorithm
+ `- [warn] using weak cipher
+ `- [info] available since OpenSSH 2.1.0
+(enc) rijndael-cbc@lysator.liu.se -- [fail] removed (in server) since OpenSSH 6.7, unsafe algorithm
+ `- [warn] disabled (in client) since OpenSSH 7.2, legacy algorithm
+ `- [warn] using weak cipher mode
+ `- [info] available since OpenSSH 2.3.0
+
+# message authentication code algorithms
+(mac) hmac-md5 -- [fail] removed (in server) since OpenSSH 6.7, unsafe algorithm
+ `- [warn] disabled (in client) since OpenSSH 7.2, legacy algorithm
+ `- [warn] using encrypt-and-MAC mode
+ `- [warn] using weak hashing algorithm
+ `- [info] available since OpenSSH 2.1.0, Dropbear SSH 0.28
+(mac) hmac-sha1 -- [warn] using encrypt-and-MAC mode
+ `- [warn] using weak hashing algorithm
+ `- [info] available since OpenSSH 2.1.0, Dropbear SSH 0.28
+(mac) umac-64@openssh.com -- [warn] using encrypt-and-MAC mode
+ `- [warn] using small 64-bit tag size
+ `- [info] available since OpenSSH 4.7
+(mac) hmac-ripemd160 -- [fail] removed (in server) since OpenSSH 6.7, unsafe algorithm
+ `- [warn] disabled (in client) since OpenSSH 7.2, legacy algorithm
+ `- [warn] using encrypt-and-MAC mode
+ `- [info] available since OpenSSH 2.5.0
+(mac) hmac-ripemd160@openssh.com -- [fail] removed (in server) since OpenSSH 6.7, unsafe algorithm
+ `- [warn] disabled (in client) since OpenSSH 7.2, legacy algorithm
+ `- [warn] using encrypt-and-MAC mode
+ `- [info] available since OpenSSH 2.1.0
+(mac) hmac-sha1-96 -- [fail] removed (in server) since OpenSSH 6.7, unsafe algorithm
+ `- [warn] disabled (in client) since OpenSSH 7.2, legacy algorithm
+ `- [warn] using encrypt-and-MAC mode
+ `- [warn] using weak hashing algorithm
+ `- [info] available since OpenSSH 2.5.0, Dropbear SSH 0.47
+(mac) hmac-md5-96 -- [fail] removed (in server) since OpenSSH 6.7, unsafe algorithm
+ `- [warn] disabled (in client) since OpenSSH 7.2, legacy algorithm
+ `- [warn] using encrypt-and-MAC mode
+ `- [warn] using weak hashing algorithm
+ `- [info] available since OpenSSH 2.5.0
+
+# fingerprints
+(fin) ssh-rsa: SHA256:YZ457EBcJTSxRKI3yXRgtAj3PBf5B9/F36b1SVooml4
+
+# algorithm recommendations (for OpenSSH 5.6)
+(rec) !diffie-hellman-group-exchange-sha256 -- kex algorithm to change (increase modulus size to 2048 bits or larger) 
+(rec) -3des-cbc -- enc algorithm to remove 
+(rec) -aes128-cbc -- enc algorithm to remove 
+(rec) -aes192-cbc -- enc algorithm to remove 
+(rec) -aes256-cbc -- enc algorithm to remove 
+(rec) -arcfour -- enc algorithm to remove 
+(rec) -arcfour128 -- enc algorithm to remove 
+(rec) -arcfour256 -- enc algorithm to remove 
+(rec) -blowfish-cbc -- enc algorithm to remove 
+(rec) -cast128-cbc -- enc algorithm to remove 
+(rec) -diffie-hellman-group-exchange-sha1 -- kex algorithm to remove 
+(rec) -diffie-hellman-group1-sha1 -- kex algorithm to remove 
+(rec) -hmac-md5 -- mac algorithm to remove 
+(rec) -hmac-md5-96 -- mac algorithm to remove 
+(rec) -hmac-ripemd160 -- mac algorithm to remove 
+(rec) -hmac-ripemd160@openssh.com -- mac algorithm to remove 
+(rec) -hmac-sha1-96 -- mac algorithm to remove 
+(rec) -rijndael-cbc@lysator.liu.se -- enc algorithm to remove 
+(rec) -ssh-dss -- key algorithm to remove 
+(rec) -ssh-rsa -- key algorithm to remove 
+(rec) -diffie-hellman-group14-sha1 -- kex algorithm to remove 
+(rec) -hmac-sha1 -- mac algorithm to remove 
+(rec) -umac-64@openssh.com -- mac algorithm to remove 
+
+# additional info
+(nfo) For hardening guides on common OSes, please see: <https://www.ssh-audit.com/hardening_guides.html>
+
diff --git a/test/docker/expected_results/openssh_5.6p1_test2.json b/test/docker/expected_results/openssh_5.6p1_test2.json
new file mode 100644
index 0000000..64d1590
--- /dev/null
+++ b/test/docker/expected_results/openssh_5.6p1_test2.json
@@ -0,0 +1 @@
+{"banner": {"comments": null, "protocol": [2, 0], "raw": "SSH-2.0-OpenSSH_5.6", "software": "OpenSSH_5.6"}, "compression": ["none", "zlib@openssh.com"], "enc": ["aes128-ctr", "aes192-ctr", "aes256-ctr", "arcfour256", "arcfour128", "aes128-cbc", "3des-cbc", "blowfish-cbc", "cast128-cbc", "aes192-cbc", "aes256-cbc", "arcfour", "rijndael-cbc@lysator.liu.se"], "fingerprints": [{"hash": "YZ457EBcJTSxRKI3yXRgtAj3PBf5B9/F36b1SVooml4", "hash_alg": "SHA256", "hostkey": "ssh-rsa"}, {"hash": "3c:c3:38:f8:55:39:c0:4a:5a:17:89:60:2c:a1:fc:6a", "hash_alg": "MD5", "hostkey": "ssh-rsa"}], "kex": [{"algorithm": "diffie-hellman-group-exchange-sha256", "keysize": 1024}, {"algorithm": "diffie-hellman-group-exchange-sha1", "keysize": 1024}, {"algorithm": "diffie-hellman-group14-sha1"}, {"algorithm": "diffie-hellman-group1-sha1"}], "key": [{"algorithm": "ssh-rsa", "keysize": 1024}, {"algorithm": "ssh-rsa-cert-v01@openssh.com", "casize": 1024, "keysize": 1024}], "mac": ["hmac-md5", "hmac-sha1", "umac-64@openssh.com", "hmac-ripemd160", "hmac-ripemd160@openssh.com", "hmac-sha1-96", "hmac-md5-96"], "target": "localhost"}
diff --git a/test/docker/expected_results/openssh_5.6p1_test2.txt b/test/docker/expected_results/openssh_5.6p1_test2.txt
new file mode 100644
index 0000000..2a6825f
--- /dev/null
+++ b/test/docker/expected_results/openssh_5.6p1_test2.txt
@@ -0,0 +1,153 @@
+# general
+(gen) banner: SSH-2.0-OpenSSH_5.6
+(gen) software: OpenSSH 5.6
+(gen) compatibility: OpenSSH 5.6-6.6, Dropbear SSH 0.53+ (some functionality from 0.52)
+(gen) compression: enabled (zlib@openssh.com)
+
+# security
+(cve) CVE-2018-15473 -- (CVSSv2: 5.3) enumerate usernames due to timing discrepencies
+(cve) CVE-2016-3115 -- (CVSSv2: 5.5) bypass command restrictions via crafted X11 forwarding data
+(cve) CVE-2016-1907 -- (CVSSv2: 5.0) cause DoS via crafted network traffic (out of bounds read)
+(cve) CVE-2015-6564 -- (CVSSv2: 6.9) privilege escalation via leveraging sshd uid
+(cve) CVE-2015-6563 -- (CVSSv2: 1.9) conduct impersonation attack
+(cve) CVE-2014-2532 -- (CVSSv2: 5.8) bypass environment restrictions via specific string before wildcard
+(cve) CVE-2014-1692 -- (CVSSv2: 7.5) cause DoS via triggering error condition (memory corruption)
+(cve) CVE-2012-0814 -- (CVSSv2: 3.5) leak data via debug messages
+(cve) CVE-2011-5000 -- (CVSSv2: 3.5) cause DoS via large value in certain length field (memory consumption)
+(cve) CVE-2010-5107 -- (CVSSv2: 5.0) cause DoS via large number of connections (slot exhaustion)
+(cve) CVE-2010-4755 -- (CVSSv2: 4.0) cause DoS via crafted glob expression (CPU and memory consumption)
+(cve) CVE-2010-4478 -- (CVSSv2: 7.5) bypass authentication check via crafted values
+
+# key exchange algorithms
+(kex) diffie-hellman-group-exchange-sha256 (1024-bit) -- [fail] using small 1024-bit modulus
+ `- [info] available since OpenSSH 4.4
+(kex) diffie-hellman-group-exchange-sha1 (1024-bit) -- [fail] using small 1024-bit modulus
+ `- [warn] using weak hashing algorithm
+ `- [info] available since OpenSSH 2.3.0
+(kex) diffie-hellman-group14-sha1 -- [warn] using weak hashing algorithm
+ `- [info] available since OpenSSH 3.9, Dropbear SSH 0.53
+(kex) diffie-hellman-group1-sha1 -- [fail] using small 1024-bit modulus
+ `- [fail] removed (in server) since OpenSSH 6.7, unsafe algorithm
+ `- [fail] disabled (in client) since OpenSSH 7.0, logjam attack
+ `- [warn] using weak hashing algorithm
+ `- [info] available since OpenSSH 2.3.0, Dropbear SSH 0.28
+
+# host-key algorithms
+(key) ssh-rsa (1024-bit) -- [fail] using weak hashing algorithm
+ `- [fail] using small 1024-bit modulus
+ `- [info] available since OpenSSH 2.5.0, Dropbear SSH 0.28
+ `- [info] a future deprecation notice has been issued in OpenSSH 8.2: https://www.openssh.com/txt/release-8.2
+(key) ssh-rsa-cert-v01@openssh.com (1024-bit cert/1024-bit CA) -- [fail] using weak hashing algorithm
+ `- [fail] using small 1024-bit modulus
+ `- [info] available since OpenSSH 5.6
+ `- [info] a future deprecation notice has been issued in OpenSSH 8.2: https://www.openssh.com/txt/release-8.2
+
+# encryption algorithms (ciphers)
+(enc) aes128-ctr -- [info] available since OpenSSH 3.7, Dropbear SSH 0.52
+(enc) aes192-ctr -- [info] available since OpenSSH 3.7
+(enc) aes256-ctr -- [info] available since OpenSSH 3.7, Dropbear SSH 0.52
+(enc) arcfour256 -- [fail] removed (in server) since OpenSSH 6.7, unsafe algorithm
+ `- [warn] disabled (in client) since OpenSSH 7.2, legacy algorithm
+ `- [warn] using weak cipher
+ `- [info] available since OpenSSH 4.2
+(enc) arcfour128 -- [fail] removed (in server) since OpenSSH 6.7, unsafe algorithm
+ `- [warn] disabled (in client) since OpenSSH 7.2, legacy algorithm
+ `- [warn] using weak cipher
+ `- [info] available since OpenSSH 4.2
+(enc) aes128-cbc -- [fail] removed (in server) since OpenSSH 6.7, unsafe algorithm
+ `- [warn] using weak cipher mode
+ `- [info] available since OpenSSH 2.3.0, Dropbear SSH 0.28
+(enc) 3des-cbc -- [fail] removed (in server) since OpenSSH 6.7, unsafe algorithm
+ `- [warn] disabled (in client) since OpenSSH 7.4, unsafe algorithm
+ `- [warn] using weak cipher
+ `- [warn] using weak cipher mode
+ `- [warn] using small 64-bit block size
+ `- [info] available since OpenSSH 1.2.2, Dropbear SSH 0.28
+(enc) blowfish-cbc -- [fail] removed (in server) since OpenSSH 6.7, unsafe algorithm
+ `- [fail] disabled since Dropbear SSH 0.53
+ `- [warn] disabled (in client) since OpenSSH 7.2, legacy algorithm
+ `- [warn] using weak cipher mode
+ `- [warn] using small 64-bit block size
+ `- [info] available since OpenSSH 1.2.2, Dropbear SSH 0.28
+(enc) cast128-cbc -- [fail] removed (in server) since OpenSSH 6.7, unsafe algorithm
+ `- [warn] disabled (in client) since OpenSSH 7.2, legacy algorithm
+ `- [warn] using weak cipher mode
+ `- [warn] using small 64-bit block size
+ `- [info] available since OpenSSH 2.1.0
+(enc) aes192-cbc -- [fail] removed (in server) since OpenSSH 6.7, unsafe algorithm
+ `- [warn] using weak cipher mode
+ `- [info] available since OpenSSH 2.3.0
+(enc) aes256-cbc -- [fail] removed (in server) since OpenSSH 6.7, unsafe algorithm
+ `- [warn] using weak cipher mode
+ `- [info] available since OpenSSH 2.3.0, Dropbear SSH 0.47
+(enc) arcfour -- [fail] removed (in server) since OpenSSH 6.7, unsafe algorithm
+ `- [warn] disabled (in client) since OpenSSH 7.2, legacy algorithm
+ `- [warn] using weak cipher
+ `- [info] available since OpenSSH 2.1.0
+(enc) rijndael-cbc@lysator.liu.se -- [fail] removed (in server) since OpenSSH 6.7, unsafe algorithm
+ `- [warn] disabled (in client) since OpenSSH 7.2, legacy algorithm
+ `- [warn] using weak cipher mode
+ `- [info] available since OpenSSH 2.3.0
+
+# message authentication code algorithms
+(mac) hmac-md5 -- [fail] removed (in server) since OpenSSH 6.7, unsafe algorithm
+ `- [warn] disabled (in client) since OpenSSH 7.2, legacy algorithm
+ `- [warn] using encrypt-and-MAC mode
+ `- [warn] using weak hashing algorithm
+ `- [info] available since OpenSSH 2.1.0, Dropbear SSH 0.28
+(mac) hmac-sha1 -- [warn] using encrypt-and-MAC mode
+ `- [warn] using weak hashing algorithm
+ `- [info] available since OpenSSH 2.1.0, Dropbear SSH 0.28
+(mac) umac-64@openssh.com -- [warn] using encrypt-and-MAC mode
+ `- [warn] using small 64-bit tag size
+ `- [info] available since OpenSSH 4.7
+(mac) hmac-ripemd160 -- [fail] removed (in server) since OpenSSH 6.7, unsafe algorithm
+ `- [warn] disabled (in client) since OpenSSH 7.2, legacy algorithm
+ `- [warn] using encrypt-and-MAC mode
+ `- [info] available since OpenSSH 2.5.0
+(mac) hmac-ripemd160@openssh.com -- [fail] removed (in server) since OpenSSH 6.7, unsafe algorithm
+ `- [warn] disabled (in client) since OpenSSH 7.2, legacy algorithm
+ `- [warn] using encrypt-and-MAC mode
+ `- [info] available since OpenSSH 2.1.0
+(mac) hmac-sha1-96 -- [fail] removed (in server) since OpenSSH 6.7, unsafe algorithm
+ `- [warn] disabled (in client) since OpenSSH 7.2, legacy algorithm
+ `- [warn] using encrypt-and-MAC mode
+ `- [warn] using weak hashing algorithm
+ `- [info] available since OpenSSH 2.5.0, Dropbear SSH 0.47
+(mac) hmac-md5-96 -- [fail] removed (in server) since OpenSSH 6.7, unsafe algorithm
+ `- [warn] disabled (in client) since OpenSSH 7.2, legacy algorithm
+ `- [warn] using encrypt-and-MAC mode
+ `- [warn] using weak hashing algorithm
+ `- [info] available since OpenSSH 2.5.0
+
+# fingerprints
+(fin) ssh-rsa: SHA256:YZ457EBcJTSxRKI3yXRgtAj3PBf5B9/F36b1SVooml4
+
+# algorithm recommendations (for OpenSSH 5.6)
+(rec) !diffie-hellman-group-exchange-sha256 -- kex algorithm to change (increase modulus size to 2048 bits or larger) 
+(rec) -3des-cbc -- enc algorithm to remove 
+(rec) -aes128-cbc -- enc algorithm to remove 
+(rec) -aes192-cbc -- enc algorithm to remove 
+(rec) -aes256-cbc -- enc algorithm to remove 
+(rec) -arcfour -- enc algorithm to remove 
+(rec) -arcfour128 -- enc algorithm to remove 
+(rec) -arcfour256 -- enc algorithm to remove 
+(rec) -blowfish-cbc -- enc algorithm to remove 
+(rec) -cast128-cbc -- enc algorithm to remove 
+(rec) -diffie-hellman-group-exchange-sha1 -- kex algorithm to remove 
+(rec) -diffie-hellman-group1-sha1 -- kex algorithm to remove 
+(rec) -hmac-md5 -- mac algorithm to remove 
+(rec) -hmac-md5-96 -- mac algorithm to remove 
+(rec) -hmac-ripemd160 -- mac algorithm to remove 
+(rec) -hmac-ripemd160@openssh.com -- mac algorithm to remove 
+(rec) -hmac-sha1-96 -- mac algorithm to remove 
+(rec) -rijndael-cbc@lysator.liu.se -- enc algorithm to remove 
+(rec) -ssh-rsa -- key algorithm to remove 
+(rec) -ssh-rsa-cert-v01@openssh.com -- key algorithm to remove 
+(rec) -diffie-hellman-group14-sha1 -- kex algorithm to remove 
+(rec) -hmac-sha1 -- mac algorithm to remove 
+(rec) -umac-64@openssh.com -- mac algorithm to remove 
+
+# additional info
+(nfo) For hardening guides on common OSes, please see: <https://www.ssh-audit.com/hardening_guides.html>
+
diff --git a/test/docker/expected_results/openssh_5.6p1_test3.json b/test/docker/expected_results/openssh_5.6p1_test3.json
new file mode 100644
index 0000000..db22eed
--- /dev/null
+++ b/test/docker/expected_results/openssh_5.6p1_test3.json
@@ -0,0 +1 @@
+{"banner": {"comments": null, "protocol": [2, 0], "raw": "SSH-2.0-OpenSSH_5.6", "software": "OpenSSH_5.6"}, "compression": ["none", "zlib@openssh.com"], "enc": ["aes128-ctr", "aes192-ctr", "aes256-ctr", "arcfour256", "arcfour128", "aes128-cbc", "3des-cbc", "blowfish-cbc", "cast128-cbc", "aes192-cbc", "aes256-cbc", "arcfour", "rijndael-cbc@lysator.liu.se"], "fingerprints": [{"hash": "YZ457EBcJTSxRKI3yXRgtAj3PBf5B9/F36b1SVooml4", "hash_alg": "SHA256", "hostkey": "ssh-rsa"}, {"hash": "3c:c3:38:f8:55:39:c0:4a:5a:17:89:60:2c:a1:fc:6a", "hash_alg": "MD5", "hostkey": "ssh-rsa"}], "kex": [{"algorithm": "diffie-hellman-group-exchange-sha256", "keysize": 1024}, {"algorithm": "diffie-hellman-group-exchange-sha1", "keysize": 1024}, {"algorithm": "diffie-hellman-group14-sha1"}, {"algorithm": "diffie-hellman-group1-sha1"}], "key": [{"algorithm": "ssh-rsa", "keysize": 1024}, {"algorithm": "ssh-rsa-cert-v01@openssh.com", "casize": 3072, "keysize": 1024}], "mac": ["hmac-md5", "hmac-sha1", "umac-64@openssh.com", "hmac-ripemd160", "hmac-ripemd160@openssh.com", "hmac-sha1-96", "hmac-md5-96"], "target": "localhost"}
diff --git a/test/docker/expected_results/openssh_5.6p1_test3.txt b/test/docker/expected_results/openssh_5.6p1_test3.txt
new file mode 100644
index 0000000..657b22e
--- /dev/null
+++ b/test/docker/expected_results/openssh_5.6p1_test3.txt
@@ -0,0 +1,153 @@
+# general
+(gen) banner: SSH-2.0-OpenSSH_5.6
+(gen) software: OpenSSH 5.6
+(gen) compatibility: OpenSSH 5.6-6.6, Dropbear SSH 0.53+ (some functionality from 0.52)
+(gen) compression: enabled (zlib@openssh.com)
+
+# security
+(cve) CVE-2018-15473 -- (CVSSv2: 5.3) enumerate usernames due to timing discrepencies
+(cve) CVE-2016-3115 -- (CVSSv2: 5.5) bypass command restrictions via crafted X11 forwarding data
+(cve) CVE-2016-1907 -- (CVSSv2: 5.0) cause DoS via crafted network traffic (out of bounds read)
+(cve) CVE-2015-6564 -- (CVSSv2: 6.9) privilege escalation via leveraging sshd uid
+(cve) CVE-2015-6563 -- (CVSSv2: 1.9) conduct impersonation attack
+(cve) CVE-2014-2532 -- (CVSSv2: 5.8) bypass environment restrictions via specific string before wildcard
+(cve) CVE-2014-1692 -- (CVSSv2: 7.5) cause DoS via triggering error condition (memory corruption)
+(cve) CVE-2012-0814 -- (CVSSv2: 3.5) leak data via debug messages
+(cve) CVE-2011-5000 -- (CVSSv2: 3.5) cause DoS via large value in certain length field (memory consumption)
+(cve) CVE-2010-5107 -- (CVSSv2: 5.0) cause DoS via large number of connections (slot exhaustion)
+(cve) CVE-2010-4755 -- (CVSSv2: 4.0) cause DoS via crafted glob expression (CPU and memory consumption)
+(cve) CVE-2010-4478 -- (CVSSv2: 7.5) bypass authentication check via crafted values
+
+# key exchange algorithms
+(kex) diffie-hellman-group-exchange-sha256 (1024-bit) -- [fail] using small 1024-bit modulus
+ `- [info] available since OpenSSH 4.4
+(kex) diffie-hellman-group-exchange-sha1 (1024-bit) -- [fail] using small 1024-bit modulus
+ `- [warn] using weak hashing algorithm
+ `- [info] available since OpenSSH 2.3.0
+(kex) diffie-hellman-group14-sha1 -- [warn] using weak hashing algorithm
+ `- [info] available since OpenSSH 3.9, Dropbear SSH 0.53
+(kex) diffie-hellman-group1-sha1 -- [fail] using small 1024-bit modulus
+ `- [fail] removed (in server) since OpenSSH 6.7, unsafe algorithm
+ `- [fail] disabled (in client) since OpenSSH 7.0, logjam attack
+ `- [warn] using weak hashing algorithm
+ `- [info] available since OpenSSH 2.3.0, Dropbear SSH 0.28
+
+# host-key algorithms
+(key) ssh-rsa (1024-bit) -- [fail] using weak hashing algorithm
+ `- [fail] using small 1024-bit modulus
+ `- [info] available since OpenSSH 2.5.0, Dropbear SSH 0.28
+ `- [info] a future deprecation notice has been issued in OpenSSH 8.2: https://www.openssh.com/txt/release-8.2
+(key) ssh-rsa-cert-v01@openssh.com (1024-bit cert/3072-bit CA) -- [fail] using weak hashing algorithm
+ `- [fail] using small 1024-bit modulus
+ `- [info] available since OpenSSH 5.6
+ `- [info] a future deprecation notice has been issued in OpenSSH 8.2: https://www.openssh.com/txt/release-8.2
+
+# encryption algorithms (ciphers)
+(enc) aes128-ctr -- [info] available since OpenSSH 3.7, Dropbear SSH 0.52
+(enc) aes192-ctr -- [info] available since OpenSSH 3.7
+(enc) aes256-ctr -- [info] available since OpenSSH 3.7, Dropbear SSH 0.52
+(enc) arcfour256 -- [fail] removed (in server) since OpenSSH 6.7, unsafe algorithm
+ `- [warn] disabled (in client) since OpenSSH 7.2, legacy algorithm
+ `- [warn] using weak cipher
+ `- [info] available since OpenSSH 4.2
+(enc) arcfour128 -- [fail] removed (in server) since OpenSSH 6.7, unsafe algorithm
+ `- [warn] disabled (in client) since OpenSSH 7.2, legacy algorithm
+ `- [warn] using weak cipher
+ `- [info] available since OpenSSH 4.2
+(enc) aes128-cbc -- [fail] removed (in server) since OpenSSH 6.7, unsafe algorithm
+ `- [warn] using weak cipher mode
+ `- [info] available since OpenSSH 2.3.0, Dropbear SSH 0.28
+(enc) 3des-cbc -- [fail] removed (in server) since OpenSSH 6.7, unsafe algorithm
+ `- [warn] disabled (in client) since OpenSSH 7.4, unsafe algorithm
+ `- [warn] using weak cipher
+ `- [warn] using weak cipher mode
+ `- [warn] using small 64-bit block size
+ `- [info] available since OpenSSH 1.2.2, Dropbear SSH 0.28
+(enc) blowfish-cbc -- [fail] removed (in server) since OpenSSH 6.7, unsafe algorithm
+ `- [fail] disabled since Dropbear SSH 0.53
+ `- [warn] disabled (in client) since OpenSSH 7.2, legacy algorithm
+ `- [warn] using weak cipher mode
+ `- [warn] using small 64-bit block size
+ `- [info] available since OpenSSH 1.2.2, Dropbear SSH 0.28
+(enc) cast128-cbc -- [fail] removed (in server) since OpenSSH 6.7, unsafe algorithm
+ `- [warn] disabled (in client) since OpenSSH 7.2, legacy algorithm
+ `- [warn] using weak cipher mode
+ `- [warn] using small 64-bit block size
+ `- [info] available since OpenSSH 2.1.0
+(enc) aes192-cbc -- [fail] removed (in server) since OpenSSH 6.7, unsafe algorithm
+ `- [warn] using weak cipher mode
+ `- [info] available since OpenSSH 2.3.0
+(enc) aes256-cbc -- [fail] removed (in server) since OpenSSH 6.7, unsafe algorithm
+ `- [warn] using weak cipher mode
+ `- [info] available since OpenSSH 2.3.0, Dropbear SSH 0.47
+(enc) arcfour -- [fail] removed (in server) since OpenSSH 6.7, unsafe algorithm
+ `- [warn] disabled (in client) since OpenSSH 7.2, legacy algorithm
+ `- [warn] using weak cipher
+ `- [info] available since OpenSSH 2.1.0
+(enc) rijndael-cbc@lysator.liu.se -- [fail] removed (in server) since OpenSSH 6.7, unsafe algorithm
+ `- [warn] disabled (in client) since OpenSSH 7.2, legacy algorithm
+ `- [warn] using weak cipher mode
+ `- [info] available since OpenSSH 2.3.0
+
+# message authentication code algorithms
+(mac) hmac-md5 -- [fail] removed (in server) since OpenSSH 6.7, unsafe algorithm
+ `- [warn] disabled (in client) since OpenSSH 7.2, legacy algorithm
+ `- [warn] using encrypt-and-MAC mode
+ `- [warn] using weak hashing algorithm
+ `- [info] available since OpenSSH 2.1.0, Dropbear SSH 0.28
+(mac) hmac-sha1 -- [warn] using encrypt-and-MAC mode
+ `- [warn] using weak hashing algorithm
+ `- [info] available since OpenSSH 2.1.0, Dropbear SSH 0.28
+(mac) umac-64@openssh.com -- [warn] using encrypt-and-MAC mode
+ `- [warn] using small 64-bit tag size
+ `- [info] available since OpenSSH 4.7
+(mac) hmac-ripemd160 -- [fail] removed (in server) since OpenSSH 6.7, unsafe algorithm
+ `- [warn] disabled (in client) since OpenSSH 7.2, legacy algorithm
+ `- [warn] using encrypt-and-MAC mode
+ `- [info] available since OpenSSH 2.5.0
+(mac) hmac-ripemd160@openssh.com -- [fail] removed (in server) since OpenSSH 6.7, unsafe algorithm
+ `- [warn] disabled (in client) since OpenSSH 7.2, legacy algorithm
+ `- [warn] using encrypt-and-MAC mode
+ `- [info] available since OpenSSH 2.1.0
+(mac) hmac-sha1-96 -- [fail] removed (in server) since OpenSSH 6.7, unsafe algorithm
+ `- [warn] disabled (in client) since OpenSSH 7.2, legacy algorithm
+ `- [warn] using encrypt-and-MAC mode
+ `- [warn] using weak hashing algorithm
+ `- [info] available since OpenSSH 2.5.0, Dropbear SSH 0.47
+(mac) hmac-md5-96 -- [fail] removed (in server) since OpenSSH 6.7, unsafe algorithm
+ `- [warn] disabled (in client) since OpenSSH 7.2, legacy algorithm
+ `- [warn] using encrypt-and-MAC mode
+ `- [warn] using weak hashing algorithm
+ `- [info] available since OpenSSH 2.5.0
+
+# fingerprints
+(fin) ssh-rsa: SHA256:YZ457EBcJTSxRKI3yXRgtAj3PBf5B9/F36b1SVooml4
+
+# algorithm recommendations (for OpenSSH 5.6)
+(rec) !diffie-hellman-group-exchange-sha256 -- kex algorithm to change (increase modulus size to 2048 bits or larger) 
+(rec) -3des-cbc -- enc algorithm to remove 
+(rec) -aes128-cbc -- enc algorithm to remove 
+(rec) -aes192-cbc -- enc algorithm to remove 
+(rec) -aes256-cbc -- enc algorithm to remove 
+(rec) -arcfour -- enc algorithm to remove 
+(rec) -arcfour128 -- enc algorithm to remove 
+(rec) -arcfour256 -- enc algorithm to remove 
+(rec) -blowfish-cbc -- enc algorithm to remove 
+(rec) -cast128-cbc -- enc algorithm to remove 
+(rec) -diffie-hellman-group-exchange-sha1 -- kex algorithm to remove 
+(rec) -diffie-hellman-group1-sha1 -- kex algorithm to remove 
+(rec) -hmac-md5 -- mac algorithm to remove 
+(rec) -hmac-md5-96 -- mac algorithm to remove 
+(rec) -hmac-ripemd160 -- mac algorithm to remove 
+(rec) -hmac-ripemd160@openssh.com -- mac algorithm to remove 
+(rec) -hmac-sha1-96 -- mac algorithm to remove 
+(rec) -rijndael-cbc@lysator.liu.se -- enc algorithm to remove 
+(rec) -ssh-rsa -- key algorithm to remove 
+(rec) -ssh-rsa-cert-v01@openssh.com -- key algorithm to remove 
+(rec) -diffie-hellman-group14-sha1 -- kex algorithm to remove 
+(rec) -hmac-sha1 -- mac algorithm to remove 
+(rec) -umac-64@openssh.com -- mac algorithm to remove 
+
+# additional info
+(nfo) For hardening guides on common OSes, please see: <https://www.ssh-audit.com/hardening_guides.html>
+
diff --git a/test/docker/expected_results/openssh_5.6p1_test4.json b/test/docker/expected_results/openssh_5.6p1_test4.json
new file mode 100644
index 0000000..a5c725d
--- /dev/null
+++ b/test/docker/expected_results/openssh_5.6p1_test4.json
@@ -0,0 +1 @@
+{"banner": {"comments": null, "protocol": [2, 0], "raw": "SSH-2.0-OpenSSH_5.6", "software": "OpenSSH_5.6"}, "compression": ["none", "zlib@openssh.com"], "enc": ["aes128-ctr", "aes192-ctr", "aes256-ctr", "arcfour256", "arcfour128", "aes128-cbc", "3des-cbc", "blowfish-cbc", "cast128-cbc", "aes192-cbc", "aes256-cbc", "arcfour", "rijndael-cbc@lysator.liu.se"], "fingerprints": [{"hash": "nsWtdJ9Z67Vrf7OsUzQov7esXhsWAfVppArGh25u244", "hash_alg": "SHA256", "hostkey": "ssh-rsa"}, {"hash": "18:e2:51:fe:21:6c:78:d0:b8:cf:32:d4:bd:56:42:e1", "hash_alg": "MD5", "hostkey": "ssh-rsa"}], "kex": [{"algorithm": "diffie-hellman-group-exchange-sha256", "keysize": 1024}, {"algorithm": "diffie-hellman-group-exchange-sha1", "keysize": 1024}, {"algorithm": "diffie-hellman-group14-sha1"}, {"algorithm": "diffie-hellman-group1-sha1"}], "key": [{"algorithm": "ssh-rsa", "keysize": 3072}, {"algorithm": "ssh-rsa-cert-v01@openssh.com", "casize": 1024, "keysize": 3072}], "mac": ["hmac-md5", "hmac-sha1", "umac-64@openssh.com", "hmac-ripemd160", "hmac-ripemd160@openssh.com", "hmac-sha1-96", "hmac-md5-96"], "target": "localhost"}
diff --git a/test/docker/expected_results/openssh_5.6p1_test4.txt b/test/docker/expected_results/openssh_5.6p1_test4.txt
new file mode 100644
index 0000000..fb48d1c
--- /dev/null
+++ b/test/docker/expected_results/openssh_5.6p1_test4.txt
@@ -0,0 +1,152 @@
+# general
+(gen) banner: SSH-2.0-OpenSSH_5.6
+(gen) software: OpenSSH 5.6
+(gen) compatibility: OpenSSH 5.6-6.6, Dropbear SSH 0.53+ (some functionality from 0.52)
+(gen) compression: enabled (zlib@openssh.com)
+
+# security
+(cve) CVE-2018-15473 -- (CVSSv2: 5.3) enumerate usernames due to timing discrepencies
+(cve) CVE-2016-3115 -- (CVSSv2: 5.5) bypass command restrictions via crafted X11 forwarding data
+(cve) CVE-2016-1907 -- (CVSSv2: 5.0) cause DoS via crafted network traffic (out of bounds read)
+(cve) CVE-2015-6564 -- (CVSSv2: 6.9) privilege escalation via leveraging sshd uid
+(cve) CVE-2015-6563 -- (CVSSv2: 1.9) conduct impersonation attack
+(cve) CVE-2014-2532 -- (CVSSv2: 5.8) bypass environment restrictions via specific string before wildcard
+(cve) CVE-2014-1692 -- (CVSSv2: 7.5) cause DoS via triggering error condition (memory corruption)
+(cve) CVE-2012-0814 -- (CVSSv2: 3.5) leak data via debug messages
+(cve) CVE-2011-5000 -- (CVSSv2: 3.5) cause DoS via large value in certain length field (memory consumption)
+(cve) CVE-2010-5107 -- (CVSSv2: 5.0) cause DoS via large number of connections (slot exhaustion)
+(cve) CVE-2010-4755 -- (CVSSv2: 4.0) cause DoS via crafted glob expression (CPU and memory consumption)
+(cve) CVE-2010-4478 -- (CVSSv2: 7.5) bypass authentication check via crafted values
+
+# key exchange algorithms
+(kex) diffie-hellman-group-exchange-sha256 (1024-bit) -- [fail] using small 1024-bit modulus
+ `- [info] available since OpenSSH 4.4
+(kex) diffie-hellman-group-exchange-sha1 (1024-bit) -- [fail] using small 1024-bit modulus
+ `- [warn] using weak hashing algorithm
+ `- [info] available since OpenSSH 2.3.0
+(kex) diffie-hellman-group14-sha1 -- [warn] using weak hashing algorithm
+ `- [info] available since OpenSSH 3.9, Dropbear SSH 0.53
+(kex) diffie-hellman-group1-sha1 -- [fail] using small 1024-bit modulus
+ `- [fail] removed (in server) since OpenSSH 6.7, unsafe algorithm
+ `- [fail] disabled (in client) since OpenSSH 7.0, logjam attack
+ `- [warn] using weak hashing algorithm
+ `- [info] available since OpenSSH 2.3.0, Dropbear SSH 0.28
+
+# host-key algorithms
+(key) ssh-rsa (3072-bit) -- [fail] using weak hashing algorithm
+ `- [info] available since OpenSSH 2.5.0, Dropbear SSH 0.28
+ `- [info] a future deprecation notice has been issued in OpenSSH 8.2: https://www.openssh.com/txt/release-8.2
+(key) ssh-rsa-cert-v01@openssh.com (3072-bit cert/1024-bit CA) -- [fail] using weak hashing algorithm
+ `- [fail] using small 1024-bit modulus
+ `- [info] available since OpenSSH 5.6
+ `- [info] a future deprecation notice has been issued in OpenSSH 8.2: https://www.openssh.com/txt/release-8.2
+
+# encryption algorithms (ciphers)
+(enc) aes128-ctr -- [info] available since OpenSSH 3.7, Dropbear SSH 0.52
+(enc) aes192-ctr -- [info] available since OpenSSH 3.7
+(enc) aes256-ctr -- [info] available since OpenSSH 3.7, Dropbear SSH 0.52
+(enc) arcfour256 -- [fail] removed (in server) since OpenSSH 6.7, unsafe algorithm
+ `- [warn] disabled (in client) since OpenSSH 7.2, legacy algorithm
+ `- [warn] using weak cipher
+ `- [info] available since OpenSSH 4.2
+(enc) arcfour128 -- [fail] removed (in server) since OpenSSH 6.7, unsafe algorithm
+ `- [warn] disabled (in client) since OpenSSH 7.2, legacy algorithm
+ `- [warn] using weak cipher
+ `- [info] available since OpenSSH 4.2
+(enc) aes128-cbc -- [fail] removed (in server) since OpenSSH 6.7, unsafe algorithm
+ `- [warn] using weak cipher mode
+ `- [info] available since OpenSSH 2.3.0, Dropbear SSH 0.28
+(enc) 3des-cbc -- [fail] removed (in server) since OpenSSH 6.7, unsafe algorithm
+ `- [warn] disabled (in client) since OpenSSH 7.4, unsafe algorithm
+ `- [warn] using weak cipher
+ `- [warn] using weak cipher mode
+ `- [warn] using small 64-bit block size
+ `- [info] available since OpenSSH 1.2.2, Dropbear SSH 0.28
+(enc) blowfish-cbc -- [fail] removed (in server) since OpenSSH 6.7, unsafe algorithm
+ `- [fail] disabled since Dropbear SSH 0.53
+ `- [warn] disabled (in client) since OpenSSH 7.2, legacy algorithm
+ `- [warn] using weak cipher mode
+ `- [warn] using small 64-bit block size
+ `- [info] available since OpenSSH 1.2.2, Dropbear SSH 0.28
+(enc) cast128-cbc -- [fail] removed (in server) since OpenSSH 6.7, unsafe algorithm
+ `- [warn] disabled (in client) since OpenSSH 7.2, legacy algorithm
+ `- [warn] using weak cipher mode
+ `- [warn] using small 64-bit block size
+ `- [info] available since OpenSSH 2.1.0
+(enc) aes192-cbc -- [fail] removed (in server) since OpenSSH 6.7, unsafe algorithm
+ `- [warn] using weak cipher mode
+ `- [info] available since OpenSSH 2.3.0
+(enc) aes256-cbc -- [fail] removed (in server) since OpenSSH 6.7, unsafe algorithm
+ `- [warn] using weak cipher mode
+ `- [info] available since OpenSSH 2.3.0, Dropbear SSH 0.47
+(enc) arcfour -- [fail] removed (in server) since OpenSSH 6.7, unsafe algorithm
+ `- [warn] disabled (in client) since OpenSSH 7.2, legacy algorithm
+ `- [warn] using weak cipher
+ `- [info] available since OpenSSH 2.1.0
+(enc) rijndael-cbc@lysator.liu.se -- [fail] removed (in server) since OpenSSH 6.7, unsafe algorithm
+ `- [warn] disabled (in client) since OpenSSH 7.2, legacy algorithm
+ `- [warn] using weak cipher mode
+ `- [info] available since OpenSSH 2.3.0
+
+# message authentication code algorithms
+(mac) hmac-md5 -- [fail] removed (in server) since OpenSSH 6.7, unsafe algorithm
+ `- [warn] disabled (in client) since OpenSSH 7.2, legacy algorithm
+ `- [warn] using encrypt-and-MAC mode
+ `- [warn] using weak hashing algorithm
+ `- [info] available since OpenSSH 2.1.0, Dropbear SSH 0.28
+(mac) hmac-sha1 -- [warn] using encrypt-and-MAC mode
+ `- [warn] using weak hashing algorithm
+ `- [info] available since OpenSSH 2.1.0, Dropbear SSH 0.28
+(mac) umac-64@openssh.com -- [warn] using encrypt-and-MAC mode
+ `- [warn] using small 64-bit tag size
+ `- [info] available since OpenSSH 4.7
+(mac) hmac-ripemd160 -- [fail] removed (in server) since OpenSSH 6.7, unsafe algorithm
+ `- [warn] disabled (in client) since OpenSSH 7.2, legacy algorithm
+ `- [warn] using encrypt-and-MAC mode
+ `- [info] available since OpenSSH 2.5.0
+(mac) hmac-ripemd160@openssh.com -- [fail] removed (in server) since OpenSSH 6.7, unsafe algorithm
+ `- [warn] disabled (in client) since OpenSSH 7.2, legacy algorithm
+ `- [warn] using encrypt-and-MAC mode
+ `- [info] available since OpenSSH 2.1.0
+(mac) hmac-sha1-96 -- [fail] removed (in server) since OpenSSH 6.7, unsafe algorithm
+ `- [warn] disabled (in client) since OpenSSH 7.2, legacy algorithm
+ `- [warn] using encrypt-and-MAC mode
+ `- [warn] using weak hashing algorithm
+ `- [info] available since OpenSSH 2.5.0, Dropbear SSH 0.47
+(mac) hmac-md5-96 -- [fail] removed (in server) since OpenSSH 6.7, unsafe algorithm
+ `- [warn] disabled (in client) since OpenSSH 7.2, legacy algorithm
+ `- [warn] using encrypt-and-MAC mode
+ `- [warn] using weak hashing algorithm
+ `- [info] available since OpenSSH 2.5.0
+
+# fingerprints
+(fin) ssh-rsa: SHA256:nsWtdJ9Z67Vrf7OsUzQov7esXhsWAfVppArGh25u244
+
+# algorithm recommendations (for OpenSSH 5.6)
+(rec) !diffie-hellman-group-exchange-sha256 -- kex algorithm to change (increase modulus size to 2048 bits or larger) 
+(rec) -3des-cbc -- enc algorithm to remove 
+(rec) -aes128-cbc -- enc algorithm to remove 
+(rec) -aes192-cbc -- enc algorithm to remove 
+(rec) -aes256-cbc -- enc algorithm to remove 
+(rec) -arcfour -- enc algorithm to remove 
+(rec) -arcfour128 -- enc algorithm to remove 
+(rec) -arcfour256 -- enc algorithm to remove 
+(rec) -blowfish-cbc -- enc algorithm to remove 
+(rec) -cast128-cbc -- enc algorithm to remove 
+(rec) -diffie-hellman-group-exchange-sha1 -- kex algorithm to remove 
+(rec) -diffie-hellman-group1-sha1 -- kex algorithm to remove 
+(rec) -hmac-md5 -- mac algorithm to remove 
+(rec) -hmac-md5-96 -- mac algorithm to remove 
+(rec) -hmac-ripemd160 -- mac algorithm to remove 
+(rec) -hmac-ripemd160@openssh.com -- mac algorithm to remove 
+(rec) -hmac-sha1-96 -- mac algorithm to remove 
+(rec) -rijndael-cbc@lysator.liu.se -- enc algorithm to remove 
+(rec) -ssh-rsa -- key algorithm to remove 
+(rec) -ssh-rsa-cert-v01@openssh.com -- key algorithm to remove 
+(rec) -diffie-hellman-group14-sha1 -- kex algorithm to remove 
+(rec) -hmac-sha1 -- mac algorithm to remove 
+(rec) -umac-64@openssh.com -- mac algorithm to remove 
+
+# additional info
+(nfo) For hardening guides on common OSes, please see: <https://www.ssh-audit.com/hardening_guides.html>
+
diff --git a/test/docker/expected_results/openssh_5.6p1_test5.json b/test/docker/expected_results/openssh_5.6p1_test5.json
new file mode 100644
index 0000000..24523b8
--- /dev/null
+++ b/test/docker/expected_results/openssh_5.6p1_test5.json
@@ -0,0 +1 @@
+{"banner": {"comments": null, "protocol": [2, 0], "raw": "SSH-2.0-OpenSSH_5.6", "software": "OpenSSH_5.6"}, "compression": ["none", "zlib@openssh.com"], "enc": ["aes128-ctr", "aes192-ctr", "aes256-ctr", "arcfour256", "arcfour128", "aes128-cbc", "3des-cbc", "blowfish-cbc", "cast128-cbc", "aes192-cbc", "aes256-cbc", "arcfour", "rijndael-cbc@lysator.liu.se"], "fingerprints": [{"hash": "nsWtdJ9Z67Vrf7OsUzQov7esXhsWAfVppArGh25u244", "hash_alg": "SHA256", "hostkey": "ssh-rsa"}, {"hash": "18:e2:51:fe:21:6c:78:d0:b8:cf:32:d4:bd:56:42:e1", "hash_alg": "MD5", "hostkey": "ssh-rsa"}], "kex": [{"algorithm": "diffie-hellman-group-exchange-sha256", "keysize": 1024}, {"algorithm": "diffie-hellman-group-exchange-sha1", "keysize": 1024}, {"algorithm": "diffie-hellman-group14-sha1"}, {"algorithm": "diffie-hellman-group1-sha1"}], "key": [{"algorithm": "ssh-rsa", "keysize": 3072}, {"algorithm": "ssh-rsa-cert-v01@openssh.com", "casize": 3072, "keysize": 3072}], "mac": ["hmac-md5", "hmac-sha1", "umac-64@openssh.com", "hmac-ripemd160", "hmac-ripemd160@openssh.com", "hmac-sha1-96", "hmac-md5-96"], "target": "localhost"}
diff --git a/test/docker/expected_results/openssh_5.6p1_test5.txt b/test/docker/expected_results/openssh_5.6p1_test5.txt
new file mode 100644
index 0000000..5ff250a
--- /dev/null
+++ b/test/docker/expected_results/openssh_5.6p1_test5.txt
@@ -0,0 +1,151 @@
+# general
+(gen) banner: SSH-2.0-OpenSSH_5.6
+(gen) software: OpenSSH 5.6
+(gen) compatibility: OpenSSH 5.6-6.6, Dropbear SSH 0.53+ (some functionality from 0.52)
+(gen) compression: enabled (zlib@openssh.com)
+
+# security
+(cve) CVE-2018-15473 -- (CVSSv2: 5.3) enumerate usernames due to timing discrepencies
+(cve) CVE-2016-3115 -- (CVSSv2: 5.5) bypass command restrictions via crafted X11 forwarding data
+(cve) CVE-2016-1907 -- (CVSSv2: 5.0) cause DoS via crafted network traffic (out of bounds read)
+(cve) CVE-2015-6564 -- (CVSSv2: 6.9) privilege escalation via leveraging sshd uid
+(cve) CVE-2015-6563 -- (CVSSv2: 1.9) conduct impersonation attack
+(cve) CVE-2014-2532 -- (CVSSv2: 5.8) bypass environment restrictions via specific string before wildcard
+(cve) CVE-2014-1692 -- (CVSSv2: 7.5) cause DoS via triggering error condition (memory corruption)
+(cve) CVE-2012-0814 -- (CVSSv2: 3.5) leak data via debug messages
+(cve) CVE-2011-5000 -- (CVSSv2: 3.5) cause DoS via large value in certain length field (memory consumption)
+(cve) CVE-2010-5107 -- (CVSSv2: 5.0) cause DoS via large number of connections (slot exhaustion)
+(cve) CVE-2010-4755 -- (CVSSv2: 4.0) cause DoS via crafted glob expression (CPU and memory consumption)
+(cve) CVE-2010-4478 -- (CVSSv2: 7.5) bypass authentication check via crafted values
+
+# key exchange algorithms
+(kex) diffie-hellman-group-exchange-sha256 (1024-bit) -- [fail] using small 1024-bit modulus
+ `- [info] available since OpenSSH 4.4
+(kex) diffie-hellman-group-exchange-sha1 (1024-bit) -- [fail] using small 1024-bit modulus
+ `- [warn] using weak hashing algorithm
+ `- [info] available since OpenSSH 2.3.0
+(kex) diffie-hellman-group14-sha1 -- [warn] using weak hashing algorithm
+ `- [info] available since OpenSSH 3.9, Dropbear SSH 0.53
+(kex) diffie-hellman-group1-sha1 -- [fail] using small 1024-bit modulus
+ `- [fail] removed (in server) since OpenSSH 6.7, unsafe algorithm
+ `- [fail] disabled (in client) since OpenSSH 7.0, logjam attack
+ `- [warn] using weak hashing algorithm
+ `- [info] available since OpenSSH 2.3.0, Dropbear SSH 0.28
+
+# host-key algorithms
+(key) ssh-rsa (3072-bit) -- [fail] using weak hashing algorithm
+ `- [info] available since OpenSSH 2.5.0, Dropbear SSH 0.28
+ `- [info] a future deprecation notice has been issued in OpenSSH 8.2: https://www.openssh.com/txt/release-8.2
+(key) ssh-rsa-cert-v01@openssh.com (3072-bit cert/3072-bit CA) -- [fail] using weak hashing algorithm
+ `- [info] available since OpenSSH 5.6
+ `- [info] a future deprecation notice has been issued in OpenSSH 8.2: https://www.openssh.com/txt/release-8.2
+
+# encryption algorithms (ciphers)
+(enc) aes128-ctr -- [info] available since OpenSSH 3.7, Dropbear SSH 0.52
+(enc) aes192-ctr -- [info] available since OpenSSH 3.7
+(enc) aes256-ctr -- [info] available since OpenSSH 3.7, Dropbear SSH 0.52
+(enc) arcfour256 -- [fail] removed (in server) since OpenSSH 6.7, unsafe algorithm
+ `- [warn] disabled (in client) since OpenSSH 7.2, legacy algorithm
+ `- [warn] using weak cipher
+ `- [info] available since OpenSSH 4.2
+(enc) arcfour128 -- [fail] removed (in server) since OpenSSH 6.7, unsafe algorithm
+ `- [warn] disabled (in client) since OpenSSH 7.2, legacy algorithm
+ `- [warn] using weak cipher
+ `- [info] available since OpenSSH 4.2
+(enc) aes128-cbc -- [fail] removed (in server) since OpenSSH 6.7, unsafe algorithm
+ `- [warn] using weak cipher mode
+ `- [info] available since OpenSSH 2.3.0, Dropbear SSH 0.28
+(enc) 3des-cbc -- [fail] removed (in server) since OpenSSH 6.7, unsafe algorithm
+ `- [warn] disabled (in client) since OpenSSH 7.4, unsafe algorithm
+ `- [warn] using weak cipher
+ `- [warn] using weak cipher mode
+ `- [warn] using small 64-bit block size
+ `- [info] available since OpenSSH 1.2.2, Dropbear SSH 0.28
+(enc) blowfish-cbc -- [fail] removed (in server) since OpenSSH 6.7, unsafe algorithm
+ `- [fail] disabled since Dropbear SSH 0.53
+ `- [warn] disabled (in client) since OpenSSH 7.2, legacy algorithm
+ `- [warn] using weak cipher mode
+ `- [warn] using small 64-bit block size
+ `- [info] available since OpenSSH 1.2.2, Dropbear SSH 0.28
+(enc) cast128-cbc -- [fail] removed (in server) since OpenSSH 6.7, unsafe algorithm
+ `- [warn] disabled (in client) since OpenSSH 7.2, legacy algorithm
+ `- [warn] using weak cipher mode
+ `- [warn] using small 64-bit block size
+ `- [info] available since OpenSSH 2.1.0
+(enc) aes192-cbc -- [fail] removed (in server) since OpenSSH 6.7, unsafe algorithm
+ `- [warn] using weak cipher mode
+ `- [info] available since OpenSSH 2.3.0
+(enc) aes256-cbc -- [fail] removed (in server) since OpenSSH 6.7, unsafe algorithm
+ `- [warn] using weak cipher mode
+ `- [info] available since OpenSSH 2.3.0, Dropbear SSH 0.47
+(enc) arcfour -- [fail] removed (in server) since OpenSSH 6.7, unsafe algorithm
+ `- [warn] disabled (in client) since OpenSSH 7.2, legacy algorithm
+ `- [warn] using weak cipher
+ `- [info] available since OpenSSH 2.1.0
+(enc) rijndael-cbc@lysator.liu.se -- [fail] removed (in server) since OpenSSH 6.7, unsafe algorithm
+ `- [warn] disabled (in client) since OpenSSH 7.2, legacy algorithm
+ `- [warn] using weak cipher mode
+ `- [info] available since OpenSSH 2.3.0
+
+# message authentication code algorithms
+(mac) hmac-md5 -- [fail] removed (in server) since OpenSSH 6.7, unsafe algorithm
+ `- [warn] disabled (in client) since OpenSSH 7.2, legacy algorithm
+ `- [warn] using encrypt-and-MAC mode
+ `- [warn] using weak hashing algorithm
+ `- [info] available since OpenSSH 2.1.0, Dropbear SSH 0.28
+(mac) hmac-sha1 -- [warn] using encrypt-and-MAC mode
+ `- [warn] using weak hashing algorithm
+ `- [info] available since OpenSSH 2.1.0, Dropbear SSH 0.28
+(mac) umac-64@openssh.com -- [warn] using encrypt-and-MAC mode
+ `- [warn] using small 64-bit tag size
+ `- [info] available since OpenSSH 4.7
+(mac) hmac-ripemd160 -- [fail] removed (in server) since OpenSSH 6.7, unsafe algorithm
+ `- [warn] disabled (in client) since OpenSSH 7.2, legacy algorithm
+ `- [warn] using encrypt-and-MAC mode
+ `- [info] available since OpenSSH 2.5.0
+(mac) hmac-ripemd160@openssh.com -- [fail] removed (in server) since OpenSSH 6.7, unsafe algorithm
+ `- [warn] disabled (in client) since OpenSSH 7.2, legacy algorithm
+ `- [warn] using encrypt-and-MAC mode
+ `- [info] available since OpenSSH 2.1.0
+(mac) hmac-sha1-96 -- [fail] removed (in server) since OpenSSH 6.7, unsafe algorithm
+ `- [warn] disabled (in client) since OpenSSH 7.2, legacy algorithm
+ `- [warn] using encrypt-and-MAC mode
+ `- [warn] using weak hashing algorithm
+ `- [info] available since OpenSSH 2.5.0, Dropbear SSH 0.47
+(mac) hmac-md5-96 -- [fail] removed (in server) since OpenSSH 6.7, unsafe algorithm
+ `- [warn] disabled (in client) since OpenSSH 7.2, legacy algorithm
+ `- [warn] using encrypt-and-MAC mode
+ `- [warn] using weak hashing algorithm
+ `- [info] available since OpenSSH 2.5.0
+
+# fingerprints
+(fin) ssh-rsa: SHA256:nsWtdJ9Z67Vrf7OsUzQov7esXhsWAfVppArGh25u244
+
+# algorithm recommendations (for OpenSSH 5.6)
+(rec) !diffie-hellman-group-exchange-sha256 -- kex algorithm to change (increase modulus size to 2048 bits or larger) 
+(rec) -3des-cbc -- enc algorithm to remove 
+(rec) -aes128-cbc -- enc algorithm to remove 
+(rec) -aes192-cbc -- enc algorithm to remove 
+(rec) -aes256-cbc -- enc algorithm to remove 
+(rec) -arcfour -- enc algorithm to remove 
+(rec) -arcfour128 -- enc algorithm to remove 
+(rec) -arcfour256 -- enc algorithm to remove 
+(rec) -blowfish-cbc -- enc algorithm to remove 
+(rec) -cast128-cbc -- enc algorithm to remove 
+(rec) -diffie-hellman-group-exchange-sha1 -- kex algorithm to remove 
+(rec) -diffie-hellman-group1-sha1 -- kex algorithm to remove 
+(rec) -hmac-md5 -- mac algorithm to remove 
+(rec) -hmac-md5-96 -- mac algorithm to remove 
+(rec) -hmac-ripemd160 -- mac algorithm to remove 
+(rec) -hmac-ripemd160@openssh.com -- mac algorithm to remove 
+(rec) -hmac-sha1-96 -- mac algorithm to remove 
+(rec) -rijndael-cbc@lysator.liu.se -- enc algorithm to remove 
+(rec) -ssh-rsa -- key algorithm to remove 
+(rec) -ssh-rsa-cert-v01@openssh.com -- key algorithm to remove 
+(rec) -diffie-hellman-group14-sha1 -- kex algorithm to remove 
+(rec) -hmac-sha1 -- mac algorithm to remove 
+(rec) -umac-64@openssh.com -- mac algorithm to remove 
+
+# additional info
+(nfo) For hardening guides on common OSes, please see: <https://www.ssh-audit.com/hardening_guides.html>
+
diff --git a/test/docker/expected_results/openssh_8.0p1_builtin_policy_test1.json b/test/docker/expected_results/openssh_8.0p1_builtin_policy_test1.json
new file mode 100644
index 0000000..f907cde
--- /dev/null
+++ b/test/docker/expected_results/openssh_8.0p1_builtin_policy_test1.json
@@ -0,0 +1 @@
+{"errors": [], "host": "localhost", "passed": true, "policy": "Hardened OpenSSH Server v8.0 (version 1)"}
diff --git a/test/docker/expected_results/openssh_8.0p1_builtin_policy_test1.txt b/test/docker/expected_results/openssh_8.0p1_builtin_policy_test1.txt
new file mode 100644
index 0000000..1b6ddee
--- /dev/null
+++ b/test/docker/expected_results/openssh_8.0p1_builtin_policy_test1.txt
@@ -0,0 +1,3 @@
+Host: localhost:2222
+Policy: Hardened OpenSSH Server v8.0 (version 1)
+Result: ✔ Passed
diff --git a/test/docker/expected_results/openssh_8.0p1_builtin_policy_test2.json b/test/docker/expected_results/openssh_8.0p1_builtin_policy_test2.json
new file mode 100644
index 0000000..49a13bc
--- /dev/null
+++ b/test/docker/expected_results/openssh_8.0p1_builtin_policy_test2.json
@@ -0,0 +1 @@
+{"errors": [{"actual": ["umac-64-etm@openssh.com", "umac-128-etm@openssh.com", "hmac-sha2-256-etm@openssh.com", "hmac-sha2-512-etm@openssh.com", "hmac-sha1-etm@openssh.com", "umac-64@openssh.com", "umac-128@openssh.com", "hmac-sha2-256", "hmac-sha2-512", "hmac-sha1"], "expected_optional": [""], "expected_required": ["hmac-sha2-256-etm@openssh.com", "hmac-sha2-512-etm@openssh.com", "umac-128-etm@openssh.com"], "mismatched_field": "MACs"}], "host": "localhost", "passed": false, "policy": "Hardened OpenSSH Server v8.0 (version 1)"}
diff --git a/test/docker/expected_results/openssh_8.0p1_builtin_policy_test2.txt b/test/docker/expected_results/openssh_8.0p1_builtin_policy_test2.txt
new file mode 100644
index 0000000..de1f3e7
--- /dev/null
+++ b/test/docker/expected_results/openssh_8.0p1_builtin_policy_test2.txt
@@ -0,0 +1,9 @@
+Host: localhost:2222
+Policy: Hardened OpenSSH Server v8.0 (version 1)
+Result: ⌠Failed!
+
+Errors:
+ * MACs did not match.
+ - Expected: hmac-sha2-256-etm@openssh.com, hmac-sha2-512-etm@openssh.com, umac-128-etm@openssh.com
+ - Actual: umac-64-etm@openssh.com, umac-128-etm@openssh.com, hmac-sha2-256-etm@openssh.com, hmac-sha2-512-etm@openssh.com, hmac-sha1-etm@openssh.com, umac-64@openssh.com, umac-128@openssh.com, hmac-sha2-256, hmac-sha2-512, hmac-sha1
+
diff --git a/test/docker/expected_results/openssh_8.0p1_custom_policy_test11.json b/test/docker/expected_results/openssh_8.0p1_custom_policy_test11.json
new file mode 100644
index 0000000..2e105ff
--- /dev/null
+++ b/test/docker/expected_results/openssh_8.0p1_custom_policy_test11.json
@@ -0,0 +1 @@
+{"errors": [], "host": "localhost", "passed": true, "policy": "Docker policy: test11 (version 1)"}
diff --git a/test/docker/expected_results/openssh_8.0p1_custom_policy_test11.txt b/test/docker/expected_results/openssh_8.0p1_custom_policy_test11.txt
new file mode 100644
index 0000000..024bcb9
--- /dev/null
+++ b/test/docker/expected_results/openssh_8.0p1_custom_policy_test11.txt
@@ -0,0 +1,3 @@
+Host: localhost:2222
+Policy: Docker policy: test11 (version 1)
+Result: ✔ Passed
diff --git a/test/docker/expected_results/openssh_8.0p1_custom_policy_test12.json b/test/docker/expected_results/openssh_8.0p1_custom_policy_test12.json
new file mode 100644
index 0000000..677fb96
--- /dev/null
+++ b/test/docker/expected_results/openssh_8.0p1_custom_policy_test12.json
@@ -0,0 +1 @@
+{"errors": [{"actual": ["3072"], "expected_optional": [""], "expected_required": ["4096"], "mismatched_field": "RSA host key (rsa-sha2-256) sizes"}, {"actual": ["3072"], "expected_optional": [""], "expected_required": ["4096"], "mismatched_field": "RSA host key (rsa-sha2-512) sizes"}, {"actual": ["3072"], "expected_optional": [""], "expected_required": ["4096"], "mismatched_field": "RSA host key (ssh-rsa) sizes"}], "host": "localhost", "passed": false, "policy": "Docker policy: test12 (version 1)"}
diff --git a/test/docker/expected_results/openssh_8.0p1_custom_policy_test12.txt b/test/docker/expected_results/openssh_8.0p1_custom_policy_test12.txt
new file mode 100644
index 0000000..6fb0561
--- /dev/null
+++ b/test/docker/expected_results/openssh_8.0p1_custom_policy_test12.txt
@@ -0,0 +1,17 @@
+Host: localhost:2222
+Policy: Docker policy: test12 (version 1)
+Result: ⌠Failed!
+
+Errors:
+ * RSA host key (rsa-sha2-256) sizes did not match.
+ - Expected: 4096
+ - Actual: 3072
+
+ * RSA host key (rsa-sha2-512) sizes did not match.
+ - Expected: 4096
+ - Actual: 3072
+
+ * RSA host key (ssh-rsa) sizes did not match.
+ - Expected: 4096
+ - Actual: 3072
+
diff --git a/test/docker/expected_results/openssh_8.0p1_custom_policy_test13.json b/test/docker/expected_results/openssh_8.0p1_custom_policy_test13.json
new file mode 100644
index 0000000..e412128
--- /dev/null
+++ b/test/docker/expected_results/openssh_8.0p1_custom_policy_test13.json
@@ -0,0 +1 @@
+{"errors": [], "host": "localhost", "passed": true, "policy": "Docker policy: test13 (version 1)"}
diff --git a/test/docker/expected_results/openssh_8.0p1_custom_policy_test13.txt b/test/docker/expected_results/openssh_8.0p1_custom_policy_test13.txt
new file mode 100644
index 0000000..b8b4b25
--- /dev/null
+++ b/test/docker/expected_results/openssh_8.0p1_custom_policy_test13.txt
@@ -0,0 +1,3 @@
+Host: localhost:2222
+Policy: Docker policy: test13 (version 1)
+Result: ✔ Passed
diff --git a/test/docker/expected_results/openssh_8.0p1_custom_policy_test14.json b/test/docker/expected_results/openssh_8.0p1_custom_policy_test14.json
new file mode 100644
index 0000000..e7a39f8
--- /dev/null
+++ b/test/docker/expected_results/openssh_8.0p1_custom_policy_test14.json
@@ -0,0 +1 @@
+{"errors": [{"actual": ["2048"], "expected_optional": [""], "expected_required": ["4096"], "mismatched_field": "Group exchange (diffie-hellman-group-exchange-sha256) modulus sizes"}], "host": "localhost", "passed": false, "policy": "Docker policy: test14 (version 1)"}
diff --git a/test/docker/expected_results/openssh_8.0p1_custom_policy_test14.txt b/test/docker/expected_results/openssh_8.0p1_custom_policy_test14.txt
new file mode 100644
index 0000000..2bb59fb
--- /dev/null
+++ b/test/docker/expected_results/openssh_8.0p1_custom_policy_test14.txt
@@ -0,0 +1,9 @@
+Host: localhost:2222
+Policy: Docker policy: test14 (version 1)
+Result: ⌠Failed!
+
+Errors:
+ * Group exchange (diffie-hellman-group-exchange-sha256) modulus sizes did not match.
+ - Expected: 4096
+ - Actual: 2048
+
diff --git a/test/docker/expected_results/openssh_8.0p1_custom_policy_test6.json b/test/docker/expected_results/openssh_8.0p1_custom_policy_test6.json
new file mode 100644
index 0000000..04620b2
--- /dev/null
+++ b/test/docker/expected_results/openssh_8.0p1_custom_policy_test6.json
@@ -0,0 +1 @@
+{"errors": [], "host": "localhost", "passed": true, "policy": "Docker policy: test6 (version 1)"}
diff --git a/test/docker/expected_results/openssh_8.0p1_custom_policy_test6.txt b/test/docker/expected_results/openssh_8.0p1_custom_policy_test6.txt
new file mode 100644
index 0000000..b0e9441
--- /dev/null
+++ b/test/docker/expected_results/openssh_8.0p1_custom_policy_test6.txt
@@ -0,0 +1,3 @@
+Host: localhost:2222
+Policy: Docker policy: test6 (version 1)
+Result: ✔ Passed
diff --git a/test/docker/expected_results/openssh_8.0p1_test1.json b/test/docker/expected_results/openssh_8.0p1_test1.json
new file mode 100644
index 0000000..cf610a3
--- /dev/null
+++ b/test/docker/expected_results/openssh_8.0p1_test1.json
@@ -0,0 +1 @@
+{"banner": {"comments": null, "protocol": [2, 0], "raw": "SSH-2.0-OpenSSH_8.0", "software": "OpenSSH_8.0"}, "compression": ["none", "zlib@openssh.com"], "enc": ["chacha20-poly1305@openssh.com", "aes128-ctr", "aes192-ctr", "aes256-ctr", "aes128-gcm@openssh.com", "aes256-gcm@openssh.com"], "fingerprints": [{"hash": "UrnXIVH+7dlw8UqYocl48yUEcKrthGDQG2CPCgp7MxU", "hash_alg": "SHA256", "hostkey": "ssh-ed25519"}, {"hash": "1e:0c:7b:34:73:bf:52:41:b0:f9:d1:a9:ab:98:c7:c9", "hash_alg": "MD5", "hostkey": "ssh-ed25519"}, {"hash": "nsWtdJ9Z67Vrf7OsUzQov7esXhsWAfVppArGh25u244", "hash_alg": "SHA256", "hostkey": "ssh-rsa"}, {"hash": "18:e2:51:fe:21:6c:78:d0:b8:cf:32:d4:bd:56:42:e1", "hash_alg": "MD5", "hostkey": "ssh-rsa"}], "kex": [{"algorithm": "curve25519-sha256"}, {"algorithm": "curve25519-sha256@libssh.org"}, {"algorithm": "ecdh-sha2-nistp256"}, {"algorithm": "ecdh-sha2-nistp384"}, {"algorithm": "ecdh-sha2-nistp521"}, {"algorithm": "diffie-hellman-group-exchange-sha256", "keysize": 2048}, {"algorithm": "diffie-hellman-group16-sha512"}, {"algorithm": "diffie-hellman-group18-sha512"}, {"algorithm": "diffie-hellman-group14-sha256"}, {"algorithm": "diffie-hellman-group14-sha1"}], "key": [{"algorithm": "rsa-sha2-512", "keysize": 3072}, {"algorithm": "rsa-sha2-256", "keysize": 3072}, {"algorithm": "ssh-rsa", "keysize": 3072}, {"algorithm": "ecdsa-sha2-nistp256"}, {"algorithm": "ssh-ed25519"}], "mac": ["umac-64-etm@openssh.com", "umac-128-etm@openssh.com", "hmac-sha2-256-etm@openssh.com", "hmac-sha2-512-etm@openssh.com", "hmac-sha1-etm@openssh.com", "umac-64@openssh.com", "umac-128@openssh.com", "hmac-sha2-256", "hmac-sha2-512", "hmac-sha1"], "target": "localhost"}
diff --git a/test/docker/expected_results/openssh_8.0p1_test1.txt b/test/docker/expected_results/openssh_8.0p1_test1.txt
new file mode 100644
index 0000000..d7e45b1
--- /dev/null
+++ b/test/docker/expected_results/openssh_8.0p1_test1.txt
@@ -0,0 +1,85 @@
+# general
+(gen) banner: SSH-2.0-OpenSSH_8.0
+(gen) software: OpenSSH 8.0
+(gen) compatibility: OpenSSH 7.4+, Dropbear SSH 2018.76+
+(gen) compression: enabled (zlib@openssh.com)
+
+# key exchange algorithms
+(kex) curve25519-sha256 -- [info] available since OpenSSH 7.4, Dropbear SSH 2018.76
+(kex) curve25519-sha256@libssh.org -- [info] available since OpenSSH 6.5, Dropbear SSH 2013.62
+(kex) ecdh-sha2-nistp256 -- [fail] using weak elliptic curves
+ `- [info] available since OpenSSH 5.7, Dropbear SSH 2013.62
+(kex) ecdh-sha2-nistp384 -- [fail] using weak elliptic curves
+ `- [info] available since OpenSSH 5.7, Dropbear SSH 2013.62
+(kex) ecdh-sha2-nistp521 -- [fail] using weak elliptic curves
+ `- [info] available since OpenSSH 5.7, Dropbear SSH 2013.62
+(kex) diffie-hellman-group-exchange-sha256 (2048-bit) -- [info] available since OpenSSH 4.4
+(kex) diffie-hellman-group16-sha512 -- [info] available since OpenSSH 7.3, Dropbear SSH 2016.73
+(kex) diffie-hellman-group18-sha512 -- [info] available since OpenSSH 7.3
+(kex) diffie-hellman-group14-sha256 -- [info] available since OpenSSH 7.3, Dropbear SSH 2016.73
+(kex) diffie-hellman-group14-sha1 -- [warn] using weak hashing algorithm
+ `- [info] available since OpenSSH 3.9, Dropbear SSH 0.53
+
+# host-key algorithms
+(key) rsa-sha2-512 (3072-bit) -- [info] available since OpenSSH 7.2
+(key) rsa-sha2-256 (3072-bit) -- [info] available since OpenSSH 7.2
+(key) ssh-rsa (3072-bit) -- [fail] using weak hashing algorithm
+ `- [info] available since OpenSSH 2.5.0, Dropbear SSH 0.28
+ `- [info] a future deprecation notice has been issued in OpenSSH 8.2: https://www.openssh.com/txt/release-8.2
+(key) ecdsa-sha2-nistp256 -- [fail] using weak elliptic curves
+ `- [warn] using weak random number generator could reveal the key
+ `- [info] available since OpenSSH 5.7, Dropbear SSH 2013.62
+(key) ssh-ed25519 -- [info] available since OpenSSH 6.5
+
+# encryption algorithms (ciphers)
+(enc) chacha20-poly1305@openssh.com -- [info] available since OpenSSH 6.5
+ `- [info] default cipher since OpenSSH 6.9.
+(enc) aes128-ctr -- [info] available since OpenSSH 3.7, Dropbear SSH 0.52
+(enc) aes192-ctr -- [info] available since OpenSSH 3.7
+(enc) aes256-ctr -- [info] available since OpenSSH 3.7, Dropbear SSH 0.52
+(enc) aes128-gcm@openssh.com -- [info] available since OpenSSH 6.2
+(enc) aes256-gcm@openssh.com -- [info] available since OpenSSH 6.2
+
+# message authentication code algorithms
+(mac) umac-64-etm@openssh.com -- [warn] using small 64-bit tag size
+ `- [info] available since OpenSSH 6.2
+(mac) umac-128-etm@openssh.com -- [info] available since OpenSSH 6.2
+(mac) hmac-sha2-256-etm@openssh.com -- [info] available since OpenSSH 6.2
+(mac) hmac-sha2-512-etm@openssh.com -- [info] available since OpenSSH 6.2
+(mac) hmac-sha1-etm@openssh.com -- [warn] using weak hashing algorithm
+ `- [info] available since OpenSSH 6.2
+(mac) umac-64@openssh.com -- [warn] using encrypt-and-MAC mode
+ `- [warn] using small 64-bit tag size
+ `- [info] available since OpenSSH 4.7
+(mac) umac-128@openssh.com -- [warn] using encrypt-and-MAC mode
+ `- [info] available since OpenSSH 6.2
+(mac) hmac-sha2-256 -- [warn] using encrypt-and-MAC mode
+ `- [info] available since OpenSSH 5.9, Dropbear SSH 2013.56
+(mac) hmac-sha2-512 -- [warn] using encrypt-and-MAC mode
+ `- [info] available since OpenSSH 5.9, Dropbear SSH 2013.56
+(mac) hmac-sha1 -- [warn] using encrypt-and-MAC mode
+ `- [warn] using weak hashing algorithm
+ `- [info] available since OpenSSH 2.1.0, Dropbear SSH 0.28
+
+# fingerprints
+(fin) ssh-ed25519: SHA256:UrnXIVH+7dlw8UqYocl48yUEcKrthGDQG2CPCgp7MxU
+(fin) ssh-rsa: SHA256:nsWtdJ9Z67Vrf7OsUzQov7esXhsWAfVppArGh25u244
+
+# algorithm recommendations (for OpenSSH 8.0)
+(rec) -ecdh-sha2-nistp256 -- kex algorithm to remove 
+(rec) -ecdh-sha2-nistp384 -- kex algorithm to remove 
+(rec) -ecdh-sha2-nistp521 -- kex algorithm to remove 
+(rec) -ecdsa-sha2-nistp256 -- key algorithm to remove 
+(rec) -ssh-rsa -- key algorithm to remove 
+(rec) -diffie-hellman-group14-sha1 -- kex algorithm to remove 
+(rec) -hmac-sha1 -- mac algorithm to remove 
+(rec) -hmac-sha1-etm@openssh.com -- mac algorithm to remove 
+(rec) -hmac-sha2-256 -- mac algorithm to remove 
+(rec) -hmac-sha2-512 -- mac algorithm to remove 
+(rec) -umac-128@openssh.com -- mac algorithm to remove 
+(rec) -umac-64-etm@openssh.com -- mac algorithm to remove 
+(rec) -umac-64@openssh.com -- mac algorithm to remove 
+
+# additional info
+(nfo) For hardening guides on common OSes, please see: <https://www.ssh-audit.com/hardening_guides.html>
+
diff --git a/test/docker/expected_results/openssh_8.0p1_test2.json b/test/docker/expected_results/openssh_8.0p1_test2.json
new file mode 100644
index 0000000..fb323eb
--- /dev/null
+++ b/test/docker/expected_results/openssh_8.0p1_test2.json
@@ -0,0 +1 @@
+{"banner": {"comments": null, "protocol": [2, 0], "raw": "SSH-2.0-OpenSSH_8.0", "software": "OpenSSH_8.0"}, "compression": ["none", "zlib@openssh.com"], "enc": ["chacha20-poly1305@openssh.com", "aes128-ctr", "aes192-ctr", "aes256-ctr", "aes128-gcm@openssh.com", "aes256-gcm@openssh.com"], "fingerprints": [{"hash": "UrnXIVH+7dlw8UqYocl48yUEcKrthGDQG2CPCgp7MxU", "hash_alg": "SHA256", "hostkey": "ssh-ed25519"}, {"hash": "1e:0c:7b:34:73:bf:52:41:b0:f9:d1:a9:ab:98:c7:c9", "hash_alg": "MD5", "hostkey": "ssh-ed25519"}], "kex": [{"algorithm": "curve25519-sha256"}, {"algorithm": "curve25519-sha256@libssh.org"}, {"algorithm": "ecdh-sha2-nistp256"}, {"algorithm": "ecdh-sha2-nistp384"}, {"algorithm": "ecdh-sha2-nistp521"}, {"algorithm": "diffie-hellman-group-exchange-sha256", "keysize": 2048}, {"algorithm": "diffie-hellman-group16-sha512"}, {"algorithm": "diffie-hellman-group18-sha512"}, {"algorithm": "diffie-hellman-group14-sha256"}, {"algorithm": "diffie-hellman-group14-sha1"}], "key": [{"algorithm": "ssh-ed25519"}, {"algorithm": "ssh-ed25519-cert-v01@openssh.com"}], "mac": ["umac-64-etm@openssh.com", "umac-128-etm@openssh.com", "hmac-sha2-256-etm@openssh.com", "hmac-sha2-512-etm@openssh.com", "hmac-sha1-etm@openssh.com", "umac-64@openssh.com", "umac-128@openssh.com", "hmac-sha2-256", "hmac-sha2-512", "hmac-sha1"], "target": "localhost"}
diff --git a/test/docker/expected_results/openssh_8.0p1_test2.txt b/test/docker/expected_results/openssh_8.0p1_test2.txt
new file mode 100644
index 0000000..a13be8a
--- /dev/null
+++ b/test/docker/expected_results/openssh_8.0p1_test2.txt
@@ -0,0 +1,77 @@
+# general
+(gen) banner: SSH-2.0-OpenSSH_8.0
+(gen) software: OpenSSH 8.0
+(gen) compatibility: OpenSSH 7.4+, Dropbear SSH 2018.76+
+(gen) compression: enabled (zlib@openssh.com)
+
+# key exchange algorithms
+(kex) curve25519-sha256 -- [info] available since OpenSSH 7.4, Dropbear SSH 2018.76
+(kex) curve25519-sha256@libssh.org -- [info] available since OpenSSH 6.5, Dropbear SSH 2013.62
+(kex) ecdh-sha2-nistp256 -- [fail] using weak elliptic curves
+ `- [info] available since OpenSSH 5.7, Dropbear SSH 2013.62
+(kex) ecdh-sha2-nistp384 -- [fail] using weak elliptic curves
+ `- [info] available since OpenSSH 5.7, Dropbear SSH 2013.62
+(kex) ecdh-sha2-nistp521 -- [fail] using weak elliptic curves
+ `- [info] available since OpenSSH 5.7, Dropbear SSH 2013.62
+(kex) diffie-hellman-group-exchange-sha256 (2048-bit) -- [info] available since OpenSSH 4.4
+(kex) diffie-hellman-group16-sha512 -- [info] available since OpenSSH 7.3, Dropbear SSH 2016.73
+(kex) diffie-hellman-group18-sha512 -- [info] available since OpenSSH 7.3
+(kex) diffie-hellman-group14-sha256 -- [info] available since OpenSSH 7.3, Dropbear SSH 2016.73
+(kex) diffie-hellman-group14-sha1 -- [warn] using weak hashing algorithm
+ `- [info] available since OpenSSH 3.9, Dropbear SSH 0.53
+
+# host-key algorithms
+(key) ssh-ed25519 -- [info] available since OpenSSH 6.5
+(key) ssh-ed25519-cert-v01@openssh.com -- [info] available since OpenSSH 6.5
+
+# encryption algorithms (ciphers)
+(enc) chacha20-poly1305@openssh.com -- [info] available since OpenSSH 6.5
+ `- [info] default cipher since OpenSSH 6.9.
+(enc) aes128-ctr -- [info] available since OpenSSH 3.7, Dropbear SSH 0.52
+(enc) aes192-ctr -- [info] available since OpenSSH 3.7
+(enc) aes256-ctr -- [info] available since OpenSSH 3.7, Dropbear SSH 0.52
+(enc) aes128-gcm@openssh.com -- [info] available since OpenSSH 6.2
+(enc) aes256-gcm@openssh.com -- [info] available since OpenSSH 6.2
+
+# message authentication code algorithms
+(mac) umac-64-etm@openssh.com -- [warn] using small 64-bit tag size
+ `- [info] available since OpenSSH 6.2
+(mac) umac-128-etm@openssh.com -- [info] available since OpenSSH 6.2
+(mac) hmac-sha2-256-etm@openssh.com -- [info] available since OpenSSH 6.2
+(mac) hmac-sha2-512-etm@openssh.com -- [info] available since OpenSSH 6.2
+(mac) hmac-sha1-etm@openssh.com -- [warn] using weak hashing algorithm
+ `- [info] available since OpenSSH 6.2
+(mac) umac-64@openssh.com -- [warn] using encrypt-and-MAC mode
+ `- [warn] using small 64-bit tag size
+ `- [info] available since OpenSSH 4.7
+(mac) umac-128@openssh.com -- [warn] using encrypt-and-MAC mode
+ `- [info] available since OpenSSH 6.2
+(mac) hmac-sha2-256 -- [warn] using encrypt-and-MAC mode
+ `- [info] available since OpenSSH 5.9, Dropbear SSH 2013.56
+(mac) hmac-sha2-512 -- [warn] using encrypt-and-MAC mode
+ `- [info] available since OpenSSH 5.9, Dropbear SSH 2013.56
+(mac) hmac-sha1 -- [warn] using encrypt-and-MAC mode
+ `- [warn] using weak hashing algorithm
+ `- [info] available since OpenSSH 2.1.0, Dropbear SSH 0.28
+
+# fingerprints
+(fin) ssh-ed25519: SHA256:UrnXIVH+7dlw8UqYocl48yUEcKrthGDQG2CPCgp7MxU
+
+# algorithm recommendations (for OpenSSH 8.0)
+(rec) -ecdh-sha2-nistp256 -- kex algorithm to remove 
+(rec) -ecdh-sha2-nistp384 -- kex algorithm to remove 
+(rec) -ecdh-sha2-nistp521 -- kex algorithm to remove 
+(rec) +rsa-sha2-256 -- key algorithm to append 
+(rec) +rsa-sha2-512 -- key algorithm to append 
+(rec) -diffie-hellman-group14-sha1 -- kex algorithm to remove 
+(rec) -hmac-sha1 -- mac algorithm to remove 
+(rec) -hmac-sha1-etm@openssh.com -- mac algorithm to remove 
+(rec) -hmac-sha2-256 -- mac algorithm to remove 
+(rec) -hmac-sha2-512 -- mac algorithm to remove 
+(rec) -umac-128@openssh.com -- mac algorithm to remove 
+(rec) -umac-64-etm@openssh.com -- mac algorithm to remove 
+(rec) -umac-64@openssh.com -- mac algorithm to remove 
+
+# additional info
+(nfo) For hardening guides on common OSes, please see: <https://www.ssh-audit.com/hardening_guides.html>
+
diff --git a/test/docker/expected_results/openssh_8.0p1_test3.json b/test/docker/expected_results/openssh_8.0p1_test3.json
new file mode 100644
index 0000000..5728136
--- /dev/null
+++ b/test/docker/expected_results/openssh_8.0p1_test3.json
@@ -0,0 +1 @@
+{"banner": {"comments": null, "protocol": [2, 0], "raw": "SSH-2.0-OpenSSH_8.0", "software": "OpenSSH_8.0"}, "compression": ["none", "zlib@openssh.com"], "enc": ["chacha20-poly1305@openssh.com", "aes256-gcm@openssh.com", "aes128-gcm@openssh.com", "aes256-ctr", "aes192-ctr", "aes128-ctr"], "fingerprints": [{"hash": "UrnXIVH+7dlw8UqYocl48yUEcKrthGDQG2CPCgp7MxU", "hash_alg": "SHA256", "hostkey": "ssh-ed25519"}, {"hash": "1e:0c:7b:34:73:bf:52:41:b0:f9:d1:a9:ab:98:c7:c9", "hash_alg": "MD5", "hostkey": "ssh-ed25519"}], "kex": [{"algorithm": "curve25519-sha256"}, {"algorithm": "curve25519-sha256@libssh.org"}, {"algorithm": "diffie-hellman-group-exchange-sha256", "keysize": 2048}], "key": [{"algorithm": "ssh-ed25519"}], "mac": ["hmac-sha2-256-etm@openssh.com", "hmac-sha2-512-etm@openssh.com", "umac-128-etm@openssh.com"], "target": "localhost"}
diff --git a/test/docker/expected_results/openssh_8.0p1_test3.txt b/test/docker/expected_results/openssh_8.0p1_test3.txt
new file mode 100644
index 0000000..5b20c3f
--- /dev/null
+++ b/test/docker/expected_results/openssh_8.0p1_test3.txt
@@ -0,0 +1,38 @@
+# general
+(gen) banner: SSH-2.0-OpenSSH_8.0
+(gen) software: OpenSSH 8.0
+(gen) compatibility: OpenSSH 7.4+, Dropbear SSH 2018.76+
+(gen) compression: enabled (zlib@openssh.com)
+
+# key exchange algorithms
+(kex) curve25519-sha256 -- [info] available since OpenSSH 7.4, Dropbear SSH 2018.76
+(kex) curve25519-sha256@libssh.org -- [info] available since OpenSSH 6.5, Dropbear SSH 2013.62
+(kex) diffie-hellman-group-exchange-sha256 (2048-bit) -- [info] available since OpenSSH 4.4
+
+# host-key algorithms
+(key) ssh-ed25519 -- [info] available since OpenSSH 6.5
+
+# encryption algorithms (ciphers)
+(enc) chacha20-poly1305@openssh.com -- [info] available since OpenSSH 6.5
+ `- [info] default cipher since OpenSSH 6.9.
+(enc) aes256-gcm@openssh.com -- [info] available since OpenSSH 6.2
+(enc) aes128-gcm@openssh.com -- [info] available since OpenSSH 6.2
+(enc) aes256-ctr -- [info] available since OpenSSH 3.7, Dropbear SSH 0.52
+(enc) aes192-ctr -- [info] available since OpenSSH 3.7
+(enc) aes128-ctr -- [info] available since OpenSSH 3.7, Dropbear SSH 0.52
+
+# message authentication code algorithms
+(mac) hmac-sha2-256-etm@openssh.com -- [info] available since OpenSSH 6.2
+(mac) hmac-sha2-512-etm@openssh.com -- [info] available since OpenSSH 6.2
+(mac) umac-128-etm@openssh.com -- [info] available since OpenSSH 6.2
+
+# fingerprints
+(fin) ssh-ed25519: SHA256:UrnXIVH+7dlw8UqYocl48yUEcKrthGDQG2CPCgp7MxU
+
+# algorithm recommendations (for OpenSSH 8.0)
+(rec) +diffie-hellman-group14-sha256 -- kex algorithm to append 
+(rec) +diffie-hellman-group16-sha512 -- kex algorithm to append 
+(rec) +diffie-hellman-group18-sha512 -- kex algorithm to append 
+(rec) +rsa-sha2-256 -- key algorithm to append 
+(rec) +rsa-sha2-512 -- key algorithm to append 
+
diff --git a/test/docker/expected_results/tinyssh_20190101_test1.json b/test/docker/expected_results/tinyssh_20190101_test1.json
new file mode 100644
index 0000000..6bf8f24
--- /dev/null
+++ b/test/docker/expected_results/tinyssh_20190101_test1.json
@@ -0,0 +1 @@
+{"banner": {"comments": "", "protocol": [2, 0], "raw": "", "software": "tinyssh_noversion"}, "compression": ["none"], "enc": ["chacha20-poly1305@openssh.com"], "fingerprints": [{"hash": "89ocln1x7KNqnMgWffGoYtD70ksJ4FrH7BMJHa7SrwU", "hash_alg": "SHA256", "hostkey": "ssh-ed25519"}, {"hash": "dd:9c:6d:f9:b0:8c:af:fa:c2:65:81:5d:5d:56:f8:21", "hash_alg": "MD5", "hostkey": "ssh-ed25519"}], "kex": [{"algorithm": "curve25519-sha256"}, {"algorithm": "curve25519-sha256@libssh.org"}, {"algorithm": "sntrup4591761x25519-sha512@tinyssh.org"}], "key": [{"algorithm": "ssh-ed25519"}], "mac": ["hmac-sha2-256"], "target": "localhost"}
diff --git a/test/docker/expected_results/tinyssh_20190101_test1.txt b/test/docker/expected_results/tinyssh_20190101_test1.txt
new file mode 100644
index 0000000..9e833c6
--- /dev/null
+++ b/test/docker/expected_results/tinyssh_20190101_test1.txt
@@ -0,0 +1,25 @@
+# general
+(gen) software: TinySSH noversion
+(gen) compatibility: OpenSSH 8.0-8.4, Dropbear SSH 2018.76+
+(gen) compression: disabled
+
+# key exchange algorithms
+(kex) curve25519-sha256 -- [info] available since OpenSSH 7.4, Dropbear SSH 2018.76
+(kex) curve25519-sha256@libssh.org -- [info] available since OpenSSH 6.5, Dropbear SSH 2013.62
+(kex) sntrup4591761x25519-sha512@tinyssh.org -- [warn] using experimental algorithm
+ `- [info] available since OpenSSH 8.0
+
+# host-key algorithms
+(key) ssh-ed25519 -- [info] available since OpenSSH 6.5
+
+# encryption algorithms (ciphers)
+(enc) chacha20-poly1305@openssh.com -- [info] available since OpenSSH 6.5
+ `- [info] default cipher since OpenSSH 6.9.
+
+# message authentication code algorithms
+(mac) hmac-sha2-256 -- [warn] using encrypt-and-MAC mode
+ `- [info] available since OpenSSH 5.9, Dropbear SSH 2013.56
+
+# fingerprints
+(fin) ssh-ed25519: SHA256:89ocln1x7KNqnMgWffGoYtD70ksJ4FrH7BMJHa7SrwU
+
diff --git a/test/docker/host_ca_ed25519 b/test/docker/host_ca_ed25519
new file mode 100644
index 0000000..7b8c41b
--- /dev/null
+++ b/test/docker/host_ca_ed25519
@@ -0,0 +1,7 @@
+-----BEGIN OPENSSH PRIVATE KEY-----
+b3BlbnNzaC1rZXktdjEAAAAABG5vbmUAAAAEbm9uZQAAAAAAAAABAAAAMwAAAAtzc2gtZW
+QyNTUxOQAAACAbM9Wp3ZPcC8Ifhu6GjNDJaoMg7KxO0el2+r9J35TltQAAAKAa0zr8GtM6
+/AAAAAtzc2gtZWQyNTUxOQAAACAbM9Wp3ZPcC8Ifhu6GjNDJaoMg7KxO0el2+r9J35TltQ
+AAAEC/j/BpfmgaZqNMTkJXO4cKZBr31N5z33IRFjh5m6IDDhsz1andk9wLwh+G7oaM0Mlq
+gyDsrE7R6Xb6v0nflOW1AAAAHWpkb2dAbG9jYWxob3N0LndvbmRlcmxhbmQubG9s
+-----END OPENSSH PRIVATE KEY-----
diff --git a/test/docker/host_ca_ed25519.pub b/test/docker/host_ca_ed25519.pub
new file mode 100644
index 0000000..01e745f
--- /dev/null
+++ b/test/docker/host_ca_ed25519.pub
@@ -0,0 +1 @@
+ssh-ed25519 AAAAC3NzaC1lZDI1NTE5AAAAIBsz1andk9wLwh+G7oaM0MlqgyDsrE7R6Xb6v0nflOW1 jdog@localhost.wonderland.lol
diff --git a/test/docker/host_ca_rsa_1024 b/test/docker/host_ca_rsa_1024
new file mode 100644
index 0000000..337b777
--- /dev/null
+++ b/test/docker/host_ca_rsa_1024
@@ -0,0 +1,15 @@
+-----BEGIN RSA PRIVATE KEY-----
+MIICXgIBAAKBgQDnRlN3AFnUe2lFf5XG9UhXLr/9POruNTFbMt0zrjOUSjmAS7hS
+6pDv5VEToT6DaR1EQUYaqSMpHYzZhuCK52vrydOm5XFbJ7712r9MyZQUhoVZx8Su
+dBHzVDIVO3jcMMWIlrfWBMnUaUHEqpmy88Y7gKDa2TWxJg1+hg51KqHrUQIDAQAB
+AoGBANALOUXRcP1tTtOP4+In/709dsONKyDBhPavGMFGsWtyIavBcbxU+bBzrq1j
+3WJFCmi99xxAjjqMNInxhMgvSaoJtsiY0/FFxqRy6l/ZnRjI6hrVKR8whrPKVgBF
+pvbjeQIn9txeCYA8kwl/Si762u7byq+qvupE53xMP94J02KBAkEA/Q4+Hn1Rjblw
+VXynF+oXIq6iZy+8PW+Y/FIL8d31ehzfcssCMdFV6S3/wBoQkWby30oGC/xGmHGR
+6ffXGilByQJBAOn3NMrBPXNkaPeQtgV3tk4s1dRDQYhbqGNz6tcgThyyPdhJCmCy
+jgUEhLwAetsDI8/+3avWbo6/csOV+BvpYUkCQQDQyEp6L1z0+FV1QqY99dZmt/yn
+89t0OLnZG/xc7osU1/OHq3TBE3y1KU2D+j1HKdAiZ9l7VAYOykzf46qmG/n5AkEA
+2kWjfcjcIIw7lULvXZh6fuI7NwTr3V/Nb8MUA1EDLqhnJCG4SdAqyKmXf6Fe/HYo
+cgKPIaIykIAxfCCsULXg6QJAOxB0CKYJlopVBdjGMlGqOEneWTmb1A2INQDE2Una
+LkSd0Rr8OiEzDeemV7j3Ec4BH0HxGMnHDxMybZwoZRnRPw==
+-----END RSA PRIVATE KEY-----
diff --git a/test/docker/host_ca_rsa_1024.pub b/test/docker/host_ca_rsa_1024.pub
new file mode 100644
index 0000000..6d861d6
--- /dev/null
+++ b/test/docker/host_ca_rsa_1024.pub
@@ -0,0 +1 @@
+ssh-rsa AAAAB3NzaC1yc2EAAAADAQABAAAAgQDnRlN3AFnUe2lFf5XG9UhXLr/9POruNTFbMt0zrjOUSjmAS7hS6pDv5VEToT6DaR1EQUYaqSMpHYzZhuCK52vrydOm5XFbJ7712r9MyZQUhoVZx8SudBHzVDIVO3jcMMWIlrfWBMnUaUHEqpmy88Y7gKDa2TWxJg1+hg51KqHrUQ== jdog@localhost.wonderland.lol
diff --git a/test/docker/host_ca_rsa_3072 b/test/docker/host_ca_rsa_3072
new file mode 100644
index 0000000..dd04653
--- /dev/null
+++ b/test/docker/host_ca_rsa_3072
@@ -0,0 +1,39 @@
+-----BEGIN RSA PRIVATE KEY-----
+MIIG4wIBAAKCAYEAqxQEIbj8w0TrBY1fDO81curijQrdLOUr8Vl8XECWc5QGd1Lk
+AG80NgdcCBPvjWxZSmYrKeqA78GUdN+KgycE0ztpxYSXKHZMaIM5Xe94BB+BocH9
+1vd/2iBzGeed1nV/zfAdq2AEHQj1TpII+a+z25yxv2PuwVTTwwo9I/6JgNq3evH4
+Hbwgr3SRfEEYZQ+YL8cOpBuNg1YZOR0k1yk23ZqAd92JybxZ4iCtOt7rcj2sFHzN
+u1U544wWBwIL5yZZKTgBhY4dqfT2Ep7IzR5HdsdrvQV9qC92GM1zDE+U3AwrVKjH
+s0YZq3jzcq/yvFDCcMMRz4/0pGFFU26oWma+n3vbAxKJoL+rhG8QM9+l2qFlLGsn
+M0kUXAJXsPKbygpaP8Z3U4eKgTuJ2GuS9eLIFnB7mrwD75V6GgN9q5mY89DfkVSk
+HaoqpY8pPdRkz9QAmMEuLtHmv29CVOpfX5v/rsm7wASAZqtUlmFu4rFGBLwvZbUl
+Wu02HmgBT47g6EIfAgMBAAECggGAKVCdKtO03yd+pomcodAHFWiaK7uq7FOwCAo3
+WUQT0Xe3FAwFmgFBF6cxV5YQ7RN0gN4poGbMmpoiUxNFLSU4KhcYFSZPJutiyn6e
+VQwm7L/7G2hw+AAvdSsPAPuJh6g6pC5Py/pVI/ns2/uyhTIkem3eEz18BF6LAXgw
+icfHx0GKu/tBk1TCg/zfwaUq0gUxGKC27XTl+QjK8JsUMY33fQ755Xiv9PMytcR0
+cVoyfBVewFffi1UqtMQ48ZpR65G743RxrP4/wcwsfD7n5LJLdyxQkh3gIMTJ8dd/
+R5V4FlueorRgjTbLTjGDxNrCAJ+locezhEEPXsPh2q0KiIXGyz2AMxaOqFmhU8oK
+aVVt8pWJ+YsrKIgc/A3s18ezO8uO5ZdtjQ+CWguduUGY7YgWezGLO1LPxhJC4d7b
+Q/xpeKveTRlcScAqOUzKgSuEhcvPgj8paUcRUoiXm4qiJBY5sXJks+YGp8BGksH0
+O94no+Ns2G58MlL+RyXk3JWrc6zRAoHBANdPplY2sIuIiiEBu95f1Qar1nCBHhB2
+i+HpnsUOdSlbxwMxoF8ffeN9N+DQqaqPu1RhFa5xbB2EUSujvOnL7b/RWqe1X9Po
+UIt5UjXctNP/HYcQDyjXY+rV5SZhHDyv6TBYurNZlvlBivliDz82THPRtqVxed3B
+w2MeaSkKAQ8rA7PE+0j3TG+YtIij0mHOhNPJgEZ/XZ9MIQOGMycRJhwOlclBI5NP
+Ak6p30ArnU2fX4qMkU3i+wqUfXS1hhDihwKBwQDLaHWPIWPVbWdcCbYQTcUmFC3i
+xkxd0UuLcfS9csk61nvdFj7m8tMExX+3fIo/fHEtzDd98Alc1i6/f6ePl0CX6NDu
+QIWLryI1QQRQidHCdw0wQ3N3VD4ZXJHDeqBxogVAkA7A/1QeXwcXE/Xj2ZgyDwhL
+3+myjmvWtw9zJsXL0F3tpPzn+Mrf0KRkWOaluOw7hMMjVjrgu6g24HMWbHHVLRTx
+dlAI7tgxCAPe2SEi+1mzaVUZ8cfgqYqC3X66UakCgcEAopxtK7+yJi/A4pzEnnYS
+FS/CjMV3R0fA7aXbW0hIBCxkaW0Zib3m/eCcSxZMjZxwBpIsJctTtBcylprbGlgB
+/1TF+tNoxEo4Sp4eEL/XciTC0Da4vEewFrPklM/S26KfovvgRYPsGeP+aco9aahA
+pVhFcT36pBiq0DkvgucjValO6n5iqgDboYzbDDdttKCcgLc2Qgf/VUfRxy+bgm3Z
+MmdxiMXBcIfDXlW9XmGSNAWhyqnPM9uxbZQoC/Tsg+QRAoHANHMcFSsz9f2+8DGk
+27FiC76aUmZ1nJ9yTmO1CwDFOMHDsK+iyqSEmy9eDm8zqsko2flVuciicWjdJw4A
+o/sJceJbtYO3q9weAwNf3HCdQPq30OEjrfpwBNQk1fYR1xtDJXHADC4Kf8ZbKq0/
+81/Rad8McZwsQ5mL3xLXDgdKa5KwFa48dIhnr6y6JxHxb3wule5W7w62Ierhpjzc
+EEUoWSLFyrmKS7Ni1cnOTbFJZR7Q831Or2Dz/E9bYwFAQ0T5AoHAM4/zU+8rsbdD
+FvvhWsj7Ivfh6pxx1Tl1Wccaauea9AJayHht0FOzkycpJrH1E+6F5MzhkFFU1SUY
+60NZxzSZgbU0HBrJRcRFyo510iMcnctdTdyh8p7nweGoD0oqXzf6cHqrUep8Y8rQ
+gkSVhPE31+NGlPbwz+NOflcaaAWYiDC6wjVt1asaZq292SJD4DF1fAUkbQ2hxgyQ
++G/6y5ovrcGnh7q63RLhW1TRf8dD2D2Av9UgXDmWZAZ5n838FS+X
+-----END RSA PRIVATE KEY-----
diff --git a/test/docker/host_ca_rsa_3072.pub b/test/docker/host_ca_rsa_3072.pub
new file mode 100644
index 0000000..b728ed7
--- /dev/null
+++ b/test/docker/host_ca_rsa_3072.pub
@@ -0,0 +1 @@
+ssh-rsa AAAAB3NzaC1yc2EAAAADAQABAAABgQCrFAQhuPzDROsFjV8M7zVy6uKNCt0s5SvxWXxcQJZzlAZ3UuQAbzQ2B1wIE++NbFlKZisp6oDvwZR034qDJwTTO2nFhJcodkxogzld73gEH4Ghwf3W93/aIHMZ553WdX/N8B2rYAQdCPVOkgj5r7PbnLG/Y+7BVNPDCj0j/omA2rd68fgdvCCvdJF8QRhlD5gvxw6kG42DVhk5HSTXKTbdmoB33YnJvFniIK063utyPawUfM27VTnjjBYHAgvnJlkpOAGFjh2p9PYSnsjNHkd2x2u9BX2oL3YYzXMMT5TcDCtUqMezRhmrePNyr/K8UMJwwxHPj/SkYUVTbqhaZr6fe9sDEomgv6uEbxAz36XaoWUsayczSRRcAlew8pvKClo/xndTh4qBO4nYa5L14sgWcHuavAPvlXoaA32rmZjz0N+RVKQdqiqljyk91GTP1ACYwS4u0ea/b0JU6l9fm/+uybvABIBmq1SWYW7isUYEvC9ltSVa7TYeaAFPjuDoQh8= jdog@localhost.wonderland.lol
diff --git a/test/docker/moduli_1024 b/test/docker/moduli_1024
new file mode 100644
index 0000000..bd81dae
--- /dev/null
+++ b/test/docker/moduli_1024
@@ -0,0 +1,44 @@
+20190821035337 2 6 100 1023 2 F0B5E9E385A451D4F46BD2E354B5FCAAC21CA960E5D3D11F877DD50541ED125161E4A5055D528D67E525115BBFAB0B2A4AB8CF5BA98A8BBA41803ED5D4CF766E9ECD39A8D8D914B6F346E0EB2BA6936082751676DCE5C4817EFC7A8105C2A094B22C25245BE13CA4085F2985D3B7A2636FF4018A7E4EA9840BF5FFBC08BE313B
+20190821035338 2 6 100 1023 2 F0B5E9E385A451D4F46BD2E354B5FCAAC21CA960E5D3D11F877DD50541ED125161E4A5055D528D67E525115BBFAB0B2A4AB8CF5BA98A8BBA41803ED5D4CF766E9ECD39A8D8D914B6F346E0EB2BA6936082751676DCE5C4817EFC7A8105C2A094B22C25245BE13CA4085F2985D3B7A2636FF4018A7E4EA9840BF5FFBC08C0B443
+20190821035338 2 6 100 1023 2 F0B5E9E385A451D4F46BD2E354B5FCAAC21CA960E5D3D11F877DD50541ED125161E4A5055D528D67E525115BBFAB0B2A4AB8CF5BA98A8BBA41803ED5D4CF766E9ECD39A8D8D914B6F346E0EB2BA6936082751676DCE5C4817EFC7A8105C2A094B22C25245BE13CA4085F2985D3B7A2636FF4018A7E4EA9840BF5FFBC08D1AF8B
+20190821035338 2 6 100 1023 2 F0B5E9E385A451D4F46BD2E354B5FCAAC21CA960E5D3D11F877DD50541ED125161E4A5055D528D67E525115BBFAB0B2A4AB8CF5BA98A8BBA41803ED5D4CF766E9ECD39A8D8D914B6F346E0EB2BA6936082751676DCE5C4817EFC7A8105C2A094B22C25245BE13CA4085F2985D3B7A2636FF4018A7E4EA9840BF5FFBC08E76DDB
+20190821035338 2 6 100 1023 2 F0B5E9E385A451D4F46BD2E354B5FCAAC21CA960E5D3D11F877DD50541ED125161E4A5055D528D67E525115BBFAB0B2A4AB8CF5BA98A8BBA41803ED5D4CF766E9ECD39A8D8D914B6F346E0EB2BA6936082751676DCE5C4817EFC7A8105C2A094B22C25245BE13CA4085F2985D3B7A2636FF4018A7E4EA9840BF5FFBC08E8F5D3
+20190821035338 2 6 100 1023 2 F0B5E9E385A451D4F46BD2E354B5FCAAC21CA960E5D3D11F877DD50541ED125161E4A5055D528D67E525115BBFAB0B2A4AB8CF5BA98A8BBA41803ED5D4CF766E9ECD39A8D8D914B6F346E0EB2BA6936082751676DCE5C4817EFC7A8105C2A094B22C25245BE13CA4085F2985D3B7A2636FF4018A7E4EA9840BF5FFBC08EE3F1B
+20190821035338 2 6 100 1023 5 F0B5E9E385A451D4F46BD2E354B5FCAAC21CA960E5D3D11F877DD50541ED125161E4A5055D528D67E525115BBFAB0B2A4AB8CF5BA98A8BBA41803ED5D4CF766E9ECD39A8D8D914B6F346E0EB2BA6936082751676DCE5C4817EFC7A8105C2A094B22C25245BE13CA4085F2985D3B7A2636FF4018A7E4EA9840BF5FFBC08F28387
+20190821035339 2 6 100 1023 5 F0B5E9E385A451D4F46BD2E354B5FCAAC21CA960E5D3D11F877DD50541ED125161E4A5055D528D67E525115BBFAB0B2A4AB8CF5BA98A8BBA41803ED5D4CF766E9ECD39A8D8D914B6F346E0EB2BA6936082751676DCE5C4817EFC7A8105C2A094B22C25245BE13CA4085F2985D3B7A2636FF4018A7E4EA9840BF5FFBC08F69A57
+20190821035339 2 6 100 1023 5 F0B5E9E385A451D4F46BD2E354B5FCAAC21CA960E5D3D11F877DD50541ED125161E4A5055D528D67E525115BBFAB0B2A4AB8CF5BA98A8BBA41803ED5D4CF766E9ECD39A8D8D914B6F346E0EB2BA6936082751676DCE5C4817EFC7A8105C2A094B22C25245BE13CA4085F2985D3B7A2636FF4018A7E4EA9840BF5FFBC0903B157
+20190821035339 2 6 100 1023 2 F0B5E9E385A451D4F46BD2E354B5FCAAC21CA960E5D3D11F877DD50541ED125161E4A5055D528D67E525115BBFAB0B2A4AB8CF5BA98A8BBA41803ED5D4CF766E9ECD39A8D8D914B6F346E0EB2BA6936082751676DCE5C4817EFC7A8105C2A094B22C25245BE13CA4085F2985D3B7A2636FF4018A7E4EA9840BF5FFBC0905C973
+20190821035339 2 6 100 1023 2 F0B5E9E385A451D4F46BD2E354B5FCAAC21CA960E5D3D11F877DD50541ED125161E4A5055D528D67E525115BBFAB0B2A4AB8CF5BA98A8BBA41803ED5D4CF766E9ECD39A8D8D914B6F346E0EB2BA6936082751676DCE5C4817EFC7A8105C2A094B22C25245BE13CA4085F2985D3B7A2636FF4018A7E4EA9840BF5FFBC0909BCD3
+20190821035339 2 6 100 1023 2 F0B5E9E385A451D4F46BD2E354B5FCAAC21CA960E5D3D11F877DD50541ED125161E4A5055D528D67E525115BBFAB0B2A4AB8CF5BA98A8BBA41803ED5D4CF766E9ECD39A8D8D914B6F346E0EB2BA6936082751676DCE5C4817EFC7A8105C2A094B22C25245BE13CA4085F2985D3B7A2636FF4018A7E4EA9840BF5FFBC090F4A2B
+20190821035340 2 6 100 1023 2 F0B5E9E385A451D4F46BD2E354B5FCAAC21CA960E5D3D11F877DD50541ED125161E4A5055D528D67E525115BBFAB0B2A4AB8CF5BA98A8BBA41803ED5D4CF766E9ECD39A8D8D914B6F346E0EB2BA6936082751676DCE5C4817EFC7A8105C2A094B22C25245BE13CA4085F2985D3B7A2636FF4018A7E4EA9840BF5FFBC0933BC13
+20190821035340 2 6 100 1023 5 F0B5E9E385A451D4F46BD2E354B5FCAAC21CA960E5D3D11F877DD50541ED125161E4A5055D528D67E525115BBFAB0B2A4AB8CF5BA98A8BBA41803ED5D4CF766E9ECD39A8D8D914B6F346E0EB2BA6936082751676DCE5C4817EFC7A8105C2A094B22C25245BE13CA4085F2985D3B7A2636FF4018A7E4EA9840BF5FFBC09395757
+20190821035340 2 6 100 1023 5 F0B5E9E385A451D4F46BD2E354B5FCAAC21CA960E5D3D11F877DD50541ED125161E4A5055D528D67E525115BBFAB0B2A4AB8CF5BA98A8BBA41803ED5D4CF766E9ECD39A8D8D914B6F346E0EB2BA6936082751676DCE5C4817EFC7A8105C2A094B22C25245BE13CA4085F2985D3B7A2636FF4018A7E4EA9840BF5FFBC093F40D7
+20190821035340 2 6 100 1023 5 F0B5E9E385A451D4F46BD2E354B5FCAAC21CA960E5D3D11F877DD50541ED125161E4A5055D528D67E525115BBFAB0B2A4AB8CF5BA98A8BBA41803ED5D4CF766E9ECD39A8D8D914B6F346E0EB2BA6936082751676DCE5C4817EFC7A8105C2A094B22C25245BE13CA4085F2985D3B7A2636FF4018A7E4EA9840BF5FFBC09478D4F
+20190821035340 2 6 100 1023 5 F0B5E9E385A451D4F46BD2E354B5FCAAC21CA960E5D3D11F877DD50541ED125161E4A5055D528D67E525115BBFAB0B2A4AB8CF5BA98A8BBA41803ED5D4CF766E9ECD39A8D8D914B6F346E0EB2BA6936082751676DCE5C4817EFC7A8105C2A094B22C25245BE13CA4085F2985D3B7A2636FF4018A7E4EA9840BF5FFBC0953A4D7
+20190821035340 2 6 100 1023 2 F0B5E9E385A451D4F46BD2E354B5FCAAC21CA960E5D3D11F877DD50541ED125161E4A5055D528D67E525115BBFAB0B2A4AB8CF5BA98A8BBA41803ED5D4CF766E9ECD39A8D8D914B6F346E0EB2BA6936082751676DCE5C4817EFC7A8105C2A094B22C25245BE13CA4085F2985D3B7A2636FF4018A7E4EA9840BF5FFBC095B5C7B
+20190821035341 2 6 100 1023 2 F0B5E9E385A451D4F46BD2E354B5FCAAC21CA960E5D3D11F877DD50541ED125161E4A5055D528D67E525115BBFAB0B2A4AB8CF5BA98A8BBA41803ED5D4CF766E9ECD39A8D8D914B6F346E0EB2BA6936082751676DCE5C4817EFC7A8105C2A094B22C25245BE13CA4085F2985D3B7A2636FF4018A7E4EA9840BF5FFBC09696573
+20190821035341 2 6 100 1023 2 F0B5E9E385A451D4F46BD2E354B5FCAAC21CA960E5D3D11F877DD50541ED125161E4A5055D528D67E525115BBFAB0B2A4AB8CF5BA98A8BBA41803ED5D4CF766E9ECD39A8D8D914B6F346E0EB2BA6936082751676DCE5C4817EFC7A8105C2A094B22C25245BE13CA4085F2985D3B7A2636FF4018A7E4EA9840BF5FFBC096BA243
+20190821035341 2 6 100 1023 2 F0B5E9E385A451D4F46BD2E354B5FCAAC21CA960E5D3D11F877DD50541ED125161E4A5055D528D67E525115BBFAB0B2A4AB8CF5BA98A8BBA41803ED5D4CF766E9ECD39A8D8D914B6F346E0EB2BA6936082751676DCE5C4817EFC7A8105C2A094B22C25245BE13CA4085F2985D3B7A2636FF4018A7E4EA9840BF5FFBC096F3903
+20190821035341 2 6 100 1023 2 F0B5E9E385A451D4F46BD2E354B5FCAAC21CA960E5D3D11F877DD50541ED125161E4A5055D528D67E525115BBFAB0B2A4AB8CF5BA98A8BBA41803ED5D4CF766E9ECD39A8D8D914B6F346E0EB2BA6936082751676DCE5C4817EFC7A8105C2A094B22C25245BE13CA4085F2985D3B7A2636FF4018A7E4EA9840BF5FFBC09850E4B
+20190821035341 2 6 100 1023 2 F0B5E9E385A451D4F46BD2E354B5FCAAC21CA960E5D3D11F877DD50541ED125161E4A5055D528D67E525115BBFAB0B2A4AB8CF5BA98A8BBA41803ED5D4CF766E9ECD39A8D8D914B6F346E0EB2BA6936082751676DCE5C4817EFC7A8105C2A094B22C25245BE13CA4085F2985D3B7A2636FF4018A7E4EA9840BF5FFBC098A1C23
+20190821035341 2 6 100 1023 5 F0B5E9E385A451D4F46BD2E354B5FCAAC21CA960E5D3D11F877DD50541ED125161E4A5055D528D67E525115BBFAB0B2A4AB8CF5BA98A8BBA41803ED5D4CF766E9ECD39A8D8D914B6F346E0EB2BA6936082751676DCE5C4817EFC7A8105C2A094B22C25245BE13CA4085F2985D3B7A2636FF4018A7E4EA9840BF5FFBC098E08E7
+20190821035342 2 6 100 1023 5 F0B5E9E385A451D4F46BD2E354B5FCAAC21CA960E5D3D11F877DD50541ED125161E4A5055D528D67E525115BBFAB0B2A4AB8CF5BA98A8BBA41803ED5D4CF766E9ECD39A8D8D914B6F346E0EB2BA6936082751676DCE5C4817EFC7A8105C2A094B22C25245BE13CA4085F2985D3B7A2636FF4018A7E4EA9840BF5FFBC09A4FF7F
+20190821035342 2 6 100 1023 5 F0B5E9E385A451D4F46BD2E354B5FCAAC21CA960E5D3D11F877DD50541ED125161E4A5055D528D67E525115BBFAB0B2A4AB8CF5BA98A8BBA41803ED5D4CF766E9ECD39A8D8D914B6F346E0EB2BA6936082751676DCE5C4817EFC7A8105C2A094B22C25245BE13CA4085F2985D3B7A2636FF4018A7E4EA9840BF5FFBC09AE4707
+20190821035342 2 6 100 1023 2 F0B5E9E385A451D4F46BD2E354B5FCAAC21CA960E5D3D11F877DD50541ED125161E4A5055D528D67E525115BBFAB0B2A4AB8CF5BA98A8BBA41803ED5D4CF766E9ECD39A8D8D914B6F346E0EB2BA6936082751676DCE5C4817EFC7A8105C2A094B22C25245BE13CA4085F2985D3B7A2636FF4018A7E4EA9840BF5FFBC09B4CE73
+20190821035342 2 6 100 1023 5 F0B5E9E385A451D4F46BD2E354B5FCAAC21CA960E5D3D11F877DD50541ED125161E4A5055D528D67E525115BBFAB0B2A4AB8CF5BA98A8BBA41803ED5D4CF766E9ECD39A8D8D914B6F346E0EB2BA6936082751676DCE5C4817EFC7A8105C2A094B22C25245BE13CA4085F2985D3B7A2636FF4018A7E4EA9840BF5FFBC09C60C6F
+20190821035342 2 6 100 1023 5 F0B5E9E385A451D4F46BD2E354B5FCAAC21CA960E5D3D11F877DD50541ED125161E4A5055D528D67E525115BBFAB0B2A4AB8CF5BA98A8BBA41803ED5D4CF766E9ECD39A8D8D914B6F346E0EB2BA6936082751676DCE5C4817EFC7A8105C2A094B22C25245BE13CA4085F2985D3B7A2636FF4018A7E4EA9840BF5FFBC09D2588F
+20190821035343 2 6 100 1023 5 F0B5E9E385A451D4F46BD2E354B5FCAAC21CA960E5D3D11F877DD50541ED125161E4A5055D528D67E525115BBFAB0B2A4AB8CF5BA98A8BBA41803ED5D4CF766E9ECD39A8D8D914B6F346E0EB2BA6936082751676DCE5C4817EFC7A8105C2A094B22C25245BE13CA4085F2985D3B7A2636FF4018A7E4EA9840BF5FFBC0A025067
+20190821035343 2 6 100 1023 2 F0B5E9E385A451D4F46BD2E354B5FCAAC21CA960E5D3D11F877DD50541ED125161E4A5055D528D67E525115BBFAB0B2A4AB8CF5BA98A8BBA41803ED5D4CF766E9ECD39A8D8D914B6F346E0EB2BA6936082751676DCE5C4817EFC7A8105C2A094B22C25245BE13CA4085F2985D3B7A2636FF4018A7E4EA9840BF5FFBC0A0E38EB
+20190821035343 2 6 100 1023 2 F0B5E9E385A451D4F46BD2E354B5FCAAC21CA960E5D3D11F877DD50541ED125161E4A5055D528D67E525115BBFAB0B2A4AB8CF5BA98A8BBA41803ED5D4CF766E9ECD39A8D8D914B6F346E0EB2BA6936082751676DCE5C4817EFC7A8105C2A094B22C25245BE13CA4085F2985D3B7A2636FF4018A7E4EA9840BF5FFBC0A213923
+20190821035344 2 6 100 1023 5 F0B5E9E385A451D4F46BD2E354B5FCAAC21CA960E5D3D11F877DD50541ED125161E4A5055D528D67E525115BBFAB0B2A4AB8CF5BA98A8BBA41803ED5D4CF766E9ECD39A8D8D914B6F346E0EB2BA6936082751676DCE5C4817EFC7A8105C2A094B22C25245BE13CA4085F2985D3B7A2636FF4018A7E4EA9840BF5FFBC0A390CA7
+20190821035344 2 6 100 1023 2 F0B5E9E385A451D4F46BD2E354B5FCAAC21CA960E5D3D11F877DD50541ED125161E4A5055D528D67E525115BBFAB0B2A4AB8CF5BA98A8BBA41803ED5D4CF766E9ECD39A8D8D914B6F346E0EB2BA6936082751676DCE5C4817EFC7A8105C2A094B22C25245BE13CA4085F2985D3B7A2636FF4018A7E4EA9840BF5FFBC0A3C7ADB
+20190821035344 2 6 100 1023 5 F0B5E9E385A451D4F46BD2E354B5FCAAC21CA960E5D3D11F877DD50541ED125161E4A5055D528D67E525115BBFAB0B2A4AB8CF5BA98A8BBA41803ED5D4CF766E9ECD39A8D8D914B6F346E0EB2BA6936082751676DCE5C4817EFC7A8105C2A094B22C25245BE13CA4085F2985D3B7A2636FF4018A7E4EA9840BF5FFBC0A44D497
+20190821035344 2 6 100 1023 2 F0B5E9E385A451D4F46BD2E354B5FCAAC21CA960E5D3D11F877DD50541ED125161E4A5055D528D67E525115BBFAB0B2A4AB8CF5BA98A8BBA41803ED5D4CF766E9ECD39A8D8D914B6F346E0EB2BA6936082751676DCE5C4817EFC7A8105C2A094B22C25245BE13CA4085F2985D3B7A2636FF4018A7E4EA9840BF5FFBC0A479B13
+20190821035345 2 6 100 1023 5 F0B5E9E385A451D4F46BD2E354B5FCAAC21CA960E5D3D11F877DD50541ED125161E4A5055D528D67E525115BBFAB0B2A4AB8CF5BA98A8BBA41803ED5D4CF766E9ECD39A8D8D914B6F346E0EB2BA6936082751676DCE5C4817EFC7A8105C2A094B22C25245BE13CA4085F2985D3B7A2636FF4018A7E4EA9840BF5FFBC0A5EF01F
+20190821035345 2 6 100 1023 2 F0B5E9E385A451D4F46BD2E354B5FCAAC21CA960E5D3D11F877DD50541ED125161E4A5055D528D67E525115BBFAB0B2A4AB8CF5BA98A8BBA41803ED5D4CF766E9ECD39A8D8D914B6F346E0EB2BA6936082751676DCE5C4817EFC7A8105C2A094B22C25245BE13CA4085F2985D3B7A2636FF4018A7E4EA9840BF5FFBC0A615D43
+20190821035345 2 6 100 1023 2 F0B5E9E385A451D4F46BD2E354B5FCAAC21CA960E5D3D11F877DD50541ED125161E4A5055D528D67E525115BBFAB0B2A4AB8CF5BA98A8BBA41803ED5D4CF766E9ECD39A8D8D914B6F346E0EB2BA6936082751676DCE5C4817EFC7A8105C2A094B22C25245BE13CA4085F2985D3B7A2636FF4018A7E4EA9840BF5FFBC0A6BEADB
+20190821035345 2 6 100 1023 5 F0B5E9E385A451D4F46BD2E354B5FCAAC21CA960E5D3D11F877DD50541ED125161E4A5055D528D67E525115BBFAB0B2A4AB8CF5BA98A8BBA41803ED5D4CF766E9ECD39A8D8D914B6F346E0EB2BA6936082751676DCE5C4817EFC7A8105C2A094B22C25245BE13CA4085F2985D3B7A2636FF4018A7E4EA9840BF5FFBC0A86309F
+20190821035345 2 6 100 1023 5 F0B5E9E385A451D4F46BD2E354B5FCAAC21CA960E5D3D11F877DD50541ED125161E4A5055D528D67E525115BBFAB0B2A4AB8CF5BA98A8BBA41803ED5D4CF766E9ECD39A8D8D914B6F346E0EB2BA6936082751676DCE5C4817EFC7A8105C2A094B22C25245BE13CA4085F2985D3B7A2636FF4018A7E4EA9840BF5FFBC0A991E8F
+20190821035346 2 6 100 1023 2 F0B5E9E385A451D4F46BD2E354B5FCAAC21CA960E5D3D11F877DD50541ED125161E4A5055D528D67E525115BBFAB0B2A4AB8CF5BA98A8BBA41803ED5D4CF766E9ECD39A8D8D914B6F346E0EB2BA6936082751676DCE5C4817EFC7A8105C2A094B22C25245BE13CA4085F2985D3B7A2636FF4018A7E4EA9840BF5FFBC0AA32C53
+20190821035346 2 6 100 1023 2 F0B5E9E385A451D4F46BD2E354B5FCAAC21CA960E5D3D11F877DD50541ED125161E4A5055D528D67E525115BBFAB0B2A4AB8CF5BA98A8BBA41803ED5D4CF766E9ECD39A8D8D914B6F346E0EB2BA6936082751676DCE5C4817EFC7A8105C2A094B22C25245BE13CA4085F2985D3B7A2636FF4018A7E4EA9840BF5FFBC0AA9FAAB
+20190821035346 2 6 100 1023 2 F0B5E9E385A451D4F46BD2E354B5FCAAC21CA960E5D3D11F877DD50541ED125161E4A5055D528D67E525115BBFAB0B2A4AB8CF5BA98A8BBA41803ED5D4CF766E9ECD39A8D8D914B6F346E0EB2BA6936082751676DCE5C4817EFC7A8105C2A094B22C25245BE13CA4085F2985D3B7A2636FF4018A7E4EA9840BF5FFBC0AAC42BB
diff --git a/test/docker/policies/policy_test1.txt b/test/docker/policies/policy_test1.txt
new file mode 100644
index 0000000..11d8e5c
--- /dev/null
+++ b/test/docker/policies/policy_test1.txt
@@ -0,0 +1,10 @@
+#
+# Docker policy: test1
+#
+
+name = "Docker policy: test1"
+version = 1
+host keys = ssh-rsa, ssh-dss
+key exchanges = diffie-hellman-group-exchange-sha256, diffie-hellman-group-exchange-sha1, diffie-hellman-group14-sha1, diffie-hellman-group1-sha1
+ciphers = aes128-ctr, aes192-ctr, aes256-ctr, arcfour256, arcfour128, aes128-cbc, 3des-cbc, blowfish-cbc, cast128-cbc, aes192-cbc, aes256-cbc, arcfour, rijndael-cbc@lysator.liu.se
+macs = hmac-md5, hmac-sha1, umac-64@openssh.com, hmac-ripemd160, hmac-ripemd160@openssh.com, hmac-sha1-96, hmac-md5-96
diff --git a/test/docker/policies/policy_test10.txt b/test/docker/policies/policy_test10.txt
new file mode 100644
index 0000000..82c821e
--- /dev/null
+++ b/test/docker/policies/policy_test10.txt
@@ -0,0 +1,39 @@
+#
+# Docker policy: test10
+#
+
+# The name of this policy (displayed in the output during scans). Must be in quotes.
+name = "Docker poliicy: test10"
+
+# The version of this policy (displayed in the output during scans). Not parsed, and may be any value, including strings.
+version = 1
+
+# The banner that must match exactly. Commented out to ignore banners, since minor variability in the banner is sometimes normal.
+# banner = "SSH-2.0-OpenSSH_5.6"
+
+# The header that must match exactly. Commented out to ignore headers, since variability in the header is sometimes normal.
+# header = "[]"
+
+# The compression options that must match exactly (order matters). Commented out to ignore by default.
+# compressions = none, zlib@openssh.com
+
+# RSA host key sizes.
+hostkey_size_rsa-sha2-256 = 3072
+hostkey_size_rsa-sha2-512 = 3072
+hostkey_size_ssh-rsa = 3072
+hostkey_size_ssh-rsa-cert-v01@openssh.com = 4096
+
+# RSA CA key sizes.
+cakey_size_ssh-rsa-cert-v01@openssh.com = 4096
+
+# The host key types that must match exactly (order matters).
+host keys = ssh-rsa, ssh-rsa-cert-v01@openssh.com
+
+# The key exchange algorithms that must match exactly (order matters).
+key exchanges = diffie-hellman-group-exchange-sha256, diffie-hellman-group-exchange-sha1, diffie-hellman-group14-sha1, diffie-hellman-group1-sha1
+
+# The ciphers that must match exactly (order matters).
+ciphers = aes128-ctr, aes192-ctr, aes256-ctr, arcfour256, arcfour128, aes128-cbc, 3des-cbc, blowfish-cbc, cast128-cbc, aes192-cbc, aes256-cbc, arcfour, rijndael-cbc@lysator.liu.se
+
+# The MACs that must match exactly (order matters).
+macs = hmac-md5, hmac-sha1, umac-64@openssh.com, hmac-ripemd160, hmac-ripemd160@openssh.com, hmac-sha1-96, hmac-md5-96
diff --git a/test/docker/policies/policy_test11.txt b/test/docker/policies/policy_test11.txt
new file mode 100644
index 0000000..d0fa4ae
--- /dev/null
+++ b/test/docker/policies/policy_test11.txt
@@ -0,0 +1,35 @@
+#
+# Docker policy: test11
+#
+
+# The name of this policy (displayed in the output during scans). Must be in quotes.
+name = "Docker policy: test11"
+
+# The version of this policy (displayed in the output during scans). Not parsed, and may be any value, including strings.
+version = 1
+
+# The banner that must match exactly. Commented out to ignore banners, since minor variability in the banner is sometimes normal.
+# banner = "SSH-2.0-OpenSSH_8.0"
+
+# The header that must match exactly. Commented out to ignore headers, since variability in the header is sometimes normal.
+# header = "[]"
+
+# The compression options that must match exactly (order matters). Commented out to ignore by default.
+# compressions = none, zlib@openssh.com
+
+# RSA host key sizes.
+hostkey_size_rsa-sha2-256 = 3072
+hostkey_size_rsa-sha2-512 = 3072
+hostkey_size_ssh-rsa = 3072
+
+# The host key types that must match exactly (order matters).
+host keys = rsa-sha2-512, rsa-sha2-256, ssh-rsa, ecdsa-sha2-nistp256, ssh-ed25519
+
+# The key exchange algorithms that must match exactly (order matters).
+key exchanges = curve25519-sha256, curve25519-sha256@libssh.org, ecdh-sha2-nistp256, ecdh-sha2-nistp384, ecdh-sha2-nistp521, diffie-hellman-group-exchange-sha256, diffie-hellman-group16-sha512, diffie-hellman-group18-sha512, diffie-hellman-group14-sha256, diffie-hellman-group14-sha1
+
+# The ciphers that must match exactly (order matters).
+ciphers = chacha20-poly1305@openssh.com, aes128-ctr, aes192-ctr, aes256-ctr, aes128-gcm@openssh.com, aes256-gcm@openssh.com
+
+# The MACs that must match exactly (order matters).
+macs = umac-64-etm@openssh.com, umac-128-etm@openssh.com, hmac-sha2-256-etm@openssh.com, hmac-sha2-512-etm@openssh.com, hmac-sha1-etm@openssh.com, umac-64@openssh.com, umac-128@openssh.com, hmac-sha2-256, hmac-sha2-512, hmac-sha1
diff --git a/test/docker/policies/policy_test12.txt b/test/docker/policies/policy_test12.txt
new file mode 100644
index 0000000..0b8a30b
--- /dev/null
+++ b/test/docker/policies/policy_test12.txt
@@ -0,0 +1,35 @@
+#
+# Docker policy: test12
+#
+
+# The name of this policy (displayed in the output during scans). Must be in quotes.
+name = "Docker policy: test12"
+
+# The version of this policy (displayed in the output during scans). Not parsed, and may be any value, including strings.
+version = 1
+
+# The banner that must match exactly. Commented out to ignore banners, since minor variability in the banner is sometimes normal.
+# banner = "SSH-2.0-OpenSSH_8.0"
+
+# The header that must match exactly. Commented out to ignore headers, since variability in the header is sometimes normal.
+# header = "[]"
+
+# The compression options that must match exactly (order matters). Commented out to ignore by default.
+# compressions = none, zlib@openssh.com
+
+# RSA host key sizes.
+hostkey_size_rsa-sha2-256 = 4096
+hostkey_size_rsa-sha2-512 = 4096
+hostkey_size_ssh-rsa = 4096
+
+# The host key types that must match exactly (order matters).
+host keys = rsa-sha2-512, rsa-sha2-256, ssh-rsa, ecdsa-sha2-nistp256, ssh-ed25519
+
+# The key exchange algorithms that must match exactly (order matters).
+key exchanges = curve25519-sha256, curve25519-sha256@libssh.org, ecdh-sha2-nistp256, ecdh-sha2-nistp384, ecdh-sha2-nistp521, diffie-hellman-group-exchange-sha256, diffie-hellman-group16-sha512, diffie-hellman-group18-sha512, diffie-hellman-group14-sha256, diffie-hellman-group14-sha1
+
+# The ciphers that must match exactly (order matters).
+ciphers = chacha20-poly1305@openssh.com, aes128-ctr, aes192-ctr, aes256-ctr, aes128-gcm@openssh.com, aes256-gcm@openssh.com
+
+# The MACs that must match exactly (order matters).
+macs = umac-64-etm@openssh.com, umac-128-etm@openssh.com, hmac-sha2-256-etm@openssh.com, hmac-sha2-512-etm@openssh.com, hmac-sha1-etm@openssh.com, umac-64@openssh.com, umac-128@openssh.com, hmac-sha2-256, hmac-sha2-512, hmac-sha1
diff --git a/test/docker/policies/policy_test13.txt b/test/docker/policies/policy_test13.txt
new file mode 100644
index 0000000..5e9bbd4
--- /dev/null
+++ b/test/docker/policies/policy_test13.txt
@@ -0,0 +1,38 @@
+#
+# Docker policy: test13
+#
+
+# The name of this policy (displayed in the output during scans). Must be in quotes.
+name = "Docker policy: test13"
+
+# The version of this policy (displayed in the output during scans). Not parsed, and may be any value, including strings.
+version = 1
+
+# The banner that must match exactly. Commented out to ignore banners, since minor variability in the banner is sometimes normal.
+# banner = "SSH-2.0-OpenSSH_8.0"
+
+# The header that must match exactly. Commented out to ignore headers, since variability in the header is sometimes normal.
+# header = "[]"
+
+# The compression options that must match exactly (order matters). Commented out to ignore by default.
+# compressions = none, zlib@openssh.com
+
+# RSA host key sizes.
+hostkey_size_rsa-sha2-256 = 3072
+hostkey_size_rsa-sha2-512 = 3072
+hostkey_size_ssh-rsa = 3072
+
+# Group exchange DH modulus sizes.
+dh_modulus_size_diffie-hellman-group-exchange-sha256 = 2048
+
+# The host key types that must match exactly (order matters).
+host keys = rsa-sha2-512, rsa-sha2-256, ssh-rsa, ecdsa-sha2-nistp256, ssh-ed25519
+
+# The key exchange algorithms that must match exactly (order matters).
+key exchanges = curve25519-sha256, curve25519-sha256@libssh.org, ecdh-sha2-nistp256, ecdh-sha2-nistp384, ecdh-sha2-nistp521, diffie-hellman-group-exchange-sha256, diffie-hellman-group16-sha512, diffie-hellman-group18-sha512, diffie-hellman-group14-sha256, diffie-hellman-group14-sha1
+
+# The ciphers that must match exactly (order matters).
+ciphers = chacha20-poly1305@openssh.com, aes128-ctr, aes192-ctr, aes256-ctr, aes128-gcm@openssh.com, aes256-gcm@openssh.com
+
+# The MACs that must match exactly (order matters).
+macs = umac-64-etm@openssh.com, umac-128-etm@openssh.com, hmac-sha2-256-etm@openssh.com, hmac-sha2-512-etm@openssh.com, hmac-sha1-etm@openssh.com, umac-64@openssh.com, umac-128@openssh.com, hmac-sha2-256, hmac-sha2-512, hmac-sha1
diff --git a/test/docker/policies/policy_test14.txt b/test/docker/policies/policy_test14.txt
new file mode 100644
index 0000000..ce1fec8
--- /dev/null
+++ b/test/docker/policies/policy_test14.txt
@@ -0,0 +1,38 @@
+#
+# Docker policy: test14
+#
+
+# The name of this policy (displayed in the output during scans). Must be in quotes.
+name = "Docker policy: test14"
+
+# The version of this policy (displayed in the output during scans). Not parsed, and may be any value, including strings.
+version = 1
+
+# The banner that must match exactly. Commented out to ignore banners, since minor variability in the banner is sometimes normal.
+# banner = "SSH-2.0-OpenSSH_8.0"
+
+# The header that must match exactly. Commented out to ignore headers, since variability in the header is sometimes normal.
+# header = "[]"
+
+# The compression options that must match exactly (order matters). Commented out to ignore by default.
+# compressions = none, zlib@openssh.com
+
+# RSA host key sizes.
+hostkey_size_rsa-sha2-256 = 3072
+hostkey_size_rsa-sha2-512 = 3072
+hostkey_size_ssh-rsa = 3072
+
+# Group exchange DH modulus sizes.
+dh_modulus_size_diffie-hellman-group-exchange-sha256 = 4096
+
+# The host key types that must match exactly (order matters).
+host keys = rsa-sha2-512, rsa-sha2-256, ssh-rsa, ecdsa-sha2-nistp256, ssh-ed25519
+
+# The key exchange algorithms that must match exactly (order matters).
+key exchanges = curve25519-sha256, curve25519-sha256@libssh.org, ecdh-sha2-nistp256, ecdh-sha2-nistp384, ecdh-sha2-nistp521, diffie-hellman-group-exchange-sha256, diffie-hellman-group16-sha512, diffie-hellman-group18-sha512, diffie-hellman-group14-sha256, diffie-hellman-group14-sha1
+
+# The ciphers that must match exactly (order matters).
+ciphers = chacha20-poly1305@openssh.com, aes128-ctr, aes192-ctr, aes256-ctr, aes128-gcm@openssh.com, aes256-gcm@openssh.com
+
+# The MACs that must match exactly (order matters).
+macs = umac-64-etm@openssh.com, umac-128-etm@openssh.com, hmac-sha2-256-etm@openssh.com, hmac-sha2-512-etm@openssh.com, hmac-sha1-etm@openssh.com, umac-64@openssh.com, umac-128@openssh.com, hmac-sha2-256, hmac-sha2-512, hmac-sha1
diff --git a/test/docker/policies/policy_test2.txt b/test/docker/policies/policy_test2.txt
new file mode 100644
index 0000000..2b7821c
--- /dev/null
+++ b/test/docker/policies/policy_test2.txt
@@ -0,0 +1,10 @@
+#
+# Docker policy: test2
+#
+
+name = "Docker policy: test2"
+version = 1
+host keys = ssh-rsa, ssh-dss
+key exchanges = kex_alg1, kex_alg2
+ciphers = aes128-ctr, aes192-ctr, aes256-ctr, arcfour256, arcfour128, aes128-cbc, 3des-cbc, blowfish-cbc, cast128-cbc, aes192-cbc, aes256-cbc, arcfour, rijndael-cbc@lysator.liu.se
+macs = hmac-md5, hmac-sha1, umac-64@openssh.com, hmac-ripemd160, hmac-ripemd160@openssh.com, hmac-sha1-96, hmac-md5-96
diff --git a/test/docker/policies/policy_test3.txt b/test/docker/policies/policy_test3.txt
new file mode 100644
index 0000000..f4ff3a0
--- /dev/null
+++ b/test/docker/policies/policy_test3.txt
@@ -0,0 +1,10 @@
+#
+# Docker policy: test3
+#
+
+name = "Docker policy: test3"
+version = 1
+host keys = ssh-rsa, ssh-dss, key_alg1
+key exchanges = diffie-hellman-group-exchange-sha256, diffie-hellman-group-exchange-sha1, diffie-hellman-group14-sha1, diffie-hellman-group1-sha1
+ciphers = aes128-ctr, aes192-ctr, aes256-ctr, arcfour256, arcfour128, aes128-cbc, 3des-cbc, blowfish-cbc, cast128-cbc, aes192-cbc, aes256-cbc, arcfour, rijndael-cbc@lysator.liu.se
+macs = hmac-md5, hmac-sha1, umac-64@openssh.com, hmac-ripemd160, hmac-ripemd160@openssh.com, hmac-sha1-96, hmac-md5-96
diff --git a/test/docker/policies/policy_test4.txt b/test/docker/policies/policy_test4.txt
new file mode 100644
index 0000000..500d96f
--- /dev/null
+++ b/test/docker/policies/policy_test4.txt
@@ -0,0 +1,10 @@
+#
+# Docker policy: test4
+#
+
+name = "Docker policy: test4"
+version = 1
+host keys = ssh-rsa, ssh-dss
+key exchanges = diffie-hellman-group-exchange-sha256, diffie-hellman-group-exchange-sha1, diffie-hellman-group14-sha1, diffie-hellman-group1-sha1
+ciphers = cipher_alg1, cipher_alg2
+macs = hmac-md5, hmac-sha1, umac-64@openssh.com, hmac-ripemd160, hmac-ripemd160@openssh.com, hmac-sha1-96, hmac-md5-96
diff --git a/test/docker/policies/policy_test5.txt b/test/docker/policies/policy_test5.txt
new file mode 100644
index 0000000..6285814
--- /dev/null
+++ b/test/docker/policies/policy_test5.txt
@@ -0,0 +1,10 @@
+#
+# Docker policy: test5
+#
+
+name = "Docker policy: test5"
+version = 1
+host keys = ssh-rsa, ssh-dss
+key exchanges = diffie-hellman-group-exchange-sha256, diffie-hellman-group-exchange-sha1, diffie-hellman-group14-sha1, diffie-hellman-group1-sha1
+ciphers = aes128-ctr, aes192-ctr, aes256-ctr, arcfour256, arcfour128, aes128-cbc, 3des-cbc, blowfish-cbc, cast128-cbc, aes192-cbc, aes256-cbc, arcfour, rijndael-cbc@lysator.liu.se
+macs = hmac-md5, hmac-sha1, umac-64@openssh.com, hmac-ripemd160, hmac-ripemd160@openssh.com, hmac_alg1, hmac-md5-96
diff --git a/test/docker/policies/policy_test6.txt b/test/docker/policies/policy_test6.txt
new file mode 100644
index 0000000..0a4aacb
--- /dev/null
+++ b/test/docker/policies/policy_test6.txt
@@ -0,0 +1,12 @@
+#
+# Docker policy: test6
+#
+
+name = "Docker policy: test6"
+version = 1
+banner = "SSH-2.0-OpenSSH_8.0"
+compressions = none, zlib@openssh.com
+host keys = rsa-sha2-512, rsa-sha2-256, ssh-rsa, ecdsa-sha2-nistp256, ssh-ed25519
+key exchanges = curve25519-sha256, curve25519-sha256@libssh.org, ecdh-sha2-nistp256, ecdh-sha2-nistp384, ecdh-sha2-nistp521, diffie-hellman-group-exchange-sha256, diffie-hellman-group16-sha512, diffie-hellman-group18-sha512, diffie-hellman-group14-sha256, diffie-hellman-group14-sha1
+ciphers = chacha20-poly1305@openssh.com, aes128-ctr, aes192-ctr, aes256-ctr, aes128-gcm@openssh.com, aes256-gcm@openssh.com
+macs = umac-64-etm@openssh.com, umac-128-etm@openssh.com, hmac-sha2-256-etm@openssh.com, hmac-sha2-512-etm@openssh.com, hmac-sha1-etm@openssh.com, umac-64@openssh.com, umac-128@openssh.com, hmac-sha2-256, hmac-sha2-512, hmac-sha1
diff --git a/test/docker/policies/policy_test7.txt b/test/docker/policies/policy_test7.txt
new file mode 100644
index 0000000..05cd27f
--- /dev/null
+++ b/test/docker/policies/policy_test7.txt
@@ -0,0 +1,39 @@
+#
+# Docker policy: test7
+#
+
+# The name of this policy (displayed in the output during scans). Must be in quotes.
+name = "Docker poliicy: test7"
+
+# The version of this policy (displayed in the output during scans). Not parsed, and may be any value, including strings.
+version = 1
+
+# The banner that must match exactly. Commented out to ignore banners, since minor variability in the banner is sometimes normal.
+# banner = "SSH-2.0-OpenSSH_5.6"
+
+# The header that must match exactly. Commented out to ignore headers, since variability in the header is sometimes normal.
+# header = "[]"
+
+# The compression options that must match exactly (order matters). Commented out to ignore by default.
+# compressions = none, zlib@openssh.com
+
+# RSA host key sizes.
+hostkey_size_rsa-sha2-256 = 3072
+hostkey_size_rsa-sha2-512 = 3072
+hostkey_size_ssh-rsa = 3072
+hostkey_size_ssh-rsa-cert-v01@openssh.com = 3072
+
+# RSA CA key sizes.
+cakey_size_ssh-rsa-cert-v01@openssh.com = 1024
+
+# The host key types that must match exactly (order matters).
+host keys = ssh-rsa, ssh-rsa-cert-v01@openssh.com
+
+# The key exchange algorithms that must match exactly (order matters).
+key exchanges = diffie-hellman-group-exchange-sha256, diffie-hellman-group-exchange-sha1, diffie-hellman-group14-sha1, diffie-hellman-group1-sha1
+
+# The ciphers that must match exactly (order matters).
+ciphers = aes128-ctr, aes192-ctr, aes256-ctr, arcfour256, arcfour128, aes128-cbc, 3des-cbc, blowfish-cbc, cast128-cbc, aes192-cbc, aes256-cbc, arcfour, rijndael-cbc@lysator.liu.se
+
+# The MACs that must match exactly (order matters).
+macs = hmac-md5, hmac-sha1, umac-64@openssh.com, hmac-ripemd160, hmac-ripemd160@openssh.com, hmac-sha1-96, hmac-md5-96
diff --git a/test/docker/policies/policy_test8.txt b/test/docker/policies/policy_test8.txt
new file mode 100644
index 0000000..6268585
--- /dev/null
+++ b/test/docker/policies/policy_test8.txt
@@ -0,0 +1,39 @@
+#
+# Docker policy: test8
+#
+
+# The name of this policy (displayed in the output during scans). Must be in quotes.
+name = "Docker poliicy: test8"
+
+# The version of this policy (displayed in the output during scans). Not parsed, and may be any value, including strings.
+version = 1
+
+# The banner that must match exactly. Commented out to ignore banners, since minor variability in the banner is sometimes normal.
+# banner = "SSH-2.0-OpenSSH_5.6"
+
+# The header that must match exactly. Commented out to ignore headers, since variability in the header is sometimes normal.
+# header = "[]"
+
+# The compression options that must match exactly (order matters). Commented out to ignore by default.
+# compressions = none, zlib@openssh.com
+
+# RSA host key sizes.
+hostkey_size_rsa-sha2-256 = 3072
+hostkey_size_rsa-sha2-512 = 3072
+hostkey_size_ssh-rsa = 3072
+hostkey_size_ssh-rsa-cert-v01@openssh.com = 3072
+
+# RSA CA key sizes.
+cakey_size_ssh-rsa-cert-v01@openssh.com = 2048
+
+# The host key types that must match exactly (order matters).
+host keys = ssh-rsa, ssh-rsa-cert-v01@openssh.com
+
+# The key exchange algorithms that must match exactly (order matters).
+key exchanges = diffie-hellman-group-exchange-sha256, diffie-hellman-group-exchange-sha1, diffie-hellman-group14-sha1, diffie-hellman-group1-sha1
+
+# The ciphers that must match exactly (order matters).
+ciphers = aes128-ctr, aes192-ctr, aes256-ctr, arcfour256, arcfour128, aes128-cbc, 3des-cbc, blowfish-cbc, cast128-cbc, aes192-cbc, aes256-cbc, arcfour, rijndael-cbc@lysator.liu.se
+
+# The MACs that must match exactly (order matters).
+macs = hmac-md5, hmac-sha1, umac-64@openssh.com, hmac-ripemd160, hmac-ripemd160@openssh.com, hmac-sha1-96, hmac-md5-96
diff --git a/test/docker/policies/policy_test9.txt b/test/docker/policies/policy_test9.txt
new file mode 100644
index 0000000..63652ce
--- /dev/null
+++ b/test/docker/policies/policy_test9.txt
@@ -0,0 +1,39 @@
+#
+# Docker policy: test9
+#
+
+# The name of this policy (displayed in the output during scans). Must be in quotes.
+name = "Docker poliicy: test9"
+
+# The version of this policy (displayed in the output during scans). Not parsed, and may be any value, including strings.
+version = 1
+
+# The banner that must match exactly. Commented out to ignore banners, since minor variability in the banner is sometimes normal.
+# banner = "SSH-2.0-OpenSSH_5.6"
+
+# The header that must match exactly. Commented out to ignore headers, since variability in the header is sometimes normal.
+# header = "[]"
+
+# The compression options that must match exactly (order matters). Commented out to ignore by default.
+# compressions = none, zlib@openssh.com
+
+# RSA host key sizes.
+hostkey_size_rsa-sha2-256 = 3072
+hostkey_size_rsa-sha2-512 = 3072
+hostkey_size_ssh-rsa = 3072
+hostkey_size_ssh-rsa-cert-v01@openssh.com = 4096
+
+# RSA CA key sizes.
+cakey_size_ssh-rsa-cert-v01@openssh.com = 1024
+
+# The host key types that must match exactly (order matters).
+host keys = ssh-rsa, ssh-rsa-cert-v01@openssh.com
+
+# The key exchange algorithms that must match exactly (order matters).
+key exchanges = diffie-hellman-group-exchange-sha256, diffie-hellman-group-exchange-sha1, diffie-hellman-group14-sha1, diffie-hellman-group1-sha1
+
+# The ciphers that must match exactly (order matters).
+ciphers = aes128-ctr, aes192-ctr, aes256-ctr, arcfour256, arcfour128, aes128-cbc, 3des-cbc, blowfish-cbc, cast128-cbc, aes192-cbc, aes256-cbc, arcfour, rijndael-cbc@lysator.liu.se
+
+# The MACs that must match exactly (order matters).
+macs = hmac-md5, hmac-sha1, umac-64@openssh.com, hmac-ripemd160, hmac-ripemd160@openssh.com, hmac-sha1-96, hmac-md5-96
diff --git a/test/docker/ssh1_host_key b/test/docker/ssh1_host_key
new file mode 100644
index 0000000..c98971c
--- /dev/null
+++ b/test/docker/ssh1_host_key
Binary files differ
diff --git a/test/docker/ssh1_host_key.pub b/test/docker/ssh1_host_key.pub
new file mode 100644
index 0000000..b66c66f
--- /dev/null
+++ b/test/docker/ssh1_host_key.pub
@@ -0,0 +1 @@
+1024 35 150823875409720459951648542224727752099073441604930026287525797402159071426070997897033651155038337251362080634963146983947007228274330777134724953282680928153520263171933106732090266742784258910450489054624715996015082463159338507115031336180486071622718809324273851629938883104520608180885444242395900180011 root@ubuntu1604server
diff --git a/test/docker/ssh_host_dsa_key b/test/docker/ssh_host_dsa_key
new file mode 100644
index 0000000..ecd47f9
--- /dev/null
+++ b/test/docker/ssh_host_dsa_key
@@ -0,0 +1,12 @@
+-----BEGIN DSA PRIVATE KEY-----
+MIIBugIBAAKBgQDth1eV+A8j191R0ey0dVXL2LGNGYM+a+PomSa7suK8xNCeVLKC
+YpQ6VSWpAf6FbRWev1UVo8IpbglwFZPcyFPK2G1H7p45ows2SN4CleszDD56e6W0
+3Plc+qMqSJ6LTjr4M5+HqTDOM3CS72d7MXUkfHQiagyrWQhXyc0kFsNJLwIVAKg7
+b5+NiIZzpg5IEH0tlYFQpuhBAoGAGcbq79QqNNZRuPCE/F05sCoTRGCmFnDjCuCg
+WN7wNRotjMz/S3pHtCCeuTT1jT6Hy0ZFHftv0t/GF8GBRgeokUbS4ytHpOkFWcTz
+8oFguDL44nq8eNfSY6bzEl84qsgEe4HP93mB4FR1ZUUgI4b7gCBOYEFl3yPiH7H1
+p7Z9E1oCgYAl1UPQkeRhElz+AgEbNsnMKu1+6O3/z95D1Wvv4OEwAImbytlBaC7p
+kwJElJNsMMfGqCC8OHdJ0e4VQQUwk/GOhD0MFhVQHBtVZYbiWmVkpfHf1ouUQg3f
+1IZmz2SSt6cPPEu+BEQ/Sn3mFRJ5XSTHLtnI0HJeDND5u1+6p1nXawIURv3Maige
+oxmfqC24VoROJEq+sew=
+-----END DSA PRIVATE KEY-----
diff --git a/test/docker/ssh_host_dsa_key.pub b/test/docker/ssh_host_dsa_key.pub
new file mode 100644
index 0000000..a32a5a0
--- /dev/null
+++ b/test/docker/ssh_host_dsa_key.pub
@@ -0,0 +1 @@
+ssh-dss AAAAB3NzaC1kc3MAAACBAO2HV5X4DyPX3VHR7LR1VcvYsY0Zgz5r4+iZJruy4rzE0J5UsoJilDpVJakB/oVtFZ6/VRWjwiluCXAVk9zIU8rYbUfunjmjCzZI3gKV6zMMPnp7pbTc+Vz6oypInotOOvgzn4epMM4zcJLvZ3sxdSR8dCJqDKtZCFfJzSQWw0kvAAAAFQCoO2+fjYiGc6YOSBB9LZWBUKboQQAAAIAZxurv1Co01lG48IT8XTmwKhNEYKYWcOMK4KBY3vA1Gi2MzP9Leke0IJ65NPWNPofLRkUd+2/S38YXwYFGB6iRRtLjK0ek6QVZxPPygWC4Mvjierx419JjpvMSXziqyAR7gc/3eYHgVHVlRSAjhvuAIE5gQWXfI+IfsfWntn0TWgAAAIAl1UPQkeRhElz+AgEbNsnMKu1+6O3/z95D1Wvv4OEwAImbytlBaC7pkwJElJNsMMfGqCC8OHdJ0e4VQQUwk/GOhD0MFhVQHBtVZYbiWmVkpfHf1ouUQg3f1IZmz2SSt6cPPEu+BEQ/Sn3mFRJ5XSTHLtnI0HJeDND5u1+6p1nXaw==
diff --git a/test/docker/ssh_host_ecdsa_key b/test/docker/ssh_host_ecdsa_key
new file mode 100644
index 0000000..69eea7b
--- /dev/null
+++ b/test/docker/ssh_host_ecdsa_key
@@ -0,0 +1,5 @@
+-----BEGIN EC PRIVATE KEY-----
+MHcCAQEEICq/YV5QenL0uW5g5tCjY3EWs+UBFmskY+Jjt2vd2aEmoAoGCCqGSM49
+AwEHoUQDQgAEdYSxDVUjOpW479L/nRDiAdxRB5Kuy2bgkP/LA2pnWPcGIWmFa4QU
+YN2U3JsFKcLIcx5cvTehQfgrHDnaSKVdKA==
+-----END EC PRIVATE KEY-----
diff --git a/test/docker/ssh_host_ecdsa_key.pub b/test/docker/ssh_host_ecdsa_key.pub
new file mode 100644
index 0000000..4e17058
--- /dev/null
+++ b/test/docker/ssh_host_ecdsa_key.pub
@@ -0,0 +1 @@
+ecdsa-sha2-nistp256 AAAAE2VjZHNhLXNoYTItbmlzdHAyNTYAAAAIbmlzdHAyNTYAAABBBHWEsQ1VIzqVuO/S/50Q4gHcUQeSrstm4JD/ywNqZ1j3BiFphWuEFGDdlNybBSnCyHMeXL03oUH4Kxw52kilXSg=
diff --git a/test/docker/ssh_host_ed25519_key b/test/docker/ssh_host_ed25519_key
new file mode 100644
index 0000000..3388574
--- /dev/null
+++ b/test/docker/ssh_host_ed25519_key
@@ -0,0 +1,7 @@
+-----BEGIN OPENSSH PRIVATE KEY-----
+b3BlbnNzaC1rZXktdjEAAAAABG5vbmUAAAAEbm9uZQAAAAAAAAABAAAAMwAAAAtzc2gtZW
+QyNTUxOQAAACC/9RD2Ao95ODDIH8i11ekTALut8AUNqWoQx0jHlP4xygAAAKDiqVOs4qlT
+rAAAAAtzc2gtZWQyNTUxOQAAACC/9RD2Ao95ODDIH8i11ekTALut8AUNqWoQx0jHlP4xyg
+AAAECTmHGkq0Qea0QqTJYMXL0bpxVU7mhgwYninfVWxrA017/1EPYCj3k4MMgfyLXV6RMA
+u63wBQ2pahDHSMeU/jHKAAAAHWpkb2dAbG9jYWxob3N0LndvbmRlcmxhbmQubG9s
+-----END OPENSSH PRIVATE KEY-----
diff --git a/test/docker/ssh_host_ed25519_key-cert.pub b/test/docker/ssh_host_ed25519_key-cert.pub
new file mode 100644
index 0000000..8eef563
--- /dev/null
+++ b/test/docker/ssh_host_ed25519_key-cert.pub
@@ -0,0 +1 @@
+ssh-ed25519-cert-v01@openssh.com AAAAIHNzaC1lZDI1NTE5LWNlcnQtdjAxQG9wZW5zc2guY29tAAAAIO1W0I8tD0c4LypvHY1XNch3BQCw9Yy28/4KmAYql80DAAAAIL/1EPYCj3k4MMgfyLXV6RMAu63wBQ2pahDHSMeU/jHKAAAAAAAAAAAAAAACAAAABHRlc3QAAAAIAAAABHRlc3QAAAAAXV7hvAAAAACBa2YhAAAAAAAAAAAAAAAAAAAAMwAAAAtzc2gtZWQyNTUxOQAAACAbM9Wp3ZPcC8Ifhu6GjNDJaoMg7KxO0el2+r9J35TltQAAAFMAAAALc3NoLWVkMjU1MTkAAABAW60bCSeIG4Ta+57zgkSbW4LIGCxtOuJJ+pP3i3S0xJJfHGnOtXbg0NQm7pulNl/wd01kgJO9A7RjbhTh7TV1AA== ssh_host_ed25519_key.pub
diff --git a/test/docker/ssh_host_ed25519_key.pub b/test/docker/ssh_host_ed25519_key.pub
new file mode 100644
index 0000000..e56a56a
--- /dev/null
+++ b/test/docker/ssh_host_ed25519_key.pub
@@ -0,0 +1 @@
+ssh-ed25519 AAAAC3NzaC1lZDI1NTE5AAAAIL/1EPYCj3k4MMgfyLXV6RMAu63wBQ2pahDHSMeU/jHK
diff --git a/test/docker/ssh_host_rsa_key_1024 b/test/docker/ssh_host_rsa_key_1024
new file mode 100644
index 0000000..e9023b6
--- /dev/null
+++ b/test/docker/ssh_host_rsa_key_1024
@@ -0,0 +1,15 @@
+-----BEGIN RSA PRIVATE KEY-----
+MIICXgIBAAKBgQDeCC1U7VqVg9AfrfWrXACiW6pzYOuP8tim68z+YN/dUU7JhFZ4
+0toteQkLcJBAD2miQ6ZJYkjVfhQ4FRFeOW5vcN0UYHn8ttb2mKdGJdt24ZYY5Z6J
+WHQhPOpSgtWyUv6RnxU2ligEeaoPaiepUUOhoyLf4WcF7voVCAKZNqeTtQIDAQAB
+AoGATGZ16s5NqDsWJ4B9k3xx/2wZZ+BGzl6a7D0habq97XLn8HGoK6UqTBFk6lnO
+WSy0hZBPrNq0AzqCDJY7RrfuZqgVAu/+HEFuXencgt8Z//ueYBaGK8yAC+OrMnDG
+LbSoIGRq8saaFtCzt47c+uSVsrhJ4TvK5gbceZuD/2uw10ECQQD79T0j+YWsLISK
+PKvYHqEXSMPN6b+lK9hRPLoF9NMksNLSjuxxhkYHz+hJPVNT+wPtRMAYmMdPXfKa
+FjuErXVFAkEA4ZgJIOeJ7OHfqGEgd29m36yFy0UaUJ+cmYuJzHAYWgW3TOanqpZm
+A8EENuXvH0DtYRVytv4m/cIRVVPxWtXzsQJBALXlQUOEc0VuSi1GScVXr3KQ3JL+
+ipWixqM3VRDRw9D8Ouc5uWbnygz/wrGFLXA2ioozlP7s5Q7eQzOMk2FgnIUCQQCz
+j5QUgLcjuVWQbF6vMhisCGImPUaIzcKT5KE1/DMl1E7mAuGJwlRIwKVeHP6L3d4T
+3EKGrRzT9lhdlocRSiLBAkEAi3xI0MOZp4xGviPc1C1TKuqdJSr8fHwbtLozwNQO
+nnF6m5S72JzZEThDBZS9zcdBp9EFpTvUGzx/O0GI454eoA==
+-----END RSA PRIVATE KEY-----
diff --git a/test/docker/ssh_host_rsa_key_1024-cert_1024.pub b/test/docker/ssh_host_rsa_key_1024-cert_1024.pub
new file mode 100644
index 0000000..17c7738
--- /dev/null
+++ b/test/docker/ssh_host_rsa_key_1024-cert_1024.pub
@@ -0,0 +1 @@
+ssh-rsa-cert-v01@openssh.com AAAAHHNzaC1yc2EtY2VydC12MDFAb3BlbnNzaC5jb20AAAAgFqxXSa9HTqCw5YW3DdIwVREPGxI+i56w32RnHWRg0NoAAAADAQABAAAAgQDeCC1U7VqVg9AfrfWrXACiW6pzYOuP8tim68z+YN/dUU7JhFZ40toteQkLcJBAD2miQ6ZJYkjVfhQ4FRFeOW5vcN0UYHn8ttb2mKdGJdt24ZYY5Z6JWHQhPOpSgtWyUv6RnxU2ligEeaoPaiepUUOhoyLf4WcF7voVCAKZNqeTtQAAAAAAAAAAAAAAAgAAAAR0ZXN0AAAACAAAAAR0ZXN0AAAAAF1evHgAAAAAgWtAtQAAAAAAAAAAAAAAAAAAAJcAAAAHc3NoLXJzYQAAAAMBAAEAAACBAOdGU3cAWdR7aUV/lcb1SFcuv/086u41MVsy3TOuM5RKOYBLuFLqkO/lUROhPoNpHURBRhqpIykdjNmG4Irna+vJ06blcVsnvvXav0zJlBSGhVnHxK50EfNUMhU7eNwwxYiWt9YEydRpQcSqmbLzxjuAoNrZNbEmDX6GDnUqoetRAAAAjwAAAAdzc2gtcnNhAAAAgFRc2g0eWXGmqSa6Z8rcMPHf4rNMornEHSnTzZ8Rdh4YBhDa9xMRRy6puaWPDzXxOfZh7eSjCwrzUMEXTgCIO4oX62xm/6kUnAmKhXle0+inR/hdPg03daE0SBJ4spBT49lJ4WIW38RKFNzjmYg70rTPAnP8oM/F3CC1GV117/Vv ssh_host_rsa_key_1024.pub
diff --git a/test/docker/ssh_host_rsa_key_1024-cert_3072.pub b/test/docker/ssh_host_rsa_key_1024-cert_3072.pub
new file mode 100644
index 0000000..ea0160a
--- /dev/null
+++ b/test/docker/ssh_host_rsa_key_1024-cert_3072.pub
@@ -0,0 +1 @@
+ssh-rsa-cert-v01@openssh.com AAAAHHNzaC1yc2EtY2VydC12MDFAb3BlbnNzaC5jb20AAAAgrJGcfyW8V6VWGT7lD1ardj2RtTP8TOjmLRNbuoGkyZQAAAADAQABAAAAgQDeCC1U7VqVg9AfrfWrXACiW6pzYOuP8tim68z+YN/dUU7JhFZ40toteQkLcJBAD2miQ6ZJYkjVfhQ4FRFeOW5vcN0UYHn8ttb2mKdGJdt24ZYY5Z6JWHQhPOpSgtWyUv6RnxU2ligEeaoPaiepUUOhoyLf4WcF7voVCAKZNqeTtQAAAAAAAAAAAAAAAgAAAAR0ZXN0AAAACAAAAAR0ZXN0AAAAAF1evHgAAAAAgWtA7gAAAAAAAAAAAAAAAAAAAZcAAAAHc3NoLXJzYQAAAAMBAAEAAAGBAKsUBCG4/MNE6wWNXwzvNXLq4o0K3SzlK/FZfFxAlnOUBndS5ABvNDYHXAgT741sWUpmKynqgO/BlHTfioMnBNM7acWElyh2TGiDOV3veAQfgaHB/db3f9ogcxnnndZ1f83wHatgBB0I9U6SCPmvs9ucsb9j7sFU08MKPSP+iYDat3rx+B28IK90kXxBGGUPmC/HDqQbjYNWGTkdJNcpNt2agHfdicm8WeIgrTre63I9rBR8zbtVOeOMFgcCC+cmWSk4AYWOHan09hKeyM0eR3bHa70FfagvdhjNcwxPlNwMK1Sox7NGGat483Kv8rxQwnDDEc+P9KRhRVNuqFpmvp972wMSiaC/q4RvEDPfpdqhZSxrJzNJFFwCV7Dym8oKWj/Gd1OHioE7idhrkvXiyBZwe5q8A++VehoDfauZmPPQ35FUpB2qKqWPKT3UZM/UAJjBLi7R5r9vQlTqX1+b/67Ju8AEgGarVJZhbuKxRgS8L2W1JVrtNh5oAU+O4OhCHwAAAY8AAAAHc3NoLXJzYQAAAYCO78ONpHp+EGWDqDLb/GPFDH32o6oaRRrIH/Bhvg1FOi5XngnHTdU7xWdnJqNE2Sl6VOrg0sTCMYcX9fZ8tVREnCo3aF7Iwow5Br67QYKayRzHANQqHaVK46lpI1gz81V00u54tX1F8oEUqm6sRmFKFuklt6CjfbR+tnpj7DrfeOTKEBOGJP2uU0jMsJr2DrBeXrzONjIJtIJ1AxWjXd2LeIWO2C6yTkcN5ggThMMaeu6QuuBPpC2PN2COfu+Mgto9g103+/SS4Wa8CzinZZn2Xe1isUxI8QRNrShy4Hl/bIZQL7mi/0rxkfw+fA7IzMk462V99gPVSp+/jK0sbUJoC3QeeglS5hWodjW+VGZfgweGQ+AE/OxkNSv+kDPMYEkjfOf4qhxS5QFvButLt6zp2UNbE5+OWvYpdjO9/DOa0ro+wCw07+dVKIcDpU2csiCcJvQ/HmKAmhch7jOHa0WaxSX0tt0xTPJWTvr6E4WZOgEnk9AvWmrKjF5tEzGYwTU= ssh_host_rsa_key_1024.pub
diff --git a/test/docker/ssh_host_rsa_key_1024.pub b/test/docker/ssh_host_rsa_key_1024.pub
new file mode 100644
index 0000000..1da6065
--- /dev/null
+++ b/test/docker/ssh_host_rsa_key_1024.pub
@@ -0,0 +1 @@
+ssh-rsa AAAAB3NzaC1yc2EAAAADAQABAAAAgQDeCC1U7VqVg9AfrfWrXACiW6pzYOuP8tim68z+YN/dUU7JhFZ40toteQkLcJBAD2miQ6ZJYkjVfhQ4FRFeOW5vcN0UYHn8ttb2mKdGJdt24ZYY5Z6JWHQhPOpSgtWyUv6RnxU2ligEeaoPaiepUUOhoyLf4WcF7voVCAKZNqeTtQ==
diff --git a/test/docker/ssh_host_rsa_key_3072 b/test/docker/ssh_host_rsa_key_3072
new file mode 100644
index 0000000..3a2c719
--- /dev/null
+++ b/test/docker/ssh_host_rsa_key_3072
@@ -0,0 +1,39 @@
+-----BEGIN RSA PRIVATE KEY-----
+MIIG5AIBAAKCAYEAzhHp7eFnrQAJqOd7aihyQyIDKgxCF7H51Q3Ft3+8af+lX3Ol
+Ie77Gi5GNNM+eRB4OzG+CBslxN5I3pM//sZ+gyylA1VuWZZkOlgtbOHutIkO2ldk
+XtoGidla0VAxLcUcUK6cCmqwBTT31Hp4Qimp2zyeg/l5q0DhWKguY13lrm5b3YZY
+rj7CW3Ktzxf8SbYz6du8KF0dHCWilzq+FLeGzXr7Yul5njVF5njkGvZ9duQ0qiVR
+zqZkrkLEWgQlCM0T+PyUbvedL1MfDZPHGh7ZhU0snOvJRsxAr31tlknq+WwauZYd
+DzJf1g1URcM65UwEsPlfgOW3ZoZogR1v57Im+KdsKhq2B3snEtJgdQh06JyO0ur4
+uUXo1mMtvBFhiptUtwP4g9v/IN4neeK+wBRom46m2Q1bMUBPneBOa8r2SY/3ynrz
+XuVIWFOQtF60aJ+BNqvgUVCKOmz1KzoJwTqGm+EFaKM5z+UQWjIbSE3Ge4X5hXtk
+Ou52v+tyDUk6boZLAgMBAAECggGAdrhxWmA7N7tG1W2Pd6iXs7+brRTk2vvpYGqP
+11kbNsJXBzf8EiG5vuqb/gEaA+uOKSRORCNHzjT2LG0POHwpFO+aneIRMkHnuolk
+mk9ME+zGhtpEdDUOAUsc/GxD+QePeZgvQ/0VLdrHUT3BnPSd7DXvaT9IbnZxnX8/
+QnYtRiJEgMrOuoxjswXNxvsdmWYEYJ38uBB1Hes80f3A1vSpECbjP6gdLh2pCM/r
+MvGBdQaipMfdar4IUTEcKHQs1fY3mlAxnWRjYCqJPmq10d3NrdUrHb2zBE1HCC4h
+aj2ycTxFhDJqGV6Y2AboHqh2c7lPJ+R2UjI9mIpALZSviHB1POcpWCAGA3NKjri9
+8jgxl3bj03ikJNfCuvlqRTa8at63W2zZTMRsxamoiO023uUOEMNBPwWXP/rVhQ8g
+ufih0SY44j0EMPIuu2PoQV4ZSOtDw8xdPrchVCa078/pP5cRa4uV0bl2K4as+cYC
+BhjEq2Org3ulDW2n6Mz5ZS7NbAkxAoHBAP/bgPGKX7rfrHa5MRHIgbLSmZtUoF51
+YGelc8ytRx6UT6wriJ1jQRXiI5mZlIXyVxMpIz9s4+h59kF+LpZuNLc3vTYpPOQn
+RUDBVY6+SPC5MancL7bfBoHahpWEJuJB/WUE7eWvQM03/LsBtU6Nq+R632t5KdqF
+A4y86qgD1vIjcBWvySLFJZGOCoNbj7ZinoBUO3ueYK6SUj8xH6TAqOJsTPvquRT3
+AFBpFBmrVc24wW7wTiLkQOhkIQs1J/ZhYwKBwQDOL07qF8wsoQBBTTXkZ59BCauz
+R8kfqe5oUBwsmGJdiIHX6gutBA07sSwzVekIvCCkJFXk3TxLoBSMHEZEIdnS+HVt
+gMIacYuhbh+XztdY0kadH/SMbVQD/2LZcL99vcZPq1QF3cHb0Buip5+fyAYjoEc7
+oVgvewD/TwdNcMjos/kMNh6l04kLi6vQG3WhoSBPWaoB669ppBNXSrWKe43nXVi6
+EvjGEiL+HCCnmD6LiD6p797Owu9AChP6fXInD/kCgcEAiLP3SRbt3yLzOtvn4+CF
+q83qVJv6s31zbO1x2cIbZbNIfm0kKTOG6vJQoxjzyj2ZWJt6QcEkZGoFsSiCK83m
+TJ5zciTGbACvd9HUrNfukO/iISeMNuEi0O65Sdm6DNnFUdw4X6grr3pihmh7PuVj
+GkisZvft7Nt08hVeKzch+W4FzRCHHxTG5eZGp7icKI64sUhQH9SXQ67aUvkkNxrZ
+IWFMIK1hBlqSyGPcYXqx9aDpeSTcGrhqFcCqBxr3pySRAoHAfJNO3delEC3yxoHN
+FwSYzyX1rOuplE0K89G7RCKKBDNPKFKL3Wx+Rluk9htpIlLwcdxWXWJiZNsCrykC
+N3YwcuyVnqTWIj4KfG3Z/tIFgPADpDnDevkvcv7iDbi2qlV4NXix2p2C3LnfiKY4
+psSnGO1lPJ0eeAmcr6VjJyIG8bqTthIY8F5gBi7Mj3+X0iFVMTxeoKxzHqP435wP
+Fe3S7kCTNFH0J1Cb/eamwDwXRhz6p5h7iXd0MMAmFAmpZ/qZAoHBAPDSIvk2ocf1
+FVW8pKtKOJFIs8iQVIaOLKwPJVP8/JsB1+7mQx5KMoROb5pNpX2edN4vvG0CgqpJ
+KekleqpH6nQCqYGFZ1BDhORElNILxeJHcNl0eAG++IJ2PfIpTZV30edDqMm0x7EI
+8POZWAx809VzcYbE2jsgpN/EuiaG30EAI5yNvyzmZRCyQykH+eltHlCx17MWBxRQ
+bb2UUfpdInTMS2vyrvkeUACkC1DGYdBVVBqqPTkHZg+Kcbs8ntQqEQ==
+-----END RSA PRIVATE KEY-----
diff --git a/test/docker/ssh_host_rsa_key_3072-cert_1024.pub b/test/docker/ssh_host_rsa_key_3072-cert_1024.pub
new file mode 100644
index 0000000..da6b9ec
--- /dev/null
+++ b/test/docker/ssh_host_rsa_key_3072-cert_1024.pub
@@ -0,0 +1 @@
+ssh-rsa-cert-v01@openssh.com AAAAHHNzaC1yc2EtY2VydC12MDFAb3BlbnNzaC5jb20AAAAgGHz1JwF/1IcxW3pdQtpqbUjIHaFuk0cR/+l50vG+9hIAAAADAQABAAABgQDOEent4WetAAmo53tqKHJDIgMqDEIXsfnVDcW3f7xp/6Vfc6Uh7vsaLkY00z55EHg7Mb4IGyXE3kjekz/+xn6DLKUDVW5ZlmQ6WC1s4e60iQ7aV2Re2gaJ2VrRUDEtxRxQrpwKarAFNPfUenhCKanbPJ6D+XmrQOFYqC5jXeWublvdhliuPsJbcq3PF/xJtjPp27woXR0cJaKXOr4Ut4bNevti6XmeNUXmeOQa9n125DSqJVHOpmSuQsRaBCUIzRP4/JRu950vUx8Nk8caHtmFTSyc68lGzECvfW2WSer5bBq5lh0PMl/WDVRFwzrlTASw+V+A5bdmhmiBHW/nsib4p2wqGrYHeycS0mB1CHTonI7S6vi5RejWYy28EWGKm1S3A/iD2/8g3id54r7AFGibjqbZDVsxQE+d4E5ryvZJj/fKevNe5UhYU5C0XrRon4E2q+BRUIo6bPUrOgnBOoab4QVooznP5RBaMhtITcZ7hfmFe2Q67na/63INSTpuhksAAAAAAAAAAAAAAAIAAAAEdGVzdAAAAAgAAAAEdGVzdAAAAABdXry0AAAAAIFrQR8AAAAAAAAAAAAAAAAAAACXAAAAB3NzaC1yc2EAAAADAQABAAAAgQDnRlN3AFnUe2lFf5XG9UhXLr/9POruNTFbMt0zrjOUSjmAS7hS6pDv5VEToT6DaR1EQUYaqSMpHYzZhuCK52vrydOm5XFbJ7712r9MyZQUhoVZx8SudBHzVDIVO3jcMMWIlrfWBMnUaUHEqpmy88Y7gKDa2TWxJg1+hg51KqHrUQAAAI8AAAAHc3NoLXJzYQAAAIB4HaEexgQ9T6rScEbiHZx+suCaYXI7ywLYyoSEO48K8o+MmO83UTLtpPa3DXlT8hSYL8Aq6Bb5AMkDawsgsC484owPqObT/5ndLG/fctNBFcCTSL0ftte+A8xH0pZaGRoKbdxxgMqX4ubrCXpbMLGF9aAeh7MRa756XzqGlsCiSA== ssh_host_rsa_key_3072.pub
diff --git a/test/docker/ssh_host_rsa_key_3072-cert_3072.pub b/test/docker/ssh_host_rsa_key_3072-cert_3072.pub
new file mode 100644
index 0000000..78f49ea
--- /dev/null
+++ b/test/docker/ssh_host_rsa_key_3072-cert_3072.pub
@@ -0,0 +1 @@
+ssh-rsa-cert-v01@openssh.com AAAAHHNzaC1yc2EtY2VydC12MDFAb3BlbnNzaC5jb20AAAAg9MVX4OlkEy3p9eC+JJp8h7j76EmI46EY/RXxCGSWTC0AAAADAQABAAABgQDOEent4WetAAmo53tqKHJDIgMqDEIXsfnVDcW3f7xp/6Vfc6Uh7vsaLkY00z55EHg7Mb4IGyXE3kjekz/+xn6DLKUDVW5ZlmQ6WC1s4e60iQ7aV2Re2gaJ2VrRUDEtxRxQrpwKarAFNPfUenhCKanbPJ6D+XmrQOFYqC5jXeWublvdhliuPsJbcq3PF/xJtjPp27woXR0cJaKXOr4Ut4bNevti6XmeNUXmeOQa9n125DSqJVHOpmSuQsRaBCUIzRP4/JRu950vUx8Nk8caHtmFTSyc68lGzECvfW2WSer5bBq5lh0PMl/WDVRFwzrlTASw+V+A5bdmhmiBHW/nsib4p2wqGrYHeycS0mB1CHTonI7S6vi5RejWYy28EWGKm1S3A/iD2/8g3id54r7AFGibjqbZDVsxQE+d4E5ryvZJj/fKevNe5UhYU5C0XrRon4E2q+BRUIo6bPUrOgnBOoab4QVooznP5RBaMhtITcZ7hfmFe2Q67na/63INSTpuhksAAAAAAAAAAAAAAAIAAAAEdGVzdAAAAAgAAAAEdGVzdAAAAABdXr0sAAAAAIFrQWwAAAAAAAAAAAAAAAAAAAGXAAAAB3NzaC1yc2EAAAADAQABAAABgQCrFAQhuPzDROsFjV8M7zVy6uKNCt0s5SvxWXxcQJZzlAZ3UuQAbzQ2B1wIE++NbFlKZisp6oDvwZR034qDJwTTO2nFhJcodkxogzld73gEH4Ghwf3W93/aIHMZ553WdX/N8B2rYAQdCPVOkgj5r7PbnLG/Y+7BVNPDCj0j/omA2rd68fgdvCCvdJF8QRhlD5gvxw6kG42DVhk5HSTXKTbdmoB33YnJvFniIK063utyPawUfM27VTnjjBYHAgvnJlkpOAGFjh2p9PYSnsjNHkd2x2u9BX2oL3YYzXMMT5TcDCtUqMezRhmrePNyr/K8UMJwwxHPj/SkYUVTbqhaZr6fe9sDEomgv6uEbxAz36XaoWUsayczSRRcAlew8pvKClo/xndTh4qBO4nYa5L14sgWcHuavAPvlXoaA32rmZjz0N+RVKQdqiqljyk91GTP1ACYwS4u0ea/b0JU6l9fm/+uybvABIBmq1SWYW7isUYEvC9ltSVa7TYeaAFPjuDoQh8AAAGPAAAAB3NzaC1yc2EAAAGAG8tCiBMSq3Of3Gmcrid2IfPmaaemYivgEEuK8ubq1rznF0vtR07/NUQ7WVzfJhUSeG0gtJ3A1ey60NjcBn0DHao4Q3ATIXnkSOIKjNolZ2urqYv9fT1LAC4I5XWGzK2aKK0NEqAYr06YPtcGOBQk5+3GPAWSJ4eQycKRz5BSuMYbKaVxU0kGSvbavG07ZntMQhia/lILyq84PjXh/JlRVpIqY+LAS0qwqkUR3gWMTmvYvYI7fXU84ReVB1ut75bY7Xx0DXHPl1Zc2MNDLGcKsByZtoO9ueZRyOlZMUJcVP5fK+OUuZKjMbCaaJnV55BQ78/rftIPYsTEEO2Sf9WT86ADa3k4S0pyWqlTxBzZcDWNt+fZFNm9wcqcYS32nDKtfixcDN8E/IJIWY7aoabPqoYnKUVQBOcIEnZf1HqsKUVmF44Dp9mKhefUs3BtcdK63j/lNXzzMrPwZQAreJqH/uV3TgYBLjMPl++ctX6tCe6Hv5zFKNhnOCSBSzcsCgIU ssh_host_rsa_key_3072.pub
diff --git a/test/docker/ssh_host_rsa_key_3072.pub b/test/docker/ssh_host_rsa_key_3072.pub
new file mode 100644
index 0000000..ad83cd1
--- /dev/null
+++ b/test/docker/ssh_host_rsa_key_3072.pub
@@ -0,0 +1 @@
+ssh-rsa AAAAB3NzaC1yc2EAAAADAQABAAABgQDOEent4WetAAmo53tqKHJDIgMqDEIXsfnVDcW3f7xp/6Vfc6Uh7vsaLkY00z55EHg7Mb4IGyXE3kjekz/+xn6DLKUDVW5ZlmQ6WC1s4e60iQ7aV2Re2gaJ2VrRUDEtxRxQrpwKarAFNPfUenhCKanbPJ6D+XmrQOFYqC5jXeWublvdhliuPsJbcq3PF/xJtjPp27woXR0cJaKXOr4Ut4bNevti6XmeNUXmeOQa9n125DSqJVHOpmSuQsRaBCUIzRP4/JRu950vUx8Nk8caHtmFTSyc68lGzECvfW2WSer5bBq5lh0PMl/WDVRFwzrlTASw+V+A5bdmhmiBHW/nsib4p2wqGrYHeycS0mB1CHTonI7S6vi5RejWYy28EWGKm1S3A/iD2/8g3id54r7AFGibjqbZDVsxQE+d4E5ryvZJj/fKevNe5UhYU5C0XrRon4E2q+BRUIo6bPUrOgnBOoab4QVooznP5RBaMhtITcZ7hfmFe2Q67na/63INSTpuhks=
diff --git a/test/stubs/colorama.pyi b/test/stubs/colorama.pyi
new file mode 100644
index 0000000..cd5a50a
--- /dev/null
+++ b/test/stubs/colorama.pyi
@@ -0,0 +1,4 @@
+from typing import Optional
+
+def init(autoreset: bool = False, convert: Optional[bool] = None, strip: Optional[bool] = None, wrap: bool = True) -> None: ...
+
diff --git a/test/test_algorithm.py b/test/test_algorithm.py
new file mode 100644
index 0000000..8003616
--- /dev/null
+++ b/test/test_algorithm.py
@@ -0,0 +1,166 @@
+import pytest
+
+from ssh_audit.algorithm import Algorithm
+from ssh_audit.timeframe import Timeframe
+
+
+# pylint: disable=attribute-defined-outside-init
+class TestAlgorithm:
+ @pytest.fixture(autouse=True)
+ def init(self, ssh_audit):
+ self.algorithm = Algorithm
+ self.timeframe = Timeframe
+
+ def _tf(self, v, s=None):
+ return self.timeframe().update(v, s)
+
+ def test_get_ssh_version(self):
+ def ver(v):
+ return self.algorithm.get_ssh_version(v)
+
+ assert ver('7.5') == ('OpenSSH', '7.5', False)
+ assert ver('7.5C') == ('OpenSSH', '7.5', True)
+ assert ver('d2016.74') == ('Dropbear SSH', '2016.74', False)
+ assert ver('l10.7.4') == ('libssh', '0.7.4', False)
+ assert ver('')[1] == ''
+
+ def test_get_since_text(self):
+ def gst(v):
+ return self.algorithm.get_since_text(v)
+
+ assert gst(['7.5']) == 'available since OpenSSH 7.5'
+ assert gst(['7.5C']) == 'available since OpenSSH 7.5 (client only)'
+ assert gst(['7.5,']) == 'available since OpenSSH 7.5'
+ assert gst(['d2016.73']) == 'available since Dropbear SSH 2016.73'
+ assert gst(['7.5,d2016.73']) == 'available since OpenSSH 7.5, Dropbear SSH 2016.73'
+ assert gst(['l10.7.4']) is None
+ assert gst([]) is None
+
+ def test_timeframe_creation(self):
+ # pylint: disable=line-too-long,too-many-statements
+ def cmp_tf(v, s, r):
+ assert str(self._tf(v, s)) == str(r)
+
+ cmp_tf(['6.2'], None, {'OpenSSH': ['6.2', None, '6.2', None]})
+ cmp_tf(['6.2'], True, {'OpenSSH': ['6.2', None, None, None]})
+ cmp_tf(['6.2'], False, {'OpenSSH': [None, None, '6.2', None]})
+ cmp_tf(['6.2C'], None, {'OpenSSH': [None, None, '6.2', None]})
+ cmp_tf(['6.2C'], True, {})
+ cmp_tf(['6.2C'], False, {'OpenSSH': [None, None, '6.2', None]})
+ cmp_tf(['6.1,6.2C'], None, {'OpenSSH': ['6.1', None, '6.2', None]})
+ cmp_tf(['6.1,6.2C'], True, {'OpenSSH': ['6.1', None, None, None]})
+ cmp_tf(['6.1,6.2C'], False, {'OpenSSH': [None, None, '6.2', None]})
+ cmp_tf(['6.2C,6.1'], None, {'OpenSSH': ['6.1', None, '6.2', None]})
+ cmp_tf(['6.2C,6.1'], True, {'OpenSSH': ['6.1', None, None, None]})
+ cmp_tf(['6.2C,6.1'], False, {'OpenSSH': [None, None, '6.2', None]})
+ cmp_tf(['6.3,6.2C'], None, {'OpenSSH': ['6.3', None, '6.2', None]})
+ cmp_tf(['6.3,6.2C'], True, {'OpenSSH': ['6.3', None, None, None]})
+ cmp_tf(['6.3,6.2C'], False, {'OpenSSH': [None, None, '6.2', None]})
+ cmp_tf(['6.2C,6.3'], None, {'OpenSSH': ['6.3', None, '6.2', None]})
+ cmp_tf(['6.2C,6.3'], True, {'OpenSSH': ['6.3', None, None, None]})
+ cmp_tf(['6.2C,6.3'], False, {'OpenSSH': [None, None, '6.2', None]})
+
+ cmp_tf(['6.2', '6.6'], None, {'OpenSSH': ['6.2', '6.6', '6.2', '6.6']})
+ cmp_tf(['6.2', '6.6'], True, {'OpenSSH': ['6.2', '6.6', None, None]})
+ cmp_tf(['6.2', '6.6'], False, {'OpenSSH': [None, None, '6.2', '6.6']})
+ cmp_tf(['6.2C', '6.6'], None, {'OpenSSH': [None, '6.6', '6.2', '6.6']})
+ cmp_tf(['6.2C', '6.6'], True, {'OpenSSH': [None, '6.6', None, None]})
+ cmp_tf(['6.2C', '6.6'], False, {'OpenSSH': [None, None, '6.2', '6.6']})
+ cmp_tf(['6.1,6.2C', '6.6'], None, {'OpenSSH': ['6.1', '6.6', '6.2', '6.6']})
+ cmp_tf(['6.1,6.2C', '6.6'], True, {'OpenSSH': ['6.1', '6.6', None, None]})
+ cmp_tf(['6.1,6.2C', '6.6'], False, {'OpenSSH': [None, None, '6.2', '6.6']})
+ cmp_tf(['6.2C,6.1', '6.6'], None, {'OpenSSH': ['6.1', '6.6', '6.2', '6.6']})
+ cmp_tf(['6.2C,6.1', '6.6'], True, {'OpenSSH': ['6.1', '6.6', None, None]})
+ cmp_tf(['6.2C,6.1', '6.6'], False, {'OpenSSH': [None, None, '6.2', '6.6']})
+ cmp_tf(['6.3,6.2C', '6.6'], None, {'OpenSSH': ['6.3', '6.6', '6.2', '6.6']})
+ cmp_tf(['6.3,6.2C', '6.6'], True, {'OpenSSH': ['6.3', '6.6', None, None]})
+ cmp_tf(['6.3,6.2C', '6.6'], False, {'OpenSSH': [None, None, '6.2', '6.6']})
+ cmp_tf(['6.2C,6.3', '6.6'], None, {'OpenSSH': ['6.3', '6.6', '6.2', '6.6']})
+ cmp_tf(['6.2C,6.3', '6.6'], True, {'OpenSSH': ['6.3', '6.6', None, None]})
+ cmp_tf(['6.2C,6.3', '6.6'], False, {'OpenSSH': [None, None, '6.2', '6.6']})
+
+ cmp_tf(['6.2', '6.6', None], None, {'OpenSSH': ['6.2', '6.6', '6.2', None]})
+ cmp_tf(['6.2', '6.6', None], True, {'OpenSSH': ['6.2', '6.6', None, None]})
+ cmp_tf(['6.2', '6.6', None], False, {'OpenSSH': [None, None, '6.2', None]})
+ cmp_tf(['6.2C', '6.6', None], None, {'OpenSSH': [None, '6.6', '6.2', None]})
+ cmp_tf(['6.2C', '6.6', None], True, {'OpenSSH': [None, '6.6', None, None]})
+ cmp_tf(['6.2C', '6.6', None], False, {'OpenSSH': [None, None, '6.2', None]})
+ cmp_tf(['6.1,6.2C', '6.6', None], None, {'OpenSSH': ['6.1', '6.6', '6.2', None]})
+ cmp_tf(['6.1,6.2C', '6.6', None], True, {'OpenSSH': ['6.1', '6.6', None, None]})
+ cmp_tf(['6.1,6.2C', '6.6', None], False, {'OpenSSH': [None, None, '6.2', None]})
+ cmp_tf(['6.2C,6.1', '6.6', None], None, {'OpenSSH': ['6.1', '6.6', '6.2', None]})
+ cmp_tf(['6.2C,6.1', '6.6', None], True, {'OpenSSH': ['6.1', '6.6', None, None]})
+ cmp_tf(['6.2C,6.1', '6.6', None], False, {'OpenSSH': [None, None, '6.2', None]})
+ cmp_tf(['6.2,6.3C', '6.6', None], None, {'OpenSSH': ['6.2', '6.6', '6.3', None]})
+ cmp_tf(['6.2,6.3C', '6.6', None], True, {'OpenSSH': ['6.2', '6.6', None, None]})
+ cmp_tf(['6.2,6.3C', '6.6', None], False, {'OpenSSH': [None, None, '6.3', None]})
+ cmp_tf(['6.3C,6.2', '6.6', None], None, {'OpenSSH': ['6.2', '6.6', '6.3', None]})
+ cmp_tf(['6.3C,6.2', '6.6', None], True, {'OpenSSH': ['6.2', '6.6', None, None]})
+ cmp_tf(['6.3C,6.2', '6.6', None], False, {'OpenSSH': [None, None, '6.3', None]})
+
+ cmp_tf(['6.2', '6.6', '7.1'], None, {'OpenSSH': ['6.2', '6.6', '6.2', '7.1']})
+ cmp_tf(['6.2', '6.6', '7.1'], True, {'OpenSSH': ['6.2', '6.6', None, None]})
+ cmp_tf(['6.2', '6.6', '7.1'], False, {'OpenSSH': [None, None, '6.2', '7.1']})
+ cmp_tf(['6.1,6.2C', '6.6', '7.1'], None, {'OpenSSH': ['6.1', '6.6', '6.2', '7.1']})
+ cmp_tf(['6.1,6.2C', '6.6', '7.1'], True, {'OpenSSH': ['6.1', '6.6', None, None]})
+ cmp_tf(['6.1,6.2C', '6.6', '7.1'], False, {'OpenSSH': [None, None, '6.2', '7.1']})
+ cmp_tf(['6.2C,6.1', '6.6', '7.1'], None, {'OpenSSH': ['6.1', '6.6', '6.2', '7.1']})
+ cmp_tf(['6.2C,6.1', '6.6', '7.1'], True, {'OpenSSH': ['6.1', '6.6', None, None]})
+ cmp_tf(['6.2C,6.1', '6.6', '7.1'], False, {'OpenSSH': [None, None, '6.2', '7.1']})
+ cmp_tf(['6.2,6.3C', '6.6', '7.1'], None, {'OpenSSH': ['6.2', '6.6', '6.3', '7.1']})
+ cmp_tf(['6.2,6.3C', '6.6', '7.1'], True, {'OpenSSH': ['6.2', '6.6', None, None]})
+ cmp_tf(['6.2,6.3C', '6.6', '7.1'], False, {'OpenSSH': [None, None, '6.3', '7.1']})
+ cmp_tf(['6.3C,6.2', '6.6', '7.1'], None, {'OpenSSH': ['6.2', '6.6', '6.3', '7.1']})
+ cmp_tf(['6.3C,6.2', '6.6', '7.1'], True, {'OpenSSH': ['6.2', '6.6', None, None]})
+ cmp_tf(['6.3C,6.2', '6.6', '7.1'], False, {'OpenSSH': [None, None, '6.3', '7.1']})
+
+ tf1 = self._tf(['6.1,d2016.72,6.2C', '6.6,d2016.73', '7.1,d2016.74'])
+ tf2 = self._tf(['d2016.72,6.2C,6.1', 'd2016.73,6.6', 'd2016.74,7.1'])
+ tf3 = self._tf(['d2016.72,6.2C,6.1', '6.6,d2016.73', '7.1,d2016.74'])
+ # check without caring for output order
+ ov = "'OpenSSH': ['6.1', '6.6', '6.2', '7.1']"
+ dv = "'Dropbear SSH': ['2016.72', '2016.73', '2016.72', '2016.74']"
+ assert len(str(tf1)) == len(str(tf2)) == len(str(tf3))
+ assert ov in str(tf1) and ov in str(tf2) and ov in str(tf3)
+ assert dv in str(tf1) and dv in str(tf2) and dv in str(tf3)
+ assert ov in repr(tf1) and ov in repr(tf2) and ov in repr(tf3)
+ assert dv in repr(tf1) and dv in repr(tf2) and dv in repr(tf3)
+
+ def test_timeframe_object(self):
+ tf = self._tf(['6.1,6.2C', '6.6', '7.1'])
+ assert 'OpenSSH' in tf
+ assert 'Dropbear SSH' not in tf
+ assert 'libssh' not in tf
+ assert 'unknown' not in tf
+ assert tf['OpenSSH'] == ('6.1', '6.6', '6.2', '7.1')
+ assert tf['Dropbear SSH'] == (None, None, None, None)
+ assert tf['libssh'] == (None, None, None, None)
+ assert tf['unknown'] == (None, None, None, None)
+ assert tf.get_from('OpenSSH', True) == '6.1'
+ assert tf.get_till('OpenSSH', True) == '6.6'
+ assert tf.get_from('OpenSSH', False) == '6.2'
+ assert tf.get_till('OpenSSH', False) == '7.1'
+
+ tf = self._tf(['6.1,d2016.72,6.2C', '6.6,d2016.73', '7.1,d2016.74'])
+ assert 'OpenSSH' in tf
+ assert 'Dropbear SSH' in tf
+ assert 'libssh' not in tf
+ assert 'unknown' not in tf
+ assert tf['OpenSSH'] == ('6.1', '6.6', '6.2', '7.1')
+ assert tf['Dropbear SSH'] == ('2016.72', '2016.73', '2016.72', '2016.74')
+ assert tf['libssh'] == (None, None, None, None)
+ assert tf['unknown'] == (None, None, None, None)
+ assert tf.get_from('OpenSSH', True) == '6.1'
+ assert tf.get_till('OpenSSH', True) == '6.6'
+ assert tf.get_from('OpenSSH', False) == '6.2'
+ assert tf.get_till('OpenSSH', False) == '7.1'
+ assert tf.get_from('Dropbear SSH', True) == '2016.72'
+ assert tf.get_till('Dropbear SSH', True) == '2016.73'
+ assert tf.get_from('Dropbear SSH', False) == '2016.72'
+ assert tf.get_till('Dropbear SSH', False) == '2016.74'
+ ov = "'OpenSSH': ['6.1', '6.6', '6.2', '7.1']"
+ dv = "'Dropbear SSH': ['2016.72', '2016.73', '2016.72', '2016.74']"
+ assert ov in str(tf)
+ assert dv in str(tf)
+ assert ov in repr(tf)
+ assert dv in repr(tf)
diff --git a/test/test_auditconf.py b/test/test_auditconf.py
new file mode 100644
index 0000000..05e7089
--- /dev/null
+++ b/test/test_auditconf.py
@@ -0,0 +1,178 @@
+import pytest
+from ssh_audit.ssh_audit import process_commandline
+
+
+# pylint: disable=attribute-defined-outside-init
+class TestAuditConf:
+ @pytest.fixture(autouse=True)
+ def init(self, ssh_audit):
+ self.AuditConf = ssh_audit.AuditConf
+ self.OutputBuffer = ssh_audit.OutputBuffer
+ self.usage = ssh_audit.usage
+ self.process_commandline = process_commandline
+
+ @staticmethod
+ def _test_conf(conf, **kwargs):
+ options = {
+ 'host': '',
+ 'port': 22,
+ 'ssh1': True,
+ 'ssh2': True,
+ 'batch': False,
+ 'colors': True,
+ 'verbose': False,
+ 'level': 'info',
+ 'ipv4': False,
+ 'ipv6': False
+ }
+ for k, v in kwargs.items():
+ options[k] = v
+ assert conf.host == options['host']
+ assert conf.port == options['port']
+ assert conf.ssh1 is options['ssh1']
+ assert conf.ssh2 is options['ssh2']
+ assert conf.batch is options['batch']
+ assert conf.colors is options['colors']
+ assert conf.verbose is options['verbose']
+ assert conf.level == options['level']
+ assert conf.ipv4 == options['ipv4']
+ assert conf.ipv6 == options['ipv6']
+
+ def test_audit_conf_defaults(self):
+ conf = self.AuditConf()
+ self._test_conf(conf)
+
+ def test_audit_conf_booleans(self):
+ conf = self.AuditConf()
+ for p in ['ssh1', 'ssh2', 'batch', 'colors', 'verbose']:
+ for v in [True, 1]:
+ setattr(conf, p, v)
+ assert getattr(conf, p) is True
+ for v in [False, 0]:
+ setattr(conf, p, v)
+ assert getattr(conf, p) is False
+
+ def test_audit_conf_port(self):
+ conf = self.AuditConf()
+ for port in [22, 2222]:
+ conf.port = port
+ assert conf.port == port
+ for port in [-1, 0, 65536, 99999]:
+ with pytest.raises(ValueError) as excinfo:
+ conf.port = port
+ excinfo.match(r'.*invalid port.*')
+
+ def test_audit_conf_ip_version_preference(self):
+ # ipv4-only
+ conf = self.AuditConf()
+ conf.ipv4 = True
+ assert conf.ipv4 is True
+ assert conf.ipv6 is False
+ assert conf.ip_version_preference == [4]
+ # ipv6-only
+ conf = self.AuditConf()
+ conf.ipv6 = True
+ assert conf.ipv4 is False
+ assert conf.ipv6 is True
+ assert conf.ip_version_preference == [6]
+ # ipv4-preferred
+ conf = self.AuditConf()
+ conf.ipv4 = True
+ conf.ipv6 = True
+ assert conf.ipv4 is True
+ assert conf.ipv6 is True
+ assert conf.ip_version_preference == [4, 6]
+ # ipv6-preferred
+ conf = self.AuditConf()
+ conf.ipv6 = True
+ conf.ipv4 = True
+ assert conf.ipv4 is True
+ assert conf.ipv6 is True
+ assert conf.ip_version_preference == [6, 4]
+ # defaults
+ conf = self.AuditConf()
+ assert conf.ipv4 is False
+ assert conf.ipv6 is False
+ assert conf.ip_version_preference == []
+
+ def test_audit_conf_level(self):
+ conf = self.AuditConf()
+ for level in ['info', 'warn', 'fail']:
+ conf.level = level
+ assert conf.level == level
+ for level in ['head', 'good', 'unknown', None]:
+ with pytest.raises(ValueError) as excinfo:
+ conf.level = level
+ excinfo.match(r'.*invalid level.*')
+
+ def test_audit_conf_process_commandline(self):
+ # pylint: disable=too-many-statements
+ c = lambda x: self.process_commandline(self.OutputBuffer, x.split(), self.usage) # noqa
+ with pytest.raises(SystemExit):
+ conf = c('')
+ with pytest.raises(SystemExit):
+ conf = c('-x')
+ with pytest.raises(SystemExit):
+ conf = c('-h')
+ with pytest.raises(SystemExit):
+ conf = c('--help')
+ with pytest.raises(SystemExit):
+ conf = c(':')
+ with pytest.raises(SystemExit):
+ conf = c(':22')
+ conf = c('localhost')
+ self._test_conf(conf, host='localhost')
+ conf = c('github.com')
+ self._test_conf(conf, host='github.com')
+ conf = c('localhost:2222')
+ self._test_conf(conf, host='localhost', port=2222)
+ conf = c('-p 2222 localhost')
+ self._test_conf(conf, host='localhost', port=2222)
+ conf = c('2001:4860:4860::8888')
+ self._test_conf(conf, host='2001:4860:4860::8888')
+ conf = c('[2001:4860:4860::8888]:22')
+ self._test_conf(conf, host='2001:4860:4860::8888')
+ conf = c('[2001:4860:4860::8888]:2222')
+ self._test_conf(conf, host='2001:4860:4860::8888', port=2222)
+ conf = c('-p 2222 2001:4860:4860::8888')
+ self._test_conf(conf, host='2001:4860:4860::8888', port=2222)
+ with pytest.raises(ValueError):
+ conf = c('localhost:abc')
+ with pytest.raises(SystemExit):
+ conf = c('-p abc localhost')
+ with pytest.raises(ValueError):
+ conf = c('localhost:-22')
+ with pytest.raises(SystemExit):
+ conf = c('-p -22 localhost')
+ with pytest.raises(ValueError):
+ conf = c('localhost:99999')
+ with pytest.raises(SystemExit):
+ conf = c('-p 99999 localhost')
+ conf = c('-1 localhost')
+ self._test_conf(conf, host='localhost', ssh1=True, ssh2=False)
+ conf = c('-2 localhost')
+ self._test_conf(conf, host='localhost', ssh1=False, ssh2=True)
+ conf = c('-12 localhost')
+ self._test_conf(conf, host='localhost', ssh1=True, ssh2=True)
+ conf = c('-4 localhost')
+ self._test_conf(conf, host='localhost', ipv4=True, ipv6=False, ipvo=(4,))
+ conf = c('-6 localhost')
+ self._test_conf(conf, host='localhost', ipv4=False, ipv6=True, ipvo=(6,))
+ conf = c('-46 localhost')
+ self._test_conf(conf, host='localhost', ipv4=True, ipv6=True, ipvo=(4, 6))
+ conf = c('-64 localhost')
+ self._test_conf(conf, host='localhost', ipv4=True, ipv6=True, ipvo=(6, 4))
+ conf = c('-b localhost')
+ self._test_conf(conf, host='localhost', batch=True, verbose=True)
+ conf = c('-n localhost')
+ self._test_conf(conf, host='localhost', colors=False)
+ conf = c('-v localhost')
+ self._test_conf(conf, host='localhost', verbose=True)
+ conf = c('-l info localhost')
+ self._test_conf(conf, host='localhost', level='info')
+ conf = c('-l warn localhost')
+ self._test_conf(conf, host='localhost', level='warn')
+ conf = c('-l fail localhost')
+ self._test_conf(conf, host='localhost', level='fail')
+ with pytest.raises(SystemExit):
+ conf = c('-l something localhost')
diff --git a/test/test_banner.py b/test/test_banner.py
new file mode 100644
index 0000000..e514134
--- /dev/null
+++ b/test/test_banner.py
@@ -0,0 +1,69 @@
+import pytest
+
+from ssh_audit.banner import Banner
+
+
+# pylint: disable=line-too-long,attribute-defined-outside-init
+class TestBanner:
+ @pytest.fixture(autouse=True)
+ def init(self, ssh_audit):
+ self.banner = Banner
+
+ def test_simple_banners(self):
+ banner = lambda x: self.banner.parse(x) # noqa
+ b = banner('SSH-2.0-OpenSSH_7.3')
+ assert b.protocol == (2, 0)
+ assert b.software == 'OpenSSH_7.3'
+ assert b.comments is None
+ assert str(b) == 'SSH-2.0-OpenSSH_7.3'
+ b = banner('SSH-1.99-Sun_SSH_1.1.3')
+ assert b.protocol == (1, 99)
+ assert b.software == 'Sun_SSH_1.1.3'
+ assert b.comments is None
+ assert str(b) == 'SSH-1.99-Sun_SSH_1.1.3'
+ b = banner('SSH-1.5-Cisco-1.25')
+ assert b.protocol == (1, 5)
+ assert b.software == 'Cisco-1.25'
+ assert b.comments is None
+ assert str(b) == 'SSH-1.5-Cisco-1.25'
+
+ def test_invalid_banners(self):
+ b = lambda x: self.banner.parse(x) # noqa
+ assert b('Something') is None
+ assert b('SSH-XXX-OpenSSH_7.3') is None
+
+ def test_banners_with_spaces(self):
+ b = lambda x: self.banner.parse(x) # noqa
+ s = 'SSH-2.0-OpenSSH_4.3p2'
+ assert str(b('SSH-2.0-OpenSSH_4.3p2 ')) == s
+ assert str(b('SSH-2.0- OpenSSH_4.3p2')) == s
+ assert str(b('SSH-2.0- OpenSSH_4.3p2 ')) == s
+ s = 'SSH-2.0-OpenSSH_4.3p2 Debian-9etch3 on i686-pc-linux-gnu'
+ assert str(b('SSH-2.0- OpenSSH_4.3p2 Debian-9etch3 on i686-pc-linux-gnu')) == s
+ assert str(b('SSH-2.0-OpenSSH_4.3p2 Debian-9etch3 on i686-pc-linux-gnu ')) == s
+ assert str(b('SSH-2.0- OpenSSH_4.3p2 Debian-9etch3 on i686-pc-linux-gnu ')) == s
+
+ def test_banners_without_software(self):
+ b = lambda x: self.banner.parse(x) # noqa
+ assert b('SSH-2.0').protocol == (2, 0)
+ assert b('SSH-2.0').software is None
+ assert b('SSH-2.0').comments is None
+ assert str(b('SSH-2.0')) == 'SSH-2.0'
+ assert b('SSH-2.0-').protocol == (2, 0)
+ assert b('SSH-2.0-').software == ''
+ assert b('SSH-2.0-').comments is None
+ assert str(b('SSH-2.0-')) == 'SSH-2.0-'
+
+ def test_banners_with_comments(self):
+ b = lambda x: self.banner.parse(x) # noqa
+ assert repr(b('SSH-2.0-OpenSSH_7.2p2 Ubuntu-1')) == '<Banner(protocol=2.0, software=OpenSSH_7.2p2, comments=Ubuntu-1)>'
+ assert repr(b('SSH-1.99-OpenSSH_3.4p1 Debian 1:3.4p1-1.woody.3')) == '<Banner(protocol=1.99, software=OpenSSH_3.4p1, comments=Debian 1:3.4p1-1.woody.3)>'
+ assert repr(b('SSH-1.5-1.3.7 F-SECURE SSH')) == '<Banner(protocol=1.5, software=1.3.7, comments=F-SECURE SSH)>'
+
+ def test_banners_with_multiple_protocols(self):
+ b = lambda x: self.banner.parse(x) # noqa
+ assert str(b('SSH-1.99-SSH-1.99-OpenSSH_3.6.1p2')) == 'SSH-1.99-OpenSSH_3.6.1p2'
+ assert str(b('SSH-2.0-SSH-2.0-OpenSSH_4.3p2 Debian-9')) == 'SSH-2.0-OpenSSH_4.3p2 Debian-9'
+ assert str(b('SSH-1.99-SSH-2.0-dropbear_0.5')) == 'SSH-1.99-dropbear_0.5'
+ assert str(b('SSH-2.0-SSH-1.99-OpenSSH_4.2p1 SSH Secure Shell (non-commercial)')) == 'SSH-1.99-OpenSSH_4.2p1 SSH Secure Shell (non-commercial)'
+ assert str(b('SSH-1.99-SSH-1.99-SSH-1.99-OpenSSH_3.9p1')) == 'SSH-1.99-OpenSSH_3.9p1'
diff --git a/test/test_buffer.py b/test/test_buffer.py
new file mode 100644
index 0000000..499dfd9
--- /dev/null
+++ b/test/test_buffer.py
@@ -0,0 +1,145 @@
+import re
+import pytest
+
+from ssh_audit.readbuf import ReadBuf
+from ssh_audit.writebuf import WriteBuf
+
+
+# pylint: disable=attribute-defined-outside-init,bad-whitespace
+class TestBuffer:
+ @pytest.fixture(autouse=True)
+ def init(self, ssh_audit):
+ self.rbuf = ReadBuf
+ self.wbuf = WriteBuf
+ self.utf8rchar = b'\xef\xbf\xbd'
+
+ @classmethod
+ def _b(cls, v):
+ v = re.sub(r'\s', '', v)
+ data = [int(v[i * 2:i * 2 + 2], 16) for i in range(len(v) // 2)]
+ return bytes(bytearray(data))
+
+ def test_unread(self):
+ w = self.wbuf().write_byte(1).write_int(2).write_flush()
+ r = self.rbuf(w)
+ assert r.unread_len == 5
+ r.read_byte()
+ assert r.unread_len == 4
+ r.read_int()
+ assert r.unread_len == 0
+
+ def test_byte(self):
+ w = lambda x: self.wbuf().write_byte(x).write_flush() # noqa
+ r = lambda x: self.rbuf(x).read_byte() # noqa
+ tc = [(0x00, '00'),
+ (0x01, '01'),
+ (0x10, '10'),
+ (0xff, 'ff')]
+ for p in tc:
+ assert w(p[0]) == self._b(p[1])
+ assert r(self._b(p[1])) == p[0]
+
+ def test_bool(self):
+ w = lambda x: self.wbuf().write_bool(x).write_flush() # noqa
+ r = lambda x: self.rbuf(x).read_bool() # noqa
+ tc = [(True, '01'),
+ (False, '00')]
+ for p in tc:
+ assert w(p[0]) == self._b(p[1])
+ assert r(self._b(p[1])) == p[0]
+
+ def test_int(self):
+ w = lambda x: self.wbuf().write_int(x).write_flush() # noqa
+ r = lambda x: self.rbuf(x).read_int() # noqa
+ tc = [(0x00, '00 00 00 00'),
+ (0x01, '00 00 00 01'),
+ (0xabcd, '00 00 ab cd'),
+ (0xffffffff, 'ff ff ff ff')]
+ for p in tc:
+ assert w(p[0]) == self._b(p[1])
+ assert r(self._b(p[1])) == p[0]
+
+ def test_string(self):
+ w = lambda x: self.wbuf().write_string(x).write_flush() # noqa
+ r = lambda x: self.rbuf(x).read_string() # noqa
+ tc = [('abc1', '00 00 00 04 61 62 63 31'),
+ (b'abc2', '00 00 00 04 61 62 63 32')]
+ for p in tc:
+ v = p[0]
+ assert w(v) == self._b(p[1])
+ if not isinstance(v, bytes):
+ v = bytes(bytearray(v, 'utf-8'))
+ assert r(self._b(p[1])) == v
+
+ def test_list(self):
+ w = lambda x: self.wbuf().write_list(x).write_flush() # noqa
+ r = lambda x: self.rbuf(x).read_list() # noqa
+ tc = [(['d', 'ef', 'ault'], '00 00 00 09 64 2c 65 66 2c 61 75 6c 74')]
+ for p in tc:
+ assert w(p[0]) == self._b(p[1])
+ assert r(self._b(p[1])) == p[0]
+
+ def test_list_nonutf8(self):
+ r = lambda x: self.rbuf(x).read_list() # noqa
+ src = self._b('00 00 00 04 de ad be ef')
+ dst = [(b'\xde\xad' + self.utf8rchar + self.utf8rchar).decode('utf-8')]
+ assert r(src) == dst
+
+ def test_line(self):
+ w = lambda x: self.wbuf().write_line(x).write_flush() # noqa
+ r = lambda x: self.rbuf(x).read_line() # noqa
+ tc = [('example line', '65 78 61 6d 70 6c 65 20 6c 69 6e 65 0d 0a')]
+ for p in tc:
+ assert w(p[0]) == self._b(p[1])
+ assert r(self._b(p[1])) == p[0]
+
+ def test_line_nonutf8(self):
+ r = lambda x: self.rbuf(x).read_line() # noqa
+ src = self._b('de ad be af')
+ dst = (b'\xde\xad' + self.utf8rchar + self.utf8rchar).decode('utf-8')
+ assert r(src) == dst
+
+ def test_bitlen(self):
+ # pylint: disable=protected-access
+ class Py26Int(int):
+ def bit_length(self):
+ raise AttributeError
+ assert self.wbuf._bitlength(42) == 6
+ assert self.wbuf._bitlength(Py26Int(42)) == 6
+
+ def test_mpint1(self):
+ mpint1w = lambda x: self.wbuf().write_mpint1(x).write_flush() # noqa
+ mpint1r = lambda x: self.rbuf(x).read_mpint1() # noqa
+ tc = [(0x0, '00 00'),
+ (0x1234, '00 0d 12 34'),
+ (0x12345, '00 11 01 23 45'),
+ (0xdeadbeef, '00 20 de ad be ef')]
+ for p in tc:
+ assert mpint1w(p[0]) == self._b(p[1])
+ assert mpint1r(self._b(p[1])) == p[0]
+
+ def test_mpint2(self):
+ mpint2w = lambda x: self.wbuf().write_mpint2(x).write_flush() # noqa
+ mpint2r = lambda x: self.rbuf(x).read_mpint2() # noqa
+ tc = [(0x0, '00 00 00 00'),
+ (0x80, '00 00 00 02 00 80'),
+ (0x9a378f9b2e332a7, '00 00 00 08 09 a3 78 f9 b2 e3 32 a7'),
+ (-0x1234, '00 00 00 02 ed cc'),
+ (-0xdeadbeef, '00 00 00 05 ff 21 52 41 11'),
+ (-0x8000, '00 00 00 02 80 00'),
+ (-0x80, '00 00 00 01 80')]
+ for p in tc:
+ assert mpint2w(p[0]) == self._b(p[1])
+ assert mpint2r(self._b(p[1])) == p[0]
+ assert mpint2r(self._b('00 00 00 02 ff 80')) == -0x80
+
+ def test_reset(self):
+ w = self.wbuf()
+ w.write_int(7)
+ w.write_int(13)
+ assert len(w.write_flush()) == 8
+
+ w.write_int(7)
+ w.write_int(13)
+ w.reset()
+ assert len(w.write_flush()) == 0
diff --git a/test/test_build_struct.py b/test/test_build_struct.py
new file mode 100644
index 0000000..6365019
--- /dev/null
+++ b/test/test_build_struct.py
@@ -0,0 +1,43 @@
+import os
+import pytest
+
+from ssh_audit.ssh2_kex import SSH2_Kex
+from ssh_audit.ssh2_kexparty import SSH2_KexParty
+
+
+@pytest.fixture
+def kex(ssh_audit):
+ kex_algs, key_algs = [], []
+ enc, mac, compression, languages = [], [], ['none'], []
+ cli = SSH2_KexParty(enc, mac, compression, languages)
+ enc, mac, compression, languages = [], [], ['none'], []
+ srv = SSH2_KexParty(enc, mac, compression, languages)
+ cookie = os.urandom(16)
+ kex = SSH2_Kex(cookie, kex_algs, key_algs, cli, srv, 0)
+ return kex
+
+
+def test_prevent_runtime_error_regression(ssh_audit, kex):
+ """Prevent a regression of https://github.com/jtesta/ssh-audit/issues/41
+
+ The following test setup does not contain any sensible data.
+ It was made up to reproduce a situation when there are several host
+ keys, and an error occurred when iterating and modifying them at the
+ same time.
+ """
+ kex.set_host_key("ssh-rsa", b"\x00\x00\x00\x07ssh-rsa\x00\x00\x00")
+ kex.set_host_key("ssh-rsa1", b"\x00\x00\x00\x07ssh-rsa\x00\x00\x00")
+ kex.set_host_key("ssh-rsa2", b"\x00\x00\x00\x07ssh-rsa\x00\x00\x00")
+ kex.set_host_key("ssh-rsa3", b"\x00\x00\x00\x07ssh-rsa\x00\x00\x00")
+ kex.set_host_key("ssh-rsa4", b"\x00\x00\x00\x07ssh-rsa\x00\x00\x00")
+ kex.set_host_key("ssh-rsa5", b"\x00\x00\x00\x07ssh-rsa\x00\x00\x00")
+ kex.set_host_key("ssh-rsa6", b"\x00\x00\x00\x07ssh-rsa\x00\x00\x00")
+ kex.set_host_key("ssh-rsa7", b"\x00\x00\x00\x07ssh-rsa\x00\x00\x00")
+ kex.set_host_key("ssh-rsa8", b"\x00\x00\x00\x07ssh-rsa\x00\x00\x00")
+
+ rv = ssh_audit.build_struct('localhost', banner=None, kex=kex)
+
+ assert len(rv["fingerprints"]) == (9 * 2) # Each host key generates two hash fingerprints: one using SHA256, and one using MD5.
+
+ for key in ['banner', 'compression', 'enc', 'fingerprints', 'kex', 'key', 'mac']:
+ assert key in rv
diff --git a/test/test_errors.py b/test/test_errors.py
new file mode 100644
index 0000000..90cfd1c
--- /dev/null
+++ b/test/test_errors.py
@@ -0,0 +1,172 @@
+import socket
+import errno
+import pytest
+
+from ssh_audit.outputbuffer import OutputBuffer
+
+
+# pylint: disable=attribute-defined-outside-init
+class TestErrors:
+ @pytest.fixture(autouse=True)
+ def init(self, ssh_audit):
+ self.AuditConf = ssh_audit.AuditConf
+ self.OutputBuffer = ssh_audit.OutputBuffer
+ self.audit = ssh_audit.audit
+
+ def _conf(self):
+ conf = self.AuditConf('localhost', 22)
+ conf.colors = False
+ conf.batch = True
+ return conf
+
+ def _audit(self, spy, conf=None, exit_expected=False):
+ if conf is None:
+ conf = self._conf()
+ spy.begin()
+
+ out = OutputBuffer()
+ if exit_expected:
+ with pytest.raises(SystemExit):
+ self.audit(out, conf)
+ else:
+ ret = self.audit(out, conf)
+ assert ret != 0
+
+ out.write()
+ lines = spy.flush()
+
+ # If the last line is empty, delete it.
+ if len(lines) > 1 and lines[-1] == '':
+ del lines[-1]
+
+ return lines
+
+ def test_connection_unresolved(self, output_spy, virtual_socket):
+ vsocket = virtual_socket
+ vsocket.gsock.addrinfodata['localhost#22'] = []
+ lines = self._audit(output_spy, exit_expected=True)
+ assert len(lines) == 1
+ assert 'has no DNS records' in lines[-1]
+
+ def test_connection_refused(self, output_spy, virtual_socket):
+ vsocket = virtual_socket
+ vsocket.errors['connect'] = socket.error(errno.ECONNREFUSED, 'Connection refused')
+ lines = self._audit(output_spy, exit_expected=True)
+ assert len(lines) == 1
+ assert 'Connection refused' in lines[-1]
+
+ def test_connection_timeout(self, output_spy, virtual_socket):
+ vsocket = virtual_socket
+ vsocket.errors['connect'] = socket.timeout('timed out')
+ lines = self._audit(output_spy, exit_expected=True)
+ assert len(lines) == 1
+ assert 'timed out' in lines[-1]
+
+ def test_recv_empty(self, output_spy, virtual_socket):
+ lines = self._audit(output_spy)
+ assert len(lines) == 1
+ assert 'did not receive banner' in lines[-1]
+
+ def test_recv_timeout(self, output_spy, virtual_socket):
+ vsocket = virtual_socket
+ vsocket.rdata.append(socket.timeout('timed out'))
+ lines = self._audit(output_spy)
+ assert len(lines) == 1
+ assert 'did not receive banner' in lines[-1]
+ assert 'timed out' in lines[-1]
+
+ def test_recv_retry_till_timeout(self, output_spy, virtual_socket):
+ vsocket = virtual_socket
+ vsocket.rdata.append(socket.error(errno.EAGAIN, 'Resource temporarily unavailable'))
+ vsocket.rdata.append(socket.error(errno.EWOULDBLOCK, 'Resource temporarily unavailable'))
+ vsocket.rdata.append(socket.error(errno.EAGAIN, 'Resource temporarily unavailable'))
+ vsocket.rdata.append(socket.timeout('timed out'))
+ lines = self._audit(output_spy)
+ assert len(lines) == 1
+ assert 'did not receive banner' in lines[-1]
+ assert 'timed out' in lines[-1]
+
+ def test_recv_retry_till_reset(self, output_spy, virtual_socket):
+ vsocket = virtual_socket
+ vsocket.rdata.append(socket.error(errno.EAGAIN, 'Resource temporarily unavailable'))
+ vsocket.rdata.append(socket.error(errno.EWOULDBLOCK, 'Resource temporarily unavailable'))
+ vsocket.rdata.append(socket.error(errno.EAGAIN, 'Resource temporarily unavailable'))
+ vsocket.rdata.append(socket.error(errno.ECONNRESET, 'Connection reset by peer'))
+ lines = self._audit(output_spy)
+ assert len(lines) == 1
+ assert 'did not receive banner' in lines[-1]
+ assert 'reset by peer' in lines[-1]
+
+ def test_connection_closed_before_banner(self, output_spy, virtual_socket):
+ vsocket = virtual_socket
+ vsocket.rdata.append(socket.error(errno.ECONNRESET, 'Connection reset by peer'))
+ lines = self._audit(output_spy)
+ assert len(lines) == 1
+ assert 'did not receive banner' in lines[-1]
+ assert 'reset by peer' in lines[-1]
+
+ def test_connection_closed_after_header(self, output_spy, virtual_socket):
+ vsocket = virtual_socket
+ vsocket.rdata.append(b'header line 1\n')
+ vsocket.rdata.append(b'\n')
+ vsocket.rdata.append(b'header line 2\n')
+ vsocket.rdata.append(socket.error(errno.ECONNRESET, 'Connection reset by peer'))
+ lines = self._audit(output_spy)
+ assert len(lines) == 3
+ assert 'did not receive banner' in lines[-1]
+ assert 'reset by peer' in lines[-1]
+
+ def test_connection_closed_after_banner(self, output_spy, virtual_socket):
+ vsocket = virtual_socket
+ vsocket.rdata.append(b'SSH-2.0-ssh-audit-test\r\n')
+ vsocket.rdata.append(socket.error(54, 'Connection reset by peer'))
+ lines = self._audit(output_spy)
+ assert len(lines) == 2
+ assert 'error reading packet' in lines[-1]
+ assert 'reset by peer' in lines[-1]
+
+ def test_empty_data_after_banner(self, output_spy, virtual_socket):
+ vsocket = virtual_socket
+ vsocket.rdata.append(b'SSH-2.0-ssh-audit-test\r\n')
+ lines = self._audit(output_spy)
+ assert len(lines) == 2
+ assert 'error reading packet' in lines[-1]
+ assert 'empty' in lines[-1]
+
+ def test_wrong_data_after_banner(self, output_spy, virtual_socket):
+ vsocket = virtual_socket
+ vsocket.rdata.append(b'SSH-2.0-ssh-audit-test\r\n')
+ vsocket.rdata.append(b'xxx\n')
+ lines = self._audit(output_spy)
+ assert len(lines) == 2
+ assert 'error reading packet' in lines[-1]
+ assert 'xxx' in lines[-1]
+
+ def test_non_ascii_banner(self, output_spy, virtual_socket):
+ vsocket = virtual_socket
+ vsocket.rdata.append(b'SSH-2.0-ssh-audit-test\xc3\xbc\r\n')
+ lines = self._audit(output_spy)
+ assert len(lines) == 3
+ assert 'error reading packet' in lines[-1]
+ assert 'ASCII' in lines[-2]
+ assert lines[-3].endswith('SSH-2.0-ssh-audit-test?')
+
+ def test_nonutf8_data_after_banner(self, output_spy, virtual_socket):
+ vsocket = virtual_socket
+ vsocket.rdata.append(b'SSH-2.0-ssh-audit-test\r\n')
+ vsocket.rdata.append(b'\x81\xff\n')
+ lines = self._audit(output_spy)
+ assert len(lines) == 2
+ assert 'error reading packet' in lines[-1]
+ assert '\\x81\\xff' in lines[-1]
+
+ def test_protocol_mismatch_by_conf(self, output_spy, virtual_socket):
+ vsocket = virtual_socket
+ vsocket.rdata.append(b'SSH-1.3-ssh-audit-test\r\n')
+ vsocket.rdata.append(b'Protocol major versions differ.\n')
+ conf = self._conf()
+ conf.ssh1, conf.ssh2 = True, False
+ lines = self._audit(output_spy, conf)
+ assert len(lines) == 4
+ assert 'error reading packet' in lines[-1]
+ assert 'major versions differ' in lines[-1]
diff --git a/test/test_outputbuffer.py b/test/test_outputbuffer.py
new file mode 100644
index 0000000..d92d7bd
--- /dev/null
+++ b/test/test_outputbuffer.py
@@ -0,0 +1,183 @@
+import pytest
+
+
+# pylint: disable=attribute-defined-outside-init
+class TestOutputBuffer:
+ @pytest.fixture(autouse=True)
+ def init(self, ssh_audit):
+ self.OutputBuffer = ssh_audit.OutputBuffer
+
+ def test_outputbuffer_no_lines(self, output_spy):
+ output_spy.begin()
+ obuf = self.OutputBuffer()
+ obuf.write()
+ assert output_spy.flush() == ['']
+ output_spy.begin()
+
+ def test_outputbuffer_defaults(self):
+ obuf = self.OutputBuffer()
+ # default: on
+ assert obuf.batch is False
+ assert obuf.use_colors is True
+ assert obuf.level == 'info'
+
+ def test_outputbuffer_colors(self, output_spy):
+ out = self.OutputBuffer()
+
+ # Test without colors.
+ out.use_colors = False
+
+ output_spy.begin()
+ out.info('info color')
+ out.write()
+ assert output_spy.flush() == ['info color']
+
+ output_spy.begin()
+ out.head('head color')
+ out.write()
+ assert output_spy.flush() == ['head color']
+
+ output_spy.begin()
+ out.good('good color')
+ out.write()
+ assert output_spy.flush() == ['good color']
+
+ output_spy.begin()
+ out.warn('warn color')
+ out.write()
+ assert output_spy.flush() == ['warn color']
+
+ output_spy.begin()
+ out.fail('fail color')
+ out.write()
+ assert output_spy.flush() == ['fail color']
+
+ # If colors aren't supported by this system, skip the color tests.
+ if not out.colors_supported:
+ return
+
+ # Test with colors.
+ out.use_colors = True
+
+ output_spy.begin()
+ out.info('info color')
+ out.write()
+ assert output_spy.flush() == ['info color']
+
+ output_spy.begin()
+ out.head('head color')
+ out.write()
+ assert output_spy.flush() in [['\x1b[0;36mhead color\x1b[0m'], ['\x1b[0;96mhead color\x1b[0m']]
+
+ output_spy.begin()
+ out.good('good color')
+ out.write()
+ assert output_spy.flush() in [['\x1b[0;32mgood color\x1b[0m'], ['\x1b[0;92mgood color\x1b[0m']]
+
+ output_spy.begin()
+ out.warn('warn color')
+ out.write()
+ assert output_spy.flush() in [['\x1b[0;33mwarn color\x1b[0m'], ['\x1b[0;93mwarn color\x1b[0m']]
+
+ output_spy.begin()
+ out.fail('fail color')
+ out.write()
+ assert output_spy.flush() in [['\x1b[0;31mfail color\x1b[0m'], ['\x1b[0;91mfail color\x1b[0m']]
+
+ def test_outputbuffer_sep(self, output_spy):
+ out = self.OutputBuffer()
+ output_spy.begin()
+ out.sep()
+ out.sep()
+ out.sep()
+ out.write()
+ assert output_spy.flush() == ['', '', '']
+
+ def test_outputbuffer_levels(self):
+ out = self.OutputBuffer()
+ assert out.get_level('info') == 0
+ assert out.get_level('good') == 0
+ assert out.get_level('warn') == 1
+ assert out.get_level('fail') == 2
+ assert out.get_level('unknown') > 2
+
+ def test_outputbuffer_level_property(self):
+ out = self.OutputBuffer()
+ out.level = 'info'
+ assert out.level == 'info'
+ out.level = 'good'
+ assert out.level == 'info'
+ out.level = 'warn'
+ assert out.level == 'warn'
+ out.level = 'fail'
+ assert out.level == 'fail'
+ out.level = 'invalid level'
+ assert out.level == 'unknown'
+
+ def test_outputbuffer_level(self, output_spy):
+ out = self.OutputBuffer()
+ # visible: all
+ out.level = 'info'
+ output_spy.begin()
+ out.info('info color')
+ out.head('head color')
+ out.good('good color')
+ out.warn('warn color')
+ out.fail('fail color')
+ out.write()
+ assert len(output_spy.flush()) == 5
+ # visible: head, warn, fail
+ out.level = 'warn'
+ output_spy.begin()
+ out.info('info color')
+ out.head('head color')
+ out.good('good color')
+ out.warn('warn color')
+ out.fail('fail color')
+ out.write()
+ assert len(output_spy.flush()) == 3
+ # visible: head, fail
+ out.level = 'fail'
+ output_spy.begin()
+ out.info('info color')
+ out.head('head color')
+ out.good('good color')
+ out.warn('warn color')
+ out.fail('fail color')
+ out.write()
+ assert len(output_spy.flush()) == 2
+ # visible: head
+ out.level = 'invalid level'
+ output_spy.begin()
+ out.info('info color')
+ out.head('head color')
+ out.good('good color')
+ out.warn('warn color')
+ out.fail('fail color')
+ out.write()
+ assert len(output_spy.flush()) == 1
+
+ def test_outputbuffer_batch(self, output_spy):
+ out = self.OutputBuffer()
+ # visible: all
+ output_spy.begin()
+ out.level = 'info'
+ out.batch = False
+ out.info('info color')
+ out.head('head color')
+ out.good('good color')
+ out.warn('warn color')
+ out.fail('fail color')
+ out.write()
+ assert len(output_spy.flush()) == 5
+ # visible: all except head
+ output_spy.begin()
+ out.level = 'info'
+ out.batch = True
+ out.info('info color')
+ out.head('head color')
+ out.good('good color')
+ out.warn('warn color')
+ out.fail('fail color')
+ out.write()
+ assert len(output_spy.flush()) == 4
diff --git a/test/test_policy.py b/test/test_policy.py
new file mode 100644
index 0000000..b78fab2
--- /dev/null
+++ b/test/test_policy.py
@@ -0,0 +1,358 @@
+import hashlib
+import pytest
+from datetime import date
+
+from ssh_audit.policy import Policy
+from ssh_audit.ssh2_kex import SSH2_Kex
+from ssh_audit.writebuf import WriteBuf
+
+
+class TestPolicy:
+ @pytest.fixture(autouse=True)
+ def init(self, ssh_audit):
+ self.Policy = Policy
+ self.wbuf = WriteBuf
+ self.ssh2_kex = SSH2_Kex
+
+
+ def _get_kex(self):
+ '''Returns an SSH2.Kex object to simulate a server connection.'''
+
+ w = self.wbuf()
+ w.write(b'\x00\x11\x22\x33\x44\x55\x66\x77\x88\x99\xaa\xbb\xcc\xdd\xee\xff')
+ w.write_list(['kex_alg1', 'kex_alg2'])
+ w.write_list(['key_alg1', 'key_alg2'])
+ w.write_list(['cipher_alg1', 'cipher_alg2', 'cipher_alg3'])
+ w.write_list(['cipher_alg1', 'cipher_alg2', 'cipher_alg3'])
+ w.write_list(['mac_alg1', 'mac_alg2', 'mac_alg3'])
+ w.write_list(['mac_alg1', 'mac_alg2', 'mac_alg3'])
+ w.write_list(['comp_alg1', 'comp_alg2'])
+ w.write_list(['comp_alg1', 'comp_alg2'])
+ w.write_list([''])
+ w.write_list([''])
+ w.write_byte(False)
+ w.write_int(0)
+ return self.ssh2_kex.parse(w.write_flush())
+
+
+ def test_builtin_policy_consistency(self):
+ '''Ensure that the BUILTIN_POLICIES struct is consistent.'''
+
+ for policy_name in Policy.BUILTIN_POLICIES:
+ # Ensure that the policy name ends with " (version X)", where X is the 'version' field.
+ version_str = " (version %s)" % Policy.BUILTIN_POLICIES[policy_name]['version']
+ assert(policy_name.endswith(version_str))
+
+ # Ensure that each built-in policy can be loaded with Policy.load_builtin_policy().
+ assert(Policy.load_builtin_policy(policy_name) is not None)
+
+ # Ensure that both server and client policy names are returned.
+ server_policy_names, client_policy_names = Policy.list_builtin_policies()
+ assert(len(server_policy_names) > 0)
+ assert(len(client_policy_names) > 0)
+
+
+ def test_policy_basic(self):
+ '''Ensure that a basic policy can be parsed correctly.'''
+
+ policy_data = '''# This is a comment
+name = "Test Policy"
+version = 1
+
+compressions = comp_alg1
+host keys = key_alg1
+key exchanges = kex_alg1, kex_alg2
+ciphers = cipher_alg1, cipher_alg2, cipher_alg3
+macs = mac_alg1, mac_alg2, mac_alg3'''
+
+ policy = self.Policy(policy_data=policy_data)
+ assert str(policy) == "Name: [Test Policy]\nVersion: [1]\nBanner: {undefined}\nCompressions: comp_alg1\nHost Keys: key_alg1\nOptional Host Keys: {undefined}\nKey Exchanges: kex_alg1, kex_alg2\nCiphers: cipher_alg1, cipher_alg2, cipher_alg3\nMACs: mac_alg1, mac_alg2, mac_alg3\nHost Key Sizes: {undefined}\nCA Key Sizes: {undefined}\nDH Modulus Sizes: {undefined}\nServer Policy: True"
+
+
+ def test_policy_invalid_1(self):
+ '''Basic policy, but with 'ciphersx' instead of 'ciphers'.'''
+
+ policy_data = '''# This is a comment
+name = "Test Policy"
+version = 1
+
+compressions = comp_alg1
+host keys = key_alg1
+key exchanges = kex_alg1, kex_alg2
+ciphersx = cipher_alg1, cipher_alg2, cipher_alg3
+macs = mac_alg1, mac_alg2, mac_alg3'''
+
+ failed = False
+ try:
+ self.Policy(policy_data=policy_data)
+ except ValueError:
+ failed = True
+
+ assert failed, "Invalid policy did not cause Policy object to throw exception"
+
+
+ def test_policy_invalid_2(self):
+ '''Basic policy, but is missing the required name field.'''
+
+ policy_data = '''# This is a comment
+#name = "Test Policy"
+version = 1
+
+compressions = comp_alg1
+host keys = key_alg1
+key exchanges = kex_alg1, kex_alg2
+ciphers = cipher_alg1, cipher_alg2, cipher_alg3
+macs = mac_alg1, mac_alg2, mac_alg3'''
+
+ failed = False
+ try:
+ self.Policy(policy_data=policy_data)
+ except ValueError:
+ failed = True
+
+ assert failed, "Invalid policy did not cause Policy object to throw exception"
+
+
+ def test_policy_invalid_3(self):
+ '''Basic policy, but is missing the required version field.'''
+
+ policy_data = '''# This is a comment
+name = "Test Policy"
+#version = 1
+
+compressions = comp_alg1
+host keys = key_alg1
+key exchanges = kex_alg1, kex_alg2
+ciphers = cipher_alg1, cipher_alg2, cipher_alg3
+macs = mac_alg1, mac_alg2, mac_alg3'''
+
+ failed = False
+ try:
+ self.Policy(policy_data=policy_data)
+ except ValueError:
+ failed = True
+
+ assert failed, "Invalid policy did not cause Policy object to throw exception"
+
+
+ def test_policy_invalid_4(self):
+ '''Basic policy, but is missing quotes in the name field.'''
+
+ policy_data = '''# This is a comment
+name = Test Policy
+version = 1
+
+compressions = comp_alg1
+host keys = key_alg1
+key exchanges = kex_alg1, kex_alg2
+ciphers = cipher_alg1, cipher_alg2, cipher_alg3
+macs = mac_alg1, mac_alg2, mac_alg3'''
+
+ failed = False
+ try:
+ self.Policy(policy_data=policy_data)
+ except ValueError:
+ failed = True
+
+ assert failed, "Invalid policy did not cause Policy object to throw exception"
+
+
+ def test_policy_invalid_5(self):
+ '''Basic policy, but is missing quotes in the banner field.'''
+
+ policy_data = '''# This is a comment
+name = "Test Policy"
+version = 1
+
+banner = 0mg
+compressions = comp_alg1
+host keys = key_alg1
+key exchanges = kex_alg1, kex_alg2
+ciphers = cipher_alg1, cipher_alg2, cipher_alg3
+macs = mac_alg1, mac_alg2, mac_alg3'''
+
+ failed = False
+ try:
+ self.Policy(policy_data=policy_data)
+ except ValueError:
+ failed = True
+
+ assert failed, "Invalid policy did not cause Policy object to throw exception"
+
+
+ def test_policy_invalid_6(self):
+ '''Basic policy, but is missing quotes in the header field.'''
+
+ policy_data = '''# This is a comment
+name = "Test Policy"
+version = 1
+
+header = 0mg
+compressions = comp_alg1
+host keys = key_alg1
+key exchanges = kex_alg1, kex_alg2
+ciphers = cipher_alg1, cipher_alg2, cipher_alg3
+macs = mac_alg1, mac_alg2, mac_alg3'''
+
+ failed = False
+ try:
+ self.Policy(policy_data=policy_data)
+ except ValueError:
+ failed = True
+
+ assert failed, "Invalid policy did not cause Policy object to throw exception"
+
+
+ def test_policy_create_1(self):
+ '''Creates a policy from a kex and ensures it is generated exactly as expected.'''
+
+ kex = self._get_kex()
+ pol_data = self.Policy.create('www.l0l.com', 'bannerX', kex, False)
+
+ # Today's date is embedded in the policy, so filter it out to get repeatable results.
+ pol_data = pol_data.replace(date.today().strftime('%Y/%m/%d'), '[todays date]')
+
+ # Instead of writing out the entire expected policy--line by line--just check that it has the expected hash.
+ assert hashlib.sha256(pol_data.encode('ascii')).hexdigest() == '4af7777fb57a1dad0cf438c899a11d4f625fd9276ea3bb5ef5c9fe8806cb47dc'
+
+
+ def test_policy_evaluate_passing_1(self):
+ '''Creates a policy and evaluates it against the same server'''
+
+ kex = self._get_kex()
+ policy_data = self.Policy.create('www.l0l.com', None, kex, False)
+ policy = self.Policy(policy_data=policy_data)
+
+ ret, errors, error_str = policy.evaluate('SSH Server 1.0', kex)
+ assert ret is True
+ assert len(errors) == 0
+ print(error_str)
+ assert len(error_str) == 0
+
+
+ def test_policy_evaluate_failing_1(self):
+ '''Ensure that a policy with a specified banner fails against a server with a different banner'''
+
+ policy_data = '''name = "Test Policy"
+version = 1
+banner = "XXX mismatched banner XXX"
+compressions = comp_alg1, comp_alg2
+host keys = key_alg1, key_alg2
+key exchanges = kex_alg1, kex_alg2
+ciphers = cipher_alg1, cipher_alg2, cipher_alg3
+macs = mac_alg1, mac_alg2, mac_alg3'''
+
+ policy = self.Policy(policy_data=policy_data)
+ ret, errors, error_str = policy.evaluate('SSH Server 1.0', self._get_kex())
+ assert ret is False
+ assert len(errors) == 1
+ assert error_str.find('Banner did not match.') != -1
+
+
+ def test_policy_evaluate_failing_2(self):
+ '''Ensure that a mismatched compressions list results in a failure'''
+
+ policy_data = '''name = "Test Policy"
+version = 1
+compressions = XXXmismatchedXXX, comp_alg1, comp_alg2
+host keys = key_alg1, key_alg2
+key exchanges = kex_alg1, kex_alg2
+ciphers = cipher_alg1, cipher_alg2, cipher_alg3
+macs = mac_alg1, mac_alg2, mac_alg3'''
+
+ policy = self.Policy(policy_data=policy_data)
+ ret, errors, error_str = policy.evaluate('SSH Server 1.0', self._get_kex())
+ assert ret is False
+ assert len(errors) == 1
+ assert error_str.find('Compression did not match.') != -1
+
+
+ def test_policy_evaluate_failing_3(self):
+ '''Ensure that a mismatched host keys results in a failure'''
+
+ policy_data = '''name = "Test Policy"
+version = 1
+compressions = comp_alg1, comp_alg2
+host keys = XXXmismatchedXXX, key_alg1, key_alg2
+key exchanges = kex_alg1, kex_alg2
+ciphers = cipher_alg1, cipher_alg2, cipher_alg3
+macs = mac_alg1, mac_alg2, mac_alg3'''
+
+ policy = self.Policy(policy_data=policy_data)
+ ret, errors, error_str = policy.evaluate('SSH Server 1.0', self._get_kex())
+ assert ret is False
+ assert len(errors) == 1
+ assert error_str.find('Host keys did not match.') != -1
+
+
+ def test_policy_evaluate_failing_4(self):
+ '''Ensure that a mismatched key exchange list results in a failure'''
+
+ policy_data = '''name = "Test Policy"
+version = 1
+compressions = comp_alg1, comp_alg2
+host keys = key_alg1, key_alg2
+key exchanges = XXXmismatchedXXX, kex_alg1, kex_alg2
+ciphers = cipher_alg1, cipher_alg2, cipher_alg3
+macs = mac_alg1, mac_alg2, mac_alg3'''
+
+ policy = self.Policy(policy_data=policy_data)
+ ret, errors, error_str = policy.evaluate('SSH Server 1.0', self._get_kex())
+ assert ret is False
+ assert len(errors) == 1
+ assert error_str.find('Key exchanges did not match.') != -1
+
+
+ def test_policy_evaluate_failing_5(self):
+ '''Ensure that a mismatched cipher list results in a failure'''
+
+ policy_data = '''name = "Test Policy"
+version = 1
+compressions = comp_alg1, comp_alg2
+host keys = key_alg1, key_alg2
+key exchanges = kex_alg1, kex_alg2
+ciphers = cipher_alg1, XXXmismatched, cipher_alg2, cipher_alg3
+macs = mac_alg1, mac_alg2, mac_alg3'''
+
+ policy = self.Policy(policy_data=policy_data)
+ ret, errors, error_str = policy.evaluate('SSH Server 1.0', self._get_kex())
+ assert ret is False
+ assert len(errors) == 1
+ assert error_str.find('Ciphers did not match.') != -1
+
+
+ def test_policy_evaluate_failing_6(self):
+ '''Ensure that a mismatched MAC list results in a failure'''
+
+ policy_data = '''name = "Test Policy"
+version = 1
+compressions = comp_alg1, comp_alg2
+host keys = key_alg1, key_alg2
+key exchanges = kex_alg1, kex_alg2
+ciphers = cipher_alg1, cipher_alg2, cipher_alg3
+macs = mac_alg1, mac_alg2, XXXmismatched, mac_alg3'''
+
+ policy = self.Policy(policy_data=policy_data)
+ ret, errors, error_str = policy.evaluate('SSH Server 1.0', self._get_kex())
+ assert ret is False
+ assert len(errors) == 1
+ assert error_str.find('MACs did not match.') != -1
+
+
+ def test_policy_evaluate_failing_7(self):
+ '''Ensure that a mismatched host keys and MACs results in a failure'''
+
+ policy_data = '''name = "Test Policy"
+version = 1
+compressions = comp_alg1, comp_alg2
+host keys = key_alg1, key_alg2, XXXmismatchedXXX
+key exchanges = kex_alg1, kex_alg2
+ciphers = cipher_alg1, cipher_alg2, cipher_alg3
+macs = mac_alg1, mac_alg2, XXXmismatchedXXX, mac_alg3'''
+
+ policy = self.Policy(policy_data=policy_data)
+ ret, errors, error_str = policy.evaluate('SSH Server 1.0', self._get_kex())
+ assert ret is False
+ assert len(errors) == 2
+ assert error_str.find('Host keys did not match.') != -1
+ assert error_str.find('MACs did not match.') != -1
diff --git a/test/test_resolve.py b/test/test_resolve.py
new file mode 100644
index 0000000..fbaa033
--- /dev/null
+++ b/test/test_resolve.py
@@ -0,0 +1,81 @@
+import socket
+import pytest
+
+
+# pylint: disable=attribute-defined-outside-init,protected-access
+class TestResolve:
+ @pytest.fixture(autouse=True)
+ def init(self, ssh_audit):
+ self.AuditConf = ssh_audit.AuditConf
+ self.audit = ssh_audit.audit
+ self.OutputBuffer = ssh_audit.OutputBuffer
+ self.ssh_socket = ssh_audit.SSH_Socket
+
+ def _conf(self):
+ conf = self.AuditConf('localhost', 22)
+ conf.colors = False
+ conf.batch = True
+ return conf
+
+ def test_resolve_error(self, output_spy, virtual_socket):
+ vsocket = virtual_socket
+ vsocket.gsock.addrinfodata['localhost#22'] = socket.gaierror(8, 'hostname nor servname provided, or not known')
+ conf = self._conf()
+ s = self.ssh_socket(self.OutputBuffer(), 'localhost', 22, conf.ip_version_preference)
+ output_spy.begin()
+ with pytest.raises(SystemExit):
+ list(s._resolve())
+ lines = output_spy.flush()
+ assert len(lines) == 1
+ assert 'hostname nor servname provided' in lines[-1]
+
+ def test_resolve_hostname_without_records(self, output_spy, virtual_socket):
+ vsocket = virtual_socket
+ vsocket.gsock.addrinfodata['localhost#22'] = []
+ conf = self._conf()
+ s = self.ssh_socket(self.OutputBuffer(), 'localhost', 22, conf.ip_version_preference)
+ output_spy.begin()
+ r = list(s._resolve())
+ assert len(r) == 0
+
+ def test_resolve_ipv4(self, virtual_socket):
+ conf = self._conf()
+ conf.ipv4 = True
+ s = self.ssh_socket(self.OutputBuffer(), 'localhost', 22, conf.ip_version_preference)
+ r = list(s._resolve())
+ assert len(r) == 1
+ assert r[0] == (socket.AF_INET, ('127.0.0.1', 22))
+
+ def test_resolve_ipv6(self, virtual_socket):
+ conf = self._conf()
+ conf.ipv6 = True
+ s = self.ssh_socket(self.OutputBuffer(), 'localhost', 22, conf.ip_version_preference)
+ r = list(s._resolve())
+ assert len(r) == 1
+ assert r[0] == (socket.AF_INET6, ('::1', 22))
+
+ def test_resolve_ipv46_both(self, virtual_socket):
+ conf = self._conf()
+ s = self.ssh_socket(self.OutputBuffer(), 'localhost', 22, conf.ip_version_preference)
+ r = list(s._resolve())
+ assert len(r) == 2
+ assert r[0] == (socket.AF_INET, ('127.0.0.1', 22))
+ assert r[1] == (socket.AF_INET6, ('::1', 22))
+
+ def test_resolve_ipv46_order(self, virtual_socket):
+ conf = self._conf()
+ conf.ipv4 = True
+ conf.ipv6 = True
+ s = self.ssh_socket(self.OutputBuffer(), 'localhost', 22, conf.ip_version_preference)
+ r = list(s._resolve())
+ assert len(r) == 2
+ assert r[0] == (socket.AF_INET, ('127.0.0.1', 22))
+ assert r[1] == (socket.AF_INET6, ('::1', 22))
+ conf = self._conf()
+ conf.ipv6 = True
+ conf.ipv4 = True
+ s = self.ssh_socket(self.OutputBuffer(), 'localhost', 22, conf.ip_version_preference)
+ r = list(s._resolve())
+ assert len(r) == 2
+ assert r[0] == (socket.AF_INET6, ('::1', 22))
+ assert r[1] == (socket.AF_INET, ('127.0.0.1', 22))
diff --git a/test/test_socket.py b/test/test_socket.py
new file mode 100644
index 0000000..9569768
--- /dev/null
+++ b/test/test_socket.py
@@ -0,0 +1,42 @@
+import pytest
+
+from ssh_audit.outputbuffer import OutputBuffer
+from ssh_audit.ssh_socket import SSH_Socket
+
+
+# pylint: disable=attribute-defined-outside-init
+class TestSocket:
+ @pytest.fixture(autouse=True)
+ def init(self, ssh_audit):
+ self.OutputBuffer = OutputBuffer
+ self.ssh_socket = SSH_Socket
+
+ def test_invalid_host(self, virtual_socket):
+ with pytest.raises(ValueError):
+ self.ssh_socket(self.OutputBuffer(), None, 22)
+
+ def test_invalid_port(self, virtual_socket):
+ with pytest.raises(ValueError):
+ self.ssh_socket(self.OutputBuffer(), 'localhost', 'abc')
+ with pytest.raises(ValueError):
+ self.ssh_socket(self.OutputBuffer(), 'localhost', -1)
+ with pytest.raises(ValueError):
+ self.ssh_socket(self.OutputBuffer(), 'localhost', 0)
+ with pytest.raises(ValueError):
+ self.ssh_socket(self.OutputBuffer(), 'localhost', 65536)
+
+ def test_not_connected_socket(self, virtual_socket):
+ sock = self.ssh_socket(self.OutputBuffer(), 'localhost', 22)
+ banner, header, err = sock.get_banner()
+ assert banner is None
+ assert len(header) == 0
+ assert err == 'not connected'
+ s, e = sock.recv()
+ assert s == -1
+ assert e == 'not connected'
+ s, e = sock.send('nothing')
+ assert s == -1
+ assert e == 'not connected'
+ s, e = sock.send_packet()
+ assert s == -1
+ assert e == 'not connected'
diff --git a/test/test_software.py b/test/test_software.py
new file mode 100644
index 0000000..0240ffc
--- /dev/null
+++ b/test/test_software.py
@@ -0,0 +1,289 @@
+import pytest
+
+from ssh_audit.banner import Banner
+from ssh_audit.software import Software
+
+
+# pylint: disable=line-too-long,attribute-defined-outside-init
+class TestSoftware:
+ @pytest.fixture(autouse=True)
+ def init(self, ssh_audit):
+ self.software = Software
+ self.banner = Banner
+
+ def test_unknown_software(self):
+ ps = lambda x: self.software.parse(self.banner.parse(x)) # noqa
+ assert ps('SSH-1.5') is None
+ assert ps('SSH-1.99-AlfaMegaServer') is None
+ assert ps('SSH-2.0-BetaMegaServer 0.0.1') is None
+
+ def test_openssh_software(self):
+ # pylint: disable=too-many-statements
+ ps = lambda x: self.software.parse(self.banner.parse(x)) # noqa
+ # common
+ s = ps('SSH-2.0-OpenSSH_7.3')
+ assert s.vendor is None
+ assert s.product == 'OpenSSH'
+ assert s.version == '7.3'
+ assert s.patch is None
+ assert s.os is None
+ assert str(s) == 'OpenSSH 7.3'
+ assert str(s) == s.display()
+ assert s.display(True) == str(s)
+ assert s.display(False) == str(s)
+ assert repr(s) == '<Software(product=OpenSSH, version=7.3)>'
+ # common, portable
+ s = ps('SSH-2.0-OpenSSH_7.2p1')
+ assert s.vendor is None
+ assert s.product == 'OpenSSH'
+ assert s.version == '7.2'
+ assert s.patch == 'p1'
+ assert s.os is None
+ assert str(s) == 'OpenSSH 7.2p1'
+ assert str(s) == s.display()
+ assert s.display(True) == str(s)
+ assert s.display(False) == 'OpenSSH 7.2'
+ assert repr(s) == '<Software(product=OpenSSH, version=7.2, patch=p1)>'
+ # dot instead of underline
+ s = ps('SSH-2.0-OpenSSH.6.6')
+ assert s.vendor is None
+ assert s.product == 'OpenSSH'
+ assert s.version == '6.6'
+ assert s.patch is None
+ assert s.os is None
+ assert str(s) == 'OpenSSH 6.6'
+ assert str(s) == s.display()
+ assert s.display(True) == str(s)
+ assert s.display(False) == str(s)
+ assert repr(s) == '<Software(product=OpenSSH, version=6.6)>'
+ # dash instead of underline
+ s = ps('SSH-2.0-OpenSSH-3.9p1')
+ assert s.vendor is None
+ assert s.product == 'OpenSSH'
+ assert s.version == '3.9'
+ assert s.patch == 'p1'
+ assert s.os is None
+ assert str(s) == 'OpenSSH 3.9p1'
+ assert str(s) == s.display()
+ assert s.display(True) == str(s)
+ assert s.display(False) == 'OpenSSH 3.9'
+ assert repr(s) == '<Software(product=OpenSSH, version=3.9, patch=p1)>'
+ # patch prefix with dash
+ s = ps('SSH-2.0-OpenSSH_7.2-hpn14v5')
+ assert s.vendor is None
+ assert s.product == 'OpenSSH'
+ assert s.version == '7.2'
+ assert s.patch == 'hpn14v5'
+ assert s.os is None
+ assert str(s) == 'OpenSSH 7.2 (hpn14v5)'
+ assert str(s) == s.display()
+ assert s.display(True) == str(s)
+ assert s.display(False) == 'OpenSSH 7.2'
+ assert repr(s) == '<Software(product=OpenSSH, version=7.2, patch=hpn14v5)>'
+ # patch prefix with underline
+ s = ps('SSH-1.5-OpenSSH_6.6.1_hpn13v11')
+ assert s.vendor is None
+ assert s.product == 'OpenSSH'
+ assert s.version == '6.6.1'
+ assert s.patch == 'hpn13v11'
+ assert s.os is None
+ assert str(s) == 'OpenSSH 6.6.1 (hpn13v11)'
+ assert str(s) == s.display()
+ assert s.display(True) == str(s)
+ assert s.display(False) == 'OpenSSH 6.6.1'
+ assert repr(s) == '<Software(product=OpenSSH, version=6.6.1, patch=hpn13v11)>'
+ # patch prefix with dot
+ s = ps('SSH-2.0-OpenSSH_5.9.CASPUR')
+ assert s.vendor is None
+ assert s.product == 'OpenSSH'
+ assert s.version == '5.9'
+ assert s.patch == 'CASPUR'
+ assert s.os is None
+ assert str(s) == 'OpenSSH 5.9 (CASPUR)'
+ assert str(s) == s.display()
+ assert s.display(True) == str(s)
+ assert s.display(False) == 'OpenSSH 5.9'
+ assert repr(s) == '<Software(product=OpenSSH, version=5.9, patch=CASPUR)>'
+
+ def test_dropbear_software(self):
+ ps = lambda x: self.software.parse(self.banner.parse(x)) # noqa
+ # common
+ s = ps('SSH-2.0-dropbear_2016.74')
+ assert s.vendor is None
+ assert s.product == 'Dropbear SSH'
+ assert s.version == '2016.74'
+ assert s.patch is None
+ assert s.os is None
+ assert str(s) == 'Dropbear SSH 2016.74'
+ assert str(s) == s.display()
+ assert s.display(True) == str(s)
+ assert s.display(False) == str(s)
+ assert repr(s) == '<Software(product=Dropbear SSH, version=2016.74)>'
+ # common, patch
+ s = ps('SSH-2.0-dropbear_0.44test4')
+ assert s.vendor is None
+ assert s.product == 'Dropbear SSH'
+ assert s.version == '0.44'
+ assert s.patch == 'test4'
+ assert s.os is None
+ assert str(s) == 'Dropbear SSH 0.44 (test4)'
+ assert str(s) == s.display()
+ assert s.display(True) == str(s)
+ assert s.display(False) == 'Dropbear SSH 0.44'
+ assert repr(s) == '<Software(product=Dropbear SSH, version=0.44, patch=test4)>'
+ # patch prefix with dash
+ s = ps('SSH-2.0-dropbear_0.44-Freesco-p49')
+ assert s.vendor is None
+ assert s.product == 'Dropbear SSH'
+ assert s.version == '0.44'
+ assert s.patch == 'Freesco-p49'
+ assert s.os is None
+ assert str(s) == 'Dropbear SSH 0.44 (Freesco-p49)'
+ assert str(s) == s.display()
+ assert s.display(True) == str(s)
+ assert s.display(False) == 'Dropbear SSH 0.44'
+ assert repr(s) == '<Software(product=Dropbear SSH, version=0.44, patch=Freesco-p49)>'
+ # patch prefix with underline
+ s = ps('SSH-2.0-dropbear_2014.66_agbn_1')
+ assert s.vendor is None
+ assert s.product == 'Dropbear SSH'
+ assert s.version == '2014.66'
+ assert s.patch == 'agbn_1'
+ assert s.os is None
+ assert str(s) == 'Dropbear SSH 2014.66 (agbn_1)'
+ assert str(s) == s.display()
+ assert s.display(True) == str(s)
+ assert s.display(False) == 'Dropbear SSH 2014.66'
+ assert repr(s) == '<Software(product=Dropbear SSH, version=2014.66, patch=agbn_1)>'
+
+ def test_libssh_software(self):
+ ps = lambda x: self.software.parse(self.banner.parse(x)) # noqa
+ # common
+ s = ps('SSH-2.0-libssh-0.2')
+ assert s.vendor is None
+ assert s.product == 'libssh'
+ assert s.version == '0.2'
+ assert s.patch is None
+ assert s.os is None
+ assert str(s) == 'libssh 0.2'
+ assert str(s) == s.display()
+ assert s.display(True) == str(s)
+ assert s.display(False) == str(s)
+ assert repr(s) == '<Software(product=libssh, version=0.2)>'
+ s = ps('SSH-2.0-libssh-0.7.4')
+ assert s.vendor is None
+ assert s.product == 'libssh'
+ assert s.version == '0.7.4'
+ assert s.patch is None
+ assert s.os is None
+ assert str(s) == 'libssh 0.7.4'
+ assert str(s) == s.display()
+ assert s.display(True) == str(s)
+ assert s.display(False) == str(s)
+ assert repr(s) == '<Software(product=libssh, version=0.7.4)>'
+
+ def test_romsshell_software(self):
+ ps = lambda x: self.software.parse(self.banner.parse(x)) # noqa
+ # common
+ s = ps('SSH-2.0-RomSShell_5.40')
+ assert s.vendor == 'Allegro Software'
+ assert s.product == 'RomSShell'
+ assert s.version == '5.40'
+ assert s.patch is None
+ assert s.os is None
+ assert str(s) == 'Allegro Software RomSShell 5.40'
+ assert str(s) == s.display()
+ assert s.display(True) == str(s)
+ assert s.display(False) == str(s)
+ assert repr(s) == '<Software(vendor=Allegro Software, product=RomSShell, version=5.40)>'
+
+ def test_hp_ilo_software(self):
+ ps = lambda x: self.software.parse(self.banner.parse(x)) # noqa
+ # common
+ s = ps('SSH-2.0-mpSSH_0.2.1')
+ assert s.vendor == 'HP'
+ assert s.product == 'iLO (Integrated Lights-Out) sshd'
+ assert s.version == '0.2.1'
+ assert s.patch is None
+ assert s.os is None
+ assert str(s) == 'HP iLO (Integrated Lights-Out) sshd 0.2.1'
+ assert str(s) == s.display()
+ assert s.display(True) == str(s)
+ assert s.display(False) == str(s)
+ assert repr(s) == '<Software(vendor=HP, product=iLO (Integrated Lights-Out) sshd, version=0.2.1)>'
+
+ def test_cisco_software(self):
+ ps = lambda x: self.software.parse(self.banner.parse(x)) # noqa
+ # common
+ s = ps('SSH-1.5-Cisco-1.25')
+ assert s.vendor == 'Cisco'
+ assert s.product == 'IOS/PIX sshd'
+ assert s.version == '1.25'
+ assert s.patch is None
+ assert s.os is None
+ assert str(s) == 'Cisco IOS/PIX sshd 1.25'
+ assert str(s) == s.display()
+ assert s.display(True) == str(s)
+ assert s.display(False) == str(s)
+ assert repr(s) == '<Software(vendor=Cisco, product=IOS/PIX sshd, version=1.25)>'
+
+ def test_software_os(self):
+ ps = lambda x: self.software.parse(self.banner.parse(x)) # noqa
+ # unknown
+ s = ps('SSH-2.0-OpenSSH_3.7.1 MegaOperatingSystem 123')
+ assert s.os is None
+ # NetBSD
+ s = ps('SSH-1.99-OpenSSH_2.5.1 NetBSD_Secure_Shell-20010614')
+ assert s.os == 'NetBSD (2001-06-14)'
+ assert str(s) == 'OpenSSH 2.5.1 running on NetBSD (2001-06-14)'
+ assert repr(s) == '<Software(product=OpenSSH, version=2.5.1, os=NetBSD (2001-06-14))>'
+ s = ps('SSH-1.99-OpenSSH_5.0 NetBSD_Secure_Shell-20080403+-hpn13v1')
+ assert s.os == 'NetBSD (2008-04-03)'
+ assert str(s) == 'OpenSSH 5.0 running on NetBSD (2008-04-03)'
+ assert repr(s) == '<Software(product=OpenSSH, version=5.0, os=NetBSD (2008-04-03))>'
+ s = ps('SSH-2.0-OpenSSH_6.6.1_hpn13v11 NetBSD-20100308')
+ assert s.os == 'NetBSD (2010-03-08)'
+ assert str(s) == 'OpenSSH 6.6.1 (hpn13v11) running on NetBSD (2010-03-08)'
+ assert repr(s) == '<Software(product=OpenSSH, version=6.6.1, patch=hpn13v11, os=NetBSD (2010-03-08))>'
+ s = ps('SSH-2.0-OpenSSH_4.4 NetBSD')
+ assert s.os == 'NetBSD'
+ assert str(s) == 'OpenSSH 4.4 running on NetBSD'
+ assert repr(s) == '<Software(product=OpenSSH, version=4.4, os=NetBSD)>'
+ s = ps('SSH-2.0-OpenSSH_3.0.2 NetBSD Secure Shell')
+ assert s.os == 'NetBSD'
+ assert str(s) == 'OpenSSH 3.0.2 running on NetBSD'
+ assert repr(s) == '<Software(product=OpenSSH, version=3.0.2, os=NetBSD)>'
+ # FreeBSD
+ s = ps('SSH-2.0-OpenSSH_7.2 FreeBSD-20160310')
+ assert s.os == 'FreeBSD (2016-03-10)'
+ assert str(s) == 'OpenSSH 7.2 running on FreeBSD (2016-03-10)'
+ assert repr(s) == '<Software(product=OpenSSH, version=7.2, os=FreeBSD (2016-03-10))>'
+ s = ps('SSH-1.99-OpenSSH_2.9 FreeBSD localisations 20020307')
+ assert s.os == 'FreeBSD (2002-03-07)'
+ assert str(s) == 'OpenSSH 2.9 running on FreeBSD (2002-03-07)'
+ assert repr(s) == '<Software(product=OpenSSH, version=2.9, os=FreeBSD (2002-03-07))>'
+ s = ps('SSH-2.0-OpenSSH_2.3.0 green@FreeBSD.org 20010321')
+ assert s.os == 'FreeBSD (2001-03-21)'
+ assert str(s) == 'OpenSSH 2.3.0 running on FreeBSD (2001-03-21)'
+ assert repr(s) == '<Software(product=OpenSSH, version=2.3.0, os=FreeBSD (2001-03-21))>'
+ s = ps('SSH-1.99-OpenSSH_4.4p1 FreeBSD-openssh-portable-overwrite-base-4.4.p1_1,1')
+ assert s.os == 'FreeBSD'
+ assert str(s) == 'OpenSSH 4.4p1 running on FreeBSD'
+ assert repr(s) == '<Software(product=OpenSSH, version=4.4, patch=p1, os=FreeBSD)>'
+ s = ps('SSH-2.0-OpenSSH_7.2-OVH-rescue FreeBSD')
+ assert s.os == 'FreeBSD'
+ assert str(s) == 'OpenSSH 7.2 (OVH-rescue) running on FreeBSD'
+ assert repr(s) == '<Software(product=OpenSSH, version=7.2, patch=OVH-rescue, os=FreeBSD)>'
+ # Windows
+ s = ps('SSH-2.0-OpenSSH_3.7.1 in RemotelyAnywhere 5.21.422')
+ assert s.os == 'Microsoft Windows (RemotelyAnywhere 5.21.422)'
+ assert str(s) == 'OpenSSH 3.7.1 running on Microsoft Windows (RemotelyAnywhere 5.21.422)'
+ assert repr(s) == '<Software(product=OpenSSH, version=3.7.1, os=Microsoft Windows (RemotelyAnywhere 5.21.422))>'
+ s = ps('SSH-2.0-OpenSSH_3.8 in DesktopAuthority 7.1.091')
+ assert s.os == 'Microsoft Windows (DesktopAuthority 7.1.091)'
+ assert str(s) == 'OpenSSH 3.8 running on Microsoft Windows (DesktopAuthority 7.1.091)'
+ assert repr(s) == '<Software(product=OpenSSH, version=3.8, os=Microsoft Windows (DesktopAuthority 7.1.091))>'
+ s = ps('SSH-2.0-OpenSSH_3.8 in RemoteSupportManager 1.0.023')
+ assert s.os == 'Microsoft Windows (RemoteSupportManager 1.0.023)'
+ assert str(s) == 'OpenSSH 3.8 running on Microsoft Windows (RemoteSupportManager 1.0.023)'
+ assert repr(s) == '<Software(product=OpenSSH, version=3.8, os=Microsoft Windows (RemoteSupportManager 1.0.023))>'
diff --git a/test/test_ssh1.py b/test/test_ssh1.py
new file mode 100644
index 0000000..d40c157
--- /dev/null
+++ b/test/test_ssh1.py
@@ -0,0 +1,173 @@
+import struct
+import pytest
+
+from ssh_audit.auditconf import AuditConf
+from ssh_audit.fingerprint import Fingerprint
+from ssh_audit.outputbuffer import OutputBuffer
+from ssh_audit.protocol import Protocol
+from ssh_audit.readbuf import ReadBuf
+from ssh_audit.ssh1 import SSH1
+from ssh_audit.ssh1_publickeymessage import SSH1_PublicKeyMessage
+from ssh_audit.ssh_audit import audit
+from ssh_audit.writebuf import WriteBuf
+
+
+# pylint: disable=line-too-long,attribute-defined-outside-init
+class TestSSH1:
+ @pytest.fixture(autouse=True)
+ def init(self, ssh_audit):
+ self.OutputBuffer = OutputBuffer
+ self.protocol = Protocol
+ self.ssh1 = SSH1
+ self.PublicKeyMessage = SSH1_PublicKeyMessage
+ self.rbuf = ReadBuf
+ self.wbuf = WriteBuf
+ self.audit = audit
+ self.AuditConf = AuditConf
+ self.fingerprint = Fingerprint
+
+ def _conf(self):
+ conf = self.AuditConf('localhost', 22)
+ conf.colors = False
+ conf.batch = True
+ conf.verbose = True
+ conf.ssh1 = True
+ conf.ssh2 = False
+ return conf
+
+ def _create_ssh1_packet(self, payload, valid_crc=True):
+ padding = -(len(payload) + 4) % 8
+ plen = len(payload) + 4
+ pad_bytes = b'\x00' * padding
+ cksum = self.ssh1.crc32(pad_bytes + payload) if valid_crc else 0
+ data = struct.pack('>I', plen) + pad_bytes + payload + struct.pack('>I', cksum)
+ return data
+
+ @classmethod
+ def _server_key(cls):
+ return (1024, 0x10001, 0xee6552da432e0ac2c422df1a51287507748bfe3b5e3e4fa989a8f49fdc163a17754939ef18ef8a667ea3b71036a151fcd7f5e01ceef1e4439864baf3ac569047582c69d6c128212e0980dcb3168f00d371004039983f6033cd785b8b8f85096c7d9405cbfdc664e27c966356a6b4eb6ee20ad43414b50de18b22829c1880b551)
+
+ @classmethod
+ def _host_key(cls):
+ return (2048, 0x10001, 0xdfa20cd2a530ccc8c870aa60d9feb3b35deeab81c3215a96557abbd683d21f4600f38e475d87100da9a4404220eeb3bb5584e5a2b5b48ffda58530ea19104a32577d7459d91e76aa711b241050f4cc6d5327ccce254f371acad3be56d46eb5919b73f20dbdb1177b700f00891c5bf4ed128bb90ed541b778288285bcfa28432ab5cbcb8321b6e24760e998e0daa519f093a631e44276d7dd252ce0c08c75e2ab28a7349ead779f97d0f20a6d413bf3623cd216dc35375f6366690bcc41e3b2d5465840ec7ee0dc7e3f1c101d674a0c7dbccbc3942788b111396add2f8153b46a0e4b50d66e57ee92958f1c860dd97cc0e40e32febff915343ed53573142bdf4b)
+
+ def _pkm_payload(self):
+ w = self.wbuf()
+ w.write(b'\x88\x99\xaa\xbb\xcc\xdd\xee\xff')
+ b, e, m = self._server_key()
+ w.write_int(b).write_mpint1(e).write_mpint1(m)
+ b, e, m = self._host_key()
+ w.write_int(b).write_mpint1(e).write_mpint1(m)
+ w.write_int(2)
+ w.write_int(72)
+ w.write_int(36)
+ return w.write_flush()
+
+ def test_crc32(self):
+ assert self.ssh1.crc32(b'') == 0x00
+ assert self.ssh1.crc32(b'The quick brown fox jumps over the lazy dog') == 0xb9c60808
+
+ def test_fingerprint(self):
+ # pylint: disable=protected-access
+ b, e, m = self._host_key()
+ fpd = self.wbuf._create_mpint(m, False)
+ fpd += self.wbuf._create_mpint(e, False)
+ fp = self.fingerprint(fpd)
+ assert b == 2048
+ assert fp.md5 == 'MD5:9d:26:f8:39:fc:20:9d:9b:ca:cc:4a:0f:e1:93:f5:96'
+ assert fp.sha256 == 'SHA256:vZdx3mhzbvVJmn08t/ruv8WDhJ9jfKYsCTuSzot+QIs'
+
+ def _assert_pkm_keys(self, pkm, skey, hkey):
+ b, e, m = skey
+ assert pkm.server_key_bits == b
+ assert pkm.server_key_public_exponent == e
+ assert pkm.server_key_public_modulus == m
+ b, e, m = hkey
+ assert pkm.host_key_bits == b
+ assert pkm.host_key_public_exponent == e
+ assert pkm.host_key_public_modulus == m
+
+ def _assert_pkm_fields(self, pkm, skey, hkey):
+ assert pkm is not None
+ assert pkm.cookie == b'\x88\x99\xaa\xbb\xcc\xdd\xee\xff'
+ self._assert_pkm_keys(pkm, skey, hkey)
+ assert pkm.protocol_flags == 2
+ assert pkm.supported_ciphers_mask == 72
+ assert pkm.supported_ciphers == ['3des', 'blowfish']
+ assert pkm.supported_authentications_mask == 36
+ assert pkm.supported_authentications == ['rsa', 'tis']
+ fp = self.fingerprint(pkm.host_key_fingerprint_data)
+ assert fp.md5 == 'MD5:9d:26:f8:39:fc:20:9d:9b:ca:cc:4a:0f:e1:93:f5:96'
+ assert fp.sha256 == 'SHA256:vZdx3mhzbvVJmn08t/ruv8WDhJ9jfKYsCTuSzot+QIs'
+
+ def test_pkm_init(self):
+ cookie = b'\x88\x99\xaa\xbb\xcc\xdd\xee\xff'
+ pflags, cmask, amask = 2, 72, 36
+ skey, hkey = self._server_key(), self._host_key()
+ pkm = self.PublicKeyMessage(cookie, skey, hkey, pflags, cmask, amask)
+ self._assert_pkm_fields(pkm, skey, hkey)
+ for skey2 in ([], [0], [0, 1], [0, 1, 2, 3]):
+ with pytest.raises(ValueError):
+ pkm = self.PublicKeyMessage(cookie, skey2, hkey, pflags, cmask, amask)
+ for hkey2 in ([], [0], [0, 1], [0, 1, 2, 3]):
+ with pytest.raises(ValueError):
+ print(hkey2)
+ pkm = self.PublicKeyMessage(cookie, skey, hkey2, pflags, cmask, amask)
+
+ def test_pkm_read(self):
+ pkm = self.PublicKeyMessage.parse(self._pkm_payload())
+ self._assert_pkm_fields(pkm, self._server_key(), self._host_key())
+
+ def test_pkm_payload(self):
+ cookie = b'\x88\x99\xaa\xbb\xcc\xdd\xee\xff'
+ skey, hkey = self._server_key(), self._host_key()
+ pflags, cmask, amask = 2, 72, 36
+ pkm1 = self.PublicKeyMessage(cookie, skey, hkey, pflags, cmask, amask)
+ pkm2 = self.PublicKeyMessage.parse(self._pkm_payload())
+ assert pkm1.payload == pkm2.payload
+
+ def test_ssh1_server_simple(self, output_spy, virtual_socket):
+ vsocket = virtual_socket
+ w = self.wbuf()
+ w.write_byte(self.protocol.SMSG_PUBLIC_KEY)
+ w.write(self._pkm_payload())
+ vsocket.rdata.append(b'SSH-1.5-OpenSSH_7.2 ssh-audit-test\r\n')
+ vsocket.rdata.append(self._create_ssh1_packet(w.write_flush()))
+ output_spy.begin()
+ out = self.OutputBuffer()
+ self.audit(out, self._conf())
+ out.write()
+ lines = output_spy.flush()
+ assert len(lines) == 17
+
+ def test_ssh1_server_invalid_first_packet(self, output_spy, virtual_socket):
+ vsocket = virtual_socket
+ w = self.wbuf()
+ w.write_byte(self.protocol.SMSG_PUBLIC_KEY + 1)
+ w.write(self._pkm_payload())
+ vsocket.rdata.append(b'SSH-1.5-OpenSSH_7.2 ssh-audit-test\r\n')
+ vsocket.rdata.append(self._create_ssh1_packet(w.write_flush()))
+ output_spy.begin()
+ out = self.OutputBuffer()
+ ret = self.audit(out, self._conf())
+ out.write()
+ assert ret != 0
+ lines = output_spy.flush()
+ assert len(lines) == 10
+ assert 'unknown message' in lines[-1]
+
+ def test_ssh1_server_invalid_checksum(self, output_spy, virtual_socket):
+ vsocket = virtual_socket
+ w = self.wbuf()
+ w.write_byte(self.protocol.SMSG_PUBLIC_KEY + 1)
+ w.write(self._pkm_payload())
+ vsocket.rdata.append(b'SSH-1.5-OpenSSH_7.2 ssh-audit-test\r\n')
+ vsocket.rdata.append(self._create_ssh1_packet(w.write_flush(), False))
+ output_spy.begin()
+ out = self.OutputBuffer()
+ with pytest.raises(SystemExit):
+ self.audit(out, self._conf())
+ out.write()
+ lines = output_spy.flush()
+ assert len(lines) == 3
+ assert ('checksum' in lines[0]) or ('checksum' in lines[1]) or ('checksum' in lines[2])
diff --git a/test/test_ssh2.py b/test/test_ssh2.py
new file mode 100644
index 0000000..fc4c0fa
--- /dev/null
+++ b/test/test_ssh2.py
@@ -0,0 +1,165 @@
+import os
+import struct
+import pytest
+
+from ssh_audit.auditconf import AuditConf
+from ssh_audit.outputbuffer import OutputBuffer
+from ssh_audit.protocol import Protocol
+from ssh_audit.readbuf import ReadBuf
+from ssh_audit.ssh2_kex import SSH2_Kex
+from ssh_audit.ssh2_kexparty import SSH2_KexParty
+from ssh_audit.ssh_audit import audit
+from ssh_audit.writebuf import WriteBuf
+
+
+# pylint: disable=line-too-long,attribute-defined-outside-init
+class TestSSH2:
+ @pytest.fixture(autouse=True)
+ def init(self, ssh_audit):
+ self.OutputBuffer = OutputBuffer
+ self.protocol = Protocol
+ self.ssh2_kex = SSH2_Kex
+ self.ssh2_kexparty = SSH2_KexParty
+ self.rbuf = ReadBuf
+ self.wbuf = WriteBuf
+ self.audit = audit
+ self.AuditConf = AuditConf
+
+ def _conf(self):
+ conf = self.AuditConf('localhost', 22)
+ conf.colors = False
+ conf.batch = True
+ conf.verbose = True
+ conf.ssh1 = False
+ conf.ssh2 = True
+ return conf
+
+ @classmethod
+ def _create_ssh2_packet(cls, payload):
+ padding = -(len(payload) + 5) % 8
+ if padding < 4:
+ padding += 8
+ plen = len(payload) + padding + 1
+ pad_bytes = b'\x00' * padding
+ data = struct.pack('>Ib', plen, padding) + payload + pad_bytes
+ return data
+
+ def _kex_payload(self):
+ w = self.wbuf()
+ w.write(b'\x00\x11\x22\x33\x44\x55\x66\x77\x88\x99\xaa\xbb\xcc\xdd\xee\xff')
+ w.write_list(['bogus_kex1', 'bogus_kex2']) # We use a bogus kex, otherwise the host key tests will kick off and fail.
+ w.write_list(['ssh-rsa', 'rsa-sha2-512', 'rsa-sha2-256', 'ssh-ed25519'])
+ w.write_list(['chacha20-poly1305@openssh.com', 'aes128-ctr', 'aes192-ctr', 'aes256-ctr', 'aes128-gcm@openssh.com', 'aes256-gcm@openssh.com', 'aes128-cbc', 'aes192-cbc', 'aes256-cbc'])
+ w.write_list(['chacha20-poly1305@openssh.com', 'aes128-ctr', 'aes192-ctr', 'aes256-ctr', 'aes128-gcm@openssh.com', 'aes256-gcm@openssh.com', 'aes128-cbc', 'aes192-cbc', 'aes256-cbc'])
+ w.write_list(['umac-64-etm@openssh.com', 'umac-128-etm@openssh.com', 'hmac-sha2-256-etm@openssh.com', 'hmac-sha2-512-etm@openssh.com', 'hmac-sha1-etm@openssh.com', 'umac-64@openssh.com', 'umac-128@openssh.com', 'hmac-sha2-256', 'hmac-sha2-512', 'hmac-sha1'])
+ w.write_list(['umac-64-etm@openssh.com', 'umac-128-etm@openssh.com', 'hmac-sha2-256-etm@openssh.com', 'hmac-sha2-512-etm@openssh.com', 'hmac-sha1-etm@openssh.com', 'umac-64@openssh.com', 'umac-128@openssh.com', 'hmac-sha2-256', 'hmac-sha2-512', 'hmac-sha1'])
+ w.write_list(['none', 'zlib@openssh.com'])
+ w.write_list(['none', 'zlib@openssh.com'])
+ w.write_list([''])
+ w.write_list([''])
+ w.write_byte(False)
+ w.write_int(0)
+ return w.write_flush()
+
+ def test_kex_read(self):
+ kex = self.ssh2_kex.parse(self._kex_payload())
+ assert kex is not None
+ assert kex.cookie == b'\x00\x11\x22\x33\x44\x55\x66\x77\x88\x99\xaa\xbb\xcc\xdd\xee\xff'
+ assert kex.kex_algorithms == ['bogus_kex1', 'bogus_kex2']
+ assert kex.key_algorithms == ['ssh-rsa', 'rsa-sha2-512', 'rsa-sha2-256', 'ssh-ed25519']
+ assert kex.client is not None
+ assert kex.server is not None
+ assert kex.client.encryption == ['chacha20-poly1305@openssh.com', 'aes128-ctr', 'aes192-ctr', 'aes256-ctr', 'aes128-gcm@openssh.com', 'aes256-gcm@openssh.com', 'aes128-cbc', 'aes192-cbc', 'aes256-cbc']
+ assert kex.server.encryption == ['chacha20-poly1305@openssh.com', 'aes128-ctr', 'aes192-ctr', 'aes256-ctr', 'aes128-gcm@openssh.com', 'aes256-gcm@openssh.com', 'aes128-cbc', 'aes192-cbc', 'aes256-cbc']
+ assert kex.client.mac == ['umac-64-etm@openssh.com', 'umac-128-etm@openssh.com', 'hmac-sha2-256-etm@openssh.com', 'hmac-sha2-512-etm@openssh.com', 'hmac-sha1-etm@openssh.com', 'umac-64@openssh.com', 'umac-128@openssh.com', 'hmac-sha2-256', 'hmac-sha2-512', 'hmac-sha1']
+ assert kex.server.mac == ['umac-64-etm@openssh.com', 'umac-128-etm@openssh.com', 'hmac-sha2-256-etm@openssh.com', 'hmac-sha2-512-etm@openssh.com', 'hmac-sha1-etm@openssh.com', 'umac-64@openssh.com', 'umac-128@openssh.com', 'hmac-sha2-256', 'hmac-sha2-512', 'hmac-sha1']
+ assert kex.client.compression == ['none', 'zlib@openssh.com']
+ assert kex.server.compression == ['none', 'zlib@openssh.com']
+ assert kex.client.languages == ['']
+ assert kex.server.languages == ['']
+ assert kex.follows is False
+ assert kex.unused == 0
+
+ def _get_empty_kex(self, cookie=None):
+ kex_algs, key_algs = [], []
+ enc, mac, compression, languages = [], [], ['none'], []
+ cli = self.ssh2_kexparty(enc, mac, compression, languages)
+ enc, mac, compression, languages = [], [], ['none'], []
+ srv = self.ssh2_kexparty(enc, mac, compression, languages)
+ if cookie is None:
+ cookie = os.urandom(16)
+ kex = self.ssh2_kex(cookie, kex_algs, key_algs, cli, srv, 0)
+ return kex
+
+ def _get_kex_variat1(self):
+ cookie = b'\x00\x11\x22\x33\x44\x55\x66\x77\x88\x99\xaa\xbb\xcc\xdd\xee\xff'
+ kex = self._get_empty_kex(cookie)
+ kex.kex_algorithms.append('bogus_kex1')
+ kex.kex_algorithms.append('bogus_kex2')
+ kex.key_algorithms.append('ssh-rsa')
+ kex.key_algorithms.append('rsa-sha2-512')
+ kex.key_algorithms.append('rsa-sha2-256')
+ kex.key_algorithms.append('ssh-ed25519')
+ kex.server.encryption.append('chacha20-poly1305@openssh.com')
+ kex.server.encryption.append('aes128-ctr')
+ kex.server.encryption.append('aes192-ctr')
+ kex.server.encryption.append('aes256-ctr')
+ kex.server.encryption.append('aes128-gcm@openssh.com')
+ kex.server.encryption.append('aes256-gcm@openssh.com')
+ kex.server.encryption.append('aes128-cbc')
+ kex.server.encryption.append('aes192-cbc')
+ kex.server.encryption.append('aes256-cbc')
+ kex.server.mac.append('umac-64-etm@openssh.com')
+ kex.server.mac.append('umac-128-etm@openssh.com')
+ kex.server.mac.append('hmac-sha2-256-etm@openssh.com')
+ kex.server.mac.append('hmac-sha2-512-etm@openssh.com')
+ kex.server.mac.append('hmac-sha1-etm@openssh.com')
+ kex.server.mac.append('umac-64@openssh.com')
+ kex.server.mac.append('umac-128@openssh.com')
+ kex.server.mac.append('hmac-sha2-256')
+ kex.server.mac.append('hmac-sha2-512')
+ kex.server.mac.append('hmac-sha1')
+ kex.server.compression.append('zlib@openssh.com')
+ for a in kex.server.encryption:
+ kex.client.encryption.append(a)
+ for a in kex.server.mac:
+ kex.client.mac.append(a)
+ for a in kex.server.compression:
+ if a == 'none':
+ continue
+ kex.client.compression.append(a)
+ return kex
+
+ def test_key_payload(self):
+ kex1 = self._get_kex_variat1()
+ kex2 = self.ssh2_kex.parse(self._kex_payload())
+ assert kex1.payload == kex2.payload
+
+ def test_ssh2_server_simple(self, output_spy, virtual_socket):
+ vsocket = virtual_socket
+ w = self.wbuf()
+ w.write_byte(self.protocol.MSG_KEXINIT)
+ w.write(self._kex_payload())
+ vsocket.rdata.append(b'SSH-2.0-OpenSSH_7.3 ssh-audit-test\r\n')
+ vsocket.rdata.append(self._create_ssh2_packet(w.write_flush()))
+ output_spy.begin()
+ out = self.OutputBuffer()
+ self.audit(out, self._conf())
+ out.write()
+ lines = output_spy.flush()
+ assert len(lines) == 70
+
+ def test_ssh2_server_invalid_first_packet(self, output_spy, virtual_socket):
+ vsocket = virtual_socket
+ w = self.wbuf()
+ w.write_byte(self.protocol.MSG_KEXINIT + 1)
+ vsocket.rdata.append(b'SSH-2.0-OpenSSH_7.3 ssh-audit-test\r\n')
+ vsocket.rdata.append(self._create_ssh2_packet(w.write_flush()))
+ output_spy.begin()
+ out = self.OutputBuffer()
+ ret = self.audit(out, self._conf())
+ out.write()
+ assert ret != 0
+ lines = output_spy.flush()
+ assert len(lines) == 5
+ assert 'unknown message' in lines[-1]
diff --git a/test/test_utils.py b/test/test_utils.py
new file mode 100644
index 0000000..d5251f7
--- /dev/null
+++ b/test/test_utils.py
@@ -0,0 +1,76 @@
+import pytest
+
+
+# pylint: disable=attribute-defined-outside-init
+class TestUtils:
+ @pytest.fixture(autouse=True)
+ def init(self, ssh_audit):
+ self.utils = ssh_audit.Utils
+
+ def test_to_bytes(self):
+ assert self.utils.to_bytes(b'fran\xc3\xa7ais') == b'fran\xc3\xa7ais'
+ assert self.utils.to_bytes('fran\xe7ais') == b'fran\xc3\xa7ais'
+ # other
+ with pytest.raises(TypeError):
+ self.utils.to_bytes(123)
+
+ def test_to_text(self):
+ assert self.utils.to_text(b'fran\xc3\xa7ais') == 'fran\xe7ais'
+ assert self.utils.to_text('fran\xe7ais') == 'fran\xe7ais'
+ # other
+ with pytest.raises(TypeError):
+ self.utils.to_text(123)
+
+ def test_is_ascii(self):
+ assert self.utils.is_ascii('francais') is True
+ assert self.utils.is_ascii('fran\xe7ais') is False
+ # other
+ assert self.utils.is_ascii(123) is False
+
+ def test_to_ascii(self):
+ assert self.utils.to_ascii('francais') == 'francais'
+ assert self.utils.to_ascii('fran\xe7ais') == 'fran?ais'
+ assert self.utils.to_ascii('fran\xe7ais', 'ignore') == 'franais'
+ with pytest.raises(TypeError):
+ self.utils.to_ascii(123)
+
+ def test_is_print_ascii(self):
+ assert self.utils.is_print_ascii('francais') is True
+ assert self.utils.is_print_ascii('francais\n') is False
+ assert self.utils.is_print_ascii('fran\xe7ais') is False
+ # other
+ assert self.utils.is_print_ascii(123) is False
+
+ def test_to_print_ascii(self):
+ assert self.utils.to_print_ascii('francais') == 'francais'
+ assert self.utils.to_print_ascii('francais\n') == 'francais?'
+ assert self.utils.to_print_ascii('fran\xe7ais') == 'fran?ais'
+ assert self.utils.to_print_ascii('fran\xe7ais\n') == 'fran?ais?'
+ assert self.utils.to_print_ascii('fran\xe7ais', 'ignore') == 'franais'
+ assert self.utils.to_print_ascii('fran\xe7ais\n', 'ignore') == 'franais'
+ with pytest.raises(TypeError):
+ self.utils.to_print_ascii(123)
+
+ def test_ctoi(self):
+ assert self.utils.ctoi(123) == 123
+ assert self.utils.ctoi('ABC') == 65
+
+ def test_parse_int(self):
+ assert self.utils.parse_int(123) == 123
+ assert self.utils.parse_int('123') == 123
+ assert self.utils.parse_int(-123) == -123
+ assert self.utils.parse_int('-123') == -123
+ assert self.utils.parse_int('abc') == 0
+
+ def test_unique_seq(self):
+ assert self.utils.unique_seq((1, 2, 2, 3, 3, 3)) == (1, 2, 3)
+ assert self.utils.unique_seq((3, 3, 3, 2, 2, 1)) == (3, 2, 1)
+ assert self.utils.unique_seq([1, 2, 2, 3, 3, 3]) == [1, 2, 3]
+ assert self.utils.unique_seq([3, 3, 3, 2, 2, 1]) == [3, 2, 1]
+
+ def test_parse_float(self):
+ assert self.utils.parse_float('5.x') == -1.0
+
+ def test_ipv6address(self):
+ assert self.utils.is_ipv6_address('1.2.3.4') is False
+ assert self.utils.is_ipv6_address('2600:1f18:420a:b500:bc4:c9c6:1d6:e3e4') is True
diff --git a/test/test_version_compare.py b/test/test_version_compare.py
new file mode 100644
index 0000000..f6f0441
--- /dev/null
+++ b/test/test_version_compare.py
@@ -0,0 +1,219 @@
+import pytest
+
+from ssh_audit.banner import Banner
+from ssh_audit.software import Software
+
+
+# pylint: disable=attribute-defined-outside-init
+class TestVersionCompare:
+ @pytest.fixture(autouse=True)
+ def init(self, ssh_audit):
+ self.software = Software
+ self.banner = Banner
+
+ def get_dropbear_software(self, v):
+ b = self.banner.parse('SSH-2.0-dropbear_{}'.format(v))
+ return self.software.parse(b)
+
+ def get_openssh_software(self, v):
+ b = self.banner.parse('SSH-2.0-OpenSSH_{}'.format(v))
+ return self.software.parse(b)
+
+ def get_libssh_software(self, v):
+ b = self.banner.parse('SSH-2.0-libssh-{}'.format(v))
+ return self.software.parse(b)
+
+ def test_dropbear_compare_version_pre_years(self):
+ s = self.get_dropbear_software('0.44')
+ assert s.compare_version(None) == 1
+ assert s.compare_version('') == 1
+ assert s.compare_version('0.43') > 0
+ assert s.compare_version('0.44') == 0
+ assert s.compare_version(s) == 0
+ assert s.compare_version('0.45') < 0
+ assert s.between_versions('0.43', '0.45')
+ assert s.between_versions('0.43', '0.43') is False
+ assert s.between_versions('0.45', '0.43') is False
+
+ def test_dropbear_compare_version_with_years(self):
+ s = self.get_dropbear_software('2015.71')
+ assert s.compare_version(None) == 1
+ assert s.compare_version('') == 1
+ assert s.compare_version('2014.66') > 0
+ assert s.compare_version('2015.71') == 0
+ assert s.compare_version(s) == 0
+ assert s.compare_version('2016.74') < 0
+ assert s.between_versions('2014.66', '2016.74')
+ assert s.between_versions('2014.66', '2015.69') is False
+ assert s.between_versions('2016.74', '2014.66') is False
+
+ def test_dropbear_compare_version_mixed(self):
+ s = self.get_dropbear_software('0.53.1')
+ assert s.compare_version(None) == 1
+ assert s.compare_version('') == 1
+ assert s.compare_version('0.53') > 0
+ assert s.compare_version('0.53.1') == 0
+ assert s.compare_version(s) == 0
+ assert s.compare_version('2011.54') < 0
+ assert s.between_versions('0.53', '2011.54')
+ assert s.between_versions('0.53', '0.53') is False
+ assert s.between_versions('2011.54', '0.53') is False
+
+ def test_dropbear_compare_version_patchlevel(self):
+ s1 = self.get_dropbear_software('0.44')
+ s2 = self.get_dropbear_software('0.44test3')
+ assert s1.compare_version(None) == 1
+ assert s1.compare_version('') == 1
+ assert s1.compare_version('0.44') == 0
+ assert s1.compare_version(s1) == 0
+ assert s1.compare_version('0.43') > 0
+ assert s1.compare_version('0.44test4') > 0
+ assert s1.between_versions('0.44test4', '0.45')
+ assert s1.between_versions('0.43', '0.44test4') is False
+ assert s1.between_versions('0.45', '0.44test4') is False
+ assert s2.compare_version(None) == 1
+ assert s2.compare_version('') == 1
+ assert s2.compare_version('0.44test3') == 0
+ assert s2.compare_version(s2) == 0
+ assert s2.compare_version('0.44') < 0
+ assert s2.compare_version('0.44test4') < 0
+ assert s2.between_versions('0.43', '0.44')
+ assert s2.between_versions('0.43', '0.44test2') is False
+ assert s2.between_versions('0.44', '0.43') is False
+ assert s1.compare_version(s2) > 0
+ assert s2.compare_version(s1) < 0
+
+ def test_dropbear_compare_version_sequential(self):
+ versions = []
+ for i in range(28, 44):
+ versions.append('0.{}'.format(i))
+ for i in range(1, 5):
+ versions.append('0.44test{}'.format(i))
+ for i in range(44, 49):
+ versions.append('0.{}'.format(i))
+ versions.append('0.48.1')
+ for i in range(49, 54):
+ versions.append('0.{}'.format(i))
+ versions.append('0.53.1')
+ for v in ['2011.54', '2012.55']:
+ versions.append(v)
+ for i in range(56, 61):
+ versions.append('2013.{}'.format(i))
+ for v in ['2013.61test', '2013.62']:
+ versions.append(v)
+ for i in range(63, 67):
+ versions.append('2014.{}'.format(i))
+ for i in range(67, 72):
+ versions.append('2015.{}'.format(i))
+ for i in range(72, 75):
+ versions.append('2016.{}'.format(i))
+ length = len(versions)
+ for i in range(length):
+ v = versions[i]
+ s = self.get_dropbear_software(v)
+ assert s.compare_version(v) == 0
+ if i - 1 >= 0:
+ vbefore = versions[i - 1]
+ assert s.compare_version(vbefore) > 0
+ if i + 1 < length:
+ vnext = versions[i + 1]
+ assert s.compare_version(vnext) < 0
+
+ def test_openssh_compare_version_simple(self):
+ s = self.get_openssh_software('3.7.1')
+ assert s.compare_version(None) == 1
+ assert s.compare_version('') == 1
+ assert s.compare_version('3.7') > 0
+ assert s.compare_version('3.7.1') == 0
+ assert s.compare_version(s) == 0
+ assert s.compare_version('3.8') < 0
+ assert s.between_versions('3.7', '3.8')
+ assert s.between_versions('3.6', '3.7') is False
+ assert s.between_versions('3.8', '3.7') is False
+
+ def test_openssh_compare_version_patchlevel(self):
+ s1 = self.get_openssh_software('2.1.1')
+ s2 = self.get_openssh_software('2.1.1p2')
+ assert s1.compare_version(s1) == 0
+ assert s2.compare_version(s2) == 0
+ assert s1.compare_version('2.1.1') == 0
+ assert s2.compare_version('2.1.1p2') == 0
+ assert s1.compare_version('2.1.1p1') == 0 # OpenBSD version and p1 version should be the same.
+ assert s1.compare_version('2.1.1p2') != 0
+ assert s2.compare_version('2.1.1') != 0
+ assert s2.compare_version('2.1.1p1') > 0
+ assert s2.compare_version('2.1.1p3') < 0
+ assert s1.compare_version(s2) != 0
+ assert s2.compare_version(s1) != 0
+
+ def test_openbsd_compare_version_sequential(self):
+ versions = []
+ for v in ['1.2.3', '2.1.0', '2.1.1', '2.2.0', '2.3.0']:
+ versions.append(v)
+ for v in ['2.5.0', '2.5.1', '2.5.2', '2.9', '2.9.9']:
+ versions.append(v)
+ for v in ['3.0', '3.0.1', '3.0.2', '3.1', '3.2.2', '3.2.3']:
+ versions.append(v)
+ for i in range(3, 7):
+ versions.append('3.{}'.format(i))
+ for v in ['3.6.1', '3.7.0', '3.7.1']:
+ versions.append(v)
+ for i in range(8, 10):
+ versions.append('3.{}'.format(i))
+ for i in range(0, 10):
+ versions.append('4.{}'.format(i))
+ for i in range(0, 10):
+ versions.append('5.{}'.format(i))
+ for i in range(0, 10):
+ versions.append('6.{}'.format(i))
+ for i in range(0, 4):
+ versions.append('7.{}'.format(i))
+ length = len(versions)
+ for i in range(length):
+ v = versions[i]
+ s = self.get_openssh_software(v)
+ assert s.compare_version(v) == 0
+ if i - 1 >= 0:
+ vbefore = versions[i - 1]
+ assert s.compare_version(vbefore) > 0
+ if i + 1 < length:
+ vnext = versions[i + 1]
+ assert s.compare_version(vnext) < 0
+
+ def test_libssh_compare_version_simple(self):
+ s = self.get_libssh_software('0.3')
+ assert s.compare_version(None) == 1
+ assert s.compare_version('') == 1
+ assert s.compare_version('0.2') > 0
+ assert s.compare_version('0.3') == 0
+ assert s.compare_version(s) == 0
+ assert s.compare_version('0.3.1') < 0
+ assert s.between_versions('0.2', '0.3.1')
+ assert s.between_versions('0.1', '0.2') is False
+ assert s.between_versions('0.3.1', '0.2') is False
+
+ def test_libssh_compare_version_sequential(self):
+ versions = []
+ for v in ['0.2', '0.3']:
+ versions.append(v)
+ for i in range(1, 5):
+ versions.append('0.3.{}'.format(i))
+ for i in range(0, 9):
+ versions.append('0.4.{}'.format(i))
+ for i in range(0, 6):
+ versions.append('0.5.{}'.format(i))
+ for i in range(0, 6):
+ versions.append('0.6.{}'.format(i))
+ for i in range(0, 5):
+ versions.append('0.7.{}'.format(i))
+ length = len(versions)
+ for i in range(length):
+ v = versions[i]
+ s = self.get_libssh_software(v)
+ assert s.compare_version(v) == 0
+ if i - 1 >= 0:
+ vbefore = versions[i - 1]
+ assert s.compare_version(vbefore) > 0
+ if i + 1 < length:
+ vnext = versions[i + 1]
+ assert s.compare_version(vnext) < 0
diff --git a/test/tools/ci-win.cmd b/test/tools/ci-win.cmd
new file mode 100644
index 0000000..103036c
--- /dev/null
+++ b/test/tools/ci-win.cmd
@@ -0,0 +1,131 @@
+@ECHO OFF
+
+IF "%PYTHON%" == "" (
+ ECHO PYTHON environment variable not set
+ EXIT 1
+)
+SET PATH=%PYTHON%;%PYTHON%\\Scripts;%PATH%"
+FOR /F %%i IN ('python -c "import platform; print(platform.python_version());"') DO (
+ SET PYTHON_VERSION=%%i
+)
+SET PYTHON_VERSION_MAJOR=%PYTHON_VERSION:~0,1%
+IF "%PYTHON_VERSION:~3,1%" == "." (
+ SET PYTHON_VERSION_MINOR=%PYTHON_VERSION:~2,1%
+) ELSE (
+ SET PYTHON_VERSION_MINOR=%PYTHON_VERSION:~2,2%
+)
+FOR /F %%i IN ('python -c "import struct; print(struct.calcsize(\"P\")*8)"') DO (
+ SET PYTHON_ARCH=%%i
+)
+CALL :devenv
+
+IF /I "%1"=="" (
+ SET target=test
+) ELSE (
+ SET target=%1
+)
+
+echo [CI] TARGET=%target%
+GOTO %target%
+
+:devenv
+SET WIN_SDK_ROOT=C:\Program Files\Microsoft SDKs\Windows
+SET VS2015_ROOT=C:\Program Files (x86)\Microsoft Visual Studio 14.0
+IF %PYTHON_VERSION_MAJOR% == 2 (
+ SET WINDOWS_SDK_VERSION="v7.0"
+) ELSE IF %PYTHON_VERSION_MAJOR% == 3 (
+ IF %PYTHON_VERSION_MAJOR% LEQ 4 (
+ SET WINDOWS_SDK_VERSION="v7.1"
+ ) ELSE (
+ SET WINDOWS_SDK_VERSION="2015"
+ )
+) ELSE (
+ ECHO Unsupported Python version: "%PYTHON_VERSION%"
+ EXIT 1
+)
+SETLOCAL ENABLEDELAYEDEXPANSION
+IF %PYTHON_ARCH% == 32 (SET PYTHON_ARCHX=x86) ELSE (SET PYTHON_ARCHX=x64)
+IF %WINDOWS_SDK_VERSION% == "2015" (
+ "%VS2015_ROOT%\VC\vcvarsall.bat" %PYTHON_ARCHX%
+) ELSE (
+ SET DISTUTILS_USE_SDK=1
+ SET MSSdk=1
+ "%WIN_SDK_ROOT%\%WINDOWS_SDK_VERSION%\Setup\WindowsSdkVer.exe" -q -version:%WINDOWS_SDK_VERSION%
+ "%WIN_SDK_ROOT%\%WINDOWS_SDK_VERSION%\Bin\SetEnv.cmd" /%PYTHON_ARCHX% /release
+)
+GOTO :eof
+
+:install
+pip install --user --upgrade pip virtualenv
+SET VENV_DIR=.venv\%PYTHON_VERSION%
+rmdir /s /q %VENV_DIR% > nul 2>nul
+mkdir .venv > nul 2>nul
+IF "%PYTHON_VERSION_MAJOR%%PYTHON_VERSION_MINOR%" == "26" (
+ python -c "import virtualenv; virtualenv.main();" %VENV_DIR%
+) ELSE (
+ python -m virtualenv %VENV_DIR%
+)
+CALL %VENV_DIR%\Scripts\activate
+python -V
+pip install tox
+deactivate
+GOTO :eof
+
+:install_deps
+SET LXML_FILE=
+SET LXML_URL=
+IF %PYTHON_VERSION_MAJOR% == 3 (
+ IF %PYTHON_VERSION_MINOR% == 3 (
+ IF %PYTHON_ARCH% == 32 (
+ SET LXML_FILE=lxml-3.7.3.win32-py3.3.exe
+ SET LXML_URL=https://pypi.python.org/packages/66/fd/b82a54e7a15e91184efeef4b659379d0581a73cf78239d70feb0f0877841/lxml-3.7.3.win32-py3.3.exe
+ ) ELSE (
+ SET LXML_FILE=lxml-3.7.3.win-amd64-py3.3.exe
+ SET LXML_URL=https://pypi.python.org/packages/dc/bc/4742b84793fa1fd991b5d2c6f2e5d32695659d6cfedf5c66aef9274a8723/lxml-3.7.3.win-amd64-py3.3.exe
+ )
+ ) ELSE IF %PYTHON_VERSION_MINOR% == 4 (
+ IF %PYTHON_ARCH% == 32 (
+ SET LXML_FILE=lxml-3.7.3.win32-py3.4.exe
+ SET LXML_URL=https://pypi.python.org/packages/88/33/265459d68d465ddc707621e6471989f5c2cb0d43f230f516800ffd629af7/lxml-3.7.3.win32-py3.4.exe
+ ) ELSE (
+ SET LXML_FILE=lxml-3.7.3.win-amd64-py3.4.exe
+ SET LXML_URL=https://pypi.python.org/packages/2d/65/e47db7f36a69a1b59b4f661e42d699d6c43e663b8fd91035e6f7681d017e/lxml-3.7.3.win-amd64-py3.4.exe
+ )
+ )
+)
+IF NOT "%LXML_FILE%" == "" (
+ CALL :download %LXML_URL% .downloads\%LXML_FILE%
+ easy_install --user .downloads\%LXML_FILE%
+)
+GOTO :eof
+
+:test
+ SET VENV_DIR=.venv\%PYTHON_VERSION%
+ CALL %VENV_DIR%\Scripts\activate
+ IF "%TOXENV%" == "" (
+ SET TOXENV=py%PYTHON_VERSION_MAJOR%%PYTHON_VERSION_MINOR%
+ )
+ IF "%PYTHON_VERSION_MAJOR%%PYTHON_VERSION_MINOR%" == "26" (
+ SET TOX=python -c "from tox import cmdline; cmdline()"
+ ) ELSE (
+ SET TOX=python -m tox
+ )
+ IF %PYTHON_VERSION_MAJOR% == 3 (
+ IF %PYTHON_VERSION_MINOR% LEQ 4 (
+ :: Python 3.3 and 3.4 does not support typed-ast (mypy dependency)
+ %TOX% --sitepackages -e %TOXENV%-test,%TOXENV%-lint,cov || EXIT 1
+ ) ELSE (
+ %TOX% --sitepackages -e %TOXENV%-test,%TOXENV%-type,%TOXENV%-lint,cov || EXIT 1
+ )
+ ) ELSE (
+ %TOX% --sitepackages -e %TOXENV%-test,%TOXENV%-lint,cov || EXIT 1
+ )
+GOTO :eof
+
+:download
+IF NOT EXIST %2 (
+ IF NOT EXIST .downloads\ mkdir .downloads
+ powershell -command "(new-object net.webclient).DownloadFile('%1', '%2')" || EXIT 1
+
+)
+GOTO :eof
diff --git a/tox.ini b/tox.ini
new file mode 100644
index 0000000..4b6d8d0
--- /dev/null
+++ b/tox.ini
@@ -0,0 +1,141 @@
+[tox]
+envlist =
+ py{py3}-{test,pylint,flake8,vulture}
+ py{36,37,38,39}-{test,mypy,pylint,flake8,vulture}
+ cov
+skip_missing_interpreters = true
+
+[testenv]
+deps =
+ test: pytest<6.0
+ test,cov: {[testenv:cov]deps}
+ test,py{36,37,38,39}-{type,mypy}: colorama
+ py{36,37,38,39}-{type,mypy}: {[testenv:mypy]deps}
+ py{py3,36,37,38,39}-{lint,pylint},lint: {[testenv:pylint]deps}
+ py{py3,36,37,38,39}-{lint,flake8},lint: {[testenv:flake8]deps}
+ py{py3,36,37,38,39}-{lint,vulture},lint: {[testenv:vulture]deps}
+setenv =
+ SSHAUDIT = {toxinidir}/src
+ test: COVERAGE_FILE = {toxinidir}/.coverage.{envname}
+ type,mypy: MYPYPATH = {toxinidir}/test/stubs
+ type,mypy: MYPYHTML = {toxinidir}/reports/html/mypy
+commands =
+ test: coverage run --source ssh_audit -m -p -- \
+ test: pytest -v --junitxml={toxinidir}/reports/junit.{envname}.xml {posargs:test}
+ test: coverage combine
+ test: coverage report --show-missing
+ test: coverage html -d {toxinidir}/reports/html/coverage.{envname}
+ py{36,37,38,39}-{type,mypy}: {[testenv:mypy]commands}
+ py{py3,36,37,38,39}-{lint,pylint},lint: {[testenv:pylint]commands}
+ py{py3,36,37,38,39}-{lint,flake8},lint: {[testenv:flake8]commands}
+ py{py3,36,37,38,39}-{lint,vulture},lint: {[testenv:vulture]commands}
+#ignore_outcome =
+# type: true
+# lint: true
+
+[testenv:cov]
+deps =
+ coverage
+setenv =
+ COVERAGE_FILE = {toxinidir}/.coverage
+commands =
+ coverage erase
+ coverage combine
+ coverage report --show-missing
+ coverage xml -i -o {toxinidir}/reports/coverage.xml
+ coverage html -d {toxinidir}/reports/html/coverage
+
+[testenv:mypy]
+deps =
+ colorama
+ lxml
+ mypy
+commands =
+ mypy \
+ --strict \
+ --show-error-context \
+ --html-report {env:MYPYHTML}.py3.{envname} \
+ {posargs:{env:SSHAUDIT}}
+
+[testenv:pylint]
+deps =
+ mccabe
+ pylint
+commands =
+ pylint \
+ --rcfile tox.ini \
+ --load-plugins=pylint.extensions.bad_builtin \
+ --load-plugins=pylint.extensions.check_elif \
+ --load-plugins=pylint.extensions.mccabe \
+ {posargs:{env:SSHAUDIT}}
+
+[testenv:flake8]
+deps =
+ flake8
+commands =
+ flake8 {posargs:{env:SSHAUDIT} {toxinidir}/setup.py {toxinidir}/test {toxinidir}/ssh-audit.py} --statistics
+
+[testenv:vulture]
+deps =
+ vulture
+commands =
+ python -c "import sys; from subprocess import Popen, PIPE; \
+ a = ['vulture', '--min-confidence', '100'] + r'{posargs:{env:SSHAUDIT}}'.split(' '); \
+ o = Popen(a, shell=False, stdout=PIPE).communicate()[0]; \
+ l = [x for x in o.split(b'\n') if x and b'Unused import' not in x]; \
+ print(b'\n'.join(l).decode('utf-8')); \
+ sys.exit(1 if len(l) > 0 else 0)"
+
+[pylint]
+reports = no
+#output-format = colorized
+indent-string = " "
+disable =
+ bad-continuation,
+ broad-except,
+ duplicate-code,
+ fixme,
+ invalid-name,
+ line-too-long,
+ missing-docstring,
+ mixed-indentation,
+ no-else-raise,
+ no-else-return,
+ super-with-arguments, # Can be re-factored, at some point.
+ too-complex,
+ too-many-boolean-expressions,
+ too-many-branches,
+ too-many-instance-attributes,
+ too-many-lines,
+ too-many-locals,
+ too-many-statements,
+ unsubscriptable-object # This seems to be a local issue on the maintainer's machine only. Could be removed, perhaps, in Jan 2021.
+max-complexity = 15
+max-args = 8
+max-locals = 20
+max-returns = 6
+max-branches = 15
+max-statements = 60
+max-parents = 7
+max-attributes = 8
+min-public-methods = 1
+max-public-methods = 20
+max-bool-expr = 5
+max-nested-blocks = 6
+max-line-length = 80
+ignore-long-lines = ^\s*(#\s+type:\s+.*|[A-Z0-9_]+\s+=\s+.*|('.*':\s+)?\[.*\],?|assert\s+.*)$
+max-module-lines = 2500
+
+[flake8]
+ignore =
+ E241, # multiple spaces after operator; should be kept for tabular data
+ E303, # too many blank lines
+ E501, # line too long
+
+[pytest]
+junit_family = xunit1
+
+[coverage:paths]
+source =
+ src
+ */site-packages
diff --git a/update_windows_man_page.sh b/update_windows_man_page.sh
new file mode 100755
index 0000000..6875030
--- /dev/null
+++ b/update_windows_man_page.sh
@@ -0,0 +1,130 @@
+#!/bin/bash
+
+#
+# The MIT License (MIT)
+#
+# Copyright (C) 2021 Joe Testa (jtesta@positronsecurity.com)
+# Copyright (C) 2021 Adam Russell (<adam[at]thecliguy[dot]co[dot]uk>)
+#
+# Permission is hereby granted, free of charge, to any person obtaining a copy
+# of this software and associated documentation files (the "Software"), to deal
+# in the Software without restriction, including without limitation the rights
+# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+# copies of the Software, and to permit persons to whom the Software is
+# furnished to do so, subject to the following conditions:
+#
+# The above copyright notice and this permission notice shall be included in
+# all copies or substantial portions of the Software.
+#
+# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+# THE SOFTWARE.
+#
+
+################################################################################
+# update_windows_man_page.sh
+#
+# PURPOSE
+# Since Windows lacks a manual reader it's necessary to provide an alternative
+# means of reading the man page.
+#
+# This script should be run as part of the ssh-audit packaging process for
+# Windows. It populates the 'WINDOWS_MAN_PAGE' variable in 'globals.py' with
+# the contents of the man page. Windows users can then print the content of
+# 'WINDOWS_MAN_PAGE' by invoking ssh-audit with the manual parameters
+# (--manual / -m).
+#
+# Cygwin is required.
+#
+# USAGE
+# update_windows_man_page.sh [-m <path-to-man-page>] [-g <path-to-globals.py>]
+#
+################################################################################
+
+function usage {
+ echo >&2 "Usage: $0 [-m <path-to-man-page>] [-g <path-to-globals.py>] [-h]"
+ echo >&2 " -m Specify an alternate man page path (default: ./ssh-audit.1)"
+ echo >&2 " -g Specify an alternate globals.py path (default: ./src/ssh_audit/globals.py)"
+ echo >&2 " -h This help message"
+}
+
+PLATFORM="$(uname -s)"
+
+# This script is intended for use on Linux and Cygwin only.
+case "$PLATFORM" in
+ Linux | CYGWIN*) ;;
+ *) echo "Platform not supported: $PLATFORM"
+ exit 1
+ ;;
+esac
+
+MAN_PAGE=./ssh-audit.1
+GLOBALS_PY=./src/ssh_audit/globals.py
+
+while getopts "m: g: h" OPTION
+do
+ case "$OPTION" in
+ m)
+ MAN_PAGE="$OPTARG"
+ ;;
+ g)
+ GLOBALS_PY="$OPTARG"
+ ;;
+ h)
+ usage
+ exit 0
+ ;;
+ *)
+ echo >&2 "Invalid parameter(s) provided"
+ usage
+ exit 1
+ ;;
+ esac
+done
+
+# Check that the specified files exist.
+[ -f "$MAN_PAGE" ] || { echo >&2 "man page file not found: $MAN_PAGE"; exit 1; }
+[ -f "$GLOBALS_PY" ] || { echo >&2 "globals.py file not found: $GLOBALS_PY"; exit 1; }
+
+# Check that the 'ul' (do underlining) binary exists.
+if [[ "$PLATFORM" = Linux ]]; then
+ command -v ul >/dev/null 2>&1 || { echo >&2 "ul not found."; exit 1; }
+fi
+
+# Check that the 'sed' (stream editor) binary exists.
+command -v sed >/dev/null 2>&1 || { echo >&2 "sed not found."; exit 1; }
+
+# Reset the globals.py file, in case it was modified from a prior run.
+git checkout $GLOBALS_PY > /dev/null 2>&1
+
+# Remove the Windows man page placeholder from 'globals.py'.
+sed -i '/^WINDOWS_MAN_PAGE/d' "$GLOBALS_PY"
+
+echo "Processing man page at ${MAN_PAGE} and placing output into ${GLOBALS_PY}..."
+
+# Append the man page content to 'globals.py'.
+# * man outputs a backspace-overwrite sequence rather than an ANSI escape
+# sequence.
+# * 'MAN_KEEP_FORMATTING' preserves the backspace-overwrite sequence when
+# redirected to a file or a pipe.
+# * sed converts unicode hyphens into an ASCI equivalent.
+# * The 'ul' command converts the backspace-overwrite sequence to an ANSI
+# escape sequence. Not required under Cygwin because man outputs ANSI escape
+# codes automatically.
+
+echo WINDOWS_MAN_PAGE = '"""' >> "$GLOBALS_PY"
+
+if [[ "$PLATFORM" = CYGWIN* ]]; then
+ MANWIDTH=80 MAN_KEEP_FORMATTING=1 man "$MAN_PAGE" | sed $'s/\u2010/-/g' >> "$GLOBALS_PY"
+else
+ MANWIDTH=80 MAN_KEEP_FORMATTING=1 man "$MAN_PAGE" | ul | sed $'s/\u2010/-/g' >> "$GLOBALS_PY"
+fi
+
+echo '"""' >> "$GLOBALS_PY"
+
+echo "Done."
+exit 0
diff --git a/windows_icon.ico b/windows_icon.ico
new file mode 100644
index 0000000..2fd381a
--- /dev/null
+++ b/windows_icon.ico
Binary files differ