[求助]Expect做ssh代理中转站的用法

npc_hp110

【问题介绍】
一般的Linux系统可以通过ssh user@ip command远程执行命令
但某些特殊的定制化系统，ssh登录后会进入定制化的人机交互进程（非shell环境），无法通过上述方式远程管理
通过expect可以实现ssh登录后的自动化控制，但是要事先把交互逻辑写入expect脚本中，通用性差。

【解决方法】
expect脚本采用tcl语法，原生态地支持socket。因此考虑用expect做一个ssh中转服务器
1，中转服务器先ssh登录系统，进入人机交互环境
2，然后启动一个socket服务器等待客户端的连接
3，当客户端发来交互命令时直接转发给spawn的ssh子进程处理
4，并将命令处理结果发送回客户端
以下是一段简单的代码，服务器端代码：

1. #!/usr/bin/expect -f
   
2. set timeout 60
   
3. set user XXX
   
4. set ip XXX
   
5. set passwd XXX
   
6. set prompt XXX
   
7. 
   
8. set sent_passwd 0
   
9. spawn ssh $user@$ip
   
10. expect {
    
11.         "yes/no" { send "yes\r";exp_continue }
    
12.         "passwd" {
    
13.                 if { $send_passwd == 0 } {
    
14.                         send "$passwd\r";
    
15.                         set send_passwd 1
    
16.                         exp_continue
    
17.                 } else {
    
18.                         puts "passwd may be wrong"
    
19.                         exit 1
    
20.                 }
    
21.         }
    
22.         "$prompt" {  }
    
23. }
    
24. 
    
25. #以上代码实现ssh登录系统
    
26. set ssh_agent [ socket -server accept 12345 ]
    
27. proc accept { client addr port } {
    
28.         puts "new client connected($addr:$port)"
    
29.         gets $client cmd
    
30.         send "$cmd\r"
    
31.         expect "(.*)$prompt"
    
32.         set result_list [ split $expect_out(1,string) "\n" ]
    
33.         for line $result_list { #似乎puts每次只能发送一行内容？？
    
34.                 puts $client $line
    
35.         }
    
36.         puts $client "<result_end>" #发送结束标识
    
37.         flush $client
    
38.         close $client
    
39. }
    
40. vwait forever

复制代码

客户端代码：

1. #!/usr/bin/expect -f
   
2. set cmd [join $argv " "]
   
3. set client [socket 127.0.0.1 12345]
   
4. puts $client $cmd
   
5. flush $client
   
6. while { 1 == 1 } {
   
7.         gets $client line
   
8.         if { $line == "<result_end>" } {
   
9.                 break
   
10.         }
    
11.         puts $line
    
12. }
    
13. close $client

复制代码

上述代码没有经过严格测试（其实没有经过任何测试^_^），请大家批评指正。

【遗留问题】
expect中启动的socket服务器，当有客户端连接时，会执行accept函数。不同的客户端的accept函数似乎是并行执行的（可能是在不同线程中）
因此如果有多个客户端同时发送命令到服务器端，可能会有多个accept函数同时运行，会产生一些问题：
1，有可能会同时发送多条命令到ssh子进程，多条命令组合可能形成非法命令
2，输出结果混在一起，无法区分是那个客户端的命令执行结果
请问大神们，如何在accept函数上加上互斥锁的机制，让它同一时刻只能处理一个客户端的命令请求？
我现在想到的方法是在shell脚本中调用客户端，调用的时候用woodie大神的脚本串行化方法加锁
http://bbs.chinaunix.net/forum.p ... st%3D1%26digest%3D1

jason680

回复 1# npc_hp110


1. 原意是...
  是解决问题还是增加问题...

2. sshpass
    sshpass -p passwrod ssh ...