/[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 207 by dpavlin, Wed Aug 12 22:56:45 2009 UTC revision 544 by dpavlin, Sat Oct 9 13:34:37 2010 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;  use store;
10    
11  use server;  use server;
12    
# Line 43  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          my $ip = $request->{_REQUEST_}->{PeerAddr};          my $ip = $request->{_REQUEST_}->{PeerAddr};
# Line 58  sub tftp_request { Line 56  sub tftp_request {
56          my $file = $request->{'_REQUEST_'}{'FileName'};          my $file = $request->{'_REQUEST_'}{'FileName'};
57          my $opcode = $OPCODES{$request->{'_REQUEST_'}{'OPCODE'}};          my $opcode = $OPCODES{$request->{'_REQUEST_'}{'OPCODE'}};
58    
59          CouchDB::audit('request', {          my $audit = {
60                  ip => $ip,                  ip => $ip,
61                  opcode => $opcode,                  opcode => $opcode,
62                  file => $file,                  path => $file,
63          });                  state => 'start',
64            };
65            store::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                  my $size = -s "$dir/$file";                  my $size = -s "$dir/$file";
72                  CouchDB::audit('completed', { file => $file, size => $size });                  $audit->{state} = 'finish';
73                    $audit->{size} = $size;
74                    store::audit( $opcode, $audit );
75          } else {          } else {
76                  CouchDB::audit('error', { file => $file, error => Net::TFTPd->error });                  $audit->{state} = 'error';
77                  $request->processRQ();                  $audit->{error} = Net::TFTPd->error;
78                    store::audit( $opcode, $audit );
79          }          }
80    
81  }  }
# Line 98  sub start { Line 101  sub start {
101          my $listener = bless {          my $listener = bless {
102                  RootDir => $dir,                  RootDir => $dir,
103    
104                  ACKtimeout  => 4,                  ACKtimeout  => 1, # 4
105                  ACKretries  => 4,                  ACKretries  => 2, # 4
106                  Readable    => 1,                  Readable    => 1,
107                  Writable    => 0,                  Writable    => 0,
108                  Timeout => 3600,                  Timeout => 3600,
109    
110                  CallBack => \&transfer_status,                  CallBack => \&transfer_status,
111  #               BlkSize => 8192,  #               BlkSize => 8192,
112  #               BlkSize => 512,                  BlkSize => 512,         # Dell's RAC doesn't like bigger packets
113                  BlkSize => 1456,        # IBM GE seems to be picky  #               BlkSize => 1456,        # IBM GE seems to be picky
114                  Debug => 99,                  Debug => 99,
115                  %params, # merge user parameters                  %params, # merge user parameters
116                  _UDPSERVER_ => $udpserver,                  _UDPSERVER_ => $udpserver,
# Line 115  sub start { Line 118  sub start {
118    
119          warn 'listener: ',dump( $listener ) if $debug;          warn 'listener: ',dump( $listener ) if $debug;
120    
121          CouchDB::audit( 'start', {          store::audit( 'start', {
122                  addr => $listener->{LocalAddr},                  addr => $listener->{LocalAddr},
123                  port => $listener->{LocalPort},                  port => $listener->{LocalPort},
124                  timeout => $listener->{Timeout},                  timeout => $listener->{Timeout},
# Line 126  sub start { Line 129  sub start {
129    
130                  # wait for any request (RRQ or WRQ)                  # wait for any request (RRQ or WRQ)
131                  if(my $request = $listener->waitRQ()) {                  if(my $request = $listener->waitRQ()) {
132                            server->refresh;
133                          tftp_request $request;                          tftp_request $request;
134                  } elsif ( my $error = Net::TFTPd->error ) {                  } elsif ( my $error = Net::TFTPd->error ) {
135                          warn $error;                          warn $error;
Legend:
Removed from v.207  
changed lines
  Added in v.544
  ViewVC Help
Powered by ViewVC 1.1.26