Pam-Python实现SSH的短信双因素认证_pam实现手机验证码登录

先自我介绍一下，小编浙江大学毕业，去过华为、字节跳动等大厂，目前阿里P7

深知大多数程序员，想要提升技能，往往是自己摸索成长，但自己不成体系的自学效果低效又漫长，而且极易碰到天花板技术停滞不前！

因此收集整理了一份《2024年最新Python全套学习资料》，初衷也很简单，就是希望能够帮助到想自学提升又不知道该从何学起的朋友。

既有适合小白学习的零基础资料，也有适合3年以上经验的小伙伴深入学习提升的进阶课程，涵盖了95%以上Python知识点，真正体系化！

由于文件比较多，这里只是将部分目录截图出来，全套包含大厂面经、学习笔记、源码讲义、实战项目、大纲路线、讲解视频，并且后续会持续更新

如果你需要这些资料，可以添加V获取：vip1024c （备注Python）

正文

key_hash = hashlib.sha512()

key_hash.update(plain_text)

return key_hash.digest()

def get_user_number(user):

“”“获取用户手机号码”“”

try:

comments = pwd.getpwnam(user).pw_gecos

except:

auth_log(“No local user (%s) found.” % user)

return -1

try:

return comments.split(‘,’)[2] # 返回用户手机号

except:

auth_log(“Invalid comment block for user %s. Phone number must belisted as Office Phone” % (user))

return -1

def gen_key(pamh, user, user_number, length):

“”“生成短信验证码并发送到用户手机”“”

pin= ‘’.join(random.choice(string.digits) for i in range(length))

msg= pamh.Message(pamh.PAM_ERROR_MSG, “The pin is: (%s)” % (pin)) # 登陆界面输出验证码，测试目的，实际使用中注释掉即可

pamh.conversation(msg)

sms= SMSOperation(pin, user_number)

try:

sms.send_text(pamh)

except:

if not user_number:

auth_log(“No phonenumber listed for user (%s).” % (user))

else:

auth_log(“Errorsending PIN to the given SMS number. (%s)” % (user_number))

return -1

return get_hash(pin)

def pam_sm_authenticate(pamh, flags, argv):

PIN_LENGTH = 6 # 短信验证码长度

try:

user = pamh.get_user()

user_number = get_user_number(user)

except pamh.exception, e:

return e.pam_result

ifuser is None or user_number == -1:

msg = pamh.Message(pamh.PAM_ERROR_MSG,“[1]Unable to get user’s phone number.\nPlease check.”)

pamh.conversation(msg)

return pamh.PAM_ABORT

pin= gen_key(pamh, user, user_number, PIN_LENGTH)

ifpin == -1:

msg = pamh.Message(pamh.PAM_ERROR_MSG, “[2]One time PIN could notbe generated.\nPlease check (%s)” % (user_number))

pamh.conversation(msg)

return pamh.PAM_ABORT

forattempt in range(0,3): # 仅允许三次错误尝试

msg = pamh.Message(pamh.PAM_PROMPT_ECHO_OFF, “Enter one time PIN:”)

resp = pamh.conversation(msg)

if get_hash(resp.resp) == pin: #用户输入与生成的验证码进行校验

return pamh.PAM_SUCCESS

else:

continue

return pamh.PAM_AUTH_ERR

def pam_sm_setcred(pamh, flags, argv):

return pamh.PAM_SUCCESS

def pam_sm_acct_mgmt(pamh, flags, argv):

return pamh.PAM_SUCCESS

def pam_sm_open_session(pamh, flags, argv):

return pamh.PAM_SUCCESS

def pam_sm_close_session(pamh, flags, argv):

return pamh.PAM_SUCCESS

def pam_sm_chauthtok(pamh, flags, argv):

return pamh.PAM_SUCCESS

保存脚本文件到/lib64/security/目录下，这里将脚本命名为：Multiauth.py。

接下来是配置SSHD，启用PAM的短信认证模块。

vi /etc/pam.d/sshd 添加一行内容如下：

auth      requisite    pam_python.so   Multiauth.py

同时需要启用ChallengeResponseAuthentication，修改/etc/ssh/sshd_config，将ChallengeResponseAuthentication设置为yes即可。

在上面的代码里面，获取用户手机号码的时候使用的方法是：pwd.getpwnam(user).pw_gecos，这里我将用户的手机号码信息存放在/etc/passwd文件中，当然可以根据需要存放在其他地方，这里只是用作演示。

一切配置完成之后，需要重启ssh服务：systemctl restart sshd.service

重启完成之后，使用gary账号ssh登陆目标主机时，就启用了短信双因素认证了。

如果在运行过程中发生错误，异常日志信息记录在/var/log/secure文件中，正常的运行日志记录在/var/log/messages中。

这里简单介绍了如何利用pam_python模块实现ssh登陆的短信验证码双因素认证，在实际的安全场景中，实现业务主机的双因素安全认证，可根据实际的安全需求进行调整，例如针对不同用户开启短信验证码，用户手机号码的存储，短信验证码的发送频率限制，短信接口访问的限制。

更多Python视频、源码、资料加群683380553免费获取

*本文作者：yuegui_2004，转载请注明FreeBuf.COM

网上学习资料一大堆，但如果学到的知识不成体系，遇到问题时只是浅尝辄止，不再深入研究，那么很难做到真正的技术提升。

需要这份系统化的资料的朋友，可以添加V获取：vip1024c （备注python）

一个人可以走的很快，但一群人才能走的更远！不论你是正从事IT行业的老鸟或是对IT行业感兴趣的新人，都欢迎加入我们的的圈子（技术交流、学习资源、职场吐槽、大厂内推、面试辅导），让我们一起学习成长！

，但如果学到的知识不成体系，遇到问题时只是浅尝辄止，不再深入研究，那么很难做到真正的技术提升。**

需要这份系统化的资料的朋友，可以添加V获取：vip1024c （备注python）
[外链图片转存中…(img-LuD4GCah-1713711650623)]

一个人可以走的很快，但一群人才能走的更远！不论你是正从事IT行业的老鸟或是对IT行业感兴趣的新人，都欢迎加入我们的的圈子（技术交流、学习资源、职场吐槽、大厂内推、面试辅导），让我们一起学习成长！