From 7ba1804af2ec34094171127d6eada72a1d1e199f Mon Sep 17 00:00:00 2001 From: rakiru Date: Thu, 11 Jan 2018 20:28:03 +0000 Subject: Fix potential error if given invalid socket type --- asciifarm/client/connection.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/asciifarm/client/connection.py b/asciifarm/client/connection.py index 251d376..20f4964 100644 --- a/asciifarm/client/connection.py +++ b/asciifarm/client/connection.py @@ -10,6 +10,8 @@ class Connection: sockType = socket.AF_UNIX elif socketType == "inet": sockType = socket.AF_INET + else: + raise ValueError("Invalid socket type: %r" % (socketType,)) self.sock = socket.socket(sockType, socket.SOCK_STREAM) def connect(self, address): -- cgit