feat: add DeviceDiscoveryRequest model

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

Author: instpe

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

@@ -0,0 +1,31 @@
+package io.github.protocol.mdtp.common.model;
+
+import io.netty.buffer.ByteBuf;
+import lombok.Data;
+
+import java.util.UUID;
+
+@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);
+    }
+
+    public short generateRequestId() {
+        UUID uuid = UUID.randomUUID();
+        return (short) (uuid.getLeastSignificantBits() & 0xFFFF);
+    }
+
+    public ByteBuf toByteBuf(ByteBuf buffer) {
+        buffer.writeShort(messageBodyHeader);
+        return buffer;
+    }
+}

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

@@ -0,0 +1,38 @@
+package io.github.protocol.mdtp.common.model;
+
+import io.netty.buffer.ByteBuf;
+import lombok.AllArgsConstructor;
+import lombok.Data;
+import lombok.NoArgsConstructor;
+
+
+@Data
+@AllArgsConstructor
+@NoArgsConstructor
+public class CDATHeader {
+
+    private byte formatType;
+
+    private byte protocolVersion;
+
+    private short messageLength;
+
+    private long timestamp;
+
+    private byte flags;
+
+    private Integer sequenceNumber;
+
+    private Integer logicalChannelId;
+
+    public ByteBuf toByteBuf(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;
+    }
+}

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

@@ -0,0 +1,41 @@
+package io.github.protocol.mdtp.common.model;
+
+import io.netty.buffer.ByteBuf;
+import lombok.Data;
+
+
+@Data
+public class DeviceDiscoveryRequest extends AbstractMessageBody {
+    private short requestId;
+
+    private byte mask;
+
+    private byte deviceTypeCount;
+
+    private int[] deviceTypes;
+
+    public ByteBuf toByteBuf(ByteBuf buffer) {
+        super.toByteBuf(buffer);
+        buffer.writeShort(requestId);
+        buffer.writeByte(mask);
+        buffer.writeByte(deviceTypeCount);
+        for (int deviceType : deviceTypes) {
+            buffer.writeInt(deviceType);
+        }
+        return buffer;
+    }
+
+    public static DeviceDiscoveryRequest fromByteBuf(ByteBuf data) {
+        DeviceDiscoveryRequest request = new DeviceDiscoveryRequest();
+        request.requestId = data.readShort();
+        request.mask = data.readByte();
+        request.deviceTypeCount = data.readByte();
+
+        int length = request.deviceTypeCount;
+        request.deviceTypes = new int[length];
+        for (int i = 0; i < length; i++) {
+            request.deviceTypes[i] = data.readInt();
+        }
+        return request;
+    }
+}

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

@@ -0,0 +1,59 @@
+package io.github.protocol.mdtp.common.model;
+
+import io.netty.buffer.ByteBuf;
+import lombok.Data;
+
+import java.nio.charset.StandardCharsets;
+
+@Data
+public class DeviceDiscoveryResponse {
+    private short messageHeader;
+
+    private short requestId;
+
+    private short responseId;
+
+    private byte mask;
+
+    private byte deviceStatus;
+
+    private byte addressCount;
+
+    private String[] addresses;
+
+    private short port;
+
+    private int deviceType;
+
+    private byte[] uniqueId;
+
+    private String deviceName;
+
+    public ByteBuf toByteBuf(ByteBuf buffer) {
+        buffer.writeShort(messageHeader);
+        buffer.writeShort(requestId);
+        buffer.writeShort(responseId);
+        buffer.writeByte(mask);
+        buffer.writeByte(deviceStatus);
+        buffer.writeByte(addressCount);
+
+        for (String address : addresses) {
+            byte[] addressBytes = address.getBytes(StandardCharsets.UTF_8);
+            buffer.writeBytes(addressBytes);
+        }
+
+        buffer.writeShort(port);
+        buffer.writeInt(deviceType);
+
+        if (uniqueId != null) {
+            buffer.writeBytes(uniqueId);
+        }
+
+        if (deviceName != null) {
+            byte[] nameBytes = deviceName.getBytes(StandardCharsets.UTF_8);
+            buffer.writeBytes(nameBytes);
+        }
+
+        return buffer;
+    }
+}

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

@@ -1,4 +1,4 @@
-package iot.github.protocol.mdtp.common.model;
+package io.github.protocol.mdtp.common.model;
 
 import lombok.Getter;
 

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

@@ -0,0 +1,27 @@
+package io.github.protocol.mdtp.common.model;
+
+import io.netty.buffer.ByteBuf;
+import lombok.Data;
+
+@Data
+public class MdtpPacket {
+
+    private CDATHeader header;
+
+    private SecurityHeader securityHeader;
+
+    private AbstractMessageBody body;
+
+    private Signature signature;
+
+    public ByteBuf toByteBuf(ByteBuf buffer) {
+        header.toByteBuf(buffer);
+        if (securityHeader != null) {
+            securityHeader.toByteBuf(buffer);
+        }
+        if (body != null) {
+            body.toByteBuf(buffer);
+        }
+        return buffer;
+    }
+}

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

@@ -0,0 +1,19 @@
+package io.github.protocol.mdtp.common.model;
+
+public enum MessageType {
+    REQUEST(0),
+
+    RESPONSE(1),
+
+    NOTIFY(2);
+
+    private final int code;
+
+    MessageType(int code) {
+        this.code = code;
+    }
+
+    public int getCode() {
+        return code;
+    }
+}

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

@@ -0,0 +1,17 @@
+package io.github.protocol.mdtp.common.model;
+
+import io.netty.buffer.ByteBuf;
+import lombok.Data;
+
+@Data
+public class SecurityHeader {
+    private byte[] encryptionData;
+
+    public ByteBuf toByteBuf(ByteBuf buffer) {
+        for (int data : encryptionData) {
+            buffer.writeInt(data);
+        }
+        return buffer;
+    }
+
+}

mdtp-common/src/main/java/iot/github/protocol/mdtp/common/model/ServiceGroup.java renamed to mdtp-common/src/main/java/io/github/protocol/mdtp/common/model/ServiceGroup.java

@@ -1,4 +1,4 @@
-package iot.github.protocol.mdtp.common.model;
+package io.github.protocol.mdtp.common.model;
 
 import lombok.Getter;
 

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

@@ -0,0 +1,9 @@
+package io.github.protocol.mdtp.common.model;
+
+import lombok.Data;
+
+@Data
+public class Signature {
+    private byte[] signatureData;
+
+}