tcl+expect+ssh实现程序版本自动更新

本文章实现程序自动更新，思路是程序启动之前 检查文件服务器的程序版本，如果大于 当前版本就更新，否则直接运行当前程序。
服务器直接用ssh就可以，自动更新想到了expect，这里的方法是使用tcl+expect来实现的。

#!/usr/bin/tclsh
package require Expect        #添加对expect的支持

#获取服务器上的可更新 文件名
#para：host 服务器 ip
#para：服务器 用户名
#para：服务器 密码
proc version_f { host user key} {
  set timeout 2

  if {[info exists host] == 0} {
    send_user "remote update host ip is $host \n"
    return
  }

  #检查设置 user 和 key 的默认值
  if {[info exists user] == 0} {
    set user "root"
  }

  if {[info exists key] == 0} {
    set key ""
  }

  #获取文件名 ，感觉这个方法有点笨，但是一时还找不到其他方法 郁闷
  #这里使用expect 的语法了
  spawn ssh ${user}@${host} "cd ~/update;ls ~/update/exefile*>/tmp/version"
  expect {
    "yes/no" {
               send "yes\r"
           expect "password"
     send "$key\r"
             }
    timeout { send_user "connect host timeout!"
               return
     }
    "password" { send "$key\r"}
    "Connection refused" {
                           send_user "host $host Connection refused!"
             return
                         }
  }
  expect {
    eof    {}
    "Permission denied" {
                          send_user "connect host auth failed!"
             return
                        }
  }

  #讲带有文件名的文件copy到本地
  spawn scp ${user}@${host}:/tmp/version /tmp/version
  expect {
    timeout { send_user "connect host timeout!"
               return
     }
    "password" { send "$key\r"}
  }
  expect eof

  unset timeout

  #读取文件名，并返回
  set f [open /tmp/version r]
  gets $f line
  set rnum [regexp {([a-z]+)[-_]([0-9]+)} $line str name index version]
  close $f

  if {$rnum == 0} {
    return
  }

  #return $version
  return $str
}

#判断是否需要更新版本
#其实就是实现了一个比较，放到主程序里显得比较繁琐，所以封装
#para: newvnum ,new version num
proc need_update {newvnum} {
  #获取当前目录下的可执行文件列表，这里的可执行文件格式是 exefile_[版本号]
  set alist [glob exefile*]
  if {[llength alist] == 0} {
    puts "local agw can not exist!\n"
    puts "update!\n"
    return 1
  }

  #用正则表达式获取版本号和文件名
  regexp {exefile[-_]([0-9]+)} [lindex $alist 0] str index version
  if {$version >= $newvnum} {
    return 0;
  }

  return 1;
}

#获取服务器上的可更新 文件名
#para：host 服务器 ip
#para：user 用户名
#para：key 密码
#para：file 服务器上的文件名
proc update {host user key file} {
  #将本地文件 放到 bak 目录备份
  set alist [glob exefile*]
  foreach agw $alist {
    file rename -force $agw ./bak/
  }
  #file rename -force exefile* ./bak/

  #下载文件
  set timeout 2
  spawn scp ${user}@${host}:~/update/$file ./
  expect {
    timeout { send_user "connect host timeout!"
               return 0
     }
    "password" { send "$key\r"}
  }
  expect eof

  unset timeout
}

#运行文件
proc run {} {
  set alist [glob exefile*]

  if {[llength alist] == 0} {
    puts "local agw do not exist!\n"
    return -1
  }
  #添加可执行权限，防止下载的文件没有权限
  exec chmod a+x [pwd]/[lindex $alist 0]
  #运行
  exec [pwd]/[lindex $alist 0] &
}

#查看最新版本号
set host 192.168.1.3
set user root
set key  111111
#获取服务器上的文件名，如果获取失败直接执行本地文件
set nname [version_f $host $user $key]
if {[info exists nname] == 0} {
  run
  exit 0;
}
puts "server newest file: $nname"
#更新文件，执行
regexp {exefile[-_]([0-9]+)} $nname str index vers
if {[need_update vers] == 1} {
  puts "update...\n"
  update $host $user $key $nname
}

run

部署
客户端部分：

普通浏览复制代码

1. set host 192.168.1.3 
2. set user root 
3. set key  111111

是客户端对主机的配置，分别是主机ip 用户名和密码
需要将上面的脚本文件与可执行文件放到同一个目录，可执行文件的命名形式是access_****,后面的****是版本号,假设是/root/access。
然后修改/etc/rc.local，添加下面的代码：
cd /root/access
./update.tcl
客户端部分部署就结束了。
服务器部分：
待更新的程序放在：~/update/目录下，文件名的格式是 access_****,后面的****是版本号。
然后开启ssh服务，并且用户名密码需要和客户端的user和key匹配。