feat: add device discover request (#11)

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

Author: instpe

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

@@ -1,5 +1,9 @@
 package io.github.protocol.mdtp.client;
 
+import io.github.protocol.mdtp.common.model.CDATHeader;
+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;
@@ -18,6 +22,9 @@ public class MdtpClient implements Closeable {
 
     private EventLoopGroup group;
 
+    private ChannelFuture channelFuture;
+
+
     public MdtpClient(MdtpClientConfig config) {
         this.config = config;
     }
@@ -37,7 +44,7 @@ public void start() throws Exception {
                     protected void initChannel(SocketChannel ch) throws Exception {
                     }
                 });
-        ChannelFuture channelFuture = bootstrap.connect().sync();
+        this.channelFuture = bootstrap.connect().sync();
         if (channelFuture.isSuccess()) {
             log.info("mdtp client started");
         } else {
@@ -55,4 +62,39 @@ public void close() throws IOException {
         this.group.shutdownGracefully();
         log.info("mdtp client closed");
     }
+
+    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());
+
+        if (deviceTypes == null) {
+            request.setDeviceTypeCount((byte) 0);
+        }
+
+        if (deviceTypes != null && deviceTypes.length > 0) {
+            request.setMask((byte) 1);
+            request.setDeviceTypeCount((byte) deviceTypes.length);
+            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);
+
+        MdtpPacket packet = new MdtpPacket();
+        packet.setHeader(cdatHeader);
+        packet.setSecurityHeader(null);
+        packet.setBody(request);
+        packet.setSignature(null);
+
+        this.channelFuture.channel().writeAndFlush(packet.toByteBuf());
+        log.info("send device discovery request success: " + packet);
+    }
 }

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

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

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

@@ -0,0 +1,34 @@
+package io.github.protocol.mdtp.common.codec;
+
+
+import io.github.protocol.mdtp.common.model.AbstractMessageBody;
+import io.github.protocol.mdtp.common.model.CDATHeader;
+import io.github.protocol.mdtp.common.model.MdtpPacket;
+import io.github.protocol.mdtp.common.model.MessageBodyHeader;
+import io.netty.buffer.ByteBuf;
+import io.netty.channel.ChannelHandlerContext;
+import io.netty.handler.codec.ByteToMessageDecoder;
+import lombok.extern.slf4j.Slf4j;
+
+import java.util.List;
+
+@Slf4j
+public class MdtpDecoder extends ByteToMessageDecoder {
+
+    @Override
+    protected void decode(ChannelHandlerContext ctx, ByteBuf in, List<Object> out) throws Exception {
+
+        MdtpPacket mdtpPacket = new MdtpPacket();
+        CDATHeader header = CDATHeader.readByteBuf(in);
+
+        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);
+    }
+}

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

@@ -0,0 +1,8 @@
+package io.github.protocol.mdtp.common.codec;
+
+import io.github.protocol.mdtp.common.model.AbstractMessageBody;
+import io.netty.buffer.ByteBuf;
+
+public interface MessageBodyDecoder {
+    AbstractMessageBody handle(ByteBuf in);
+}

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

@@ -0,0 +1,19 @@
+package io.github.protocol.mdtp.common.codec;
+
+import io.github.protocol.mdtp.common.model.MessageBodyHeader;
+
+import java.util.HashMap;
+import java.util.Map;
+
+public class MessageDecoderFactory {
+
+    private static final Map<Short, MessageBodyDecoder> decoders = new HashMap<>();
+
+    static {
+        decoders.put(MessageBodyHeader.DEVICE_DISCOVERY_REQUEST.toShort(), new DeviceDiscoveryRequestDecoder());
+    }
+
+    public static MessageBodyDecoder getDecoder(MessageBodyHeader header) {
+        return decoders.get(header.toShort());
+    }
+}

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

@@ -7,22 +7,14 @@
 
 @Data
 public abstract class AbstractMessageBody {
-    private short messageBodyHeader;
-
-    public void setMessageBodyHeader(MessageType messageType, ServiceGroup serviceGroup, DiscoveryServiceCode serviceCode) {
-        this.messageBodyHeader = 0;
-        this.messageBodyHeader |= (short) (messageType.getCode() & 0b111);
-        this.messageBodyHeader |= (short) ((serviceGroup.getCode() & 0b1111111) << 3);
-        this.messageBodyHeader |= (short) ((serviceCode.getCode() & 0b111111) << 10);
-    }
+    private MessageBodyHeader messageBodyHeader;
 
     public short generateRequestId() {
         UUID uuid = UUID.randomUUID();
         return (short) (uuid.getLeastSignificantBits() & 0xFFFF);
     }
 
-    public ByteBuf toByteBuf(ByteBuf buffer) {
-        buffer.writeShort(messageBodyHeader);
-        return buffer;
+    public void writeByteBuf(ByteBuf buffer) {
+        messageBodyHeader.writeByteBuf(buffer);
     }
 }

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

@@ -25,14 +25,27 @@ public class CDATHeader {
 
     private Integer logicalChannelId;
 
-    public ByteBuf toByteBuf(ByteBuf buffer) {
+    public void writeByteBuf(ByteBuf buffer) {
         buffer.writeByte(formatType);
         buffer.writeByte(protocolVersion);
         buffer.writeShort(messageLength);
         buffer.writeLong(timestamp);
         buffer.writeByte(flags);
         buffer.writeInt(sequenceNumber);
         buffer.writeInt(logicalChannelId);
-        return buffer;
+    }
+
+    public static CDATHeader readByteBuf(ByteBuf buffer) {
+        CDATHeader header = new CDATHeader();
+
+        header.setFormatType(buffer.readByte());
+        header.setProtocolVersion(buffer.readByte());
+        header.setMessageLength(buffer.readShort());
+        header.setTimestamp(buffer.readLong());
+        header.setFlags(buffer.readByte());
+        header.setSequenceNumber(buffer.readInt());
+        header.setLogicalChannelId(buffer.readInt());
+
+        return header;
     }
 }

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

@@ -5,7 +5,8 @@
 
 
 @Data
-public class DeviceDiscoveryRequest extends AbstractMessageBody {
+public class DeviceDiscoveryRequest extends AbstractMessageBody{
+
     private short requestId;
 
     private byte mask;
@@ -14,15 +15,16 @@ public class DeviceDiscoveryRequest extends AbstractMessageBody {
 
     private int[] deviceTypes;
 
-    public ByteBuf toByteBuf(ByteBuf buffer) {
-        super.toByteBuf(buffer);
+    public void writeByteBuf(ByteBuf buffer) {
+        super.writeByteBuf(buffer);
         buffer.writeShort(requestId);
         buffer.writeByte(mask);
         buffer.writeByte(deviceTypeCount);
-        for (int deviceType : deviceTypes) {
-            buffer.writeInt(deviceType);
+        if (deviceTypeCount > 0) {
+            for (int deviceType : deviceTypes) {
+                buffer.writeInt(deviceType);
+            }
         }
-        return buffer;
     }
 
     public static DeviceDiscoveryRequest fromByteBuf(ByteBuf data) {

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

@@ -29,7 +29,7 @@ public class DeviceDiscoveryResponse {
 
     private String deviceName;
 
-    public ByteBuf toByteBuf(ByteBuf buffer) {
+    public void writeByteBuf(ByteBuf buffer) {
         buffer.writeShort(messageHeader);
         buffer.writeShort(requestId);
         buffer.writeShort(responseId);
@@ -53,7 +53,5 @@ public ByteBuf toByteBuf(ByteBuf buffer) {
             byte[] nameBytes = deviceName.getBytes(StandardCharsets.UTF_8);
             buffer.writeBytes(nameBytes);
         }
-
-        return buffer;
     }
 }

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

@@ -16,4 +16,12 @@ public enum DiscoveryServiceCode {
         this.description = description;
     }
 
+    public static DiscoveryServiceCode fromCode(int value) {
+        for (DiscoveryServiceCode serviceCode : DiscoveryServiceCode.values()) {
+            if (serviceCode.getCode() == value) {
+                return serviceCode;
+            }
+        }
+        throw new IllegalArgumentException("Invalid DiscoveryServiceCode: " + value);
+    }
 }