調(diào)用 passwd 命令自動修改密碼，非 Expect

黑色陽光_cu

1. #!/usr/bin/env perl
   
2. 
   
3. use strict;
   
4. use warnings;
   
5. use IO::Pty;
   
6. use IO::Select;
   
7. # use Term::ReadKey;
   
8. 
   
9. usage() if ($#ARGV < 1);
   
10. 
    
11. my $pty = IO::Pty->new();
    
12. my $select = IO::Select->new($pty);
    
13. my $buffer = "";
    
14. 
    
15. my $pid = fork();
    
16. die "Fork fail!\n" if (not defined $pid);
    
17. if ($pid != 0) {
    
18.     my $loop = 1;
    
19.     local $SIG{CHLD} = sub { $loop = 0; };
    
20.     while ($loop) {
    
21.         expect();
    
22.     }
    
23. 
    
24.     expect();
    
25.     waitpid($pid, 0);
    
26. }
    
27. else {
    
28.     my $slave = $pty->slave();
    
29.     open(STDIN, "<&", $slave);
    
30.     open(STDOUT, ">&", $slave);
    
31.     open(STDERR, ">&", $slave);
    
32.     close $pty;
    
33.     close $slave;
    
34.     exec("passwd");
    
35.     exit 1;
    
36. }
    
37. 
    
38. ################################
    
39. 
    
40. sub usage {
    
41.     print STDERR "$0 Password NewPassword\n";
    
42.     exit 0;
    
43. }
    
44. 
    
45. sub expect {
    
46.     my ($r, $w, $e) = IO::Select->select($select, $select, undef, 0.02);
    
47.     if ($#$r > -1) {
    
48.         if (sysread($pty, my $buf, 8196)) {
    
49.             $buffer .= $buf;
    
50.             syswrite(STDOUT, $buf);
    
51.         }
    
52.     }
    
53.     else {
    
54.         return;
    
55.     }
    
56.    
    
57.     if ($#$w > -1) {
    
58.         if ($buffer =~ /\(current\) UNIX password:/) {
    
59.             $buffer = "";
    
60.             syswrite($pty, "$ARGV[0]\r");
    
61.         }
    
62.         elsif ($buffer =~ /New UNIX password:/) {
    
63.             $buffer = "";
    
64.             syswrite($pty, "$ARGV[1]\r");
    
65.         }
    
66.         elsif ($buffer =~ /Retype new UNIX password:/) {
    
67.             $buffer = "";
    
68.             syswrite($pty, "$ARGV[1]\r");
    
69.         }
    
70.         else {
    
71.             ;
    
72.         }
    
73.     }
    
74. }
    

復(fù)制代碼

flw

這個其實就是 Expect 的簡易版，
以及 screen 實用工具的主要工作原理。

支持一下，加個精華。

flw

發(fā)布代碼時，
如果能夠?qū)Υ�碼的主要意圖加以說明，并且指出一些注意事項，就更加完美了。
就像下面這樣：

http://www.72891.cn/redirec ... 203&pid=8215506

如同標(biāo)題所說，本程序旨在演示如何操縱偽終端和交互式應(yīng)用程序通信，
因此對于 passwd 實用工具的輸出解析并不多，只是做了一些簡單的判斷。

事實上，根據(jù)輸入內(nèi)容的不同，passwd 實用工具的輸出信息多達(dá)十多種，PAM 的到來使得 passwd 的輸出更加復(fù)雜，
因此如果希望能夠自動控制 passwd，那么還是用 expect/perl 等工具較為方便。

另外需要說明的一點(diǎn)是，自動控制交互式應(yīng)用程序運(yùn)行時，盡量運(yùn)行其英文版本，這樣信息會規(guī)范一些從而有利于捕獲并匹配。
中文或者其它語種的信息不僅要考慮到編碼的問題，而且信息用語經(jīng)常會有變化。

通過設(shè)定環(huán)境變量 LANG=C，可以使 passwd 實用工具的輸出變成英文。
flw 發(fā)表于 2008-04-14 19:03

黑色陽光_cu

發(fā)布代碼時，
如果能夠?qū)Υ�碼的主要意圖加以說明，并且指出一些注意事項，就更加完美了。
就像下面這樣： ...
flw 發(fā)表于 2011-03-05 08:42

  flw 說得是

seufy88

flw 說得是
黑色陽光_cu 發(fā)表于 2011-03-06 08:58

    請問一下,expect()是什么子例程?是perl內(nèi)建的還是從哪個模塊導(dǎo)入的?

minority

學(xué)習(xí)！

yanhuo_149

支持一下