perl 5.0 alpha 9
[p5sagit/p5-mst-13.2.git] / lib / chat2.pl
CommitLineData
79072805 1# chat.pl: chat with a server
2# Based on: V2.01.alpha.7 91/06/16
3# Randal L. Schwartz (was <merlyn@iwarp.intel.com>)
4# multihome additions by A.Macpherson@bnr.co.uk
5# allow for /dev/pts based systems by Joe Doupnik <JRD@CC.USU.EDU>
11aea360 6
7package chat;
8
79072805 9if( defined( &main'PF_INET ) ){
10 $pf_inet = &main'PF_INET;
11 $sock_stream = &main'SOCK_STREAM;
12 local($name, $aliases, $proto) = getprotobyname( 'tcp' );
13 $tcp_proto = $proto;
14}
15else {
16 # XXX hardwired $PF_INET, $SOCK_STREAM, 'tcp'
17 # but who the heck would change these anyway? (:-)
18 $pf_inet = 2;
19 $sock_stream = 1;
20 $tcp_proto = 6;
21}
22
23
11aea360 24$sockaddr = 'S n a4 x8';
79072805 25chop($thishost = `hostname`);
11aea360 26
27# *S = symbol for current I/O, gets assigned *chatsymbol....
28$next = "chatsymbol000000"; # next one
29$nextpat = "^chatsymbol"; # patterns that match next++, ++, ++, ++
30
31
32## $handle = &chat'open_port("server.address",$port_number);
33## opens a named or numbered TCP server
34
35sub open_port { ## public
36 local($server, $port) = @_;
37
38 local($serveraddr,$serverproc);
39
79072805 40 # We may be multi-homed, start with 0, fixup once connexion is made
41 $thisaddr = "\0\0\0\0" ;
42 $thisproc = pack($sockaddr, 2, 0, $thisaddr);
43
11aea360 44 *S = ++$next;
45 if ($server =~ /^(\d+)+\.(\d+)\.(\d+)\.(\d+)$/) {
46 $serveraddr = pack('C4', $1, $2, $3, $4);
47 } else {
48 local(@x) = gethostbyname($server);
49 return undef unless @x;
50 $serveraddr = $x[4];
51 }
52 $serverproc = pack($sockaddr, 2, $port, $serveraddr);
79072805 53 unless (socket(S, $pf_inet, $sock_stream, $tcp_proto)) {
11aea360 54 ($!) = ($!, close(S)); # close S while saving $!
55 return undef;
56 }
57 unless (bind(S, $thisproc)) {
58 ($!) = ($!, close(S)); # close S while saving $!
59 return undef;
60 }
61 unless (connect(S, $serverproc)) {
62 ($!) = ($!, close(S)); # close S while saving $!
63 return undef;
64 }
79072805 65# We opened with the local address set to ANY, at this stage we know
66# which interface we are using. This is critical if our machine is
67# multi-homed, with IP forwarding off, so fix-up.
68 local($fam,$lport);
69 ($fam,$lport,$thisaddr) = unpack($sockaddr, getsockname(S));
70 $thisproc = pack($sockaddr, 2, 0, $thisaddr);
71# end of post-connect fixup
11aea360 72 select((select(S), $| = 1)[0]);
73 $next; # return symbol for switcharound
74}
75
76## ($host, $port, $handle) = &chat'open_listen([$port_number]);
77## opens a TCP port on the current machine, ready to be listened to
78## if $port_number is absent or zero, pick a default port number
79## process must be uid 0 to listen to a low port number
80
81sub open_listen { ## public
82
83 *S = ++$next;
84 local($thisport) = shift || 0;
85 local($thisproc_local) = pack($sockaddr, 2, $thisport, $thisaddr);
86 local(*NS) = "__" . time;
79072805 87 unless (socket(NS, $pf_inet, $sock_stream, $tcp_proto)) {
11aea360 88 ($!) = ($!, close(NS));
89 return undef;
90 }
91 unless (bind(NS, $thisproc_local)) {
92 ($!) = ($!, close(NS));
93 return undef;
94 }
95 unless (listen(NS, 1)) {
96 ($!) = ($!, close(NS));
97 return undef;
98 }
99 select((select(NS), $| = 1)[0]);
100 local($family, $port, @myaddr) =
101 unpack("S n C C C C x8", getsockname(NS));
102 $S{"needs_accept"} = *NS; # so expect will open it
103 (@myaddr, $port, $next); # returning this
104}
105
106## $handle = &chat'open_proc("command","arg1","arg2",...);
107## opens a /bin/sh on a pseudo-tty
108
109sub open_proc { ## public
110 local(@cmd) = @_;
111
112 *S = ++$next;
113 local(*TTY) = "__TTY" . time;
114 local($pty,$tty) = &_getpty(S,TTY);
115 die "Cannot find a new pty" unless defined $pty;
79072805 116 $pid = fork;
11aea360 117 die "Cannot fork: $!" unless defined $pid;
118 unless ($pid) {
119 close STDIN; close STDOUT; close STDERR;
120 setpgrp(0,$$);
121 if (open(DEVTTY, "/dev/tty")) {
122 ioctl(DEVTTY,0x20007471,0); # XXX s/b &TIOCNOTTY
123 close DEVTTY;
124 }
125 open(STDIN,"<&TTY");
126 open(STDOUT,">&TTY");
127 open(STDERR,">&STDOUT");
128 die "Oops" unless fileno(STDERR) == 2; # sanity
129 close(S);
130 exec @cmd;
131 die "Cannot exec @cmd: $!";
132 }
133 close(TTY);
134 $next; # return symbol for switcharound
135}
136
137# $S is the read-ahead buffer
138
139## $return = &chat'expect([$handle,] $timeout_time,
140## $pat1, $body1, $pat2, $body2, ... )
141## $handle is from previous &chat'open_*().
142## $timeout_time is the time (either relative to the current time, or
143## absolute, ala time(2)) at which a timeout event occurs.
144## $pat1, $pat2, and so on are regexs which are matched against the input
145## stream. If a match is found, the entire matched string is consumed,
146## and the corresponding body eval string is evaled.
147##
148## Each pat is a regular-expression (probably enclosed in single-quotes
149## in the invocation). ^ and $ will work, respecting the current value of $*.
150## If pat is 'TIMEOUT', the body is executed if the timeout is exceeded.
151## If pat is 'EOF', the body is executed if the process exits before
152## the other patterns are seen.
153##
154## Pats are scanned in the order given, so later pats can contain
155## general defaults that won't be examined unless the earlier pats
156## have failed.
157##
158## The result of eval'ing body is returned as the result of
159## the invocation. Recursive invocations are not thought
160## through, and may work only accidentally. :-)
161##
162## undef is returned if either a timeout or an eof occurs and no
163## corresponding body has been defined.
164## I/O errors of any sort are treated as eof.
165
166$nextsubname = "expectloop000000"; # used for subroutines
167
168sub expect { ## public
169 if ($_[0] =~ /$nextpat/) {
170 *S = shift;
171 }
172 local($endtime) = shift;
173
174 local($timeout,$eof) = (1,1);
175 local($caller) = caller;
176 local($rmask, $nfound, $timeleft, $thisbuf);
177 local($cases, $pattern, $action, $subname);
178 $endtime += time if $endtime < 600_000_000;
179
180 if (defined $S{"needs_accept"}) { # is it a listen socket?
181 local(*NS) = $S{"needs_accept"};
182 delete $S{"needs_accept"};
183 $S{"needs_close"} = *NS;
184 unless(accept(S,NS)) {
185 ($!) = ($!, close(S), close(NS));
186 return undef;
187 }
188 select((select(S), $| = 1)[0]);
189 }
190
191 # now see whether we need to create a new sub:
192
193 unless ($subname = $expect_subname{$caller,@_}) {
194 # nope. make a new one:
195 $expect_subname{$caller,@_} = $subname = $nextsubname++;
196
197 $cases .= <<"EDQ"; # header is funny to make everything elsif's
198sub $subname {
199 LOOP: {
200 if (0) { ; }
201EDQ
202 while (@_) {
203 ($pattern,$action) = splice(@_,0,2);
204 if ($pattern =~ /^eof$/i) {
205 $cases .= <<"EDQ";
206 elsif (\$eof) {
207 package $caller;
208 $action;
209 }
210EDQ
211 $eof = 0;
212 } elsif ($pattern =~ /^timeout$/i) {
213 $cases .= <<"EDQ";
214 elsif (\$timeout) {
215 package $caller;
216 $action;
217 }
218EDQ
219 $timeout = 0;
220 } else {
221 $pattern =~ s#/#\\/#g;
222 $cases .= <<"EDQ";
223 elsif (\$S =~ /$pattern/) {
224 \$S = \$';
225 package $caller;
226 $action;
227 }
228EDQ
229 }
230 }
231 $cases .= <<"EDQ" if $eof;
232 elsif (\$eof) {
233 undef;
234 }
235EDQ
236 $cases .= <<"EDQ" if $timeout;
237 elsif (\$timeout) {
238 undef;
239 }
240EDQ
241 $cases .= <<'ESQ';
242 else {
243 $rmask = "";
244 vec($rmask,fileno(S),1) = 1;
245 ($nfound, $rmask) =
246 select($rmask, undef, undef, $endtime - time);
247 if ($nfound) {
248 $nread = sysread(S, $thisbuf, 1024);
249 if ($nread > 0) {
250 $S .= $thisbuf;
251 } else {
252 $eof++, redo LOOP; # any error is also eof
253 }
254 } else {
255 $timeout++, redo LOOP; # timeout
256 }
257 redo LOOP;
258 }
259 }
260}
261ESQ
262 eval $cases; die "$cases:\n$@" if $@;
263 }
264 $eof = $timeout = 0;
265 do $subname();
266}
267
268## &chat'print([$handle,] @data)
269## $handle is from previous &chat'open().
270## like print $handle @data
271
272sub print { ## public
273 if ($_[0] =~ /$nextpat/) {
274 *S = shift;
275 }
276 print S @_;
79072805 277 if( $chat'debug ){
278 print STDERR "printed:";
279 print STDERR @_;
280 }
11aea360 281}
282
283## &chat'close([$handle,])
284## $handle is from previous &chat'open().
285## like close $handle
286
287sub close { ## public
288 if ($_[0] =~ /$nextpat/) {
289 *S = shift;
290 }
291 close(S);
292 if (defined $S{"needs_close"}) { # is it a listen socket?
293 local(*NS) = $S{"needs_close"};
294 delete $S{"needs_close"};
295 close(NS);
296 }
297}
298
299## @ready_handles = &chat'select($timeout, @handles)
300## select()'s the handles with a timeout value of $timeout seconds.
301## Returns an array of handles that are ready for I/O.
302## Both user handles and chat handles are supported (but beware of
303## stdio's buffering for user handles).
304
305sub select { ## public
306 local($timeout) = shift;
307 local(@handles) = @_;
308 local(%handlename) = ();
309 local(%ready) = ();
310 local($caller) = caller;
311 local($rmask) = "";
312 for (@handles) {
313 if (/$nextpat/o) { # one of ours... see if ready
314 local(*SYM) = $_;
315 if (length($SYM)) {
316 $timeout = 0; # we have a winner
317 $ready{$_}++;
318 }
319 $handlename{fileno($_)} = $_;
320 } else {
321 $handlename{fileno(/'/ ? $_ : "$caller\'$_")} = $_;
322 }
323 }
324 for (sort keys %handlename) {
325 vec($rmask, $_, 1) = 1;
326 }
327 select($rmask, undef, undef, $timeout);
328 for (sort keys %handlename) {
329 $ready{$handlename{$_}}++ if vec($rmask,$_,1);
330 }
331 sort keys %ready;
332}
333
334# ($pty,$tty) = $chat'_getpty(PTY,TTY):
335# internal procedure to get the next available pty.
336# opens pty on handle PTY, and matching tty on handle TTY.
337# returns undef if can't find a pty.
79072805 338# Modify "/dev/pty" to "/dev/pts" for Dell Unix v2.2 (aka SVR4.04). Joe Doupnik.
11aea360 339
340sub _getpty { ## private
341 local($_PTY,$_TTY) = @_;
342 $_PTY =~ s/^([^']+)$/(caller)[$[]."'".$1/e;
343 $_TTY =~ s/^([^']+)$/(caller)[$[]."'".$1/e;
79072805 344 local($pty, $tty, $kind);
345 if( -e "/dev/pts000" ){ ## mods by Joe Doupnik Dec 1992
346 $kind = "pts"; ## SVR4 Streams
347 } else {
348 $kind = "pty"; ## BSD Clist stuff
349 }
11aea360 350 for $bank (112..127) {
79072805 351 next unless -e sprintf("/dev/$kind%c0", $bank);
11aea360 352 for $unit (48..57) {
79072805 353 $pty = sprintf("/dev/$kind%c%c", $bank, $unit);
11aea360 354 open($_PTY,"+>$pty") || next;
355 select((select($_PTY), $| = 1)[0]);
356 ($tty = $pty) =~ s/pty/tty/;
357 open($_TTY,"+>$tty") || next;
358 select((select($_TTY), $| = 1)[0]);
359 system "stty nl>$tty";
360 return ($pty,$tty);
361 }
362 }
363 undef;
364}
365
3661;