/[pxelator]/lib/PXElator/tftpd.pm
This is repository of my old source code which isn't updated any more. Go to git.rot13.org for current projects!
ViewVC logotype

Diff of /lib/PXElator/tftpd.pm

Parent Directory Parent Directory | Revision Log Revision Log | View Patch Patch

revision 141 by dpavlin, Tue Aug 4 17:29:59 2009 UTC revision 261 by dpavlin, Wed Aug 19 10:58:02 2009 UTC
# Line 6  use strict; Line 6  use strict;
6  use Net::TFTPd 0.03 qw(%OPCODES);  use Net::TFTPd 0.03 qw(%OPCODES);
7  use IO::Socket::INET;  use IO::Socket::INET;
8  use Data::Dump qw/dump/;  use Data::Dump qw/dump/;
9    use CouchDB;
10    
11  use server;  use server;
12    
# Line 42  use config; Line 43  use config;
43  sub tftp_request {  sub tftp_request {
44          my $request = shift;          my $request = shift;
45    
         server->refresh;  
   
46          warn 'request: ', dump( $request ) if $debug;          warn 'request: ', dump( $request ) if $debug;
47    
48          config::for_ip( $request->{_REQUEST_}->{PeerAddr} );          my $ip = $request->{_REQUEST_}->{PeerAddr};
49            config::for_ip( $ip );
50    
51          if ( $request->{RootDir} ne $dir ) {          if ( $request->{RootDir} ne $dir ) {
52                  $request->{RootDir} = $dir;                  $request->{RootDir} = $dir;
# Line 54  sub tftp_request { Line 54  sub tftp_request {
54          }          }
55    
56          my $file = $request->{'_REQUEST_'}{'FileName'};          my $file = $request->{'_REQUEST_'}{'FileName'};
57          # received request          my $opcode = $OPCODES{$request->{'_REQUEST_'}{'OPCODE'}};
58          print $OPCODES{$request->{'_REQUEST_'}{'OPCODE'}}, " $file\n";  
59            my $audit = {
60                    ip => $ip,
61                    opcode => $opcode,
62                    file => $file,
63                    state => 'start',
64            };
65            CouchDB::audit( $opcode, $audit );
66    
67          progress_bar::start;          progress_bar::start;
68    
69          # process the request          # process the request
70          if( $request->processRQ() ) {          if( $request->processRQ() ) {
71                  print "\nOK completed $file ", -s "$dir/$file", "\n";                  my $size = -s "$dir/$file";
72                    $audit->{state} = 'finish';
73                    $audit->{size} = $size;
74                    CouchDB::audit( $opcode, $audit );
75          } else {          } else {
76                  print "ERROR ", Net::TFTPd->error, "\n";                  $audit->{state} = 'error';
77                    $audit->{error} = Net::TFTPd->error;
78                    CouchDB::audit( $opcode, $audit );
79                  $request->processRQ();                  $request->processRQ();
80          }          }
81    
# Line 107  sub start { Line 119  sub start {
119    
120          warn 'listener: ',dump( $listener ) if $debug;          warn 'listener: ',dump( $listener ) if $debug;
121    
122          printf "TFTP listen %s:%d timeout: %d dir: $dir\n",          CouchDB::audit( 'start', {
123                  $listener->{LocalAddr},                  addr => $listener->{LocalAddr},
124                  $listener->{LocalPort},                  port => $listener->{LocalPort},
125                  $listener->{Timeout};                  timeout => $listener->{Timeout},
126                    params => { %params },
127            });
128    
129          while(1) {          while(1) {
130    
131                  # wait for any request (RRQ or WRQ)                  # wait for any request (RRQ or WRQ)
132                  if(my $request = $listener->waitRQ()) {                  if(my $request = $listener->waitRQ()) {
133                            server->refresh;
134                          tftp_request $request;                          tftp_request $request;
135                  } elsif ( my $error = Net::TFTPd->error ) {                  } elsif ( my $error = Net::TFTPd->error ) {
136                          warn $error;                          warn $error;
Legend:
Removed from v.141  
changed lines
  Added in v.261
  ViewVC Help
Powered by ViewVC 1.1.26