From 100d0aa284fc40e20eba31c4ef54a95ce56a18f5 Mon Sep 17 00:00:00 2001
From: messyidea <the.mid.sea@gmail.com>
Date: Fri, 29 Apr 2016 21:35:17 +0800
Subject: [PATCH] add socket timeout

---
 from_dict/check_from_dict.py | 9 ++++++++-
 reverse/check_domain.py      | 7 ++++++-
 2 files changed, 14 insertions(+), 2 deletions(-)

diff --git a/from_dict/check_from_dict.py b/from_dict/check_from_dict.py
index e701ab5..0f20f46 100644
--- a/from_dict/check_from_dict.py
+++ b/from_dict/check_from_dict.py
@@ -5,6 +5,9 @@
 import urllib
 import os
 import time
+import socket
+
+socket.setdefaulttimeout(60)
 
 domain_suffix = '.com'
 
@@ -33,7 +36,11 @@ def get_words_from_file(words_file=words_file, init_word_line=0):
 
 def is_ava(domain):
     """判断该域名是否被注册"""
-    data = urllib.urlopen(api % domain).read()
+    try:
+        data = urllib.urlopen(api % domain).read()
+    except:
+        return False
+
     ava_pattern = re.compile(r'<original>(.*) : .*</original>')
     result = ava_pattern.findall(data)
     if '210' in result:
diff --git a/reverse/check_domain.py b/reverse/check_domain.py
index 932dcdb..2b10615 100644
--- a/reverse/check_domain.py
+++ b/reverse/check_domain.py
@@ -9,7 +9,9 @@
 import re
 import os
 import pickle
+import socket
 
+socket.setdefaulttimeout(60)
 api = "http://panda.www.net.cn/cgi-bin/check.cgi?area_domain=%s"  # api地址
 string = "abcdefghigklmnopqrstuvwxyz"                                            # 所有字母
 string_len = len(string)                                          # 长度
@@ -48,7 +50,10 @@ def num_2_string(name, string):
 
 def is_ava(domain):
     """判断该域名是否被注册"""
-    data = urllib.urlopen(api % domain).read()
+    try:
+        data = urllib.urlopen(api % domain).read()
+    except:
+        return False
     ava_pattern = re.compile(r'<original>(.*) : .*</original>')
     perm_pattern = re.compile(r'Forbidden')
     result = ava_pattern.findall(data)