TFTP源代码

最新推荐文章于 2024-04-30 01:44:13 发布
最新推荐文章于 2024-04-30 01:44:13 发布
阅读量8.9k 收藏 14
点赞数
分类专栏： 嵌入式技术 文章标签： cmd file buffer socket null byte

最近在研究计算机网络，TFTP协议，简单文件传输协议，是基于UDP协议的。协议本身非常简单，很像停止等待协议，即服务器发一次数据，然后等待客户端应答，然后继续发，继续应答...TFTP主要有两种模式，netascii和octet，数据每次传512字节，报文编码从1开始，打算把TFTP协议改造一下，加入些自己的东西，用于自己的程序，参考了下别人的源码，如下

/**************************************************
* TFTP client compatible with RFC-1350
* compile under visiual c++ or borland c++
* author email: helloworld1@163.com
***************************************************/

#define _VC /* if compile under visiual c++ else undefine this*/
#include <stdafx.h>
#include <stdio.h>
#include <winsock.h>
#include <conio.h>

#ifndef MAKEWORD
#define MAKEWORD(l,h) ((WORD)(((BYTE)(l))|(((WORD)(BYTE)(h))<<8)))
#endif
#define WSA_MAJOR_VERSION 1
#define WSA_MINOR_VERSION 1
#define WSA_VERSION MAKEWORD(WSA_MAJOR_VERSION, WSA_MINOR_VERSION)

/* read/write request packet format
2 bytes     string    1 byte     string   1 byte
------------------------------------------------
| Opcode | Filename |   0 |    Mode    |   0 |
------------------------------------------------

*/
#define TFTP_RRQ 1   /*Read request (RRQ)*/
#define TFTP_WRQ 2   /*Write request (WRQ) */

/* DATA packet format
2 bytes     2 bytes      n bytes
----------------------------------
| Opcode |   Block # |   Data     |
----------------------------------
*/
#define TFTP_DATA 3 /*Data (DATA)*/

/* ACK packet format
2 bytes     2 bytes
---------------------
| Opcode |   Block # |
---------------------
*/
#define TFTP_ACK 4   /*Acknowledgment (ACK)*/

/*ERROR packet format
2 bytes     2 bytes      string    1 byte
-----------------------------------------
| Opcode | ErrorCode |   ErrMsg   |   0 |
-----------------------------------------
*/
#define TFTP_ERROR 5 /*Error (ERROR)*/

#define TFTP_NETASCII 0
#define TFTP_OCTET 1
#define TFTP_WSTAT_FIRSTACK 0
#define TFTP_WSTAT_NEXTACK 1
#define TFTP_WSTAT_LASTACK 2
#define MAX_RETRY 3
#define TFTP_NOTEND_DATALEN 512+2+2

#ifdef _VC
#pragma comment( lib, "Wsock32.lib" )
#endif

typedef void (* CMDFUNC)(char [][256],int pcount);
typedef struct _cmdnum{
char *cmd;
int num;
int paramcount;
CMDFUNC callback;
}CMDNUM,*PCMDNUM;

void connectto(char cmd[][256],int pcount);
void setoctet(char cmd[][256],int pcount);
void setascii(char cmd[][256],int pcount);
void quit(char cmd[][256],int pcount);
void showhelp(char cmd[][256],int pcount);
void test(char cmd[][256],int pcount);
void getfile(char cmd[][256],int pcount);
void putfile(char cmd[][256],int pcount);
int stripcmd(char *s,char cmd[][256]);
void parsecmd(char *s);
int getcmdnum(char *s);
int makereq(char type,int mode,char *filename,char *buffer,int size);
int makeack(unsigned short num,char *buffer,int size );
void showsysinfo();

CMDNUM cmdlist[] = {
{"help",1,0,showhelp},
{"exit",2,0,quit},
{"test",3,0,test},
{"get",4,1,getfile},
{"put",5,2,putfile},
{"octet",6,0,setoctet},
{"ascii",7,0,setascii},
{"connect",8,1,connectto}};

char *helptext = "help: show this text/n/
             exit: exit pragram/n/
             ctet: set file mode to octet/n/
             ascii: set file mode to netascii/n/
             connect remoteip: connect to server/n/
             get filename: get file from server/n/
             put localname remotefilename: upload file to server/n";

SOCKET sock = INVALID_SOCKET;
char desthost[256] = "127.0.0.1";
int filemode = TFTP_OCTET;

int main(int argc, char* argv[])
{
  
   char cmd[256];
   WSADATA stWSAData;
   int ret = 0;
   sockaddr_in addr;
   showsysinfo();
   if(WSAStartup(WSA_VERSION, &stWSAData)!=0)
   {
    printf("Can't start Socket /n");
    exit(0);
   }
  
   sock = socket(PF_INET,SOCK_DGRAM,0);
   if(sock==INVALID_SOCKET)
   {
    printf("Can't create socket /n");
    exit(0);
   }
   addr.sin_family = PF_INET;
   addr.sin_port = INADDR_ANY;
   addr.sin_addr.s_addr = INADDR_ANY;
   if(bind(sock,(struct sockaddr *)&addr,sizeof(addr))!=0)
   {
    printf("Can't bind socket /n");
    exit(0);
   }
   while(true)
   {
    fflush( stdin );
    printf("#");
    gets(cmd);
    parsecmd(cmd);
   }
  
   return 0;
}

void showsysinfo()
{
   printf("TFTP client version 1.0/n");
  
}
int stripcmd(char *s,char cmd[][256])
{
   int i=0;
   char *token=NULL;
   char seps[] = " ,/t/n";
   token = strtok( s, seps );
   while(token!=NULL)
   {
    if (i>2) break;
    strcpy(cmd[i],token);
    token = strtok( NULL, seps );
    i++;
   }
   return i;
}


int getcmdnum(char *s)
{
   int i = 0;
   for(i=0;i<sizeof(cmdlist)/sizeof(CMDNUM);i++)
   {
    if(stricmp(s,cmdlist[i].cmd)==0)
    {
     return i;
    }
   }
   return -1;
}

void parsecmd(char *s)
{
   char cmd[3][256];
   int pcount = 0;
   int num = -1;
   pcount = stripcmd(s,cmd);
   num=getcmdnum(cmd[0]);
   if(num==-1)
   {
    printf("No such commond /n");
    return;
   }
   else
   {
    cmdlist[num].callback(cmd,pcount-1);
   }
}

void quit(char cmd[][256],int pcount)
{
   printf("exit to system /n");
   closesocket(sock);
   exit(0);
}

void showhelp(char cmd[][256],int pcount)
{
   printf(helptext);
}

void test(char cmd[][256],int pcount)
{
}

void setoctet(char cmd[][256],int pcount)
{
   filemode = TFTP_OCTET;
   printf("Set file mode to octet/n");
}

void setascii(char cmd[][256],int pcount)
{
   filemode = TFTP_NETASCII;
   printf("Set file mode to netascii/n");
}

void connectto(char cmd[][256],int pcount)
{
   if(pcount<1)
   {
    printf("usage: connect remoteip /n");
    return;
   }
   strcpy(desthost,cmd[1]);
}

int makeack(unsigned short num,char *buffer,int size )
{
   int pos = 0;
   buffer[pos] = 0;
   pos++;
   buffer[pos] = TFTP_ACK;
   pos++;
   buffer[pos] = (char)(num>>8);
   pos++;
   buffer[pos] = (char)num;
   pos++;
   return pos;
}

int makereq(char type,int mode,char *filename,char *buffer,int size)
{
   int pos = 0;
   unsigned int i = 0;
   char s[32] = "";
   if(mode==TFTP_NETASCII)
    strcpy(s,"netascii");
   else
    strcpy(s,"octet");
   buffer[pos] = 0;
   pos++;
   buffer[pos] = type;
   pos++;
   for(i=0;i<strlen(filename);i++)
   {
    buffer[pos] = filename[i];
    pos++;
   }
   buffer[pos] = 0;
   pos++;
   for(i=0;i<strlen(s);i++)
   {
    buffer[pos] = s[i];
    pos++;
   }
   buffer[pos] = 0;
   pos++;
   return pos;
}

int makedata(int num,char *data,int datasize,char *buffer,int bufsize)
{
   int pos = 0;
   buffer[pos] = 0;
   pos++;
   buffer[pos] = TFTP_DATA;
   pos++;
   buffer[pos] = (char)(num>>8);
   pos++;
   buffer[pos] = (char)num;
   pos++;
   memcpy(&buffer[pos],data,datasize);
   pos = pos + datasize;
   return pos;
  
}

void getfile(char cmd[][256],int pcount)
{
   char sendbuf[1024] = {0};
   char recvbuf[1024] = {0};
   sockaddr_in addr;
   sockaddr_in from;
   int fromlen = 0;
   int ret = 0;
   int len = 0 ;
   fd_set fdr;
   int retry = 0;
   struct timeval timeout = {5,0};
   int stat = 0;
   int lastdata = 0;
   FILE *file;
   int flen = 0;
   int c;
   if(pcount!=1)
   {
    printf("usage: get filename/n");
    return;
   }
   if((file=fopen(cmd[1],"r"))!=NULL)
   {
    printf("File %s already exits,overwrite? y/n ",cmd[1]);
    while(true)
    {
     c = getch();
     if('Y'==toupper(c))
     {
      printf("/n");
      fclose(file);
      break;
     }
     else if('N'==toupper(c))
     {
      printf("/n");
      fclose(file);
      return;
     }
    }
   }
   if((file=fopen(cmd[1],"w+b"))==NULL)
   {
    printf("Can't create file/n");
    return;
   }
   len = makereq(TFTP_RRQ,filemode,cmd[1],sendbuf,sizeof(sendbuf));
   addr.sin_family =PF_INET;
   from.sin_family =PF_INET;
   addr.sin_port = htons(69);
   addr.sin_addr.s_addr   = inet_addr(desthost);
   ret = sendto(sock,sendbuf,len,0,(sockaddr *)&addr,sizeof(addr));
   while(true)
   {
    FD_ZERO(&fdr);
    FD_SET(sock, &fdr);
    ret = select(sock, &fdr, NULL,NULL, &timeout);
    if(SOCKET_ERROR==ret)
    {
     printf("Socket error /n");
     fclose(file);
     return;
    }
    else if(0==ret)
    {
     if(MAX_RETRY==retry)
     {
      printf("Time Out /n");
      fclose(file);
      return;
     }
     sendto(sock,sendbuf,len,0,(sockaddr *)&addr,sizeof(addr));
     retry++;
    }
    else
    {
     if (FD_ISSET(sock,&fdr))
     {
      retry = 0;
      fromlen = sizeof(sockaddr);
      ret = recvfrom(sock,recvbuf,sizeof(recvbuf),0,(sockaddr *)&from,&fromlen);
      if(TFTP_ERROR==recvbuf[1])
      {
       fclose(file);
       printf("Error %d: %s /n",recvbuf[3],&recvbuf[4]);
       return;
      }
      if(0==stat)
      {
       addr.sin_port = from.sin_port ;
       stat = 1;
      }
      if(TFTP_DATA==recvbuf[1])
      {
       lastdata = recvbuf[2]*256 + recvbuf[3];
       len = makeack(lastdata,sendbuf,sizeof(sendbuf));
       sendto(sock,sendbuf,len,0,(sockaddr *)&addr,sizeof(addr));
       if(ret<TFTP_NOTEND_DATALEN)
       {
        fwrite(&recvbuf[4],1,ret-4,file);
        flen = flen + ret -4;
        fclose(file);
        printf("total %d byte received/n",flen);
        return;
       }
       else
       {
        fwrite(&recvbuf[4],1,512,file);
        flen = flen + 512;
        printf("%d byte received/r",flen);
       }
      }
     }
    }
   
   }
}

void putfile(char cmd[][256],int pcount)
{
char sendbuf[1024] = {0};
char recvbuf[1024] = {0};
char databuf[1024] = {0};
sockaddr_in addr;
sockaddr_in from;
int fromlen = 0;
int ret = 0;
int len = 0 ;
fd_set fdr;
int retry = 0;
struct timeval timeout = {5,0};
int stat = TFTP_WSTAT_FIRSTACK;
int lastack= 0;
FILE *file;
int flen = 0;
int blocknum = 0;
size_t rlen = 0;
if(pcount!=2)
{
   printf("usage: put localfilename remotefilename /n");
   return;
}
if((file=fopen(cmd[1],"r"))==NULL)
{
   printf("File %s not found /n",cmd[1]);
   return;
}
len = makereq(TFTP_WRQ,filemode,cmd[2],sendbuf,sizeof(sendbuf));
addr.sin_family =PF_INET;
addr.sin_port = htons(69);
addr.sin_addr.s_addr   = inet_addr(desthost);
ret = sendto(sock,sendbuf,len,0,(sockaddr *)&addr,sizeof(addr));
if((file=fopen(cmd[1],"r"))==NULL)
{
   printf("Can't Open file %s/n",cmd[1]);
   return;
}
while(true)
{
   FD_ZERO(&fdr);
   FD_SET(sock, &fdr);
   ret = select(sock, &fdr, NULL,NULL, &timeout);
   if(SOCKET_ERROR==ret)
   {
    printf("Socket error /n");
    fclose(file);
    return;
   }
   else if(0==ret)
   {
    if(MAX_RETRY==retry)
    {
     printf("Time Out /n");
     fclose(file);
     return;
    }
    sendto(sock,sendbuf,len,0,(sockaddr *)&addr,sizeof(addr));
    retry++;
   }
   else
   {
    retry = 0;
    fromlen = sizeof(sockaddr);
    ret = recvfrom(sock,recvbuf,sizeof(recvbuf),0,(sockaddr *)&from,&fromlen);
    if(TFTP_ERROR==recvbuf[1])
    {
     fclose(file);
     printf("Error %d: %s /n",recvbuf[3],&recvbuf[4]);
     return;
    }
    if(TFTP_ACK==recvbuf[1])
    {
     lastack = recvbuf[2]*256 + recvbuf[3];
     switch(stat)
     {
     case TFTP_WSTAT_FIRSTACK:
      if(0==lastack)
      {
       stat = TFTP_WSTAT_NEXTACK;
       addr.sin_port = from.sin_port ;
       rlen = fread(databuf,1,512,file);
       flen = flen + rlen;
       if(rlen<512 && feof(file))
       {
        stat = TFTP_WSTAT_LASTACK;
       }
       else if(ferror(file))
       {
        printf("Error: read file/n");
        fclose(file);
        return;
       }
       blocknum++;
       len = makedata(blocknum,databuf,rlen,sendbuf,sizeof(sendbuf));
       sendto(sock,sendbuf,len,0,(sockaddr *)&addr,sizeof(addr));
       printf("%d byte send/r",flen);
      }
      else
      {
       fclose(file);
       printf("Error Ack Number");
       return;
      }
      break;
     case TFTP_WSTAT_NEXTACK:
      if(lastack==blocknum)
      {
       rlen = fread(databuf,1,512,file);
       flen = flen + rlen;
       if(rlen<512 && feof(file))
       {
        stat = TFTP_WSTAT_LASTACK;
       }
       else if(ferror(file))
       {
        printf("Error: read file/n");
        fclose(file);
        return;
       }
       blocknum++;
       len = makedata(blocknum,databuf,rlen,sendbuf,sizeof(sendbuf));
       sendto(sock,sendbuf,len,0,(sockaddr *)&addr,sizeof(addr));
       printf("%d byte send/r",flen);
      }
      else
      {
       fclose(file);
       printf("Error Ack Number");
       return;
      }
      break;
     case TFTP_WSTAT_LASTACK:
      if(lastack==blocknum)
      {
       printf("%d byte send/n",flen);
       return;
      }
      else
      {
       fclose(file);
       printf("Error Ack Number");
       return;
      }
      break;
     }
    }
   }
}
}

zhwfnh
关注
  • 0
    点赞
  • 踩
  • 14
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
专栏目录
tftp 源代码
05-02
python 实现的tftp源代码，可使用，经过测试上传和下载没问题
tftp源代码（从busybox提取，编译通过）
07-25
从busybox中提取的tftp工具源码，测试通过，可以直接编译。对端开启tftp服务后，可以通过tftp进行文件的上传和下载。下载请注意busybox版tftp工具和常用tftp使用方法不同。
TFTP服务器端源代码，可同时支持多个客户端同时下载上传
fanxiushu的专栏
09-22 4910
By Fanxiushu   转载或引用请注明原作者。 TFTP是个十分简单的基于UDP的传输协议，因为TFTP本身很大的局限性，比如传输的文件本身受32M的限制，传输效率低下等等， 所以平时也没什么大用处。主要用于电脑（或者各种嵌入式设备）启动时候，下载引导程序之用。 最近刚好用上TFTP的服务器端，本来想到协议十分简单，随便从网上下载一个源代码来使用，最终没找到合适的，所以求人还
TFTP in C#
流男@水孩
08-29 2711
The following source code is for a TFTP application. I searched hard and long and all I found was controls for sale. The library I currently have only does file "gets" as that is all I needed for the
tftp协议（客户端和服务端）c语言实现
最新发布
biao2488890051的博客
04-30 248
网上搜索该资料，有很多相关博客，但是很多都是复制来的，代码不全，无法运行。这里给大家推荐这个我找到的开源代码，简单又完整（其它一些开源博客，要么代码量太大太复杂，要么拆成很多个c文件，不便于理解）。
TFTP(1-1示例程序).zip_TFTP c++_tftp_tftp 客户端_tftp客户端_示例程序
07-15
tftp客户端和服务器用于c++语言assssss
TFTP协议
似水年华－－嵌入式Linux
09-22 1242
 1. 目的TFTP是一个传输文件的简单协议，它其于UDP协议而实现，但是我们也不能确定有些TFTP协议是基于其它传输协议完成的。此协议设计的时候是进行小文件传输的。因此它不具备通常的FTP的许多功能，它只能从文件服务器上获得或写入文件，不能列出目录，不进行认证，它传输8位数据。传输中有三种模式：netascii，这是8位的ASCII码形式，另一种是octet，这是8位源数据类型；最后一种
常用ftp、tftp服务器软件
fengruoying93的专栏
12-22 3296
这里分享几个非常不错的FTP服务器软件，分别是Wing FTP Server、Serv-U和Filezilla Server，都支持一键安装，而且使用起来非常容易。 一、FileZilla Server 这是Windows平台下一个免费的FTP服务器软件，非常轻巧灵活，最新版的安装包也就2M左右，功能强大、操作简单、界面清新、实用性强，占用系统资源非常少，如果你想玩玩简单的FTP服务器，那么F...
TFTP客户端源代码
12-18
#include #include #include #include #include #pragma hdrstop #include "Unit1.h" #define TIMEOUT_USECS 1000000 #define MAX_TIMEOUT 32000000 //----------------------------------------------------...
TFTP RFC1350 源代码
03-19
TFTP RFC1350 实现 C++源代码， 服务器端和客户端都有
tftp协议开源代码
07-10
tftp开源的代码，协议本身就很简单，先看看协议，再对代码看看，促进学习
tftp-ftp开源软件
04-17
开源软件，支持tftp和ftp server端、client端，方便用户进行文件上传和下载
tftpd的源码实现
10-22
TFTPd的实现，其中包括了tftp服务端和客户端，以及DHCP服务的实现
tftp超简单的源码，可以看一下！
03-14
tftp简单源码！
tftp_proxy源码
05-23
Tftp ProxyServer 代理服务器 源码 *=========================================================================== * * Project: tftp_proxy, a proxy for TFTP transfers through firewalls * File: tftp_proxy.c * Purpose: tftp protocol relay * *=========================================================================== * This source code is free software; you can redistribute it and/or * modify it under the terms of the European Union Public Licence * as published at http://www.osor.eu/eupl/ * * This code is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
Qt写的TFTP客户端源代码
04-18
用Qt写了一个TFTP的测试程序 可以上传和下载TFTP文件
tftp linux 源代码
01-01
TFTP是一个简单文件传输协议，专门用于在网络上传输文件。...总之，TFTPD的源代码实现了TFTP协议的各种功能，通过阅读和理解源代码，可以深入了解TFTP协议的实现细节，为定制化的开发提供参考和支持。

“相关推荐”对你有帮助么？

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值