Category

Development Platform
Home > SourceCode/Document > OS Develop > View Code
TFTPD.C
Package: ertos.rar　[view]
Upload User: sunrenlu
Upload Date: 2022-06-13
Package Size: 1419k
Code Size: 4k
Category:

OS Develop

Development Platform:

DOS

TFTPD.C：Code Content
  1. /*
  2.  * eRTOS TFTP Server
  3.  * Copyright (c) 2000 Erick Engelke
  4.  */
  5. #include <stdio.h>
  6. #include <ctype.h>
  7. #include <string.h>
  8. #include <memory.h>
  9. #include <rtos.h>
  10. #include <net.h>
  11. #include <tftp.h>
  13. #define TFTP_PORT 69
  14. #define TFTP_BUF_LEN 1024
  16. #define MAXPATHLEN 81
  18. int _tftpd_readonly = 0;
  19. //---------------------------------------------------------------------
  20. // getblock - returns received length
  21. static int getblock( char *filename, int blocknum, void *buffer )
  22. {
  23.     FILE *f;
  24.     int result = 0;
  26.     dos_enter();
  27.     f = fopen( filename, "rb");
  28.     if ( f != NULL ) {
  29.         if ( 0 == fseek( f, (blocknum - 1) * 512, SEEK_SET ))
  30.             result = fread( buffer, 1, 512, f );
  31.         fclose( f );
  32.     }
  33.     dos_exit();
  34.     return( result );
  35. }
  37. // putblock - returns 0 on success
  38. static int putblock( char *filename, int blocknum, void *buffer, int length )
  39. {
  40.     FILE *f;
  41.     int result = 0;
  43.     dos_enter();
  44.     f = fopen( filename, "a+b");
  45.     if ( f != NULL ) {
  46.         if ( 0 == fseek( f, (blocknum - 1) * 512, SEEK_SET ))
  47.             fwrite( buffer, 1, length, f );
  48.         else result = 3;    // disk full or alloc exceeded
  49.         fclose( f );
  50.     } else result = 3;
  51.     dos_exit();
  53.     return( result ? 3 : 0 );
  54. }
  57. #pragma argsused
  58. void tftpdthread( DWORD remotehost )
  59. {
  60.     char *buf;
  61.     unsigned buflen;
  62.     int socklistening;
  63.     static udp_Socket sock;
  64.     DATA_Packet *d;
  65.     ACK_Packet *a;
  66.     RRQ_Packet *r;
  67.     ERROR_Packet *e;
  68.     char filename[ MAXPATHLEN ];
  69.     int block;
  70.     *filename = 0;
  71.     if ( ( buf = kcalloc( TFTP_BUF_LEN, 1 )) == NULL )
  72.         rt_halt( "TFTPD insufficient memory");
  75.     d = a = r = e = buf;
  77.     socklistening = 0;
  78.     do {
  79.         if ( socklistening == 0 ) {
  80.             /* udp_open - use remote_IP=0, bind to first incoming IP address */
  81.             /* if remotehost = 0  listen to anyone */
  82.             /* otherwise directed listen */
  83.             if (!udp_open( &sock, TFTP_PORT, remotehost, -1, NULL ))
  84.                 rt_halt("unable to open TFTP server socket");
  85.             socklistening = 1;
  86.         }
  87.         rt_sleep( 10 );
  88.         tcp_tick( NULL );
  90.         /* look for input */
  91.         if ( (buflen = sock_dataready( &sock )) == 0 ) {
  92.             continue;   /* no input */
  93.         }
  95.         /* we received an TFTP request, process it */
  96.         buflen = sock_fastread( &sock, buf, buflen );
  98.         /* remote GET */
  99.         if ( d->op2 == TFTP_RRQ ) {
  100.             memset( filename, 0, MAXPATHLEN );
  101.             strncpy( filename, r->filename, MAXPATHLEN - 1 );
  103.             d->op2 = TFTP_DATA;
  104.             block = 1;
  105.             buflen = sizeof( ACK_Packet ) + getblock( filename, block, d->data );
  106.             d->block = intel16( block );
  107.         }
  108.         else if ( d->op2 == TFTP_ACK ) {
  109.             d->op2 = TFTP_DATA;
  110.             block = intel16( d->block ) + 1;
  111.             buflen = sizeof( ACK_Packet ) + getblock( filename, block, d->data );
  112.             d->block = intel16( block );
  113.         }
  114.         /* remote PUT */
  115.         else if ( d->op2 == TFTP_WRQ ) {
  116.             memset( filename, 0, MAXPATHLEN );
  117.             strncpy( filename, r->filename, MAXPATHLEN - 1 );
  118.             unlink( filename );     // reset existing file
  119.             d->op2 = TFTP_ACK;
  120.             d->block = intel16( 0 );
  121.             buflen = sizeof( ACK_Packet );
  122.         }
  123.         else if ( d->op2 == TFTP_DATA ) {
  124.             if ( _tftpd_readonly != 0 ) {
  125.                e->op2 = TFTP_ERROR;
  126.                e->err1 = 0;
  127.                e->err2 = 3;
  128.                e->errmsg[0] = 0;
  129.                buflen = sizeof( ERROR_Packet );
  130.             } else {
  131.                 block = intel16( d->block );
  132.                 if ( putblock( filename, block, d->data, buflen - sizeof( ACK_Packet )) != 0 ) {
  133.                     e->op2 = TFTP_ERROR;
  134.                     e->err1 = 0;
  135.                     e->err2 = 3;
  136.                     e->errmsg[0] = 0;
  137.                     buflen = sizeof( ERROR_Packet );
  138.                 } else {
  139.                     d->op2 = TFTP_ACK;
  140.                     buflen = sizeof( ACK_Packet );
  141.                     // d->block = intel16( block ) ; // redundant
  142.                 }
  143.             }
  144.         }
  147.         sock_fastwrite( &sock, buf, buflen );
  148.         sock_close( &sock );
  150.         /* open next tftpd server */
  151.         socklistening = 0;
  153.     } while( 1 );
  154. }