如何在TFTP客户端和服务器通信中结束小端到大端问题。

时间:2011-09-12 12:23:41

标签: c++ endianness tftp

我正在为cPP中的TFTP客户端创建WRQ数据包。该代码在Little endian系统(PC)中运行良好,并且在创建数据包时遇到Big Endian系统的问题。

代码是

  #define TFTP_OPCODE_READ     1
  #define TFTP_OPCODE_WRITE    2
  #define TFTP_OPCODE_DATA     3
  #define TFTP_OPCODE_ACK      4
  #define TFTP_OPCODE_ERROR    5

  #define cTFTPPacket_MAX_SIZE 1024
  #define cTFTPPacket_DATA_SIZE 512

   #define TFTP_DEFAULT_TRANSFER_MODE "octet"           //"netascii", "octet", or "mail"

   typedef unsigned char BYTE;
   typedef unsigned short WORD;    

/////////////////////////      below is Packet.cpp file       ///////

bool cTFTPPacket::addByte(BYTE b) 
{
         if (mCurPacketSize >= cTFTPPacket_MAX_SIZE) 
          {
                return false;
          }
        mData[mCurPacketSize] = (unsigned char)b;
        mCurPacketSize++;
        return true;
}

 bool cTFTPPacket::addWord(WORD w) 
{
      if (!addByte(*(((BYTE*)&w)+1)))
    { 
    return false;
    }
    return (!addByte(*((BYTE*)&w)));
}

bool cTFTPPacket::addString(char* str) 
{
    int n = strlen(str);
    int i=0;
    for (i=0;i<n;i++) 
    {
      if (!addByte(*(str + i))) 
      {
          return false;
      }
    }
    return true;
}

预期的数据包是

0x00 0x02 string 0x00 octet 0x00

获得big endian

0x02 0x00 string 0x00 octet 0x00

创建数据包的代码是

 bool cTFTPPacket::createWRQ(char* filename) 
 {
 /*      structure is the same as RRQ  */
   clear();
   addWord(TFTP_OPCODE_WRITE);
   addString(filename);
   addByte(0);
   addString(TFTP_DEFAULT_TRANSFER_MODE);
   addByte(0);
   return true;
 }

  bool cTFTPPacket::createACK(int packet_num)
 {
   clear();
   addWord(TFTP_OPCODE_ACK);
   addWord(packet_num);
   return true;
 }


   bool cTFTPPacket::createData(int block, char* mData, int data_size) 
   {
    /*         2 bytes    2 bytes       n bytes
    ----------------------------------------
     DATA  | 03    |   Block #  |    Data    |
    ---------------------------------------- */
   clear();                     // to clean the memory location
   addWord(TFTP_OPCODE_DATA);
   addWord(block);
   addMemory(mData, data_size);
   return true;
 }

 bool cTFTPPacket::addMemory(char* buffer, int len) 
 {
   bool oStatus=false;
   if (mCurPacketSize + len >= cTFTPPacket_MAX_SIZE)  
   {
      cout<<("Packet max size exceeded");
      oStatus= false;
   }
   else
   {
   memcpy(&(mData[mCurPacketSize]), buffer, len);
   mCurPacketSize += len;
   oStatus= true;
    }
    return oStatus;
  }


      BYTE cTFTPPacket::getByte(int offset) 
  {
       return (BYTE)mData[offset];
  }

 WORD cTFTPPacket::getWord(int offset) 
{
    WORD hi = getByte(offset);
    //WORD lo = getByte(offset + 1);
    WORD lo = getByte(offset + 1);
    return ((hi<<8)|lo);
}

WORD cTFTPPacket::getNumber() 
{
    if (this->isData() || this->isACK()) 
    {
        return this->getWord(2);
    } 
    else        
    {
        return 0;
    }
}

 bool cTFTPPacket::getString(int offset, char* buffer, int len)
 {
bool oStatus=false;
    if (offset > mCurPacketSize)
    {
        oStatus=false;
    }
    else if (len < mCurPacketSize - offset) 
    {
        oStatus= false;
    }
    else
    {
    memcpy(buffer, &(mData[offset]), mCurPacketSize - offset);
    oStatus= true;
    }
     return oStatus;
 }

 bool cTFTPPacket::createError(int error_code, char* message) {

 /*        2 bytes  2 bytes        string    1 byte
      ----------------------------------------
    ERROR | 05    |  ErrorCode |   ErrMsg   |   0  |
      ----------------------------------------  */
    clear();
    addWord(TFTP_OPCODE_ERROR);
    addWord(error_code);
    addString(message);
    addByte(0);
    return true;

 }

 int cTFTPPacket::getSize() 
 {
    return mCurPacketSize;
 }

  bool cTFTPPacket::setSize(int size) 
  {
    if (size <= cTFTPPacket_MAX_SIZE) 
    {
            mCurPacketSize = size;
            return true;
    }
    else 
    {
            return false;
    }
  }

 bool cTFTPPacket::isRRQ() 
 {
    return (this->getWord(0) == TFTP_OPCODE_READ);
 }

  bool cTFTPPacket::isWRQ() 
  {
     return (this->getWord(0) == TFTP_OPCODE_WRITE);
   }

   bool cTFTPPacket::isACK() 
   {
     return (this->getWord(0) == TFTP_OPCODE_ACK);
     }
    bool cTFTPPacket::isData() {
      return (this->getWord(0) == TFTP_OPCODE_DATA);
    }

    bool cTFTPPacket::isError() 
    {
    return (this->getWord(0) == TFTP_OPCODE_ERROR);
     }

    void cTFTPPacket::clear()      
    {
         mCurPacketSize = 0;
         memset(mData, mCurPacketSize, cTFTPPacket_MAX_SIZE);
     }

     unsigned char* cTFTPPacket::getData(int offset) 
    {
      return &(mData[offset]);
      }

    bool cTFTPPacket::copyData(int offset, char* dest, int length) 
    {
     bool oStatus=false;
         if (offset > this->getSize()) 
        {
        oStatus= false;
    }
    else if (length < (this->getSize() - offset)) 
    {
        oStatus= false; 
    }
    else
    {
        memcpy(dest, &(mData[offset]), (this->getSize()-offset));
        oStatus= true;
    }
    return oStatus;
 }

  void cTFTPPacket::dumpData() {
    xRM_DEBUG("\n--------------DATA DUMP---------------------\n");
    xRM_DEBUG("Size: " << mCurPacketSize );
    for (int i = 0; i < mCurPacketSize; i++) 
    {
            xRM_DEBUG(mData[i]);
            cout<<mData[i];
    }
   }

    cTFTPPacket::~cTFTPPacket() {

     }

我认为在Addword中出现问题,并让Word如何解决这个问题。 还有一件事我的服务器是小端(普通的linux机器)和硬件是大端

3 个答案:

答案 0 :(得分:1)

当您处理其中一台机器是大端的可能性时,您需要在发送时将所有内容转换为网络字节顺序。

您需要的功能是htonlhtons(主机到网络长/短)

接收数据时,使用反向转换为本地机器所需的数据; ntohlntohs。在大端机器上,这些都是无操作的,数据保持不变。在小端机器上,它们转换为正确的(小端)字节顺序。

修改:回复OP的评论如下:

您的WORDunsigned short。这意味着您需要使用htons()函数中的addWord()进行转换:

bool cTFTPPacket::addWord(WORD w) 
{
    w = htons(w);
    if (!addByte(*(((BYTE*)&w)+1)))
    { 
        return false;
    }
    return (!addByte(*((BYTE*)&w)));
}

然后在getWord()函数中反转该过程:

WORD cTFTPPacket::getWord(int offset) 
{
    WORD hi = getByte(offset);
    //WORD lo = getByte(offset + 1);
    WORD lo = getByte(offset + 1);
    return nstoh((hi<<8)|lo);
}

答案 1 :(得分:0)

传统方法是使用htons(将16位值从主机转换为网络字节顺序)和ntohs(从网络转换为主机字节顺序);对于32位值,类似ntohlhtonl

答案 2 :(得分:0)

“传统”方法是假装只有两个 可能的排序(虽然我已经看过至少三个),以及所有 机器是8位字节的二进制补码。更好的方法是 逻辑地处理数字格式。对于未签名的,例如:

void
insertUnsigned16Bits( char* dest, unsigned value )
{
    *dest ++ = (value >> 8) & 0xFF;
    *dest ++ = (value     ) & 0xFF;
}

unsigned
extractUnsigned16Bits( char const* source )
{
    unsigned result = 0;
    result |= (*source ++ & 0xFF) << 8;
    result |= (*source ++ & 0xFF);
    return result;
}

这很容易扩展到任何大小的无符号整数 在你的机器上显示。

对于带符号输出,如果格式指定2的补码,则只需转换 未签名 - 标准要求此类转换才能正确执行 事情。在输入上,要真正便携,你必须阅读更大的 键入,测试您读取的无符号值是否大于 最大有符号值,如果是,则减去相应的无符号值 最大;然而,在大多数机器上,只是阅读一个未签名的 转换为签名类型将起作用。