git.rot13.org Git - pxelator/blob - bin/tftpd.pl

   1 #!/usr/bin/perl
   2 use strict;
   3 use lib 'lib';
   4 use Net::TFTPd 0.03 qw(%OPCODES);
   5
   6 die "need to run $0 as root like this\nsudo $0\n" unless $< == 0;
   7
   8 sub tftp_dir {
   9         my $dir = 'conf/' . readlink('conf/tftp.dir');
  10         $dir =~ s{[^/]+/\.\./([^/]+)}{$1};
  11         return $dir;
  12 }
  13
  14 my $tftp_dir = tftp_dir;
  15
  16 die "no $tftp_dir\n" unless -e $tftp_dir;
  17
  18 # callback sub used to print transfer status
  19 sub callback
  20 {
  21         my $req = shift;
  22         if($req->{'_REQUEST_'}{'OPCODE'} eq $OPCODES{'RRQ'})
  23         {
  24                 # RRQ
  25                 printf "block: %u\/%u\n", $req->{'_REQUEST_'}{'LASTACK'}, $req->{'_REQUEST_'}{'LASTBLK'};
  26         }
  27         elsif($req->{'_REQUEST_'}{'OPCODE'} eq $OPCODES{'WRQ'})
  28         {
  29                 # WRQ
  30                 printf "block: %u\/%u\n", $req->{'_REQUEST_'}{'LASTBLK'}, $req->{'_REQUEST_'}{'LASTACK'};
  31         } else {
  32                 warn "IGNORED: ", dump( $req );
  33         }
  34 }
  35
  36 # create the listener
  37 my $listener = Net::TFTPd->new(
  38         'RootDir' => $tftp_dir,
  39         'Writable' => 0,
  40         'Timeout' => 3600,
  41         'CallBack' => \&callback,
  42 #       LocalAddr => '10.0.0.100',
  43 #       BlkSize => 8192,
  44 #       BlkSize => 512,
  45         BlkSize => 1456,        # IBM GE seems to be picky
  46         Debug => 99,
  47 ) or die Net::TFTPd->error;
  48 printf "TFTP on %s:%d timeout: %d dir: $tftp_dir\n", $listener->{'LocalAddr'} ? $listener->{'LocalAddr'} : "'any address'",  $listener->{'LocalPort'}, $listener->{'Timeout'};
  49
  50 while(1) {
  51
  52         # wait for any request (RRQ or WRQ)
  53         if(my $request = $listener->waitRQ()) {
  54
  55                 $tftp_dir = tftp_dir;
  56
  57                 if ( $request->{RootDir} ne $tftp_dir ) {
  58                         $request->{RootDir} = $tftp_dir;
  59                         warn "new root: $tftp_dir\n";
  60                 }
  61
  62                 # received request
  63                 printf "Received a %s for file '%s'\n", $OPCODES{$request->{'_REQUEST_'}{'OPCODE'}}, $request->{'_REQUEST_'}{'FileName'};
  64
  65                 # process the request
  66                 if($request->processRQ()) {
  67                         print "OK, transfer completed successfully\n";
  68                 } else {
  69                         warn Net::TFTPd->error;
  70                         $request->processRQ();
  71                 }
  72         } else {
  73                 # request not received (timed out waiting for request etc.)
  74                 warn Net::TFTPd->error;
  75         }
  76
  77 }