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

之前写了一篇关于 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);
    }

 

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

 

请您到ITEYE看我的原创：http://cuisuqiang.iteye.com

或支持我的个人博客，地址：http://www.javacui.com

 

评论

1 楼 cuisuqiang 2012-08-24  

不好意思经过测试，以上代码存在问题！
由于TCP传输可能不是一次性传输的，另外，运行设备在一些情况下会产生垃圾数据，因此必须对接收的数据进行验证，所以在复写MessageLengthDecoder接口时，不能过分改动：

  public static class SnmpMesssageLengthDecoderMe implements MessageLengthDecoder {
	  
	    public int getMinHeaderLength() {
	      return MIN_SNMP_HEADER_LENGTH + 2;
	    }
	    
	    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() + 2;
	        MessageLength messageLength = new MessageLength(hl, ml);
	        return messageLength;
	    }
  }

我们只改动头字节长度一项，另外getMessageLength方法也稍微修改


而在调用方法时传递的字节流我们把头给截取即可

// TODO 信息转换
byte[] btnew = new byte[buf.length];
System.arraycopy(buf, 2, btnew, 0, buf.length - 2);
messageLength = messageLengthDecoder.getMessageLength(ByteBuffer.wrap(btnew));