Shell脚本实现自动登录服务器

609 阅读4分钟

Shell脚本实现自动登录服务器

本文正在参与 “走过Linux三十年”话题征文活动


0x00.语法基础

1.Expect语言实现自动化交互

Expect是一个免费的编程工具语言,用来实现自动和交互式任务进行通信,而无需人的干预。Expect的作者Don Libes在1990年开始编写Expect时对Expect做有如下定义:Expect是一个用来实现自动交互功能的软件套件(Expect [is a] software suite for automating interactive tools)。使用它,系统管理员可以创建脚本来对命令或程序进行输入,而这些命令和程序是期望从终端(terminal)得到输入,一般来说这些输入都需要手工输入进行的。Expect则可以根据程序的提示模拟标准输入提供给程序需要的输入来实现交互程序执行。甚至可以实现简单的BBS聊天机器人。 :)

Expect是不断发展的,随着时间的流逝,其功能越来越强大,已经成为系统管理员的的一个强大助手。Expect需要Tcl编程语言的支持,要在系统上运行Expect必须首先安装Tcl。

Expect是一种自动交互语言,能实现在shell脚本中为scp和ssh等自动输入密码自动登陆,我们通过shell可以实现简单的控制流功能,如:循环、判断等。但是对于需要交互的场合必须通过人工干预,expect就时是用来实现这种功能的工具。

从最简单的层次来说,expect的工作方式像是一个通用化的chat脚本工具,chat脚本最早用于uucp网路i内,以用来实现实现计算机之间需要建立连接时进行特定的登陆会话自动化。

Chat脚本由一系列的expect-send对组成,expect等待输出中输出的特定字符,通常是一个提示符,然后发送特定的响应。

2.Expect 相关的命令

2.1 spawn

spawn命令是Expect的初始命令,它用于启动一个进程,之后所有expect操作都在这个进程中进行,如果没有spawn语句,整个expect就无法再进行下去了,使用方法就像下面这样:

spawn ssh  root@192.168.0.1

在spawn命令后面,直接加上要启动的进程等信息。当然,如果真的不要spawn过程也没有关系,虽然这样就没有办法单独执行,但是这个脚本可以与任何调用它的进程进行交互。

2.2 expect

expect命令用于等候一个相匹配的输出,一旦匹配就执行后面的动作,这个命令接受几个特有参数,用的最多的就是-re,表示使用正则表达式的方式匹配。expect命令还有一种用法,它可以在一个expect匹配中同时匹配多个关键字,只需要将关键字放在一个大括号中就可以了:

spawn  ssh  root@192.168.0.1
expect  {
 
   -re  “password:” {exp_send  “word\r”}
 
   -re  “jzcj@” { }
   }

2.3 send

一般是expect中的动作命令,向进程发送输入内容,类似的是send_user表示把后面的内容输出到标准输出中,send_error表示输出到标准错误中。

2.4 exp_continue:

exp_continue:需处于expect的动作中,表示expect的匹配从头开始继续匹配。

2.5 exit,close,wait

exit,close,waitexit表示退出脚本,close表示立即关闭过程,而wait则是等待过程返回eof时关闭。

2.6 interact

ineract //留在环境中不退出 

interact:运行表示将控制权交给用户,与spawn生成的进程进行交互。由用户与spawn生成的进程进行交互,比如登录ftp服务器并下载的过程中,登录ftp服务器的过程可以由用户输入自己的用户名和密码,然后用户再输入q字符将控制权交给脚本,由脚本完成后面的交互动作。

2.7 expect eof //退出环境

expect eof //退出环境

0x01.登录脚本

touch login_server.sh
chmod 777 
#!/bin/bash

# ReferenceLink:https://yq.aliyun.com/articles/516347

#show all host infos of serverList.txt

if [[ -f ./serverList.txt ]]
then
  hostNum=`cat ./serverList.txt | wc -l`
else
  echo "No .serverList.txt in ./ dir, please create it and add server infos."
  exit
fi
while [ True ]
do
  echo -e "+++++++++++ Host List ++++++++++++++++"
  awk -F' ' '{printf("%3d -> %s@%s\n", NR,$1,$2)}'  ./serverList.txt
  echo -e "++++++++++++++++++++++++++++++++++++++"
  echo -e "Enter hostID at first column."
  echo -e "Enter q or Q to quit."
  read hostID
  if [[ "$hostID" == 'q' ]] || [[ "$hostID" == 'Q' ]]
  then
    exit
  elif [[ $hostID -lt 1 ]] || [[ $hostID -gt $hostNum ]]
  then
    echo "Wrong hostID is selected, Only $hostNum hosts are listed, please check."
    continue
  else
    break
  fi
done
user=""
host=""
passwd=""
eval $(awk -v hostID=$hostID -F' ' '{if (NR==hostID) {printf("user=%s;host=%s;passwd=%s;",$1,$2,$3);}}'  ./serverList.txt)
#echo $user, $host, $passwd
echo "login in $user@$host"
expect -c "
set timeout 30
  spawn ssh $user@$host
  expect {
  \"*yes/no\" { send \"yes\r\"; exp_continue }
  \"*?assword:\" { send \"$passwd\r\" }
  }
  interact
"

0x02.服务器列表

serverlist.txt
第一列:系统用户名称;
第二列:服务器ip地址;
第三列:服务器密码。

test1	192.168.1.222	123456
bt		192.168.1.101	123456
st		192.168.1.103	123456

0x03.扩展阅读

csdn1

csdn2