elie/ssh-audit
1
0
Fork 0
mirror of https://github.com/jtesta/ssh-audit.git synced 2024-12-22 22:15:22 +01:00
Code Issues Packages Projects Releases Wiki Activity

Now prints a graceful error message when policy file is not found.

Browse Source
This commit is contained in:
Joe Testa 2020-10-20 23:26:21 -04:00
parent 1730126af8
commit f0db035044
1 changed files with 9 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
src/ssh_audit/policy.py
View File

@ -21,10 +21,13 @@
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
THE SOFTWARE. THE SOFTWARE.
""" """
import sys
from typing import Dict, List, Tuple from typing import Dict, List, Tuple
from typing import Optional, Any, Union, cast from typing import Optional, Any, Union, cast
from datetime import date from datetime import date
from ssh_audit import exitcodes
from ssh_audit.ssh2_kex import SSH2_Kex # pylint: disable=unused-import from ssh_audit.ssh2_kex import SSH2_Kex # pylint: disable=unused-import
from ssh_audit.banner import Banner from ssh_audit.banner import Banner
@ -107,8 +110,12 @@ class Policy:
return return
if policy_file is not None: if policy_file is not None:
with open(policy_file, "r") as f: try:
policy_data = f.read() with open(policy_file, "r") as f:
policy_data = f.read()
except FileNotFoundError:
print("Error: policy file not found: %s" % policy_file)
sys.exit(exitcodes.UNKNOWN_ERROR)
lines = [] lines = []
if policy_data is not None: if policy_data is not None: