SNMP 修改SNMP4J消息内容 之TCP发送模式

73641 开源 | 2014-11-10

之前写了一篇关于 SNMP 修改SNMP4J消息内容 的博客，但是当时只是针对UDP的，后来应用到TCP上后发现有一些问题，现在说一下解决方法

上一篇连接地址：http://cuisuqiang.iteye.com/blog/1584391 

 

使用TCP发送时，由于Socket不知道数据流的长度，所以要增加数据包长度到包上，这才是真正的需求，是我上次理解错了

增加头信息与上一次一样，直接在发送时增加即可

 

找到类：DefaultTcpTransportMapping

找到发送方法：

public void sendMessage(Address address, byte[] message)

在发送时增加消息头 

  /**
   * Sends a SNMP message to the supplied address.
   * @param address
   *    an <code>TcpAddress</code>. A <code>ClassCastException</code> is thrown
   *    if <code>address</code> is not a <code>TcpAddress</code> instance.
   * @param message byte[]
   *    the message to sent.
   * @throws IOException
   */
  public void sendMessage(Address address, byte[] message)
      throws java.io.IOException
  {
    if (server == null) {
      listen();
    }
 // TODO 增加头字段
 short length = (short)(message.length + 2);
 byte[] btlength = shortToByte(length);
 byte[] relmess = new byte[length];
 System.arraycopy(btlength, 0, relmess, 0, 2);
 System.arraycopy(message, 0, relmess, 2, message.length);
 
 serverThread.sendMessage(address, relmess);
  }

但是在接收时就遇到一个问题，底层接收后会直接验证数据流，如果你增加头信息，就不算是标准SNMP包了，SNMP4J就不会接收不会处理

 

在类：DefaultTcpTransportMapping 中有一个内部类：

class ServerThread implements WorkerTask

方法

private void readMessage(SelectionKey sk, SocketChannel readChannel,
                             TcpAddress incomingAddress) throws IOException

提供对于数据流的处理，处理后将数据流返回到上层进行解析

 

在方法中有这样有这样的代码：

  ByteBuffer byteBuffer = ByteBuffer.wrap(buf);
  byteBuffer.limit(messageLengthDecoder.getMinHeaderLength());

MessageLengthDecoder 接口用于标识SNMP数据包的一些属性解析方法，包括最小头字节和获得 MessageLength 对象

如果数据小于最小字节，那么直接抛弃不解析

同样在TCP发送类中有一个内部类来实现这个接口：

  public static class SnmpMesssageLengthDecoder implements MessageLengthDecoder {
    public int getMinHeaderLength() {
      return MIN_SNMP_HEADER_LENGTH;
    }
    public MessageLength getMessageLength(ByteBuffer buf) throws IOException {
      MutableByte type = new MutableByte();
      BERInputStream is = new BERInputStream(buf);
      int ml = BER.decodeHeader(is, type);
      int hl = (int)is.getPosition();
      MessageLength messageLength = new MessageLength(hl, ml);
      return messageLength;
    }
  }

由于我们增加两个字节的长度，所以要重新写一个实现的接口

  // TODO 自己定义的头解析
  public static class SnmpMesssageLengthDecoderMe implements MessageLengthDecoder {   
     public int getMinHeaderLength() {
       return MIN_SNMP_HEADER_LENGTH + 2;
     }
     public MessageLength getMessageLength(ByteBuffer buf) throws IOException {
          int hl = 4;
          int ml = decodeHeader(buf);
          MessageLength messageLength = new MessageLength(hl, ml);
          return messageLength;
     }     
     public int decodeHeader(ByteBuffer is){
      byte[] bt = is.array();
      byte[] len = new byte[]{ bt[0], bt[1] };
      return byteToShort(len) - 4;
     }     
     /**
   * @功能 字节的转换与短整型
   * @return 短整型
   */
  public synchronized static short byteToShort(byte[] b) {
   short s = 0;
   short s0 = (short) (b[1] & 0xff);// 最低位
   short s1 = (short) (b[0] & 0xff);
   s1 <<= 8;
   s = (short) (s0 | s1);
   return s;
  }
  }

长度就是原来的 +2 ，MessageLength 的属性有两个，按照我的写法写可以解决问题，但是我不明白

int ml = BER.decodeHeader(is, type);

的含义，希望有明白的给说明一下

消息的长度实际上就是我们定义的头表示的长度 -4

这样流读取方法才能把我们发送的数据流完整读取出来

 

上次在转发到上层解析时的方法也要修改，因为现在修改的只是正常读取到所有的流，而返给上层的数据流是要截取掉这两个字节的

    @SuppressWarnings("static-access")
 private void dispatchMessage(TcpAddress incomingAddress,
                                 ByteBuffer byteBuffer, long bytesRead) {
      byteBuffer.flip();
      if (logger.isDebugEnabled()) {
        logger.debug("Received message from " + incomingAddress +
                     " with length " + bytesRead + ": " +
                     new OctetString(byteBuffer.array(), 0,
                                     (int)bytesRead).toHexString());
      }
      ByteBuffer bis;
      if (isAsyncMsgProcessingSupported()) {
        byte[] bytes = new byte[(int)bytesRead - 2];
        System.arraycopy(byteBuffer.array(), 2, bytes, 0, (int)bytesRead - 2);
        bis = ByteBuffer.wrap(bytes);
      }
      else {
       byte[] bytes = new byte[(int)bytesRead - 2];
          System.arraycopy(byteBuffer.array(), 2, bytes, 0, (int)bytesRead - 2);
          bis = ByteBuffer.wrap(bytes);
      }
        
      // TODO 调用解析数据流
      fireProcessMessage(incomingAddress, bis);
    }

这样就可以正常解析增加两个字节的数据包

推荐您阅读更多有关于“ tcp snmp socket UDP 报文 修改 ”的文章

上一篇：SNMP 在Windows上建一个SNMP服务并获得该机器的机器名 下一篇：SNMP4J 服务端连接的超时时间

猜你喜欢

• 11-10HTTP POST数据和数据接收
• 11-10常用SpringJPA主键生成策略，UUID和自增
• 11-10Apache Commons Jar包下载
• 11-10IIS服务器返回Bad Request (Invalid Hostname)解决办法
• 11-10SpringJPA 慎重 findAll() 切记分页

评论：