Linux 北大网关断网重连

概述

把台式机搬到实验室当服务器用了，买了个极路由插在公网IP上，做一个端口转发，可以随时随地远程ssh这台电脑了。但是问题来了，PKU的网关不稳定，有的时候会断，断了的话这台电脑就连不上网了。

于是我利用北大提供的linux shell联网客户端connect, 写了一个脚本，每隔一定时间检测一下网络是否通畅，如果网断了，就自动重新连接。

下载connect客户端

去https://its.pku.edu.cn/download_ipgwclient.jsp下载北大网关linux命令行客户端，chmod 755 connect，然后mv到 /usr/local/bin/目录下：

$ wget https://its.pku.edu.cn/pku_gateway_apps/download/linux/connect $ sudo chmod 755 connect
$ sudo mv ./connect /usr/local/bin/

编写脚本

$ vim pkunet.py

#!/usr/bin/python
#coding=utf8
'''
本脚本用于定时检查北大校园网关是否联通，如果不联通，则重新连接。
前提是在/usr/local/bin/目录下存在connect客户端。
connect是北大计算中心提供的linux shell客户端，下载地址如下：
 https://its.pku.edu.cn/download_ipgwclient.jsp @作者： 刘伟杰
@日期： 2017-03-14
'''
import os
import time
import urllib2 as ulb
import logging

# settings
CHECK_SITE = 'https://www.baidu.com/'
TIME_OUT = 3  # 连接超时时间
USER_NAME = 'studentid'
PASSWORD = 'password'
TIME_INTERVAL = 600  # 每600秒检查一次网络
LOG_FILE = '/home/liuweijie/.pkunet.log'

# configuration
this_dir = os.path.dirname(os.path.abspath(__file__))
connect_client = '/usr/local/bin/connect'
logging.basicConfig(
level=logging.DEBUG,
format='%(asctime)s %(filename)s[line:%(lineno)d] %(levelname)s %(message)s',
datefmt='%a, %d %b %Y %H:%M:%S',
filename=LOG_FILE,
filemode='a'
)

def is_connect():
'''
检测网络是否通畅
'''
try:
ulb.urlopen(CHECK_SITE, timeout=TIME_OUT)
return True
except:
return False

def connect():
'''
连接网关
'''
try:
os.system('%s -u %s -p %s'%(connect_client, USER_NAME, PASSWORD))
return True
except:
return False

def run():
now = time.strftime('%Y-%m-%d %X', time.localtime())
logging.info('%s: PKUNet start!'%(now))
connect()
while True:

time_remaining = TIME_INTERVAL - time.time()%TIME_INTERVAL
time.sleep(time_remaining)

if not is_connect():
now = time.strftime('%Y-%m-%d %X', time.localtime())
logging.warning("%s: Network wrong!"%(now))
is_success = connect()
if is_success:
logging.info("Network reconnect successfully!")
else:
logging.warning("Network reconnect Falied!")

if __name__ == '__main__':
run()

保存脚本, 改变脚本权限, 把脚本移动到/usr/local/bin/目录下

$ sudo chmod 777 pkunet.py
$ sudo mv pkunet.py /usr/local/bin/

设置开机执行该脚本

在/etc/init.d/写一个脚本

$ cd /etc/init.d/
$ sudo vim pkunet

添加如下内容：

#! /bin/sh
/usr/local/bin/pkunet.py &

保存退出，然后改变该脚本权限

$ chmod 755 pkunet

添加到开机启动服务中

$ sudo update-rc.d pkunet defaults 99

重启，这样就好了，每次的log会保存到/home/liuweijie/.pkunet.log中

如果要关闭这个服务

$ sudo update-rc.d -f pkunet.py remove