def run(self):
while self.is_alive: # TODO: and not self.queue.empty()?
try:
seq = int(time.time())
if self.queue.qsize() > 0:
logging.info("[Worker] Before get packet from queue, seq:%s", seq)
packet = self.queue.get(True, self.BLOCK_TIMEOUT)
logging.debug("[Worker] After get packet: %s from queue, seq:%s", packet, seq)
queue_len = self.queue.qsize()
# NOTE: Send alarm message if need
if queue_len >= self.alarm_size:
content = SMSCode.SMS_EVENTER_QUEUE_REPORT % ConfHelper.UWEB_CONF.url_out
notify_maintainer(self.db, self.redis, content, 2)
logging.info("[EVENTER] Notify EVENTER queue exception to administrator!")
# NOTE: Deal with the packet
if self.name == EVENTER.INFO_TYPE.REPORT:
self._run_callback(PacketTask(packet, self.db, self.redis).run_report)
else:
self._run_callback(PacketTask(packet, self.db, self.redis).run_position)
else:
time.sleep(0.1)
except Exception as e:
logging.info("[Worker] get exception:%s", e.args)
def post(self):
"""Generate a captcha for retrieving the password."""
status = ErrorCode.SUCCESS
try:
data = DotDict(json_decode(self.request.body))
umobile = data.mobile
captcha_psd = data.captcha_psd
logging.info("[UWEB] Get captcha request: %s", data)
except Exception as e:
status = ErrorCode.ILLEGAL_DATA_FORMAT
logging.exception("[UWEB] Invalid data format. body: %s, Exception: %s",
self.request.body, e.args)
self.write_ret(status)
return
try:
status = self.check_privilege(umobile)
if status != ErrorCode.SUCCESS:
logging.error("[UWEB] User: %s is just for test, has no right to access the function.",
umobile)
self.write_ret(status)
return
captchahash = self.get_secure_cookie("captchahash_password")
m = hashlib.md5()
m.update(captcha_psd.lower())
m.update(UWEB.HASH_SALT)
hash_ = m.hexdigest()
if hash_.lower() != captchahash.lower():
status = ErrorCode.WRONG_CAPTCHA_IMAGE
logging.info("[UWEB] Come from browser, captcha-check failed.")
self.write_ret(status)
return
user = self.db.get("SELECT mobile"
" FROM T_USER"
" WHERE mobile = %s"
" LIMIT 1",
umobile)
if user:
remote_ip = self.request.remote_ip
remote_ip_key = "register_remote_ip:%s" % remote_ip
umobile_key = "register_umobile:%s" % umobile
remote_ip_times = self.redis.getvalue(remote_ip_key)
umobile_times = self.redis.getvalue(umobile_key)
if remote_ip_times is None:
remote_ip_times = 0
if umobile_times is None:
umobile_times = 0
logging.info("[UWEB] Register. umobile: %s, umobile_times: %s, remote_ip: %s, remote_ip_times: %s",
umobile, umobile_times, remote_ip, remote_ip_times)
#NOTE: In current day, the same remote_ip allows 10 times, the umobile, 3 times
current_time = int(time.time())
date = get_date_from_utc(current_time)
year, month, day = date.year, date.month, date.day
start_time_, end_time_ = start_end_of_day(year=year, month=month, day=day)
if umobile_times >= 3: # <= 3 is ok
status = ErrorCode.REGISTER_EXCESS
if remote_ip_times >= 10: # <= 10 is ok
status = ErrorCode.REGISTER_EXCESS
if status == ErrorCode.REGISTER_EXCESS:
body = u'管理员您好：检测到频繁注册，请查看. umobile: %s, umobile_times: %s, remote_ip: %s, remote_ip_times: %s' % (
umobile, umobile_times, remote_ip, remote_ip_times)
notify_maintainer(self.db, self.redis, body, 'password')
self.write_ret(status)
return
captcha = ''.join(random.choice(string.digits) for x in range(4))
getcaptcha_sms = SMSCode.SMS_CAPTCHA % (captcha)
ret = SMSHelper.send(umobile, getcaptcha_sms)
ret = DotDict(json_decode(ret))
if ret.status == ErrorCode.SUCCESS:
logging.info("[UWEB] user uid: %s get captcha success, the captcha: %s",
umobile, captcha)
captcha_key = get_captcha_key(umobile)
self.redis.setvalue(captcha_key, captcha, UWEB.SMS_CAPTCHA_INTERVAL)
self.redis.set(umobile_key, umobile_times+1)
self.redis.expireat(umobile_key, end_time_)
self.redis.set(remote_ip_key, remote_ip_times+1)
self.redis.expireat(remote_ip_key, end_time_)
else:
status = ErrorCode.SERVER_BUSY
logging.error("[UWEB] user uid: %s get captcha failed.", umobile)
else:
status = ErrorCode.USER_NOT_ORDERED
logging.error("[UWEB] user uid: %s does not exist, get captcha failed.",
umobile)
self.write_ret(status)
except Exception as e:
logging.exception("[UWEB] user uid: %s retrieve password failed. Exception: %s",
umobile, e.args)
status = ErrorCode.SERVER_BUSY
self.write_ret(status)
def get(self):
"""Send captcha to user's phone through sms.
"""
status = ErrorCode.SUCCESS
try:
umobile = self.get_argument('umobile', '')
tmobile = self.get_argument('tmobile', '')
remote_ip = self.request.remote_ip
captcha_image = self.get_argument('captcha_img', '')
captchahash = self.get_secure_cookie("captchahash_image")
logging.info("[UWEB] Get captcha-sms request. umobile:%s, tmobile: %s, captcha_img: %s",
umobile, tmobile, captcha_image)
m = hashlib.md5()
m.update(captcha_image.lower())
m.update(UWEB.HASH_SALT)
hash_ = m.hexdigest()
if hash_.lower() != captchahash.lower():
status = ErrorCode.WRONG_CAPTCHA_IMAGE
logging.info(
"[UWEB] Come from browser, captcha-check failed.")
self.write_ret(status)
return
# check tmobile is whitelist or not
white_list = check_zs_phone(tmobile, self.db)
if not white_list:
logging.info("[UWEB] %s is not whitelist", tmobile)
status = ErrorCode.MOBILE_NOT_ORDERED
message = ErrorCode.ERROR_MESSAGE[status] % tmobile
self.write_ret(status, message=message)
return
# NOTE: check times
remote_ip_key = "register_remote_ip:%s" % remote_ip
umobile_key = "register_umobile:%s" % umobile
remote_ip_times = self.redis.getvalue(remote_ip_key)
umobile_times = self.redis.getvalue(umobile_key)
if remote_ip_times is None:
remote_ip_times = 0
if umobile_times is None:
umobile_times = 0
logging.info("[UWEB] Register. umobile: %s, umobile_times: %s, remote_ip: %s, remote_ip_times: %s",
umobile, umobile_times, remote_ip, remote_ip_times)
# NOTE: In current day, the same remote_ip allows 10 times, the
# umobile, 3 times
current_time = int(time.time())
date = get_date_from_utc(current_time)
year, month, day = date.year, date.month, date.day
start_time_, end_time_ = start_end_of_day(
year=year, month=month, day=day)
if umobile_times >= 3: # <= 3 is ok
status = ErrorCode.REGISTER_EXCESS
if remote_ip_times >= 10: # <= 10 is ok
status = ErrorCode.REGISTER_EXCESS
if status == ErrorCode.REGISTER_EXCESS:
body = u'管理员您好：检测到频繁注册，请查看. umobile: %s, umobile_times: %s, remote_ip: %s, remote_ip_times: %s' % (
umobile, umobile_times, remote_ip, remote_ip_times)
notify_maintainer(self.db, self.redis, body, 'register')
self.write_ret(status)
return
psd = ''.join(random.choice(string.digits) for x in range(4))
captcha_sms = SMSCode.SMS_REG % (psd)
ret = SMSHelper.send(umobile, captcha_sms)
ret = DotDict(json_decode(ret))
if ret.status == ErrorCode.SUCCESS:
logging.info("[UWEB] Get sms captcha successfully. umobile: %s, captcha: %s.",
umobile, psd)
captcha_key = get_captcha_key(umobile)
self.redis.setvalue(
captcha_key, psd, UWEB.SMS_CAPTCHA_INTERVAL)
self.redis.set(umobile_key, umobile_times + 1)
self.redis.expireat(umobile_key, end_time_)
self.redis.set(remote_ip_key, remote_ip_times + 1)
self.redis.expireat(remote_ip_key, end_time_)
else:
status = ErrorCode.SERVER_BUSY
logging.error(
"[UWEB] Get sms captcha failed. umobile: %s.", umobile)
self.write_ret(status)
except Exception as e:
logging.exception("[UWEB] Get sms captcha failed. umobile:%s. Exception: %s",
umobile, e.args)
status = ErrorCode.SERVER_BUSY
self.write_ret(status)
def post(self):
"""Retrieve the password."""
status = ErrorCode.SUCCESS
try:
data = DotDict(json_decode(self.request.body))
umobile = data.mobile
captcha_psd = data.get('captcha_psd','')
captchahash = self.get_secure_cookie("captchahash_password")
logging.info("[UWEB] Corp retrieve password request: %s", data)
except Exception as e:
status = ErrorCode.ILLEGAL_DATA_FORMAT
logging.exception("[UWEB] Invalid data format. body: %s, Exception: %s",
self.request.body, e.args)
self.write_ret(status)
return
try:
# check the umobile whether belongs to guandong
is_guandong = check_gd_phone(umobile)
if is_guandong:
pass
else:
logging.info("[UWEB] Mobile is not come from GuanDong, reject it.")
status = ErrorCode.UMOBILE_REGISTER_EXCESS
self.write_ret(status)
return
#NOTE: check captcha-sms for brower
from_brower = False
if self.request.headers.get('User-Agent',None):
user_agent = self.request.headers.get('User-Agent').lower()
if re.search('darwin', user_agent): # Ios client
logging.info("[UWEB] Come from IOS client, do not check captcha-image, User-Agent: %s",
user_agent)
from_brower = False
else:
logging.info("[UWEB] Come from browser, check captcha-image, User-Agent: %s",
user_agent)
from_brower = True
else: # Android client
from_brower = False
logging.info("[UWEB] Come from Android client, do not check captcha-image")
if from_brower:
m = hashlib.md5()
m.update(captcha_psd.lower())
m.update(UWEB.HASH_SALT)
hash_ = m.hexdigest()
if hash_.lower() != captchahash.lower():
status = ErrorCode.WRONG_CAPTCHA_IMAGE
logging.info("[UWEB] Come from browser, captcha-check failed.")
self.write_ret(status)
return
user = self.db.get("SELECT mobile"
" FROM T_CORP"
" WHERE cid = %s"
" LIMIT 1",
umobile)
if not user:
user = self.db.get("SELECT mobile"
" FROM T_OPERATOR"
" WHERE oid = %s"
" LIMIT 1",
umobile)
if user:
remote_ip = self.request.remote_ip
remote_ip_key = "register_remote_ip:%s" % remote_ip
umobile_key = "register_umobile:%s" % umobile
remote_ip_times = self.redis.getvalue(remote_ip_key)
umobile_times = self.redis.getvalue(umobile_key)
if remote_ip_times is None:
remote_ip_times = 0
if umobile_times is None:
umobile_times = 0
logging.info("[UWEB] Register. umobile: %s, umobile_times: %s, remote_ip: %s, remote_ip_times: %s",
umobile, umobile_times, remote_ip, remote_ip_times)
#NOTE: In current day, the same remote_ip allows 10 times, the umobile, 3 times
current_time = int(time.time())
date = get_date_from_utc(current_time)
year, month, day = date.year, date.month, date.day
start_time_, end_time_ = start_end_of_day(year=year, month=month, day=day)
if umobile_times >= 3: # <= 3 is ok
status = ErrorCode.REGISTER_EXCESS
if remote_ip_times >= 10: # <= 10 is ok
status = ErrorCode.REGISTER_EXCESS
if status == ErrorCode.REGISTER_EXCESS:
body = u'管理员您好：检测到频繁注册，请查看. umobile: %s, umobile_times: %s, remote_ip: %s, remote_ip_times: %s' % (
umobile, umobile_times, remote_ip, remote_ip_times)
notify_maintainer(self.db, self.redis, body, 'password')
self.write_ret(status)
return
captcha = ''.join(random.choice(string.digits) for x in range(4))
getcaptcha_sms = SMSCode.SMS_CAPTCHA % (captcha)
ret = SMSHelper.send(umobile, getcaptcha_sms)
ret = DotDict(json_decode(ret))
if ret.status == ErrorCode.SUCCESS:
logging.info("[UWEB] corp mobile: %s get captcha success, the captcha: %s",
umobile, captcha)
captcha_key = get_captcha_key(umobile)
self.redis.setvalue(captcha_key, captcha, UWEB.SMS_CAPTCHA_INTERVAL)
self.redis.set(umobile_key, umobile_times+1)
self.redis.expireat(umobile_key, end_time_)
self.redis.set(remote_ip_key, remote_ip_times+1)
self.redis.expireat(remote_ip_key, end_time_)
else:
status = ErrorCode.SERVER_BUSY
logging.error("[UWEB] Get captcha failed. corp mobile: %s",
umobile)
else:
logging.error("[UWEB] Get captcha failed. corp mobile: %s does not exist.",
umobile)
status = ErrorCode.USER_NOT_ORDERED
self.write_ret(status)
except Exception as e:
logging.exception("[UWEB] Get captcha failed. corp mobile: %s, Exception: %s",
umobile, e.args)
status = ErrorCode.SERVER_BUSY
self.write_ret(status)
