Commands and result support

1 files changed, 31 insertions, 5 deletions

diff --git a/src/main/java/org/traccar/protocol/KhdProtocolEncoder.java b/src/main/java/org/traccar/protocol/KhdProtocolEncoder.java
index 4a8df26c8..8aeb9660d 100644
--- a/src/main/java/org/traccar/protocol/KhdProtocolEncoder.java
+++ b/src/main/java/org/traccar/protocol/KhdProtocolEncoder.java
@@ -1,5 +1,5 @@
 /*
- * Copyright 2015 - 2019 Anton Tananaev (anton@traccar.org)
+ * Copyright 2015 - 2021 Anton Tananaev (anton@traccar.org)
  *
  * Licensed under the Apache License, Version 2.0 (the "License");
  * you may not use this file except in compliance with the License.
@@ -24,14 +24,19 @@ import org.traccar.Protocol;
 
 public class KhdProtocolEncoder extends BaseProtocolEncoder {
 
+    public static final int MSG_ON_DEMAND_TRACK = 0x30;
     public static final int MSG_CUT_OIL = 0x39;
     public static final int MSG_RESUME_OIL = 0x38;
+    public static final int MSG_CHECK_VERSION = 0x3D;
+    public static final int MSG_FACTORY_RESET = 0xC3;
+    public static final int MSG_SET_OVERSPEED = 0x3F;
+    public static final int MSG_DELETE_MILEAGE = 0x66;
 
     public KhdProtocolEncoder(Protocol protocol) {
         super(protocol);
     }
 
-    private ByteBuf encodeCommand(int command, String uniqueId) {
+    private ByteBuf encodeCommand(int command, String uniqueId, ByteBuf content) {
 
         ByteBuf buf = Unpooled.buffer();
 
@@ -39,7 +44,12 @@ public class KhdProtocolEncoder extends BaseProtocolEncoder {
         buf.writeByte(0x29);
 
         buf.writeByte(command);
-        buf.writeShort(6); // size
+
+        int length = 6;
+        if (content != null) {
+            length += content.readableBytes();
+        }
+        buf.writeShort(length);
 
         uniqueId = "00000000".concat(uniqueId);
         uniqueId = uniqueId.substring(uniqueId.length() - 8);
@@ -48,6 +58,10 @@ public class KhdProtocolEncoder extends BaseProtocolEncoder {
         buf.writeByte(Integer.parseInt(uniqueId.substring(4, 6)) + 0x80);
         buf.writeByte(Integer.parseInt(uniqueId.substring(6, 8)));
 
+        if (content != null) {
+            buf.writeBytes(content);
+        }
+
         buf.writeByte(Checksum.xor(buf.nioBuffer()));
         buf.writeByte(0x0D); // ending
 
@@ -61,9 +75,21 @@ public class KhdProtocolEncoder extends BaseProtocolEncoder {
 
         switch (command.getType()) {
             case Command.TYPE_ENGINE_STOP:
-                return encodeCommand(MSG_CUT_OIL, uniqueId);
+                return encodeCommand(MSG_CUT_OIL, uniqueId, null);
             case Command.TYPE_ENGINE_RESUME:
-                return encodeCommand(MSG_RESUME_OIL, uniqueId);
+                return encodeCommand(MSG_RESUME_OIL, uniqueId, null);
+            case Command.TYPE_GET_VERSION:
+                return encodeCommand(MSG_CHECK_VERSION, uniqueId, null);
+            case Command.TYPE_FACTORY_RESET:
+                return encodeCommand(MSG_FACTORY_RESET, uniqueId, null);
+            case Command.TYPE_SET_SPEED_LIMIT:
+                ByteBuf content = Unpooled.buffer();
+                content.writeByte(Integer.parseInt(command.getString(Command.KEY_DATA)));
+                return encodeCommand(MSG_RESUME_OIL, uniqueId, content);
+            case Command.TYPE_SET_ODOMETER:
+                return encodeCommand(MSG_DELETE_MILEAGE, uniqueId, null);
+            case Command.TYPE_POSITION_SINGLE:
+                return encodeCommand(MSG_ON_DEMAND_TRACK, uniqueId, null);
             default:
                 return null;
         }