feat: add device discover response

Signed-off-by: wei <[email protected]>

Author: instpe

mdtp-client/src/main/java/io/github/protocol/mdtp/client/MdtpClient.java

@@ -1,12 +1,14 @@
 package io.github.protocol.mdtp.client;
 
+import io.github.protocol.mdtp.common.codec.MdtpDecoder;
 import io.github.protocol.mdtp.common.model.CDATHeader;
+import io.github.protocol.mdtp.common.model.CDATHeaderFactory;
 import io.github.protocol.mdtp.common.model.DeviceDiscoveryRequest;
 import io.github.protocol.mdtp.common.model.MdtpPacket;
-import io.github.protocol.mdtp.common.model.MessageBodyHeader;
 import io.netty.bootstrap.Bootstrap;
 import io.netty.channel.ChannelFuture;
 import io.netty.channel.ChannelInitializer;
+import io.netty.channel.ChannelPipeline;
 import io.netty.channel.EventLoopGroup;
 import io.netty.channel.nio.NioEventLoopGroup;
 import io.netty.channel.socket.SocketChannel;
@@ -42,6 +44,8 @@ public void start() throws Exception {
                 .handler(new ChannelInitializer<SocketChannel>() {
                     @Override
                     protected void initChannel(SocketChannel ch) throws Exception {
+                        ChannelPipeline pipeline = ch.pipeline();
+                        pipeline.addLast(new MdtpDecoder());
                     }
                 });
         this.channelFuture = bootstrap.connect().sync();
@@ -66,8 +70,7 @@ public void close() throws IOException {
     public void sendDeviceDiscoveryRequest(int[] deviceTypes) {
         log.info("start to send device discovery request.");
         DeviceDiscoveryRequest request = new DeviceDiscoveryRequest();
-        request.setMessageBodyHeader(MessageBodyHeader.DEVICE_DISCOVERY_REQUEST);
-        request.setRequestId(request.generateRequestId());
+        request.setRequestId(request.generateId());
 
         if (deviceTypes == null) {
             request.setDeviceTypeCount((byte) 0);
@@ -79,14 +82,7 @@ public void sendDeviceDiscoveryRequest(int[] deviceTypes) {
             request.setDeviceTypes(deviceTypes);
         }
 
-        CDATHeader cdatHeader = new CDATHeader();
-        cdatHeader.setFormatType((byte) 0x02);
-        cdatHeader.setProtocolVersion((byte) 1);
-        cdatHeader.setMessageLength((short) 0);
-        cdatHeader.setTimestamp(System.currentTimeMillis());
-        cdatHeader.setFlags((byte) 0b01100000);
-        cdatHeader.setSequenceNumber(0);
-        cdatHeader.setLogicalChannelId(0);
+        CDATHeader cdatHeader = CDATHeaderFactory.createDeviceDiscoveryCDATHeader();
 
         MdtpPacket packet = new MdtpPacket();
         packet.setHeader(cdatHeader);
@@ -95,6 +91,6 @@ public void sendDeviceDiscoveryRequest(int[] deviceTypes) {
         packet.setSignature(null);
 
         this.channelFuture.channel().writeAndFlush(packet.toByteBuf());
-        log.info("send device discovery request success: " + packet);
+        log.info("send device discovery request success.");
     }
 }

mdtp-common/src/main/java/io/github/protocol/mdtp/common/codec/DeviceDiscoveryReponseDecoder.java

@@ -0,0 +1,13 @@
+package io.github.protocol.mdtp.common.codec;
+
+import io.github.protocol.mdtp.common.model.AbstractMessageBody;
+import io.github.protocol.mdtp.common.model.DeviceDiscoveryResponse;
+import io.netty.buffer.ByteBuf;
+
+public class DeviceDiscoveryReponseDecoder implements MessageBodyDecoder {
+    @Override
+    public AbstractMessageBody handle(ByteBuf in) {
+        return DeviceDiscoveryResponse.readFromBuffer(in);
+    }
+
+}

mdtp-common/src/main/java/io/github/protocol/mdtp/common/codec/MdtpDecoder.java

@@ -24,11 +24,10 @@ protected void decode(ChannelHandlerContext ctx, ByteBuf in, List<Object> out) t
         MessageBodyHeader messageBodyHeader = MessageBodyHeader.readByteBuf(in);
         MessageBodyDecoder messageDecode = MessageDecoderFactory.getDecoder(messageBodyHeader);
         AbstractMessageBody messageBody = messageDecode.handle(in);
-        messageBody.setMessageBodyHeader(messageBodyHeader);
-
         mdtpPacket.setHeader(header);
         mdtpPacket.setBody(messageBody);
 
         out.add(mdtpPacket);
+        log.info("decode packet success: {}", mdtpPacket);
     }
 }

mdtp-common/src/main/java/io/github/protocol/mdtp/common/codec/MessageDecoderFactory.java

@@ -11,6 +11,7 @@ public class MessageDecoderFactory {
 
     static {
         decoders.put(MessageBodyHeader.DEVICE_DISCOVERY_REQUEST.toShort(), new DeviceDiscoveryRequestDecoder());
+        decoders.put(MessageBodyHeader.DEVICE_DISCOVERY_RESPONSE.toShort(), new DeviceDiscoveryReponseDecoder());
     }
 
     public static MessageBodyDecoder getDecoder(MessageBodyHeader header) {

mdtp-common/src/main/java/io/github/protocol/mdtp/common/handler/DeviceDiscoveryRequestHandler.java

@@ -0,0 +1,36 @@
+package io.github.protocol.mdtp.common.handler;
+
+import io.github.protocol.mdtp.common.model.Attributes;
+import io.github.protocol.mdtp.common.model.CDATHeader;
+import io.github.protocol.mdtp.common.model.CDATHeaderFactory;
+import io.github.protocol.mdtp.common.model.Device;
+import io.github.protocol.mdtp.common.model.DeviceDiscoveryRequest;
+import io.github.protocol.mdtp.common.model.DeviceDiscoveryResponse;
+import io.github.protocol.mdtp.common.model.MdtpPacket;
+import io.netty.channel.ChannelHandlerContext;
+import lombok.extern.slf4j.Slf4j;
+
+@Slf4j
+public class DeviceDiscoveryRequestHandler implements MessageBodyHandler {
+
+    @Override
+    public void handle (ChannelHandlerContext ctx, MdtpPacket requestPacket) {
+        log.info("start to send device discovery response.");
+        DeviceDiscoveryRequest deviceDiscoveryRequest = (DeviceDiscoveryRequest) requestPacket.getBody();
+        DeviceDiscoveryResponse deviceDiscoveryResponse = new DeviceDiscoveryResponse();
+        deviceDiscoveryResponse.setRequestId(deviceDiscoveryRequest.getRequestId());
+        deviceDiscoveryResponse.setResponseId(deviceDiscoveryResponse.generateId());
+        Device device = ctx.channel().attr(Attributes.DEVICE_KEY).get();
+        deviceDiscoveryResponse.setDevice(device);
+
+        CDATHeader cdatHeader = CDATHeaderFactory.createDeviceDiscoveryCDATHeader();
+
+        MdtpPacket packet = new MdtpPacket();
+        packet.setHeader(cdatHeader);
+        packet.setSecurityHeader(null);
+        packet.setBody(deviceDiscoveryResponse);
+        packet.setSignature(null);
+        ctx.channel().writeAndFlush(packet.toByteBuf());
+        log.info("send device discovery response success.");
+    }
+}

mdtp-common/src/main/java/io/github/protocol/mdtp/common/handler/MessageBodyHandler.java

@@ -0,0 +1,9 @@
+package io.github.protocol.mdtp.common.handler;
+
+import io.github.protocol.mdtp.common.model.MdtpPacket;
+import io.netty.channel.ChannelHandlerContext;
+
+public interface MessageBodyHandler {
+
+    void handle (ChannelHandlerContext ctx, MdtpPacket mdtpPacket);
+}

mdtp-common/src/main/java/io/github/protocol/mdtp/common/handler/MessageHandlerFactory.java

@@ -0,0 +1,19 @@
+package io.github.protocol.mdtp.common.handler;
+
+
+import io.github.protocol.mdtp.common.model.MessageBodyHeader;
+
+import java.util.HashMap;
+import java.util.Map;
+
+public class MessageHandlerFactory {
+    private static final Map<Short, MessageBodyHandler> handlers = new HashMap<>();
+
+    static {
+        handlers.put(MessageBodyHeader.DEVICE_DISCOVERY_REQUEST.toShort(), new DeviceDiscoveryRequestHandler());
+    }
+
+    public static MessageBodyHandler getHandler(MessageBodyHeader header) {
+        return handlers.get(header.toShort());
+    }
+}

mdtp-common/src/main/java/io/github/protocol/mdtp/common/model/AbstractMessageBody.java

@@ -9,9 +9,16 @@
 public abstract class AbstractMessageBody {
     private MessageBodyHeader messageBodyHeader;
 
-    public short generateRequestId() {
+    protected AbstractMessageBody () {
+    }
+
+    protected AbstractMessageBody (MessageBodyHeader messageBodyHeader) {
+        this.messageBodyHeader = messageBodyHeader;
+    }
+
+    public short generateId() {
         UUID uuid = UUID.randomUUID();
-        return (short) (uuid.getLeastSignificantBits() & 0xFFFF);
+        return (short) (uuid.getLeastSignificantBits() & 0x7FFF);
     }
 
     public void writeByteBuf(ByteBuf buffer) {

mdtp-common/src/main/java/io/github/protocol/mdtp/common/model/Address.java

@@ -0,0 +1,36 @@
+package io.github.protocol.mdtp.common.model;
+
+import io.netty.buffer.ByteBuf;
+
+import java.net.InetAddress;
+import java.net.UnknownHostException;
+
+public class Address {
+    public static final byte IPV4_TYPE = 4;
+    public static final byte IPV6_TYPE = 6;
+
+    private final byte type;
+    private final byte[] value;
+
+    public Address(byte type, byte[] value) {
+        this.type = type;
+        this.value = value;
+    }
+
+    public String getIpString() throws UnknownHostException {
+        return InetAddress.getByAddress(value).getHostAddress();
+    }
+
+    public void writeByteBuf(ByteBuf buffer) {
+        buffer.writeByte(type);
+        buffer.writeBytes(value);
+    }
+
+    public static Address readByteBuf(ByteBuf buffer) {
+        byte type = buffer.readByte();
+        int length = (type == IPV4_TYPE) ? 4 : 16;
+        byte[] value = new byte[length];
+        buffer.readBytes(value);
+        return new Address(type, value);
+    }
+}

mdtp-common/src/main/java/io/github/protocol/mdtp/common/model/Attributes.java

@@ -0,0 +1,7 @@
+package io.github.protocol.mdtp.common.model;
+
+import io.netty.util.AttributeKey;
+
+public class Attributes {
+    public static final AttributeKey<Device> DEVICE_KEY = AttributeKey.valueOf("device");
+}