Fix improper length of sms message stored in User Data Length field

[project/uqmi.git] / commands-wms.c

diff --git a/commands-wms.c b/commands-wms.c
index da4aeaf..2e4e08f 100644 (file)
--- a/commands-wms.c
+++ b/commands-wms.c
@@ -1,5 +1,8 @@
 #include "qmi-message.h"
 
+#define MIN(a,b) (((a)<(b))?(a):(b))
+#define CEILDIV(x,y) (((x) + (y) - 1) / (y))
+
 static void cmd_wms_list_messages_cb(struct qmi_dev *qmi, struct qmi_request *req, struct qmi_msg *msg)
 {
        struct qmi_wms_list_messages_response res;
@@ -67,6 +70,7 @@ pdu_decode_7bit_char(char *dest, int len, unsigned char c, bool *escape)
        fprintf(stderr, " %02x", c);
        dest += len;
        if (*escape) {
+               *escape = false;
                switch(c) {
                case 0x0A:
                        *dest = 0x0C;
@@ -158,13 +162,13 @@ pdu_decode_7bit_str(char *dest, const unsigned char *data, int data_len, int bit
        return len;
 }
 
-static void decode_udh(const unsigned char *data)
+static int decode_udh(const unsigned char *data)
 {
        const unsigned char *end;
-       unsigned int type, len;
+       unsigned int type, len, udh_len;
 
-       len = *(data++);
-       end = data + len;
+       udh_len = *(data++);
+       end = data + udh_len;
        while (data < end) {
                const unsigned char *val;
 
@@ -176,38 +180,30 @@ static void decode_udh(const unsigned char *data)
                        break;
 
                switch (type) {
-               case 0:
+               case 0x00:
                        blobmsg_add_u32(&status, "concat_ref", (uint32_t) val[0]);
                        blobmsg_add_u32(&status, "concat_part", (uint32_t) val[2]);
                        blobmsg_add_u32(&status, "concat_parts", (uint32_t) val[1]);
                        break;
+               case 0x08:
+                       blobmsg_add_u32(&status, "concat_ref", (uint32_t) (val[0] << 8 | val[1]));
+                       blobmsg_add_u32(&status, "concat_part", (uint32_t) val[3]);
+                       blobmsg_add_u32(&status, "concat_parts", (uint32_t) val[2]);
+                       break;
                default:
                        break;
                }
        }
+
+       return udh_len + 1;
 }
 
-static void decode_7bit_field(char *name, const unsigned char *data, int data_len, bool udh)
+static void decode_7bit_field(char *name, const unsigned char *data, int data_len, int bit_offset)
 {
-       const unsigned char *udh_start;
-       char *dest;
-       int pos_offset = 0;
-
-       if (udh) {
-               int len = data[0] + 1;
-
-               udh_start = data;
-               data += len;
-               data_len -= len;
-               pos_offset = len % 7;
-       }
-
-       dest = blobmsg_alloc_string_buffer(&status, name, 3 * (data_len * 8 / 7) + 2);
-       pdu_decode_7bit_str(dest, data, data_len, pos_offset);
+       char *dest = blobmsg_alloc_string_buffer(&status, name, 3 * data_len + 2);
+       pdu_decode_7bit_str(dest, data, CEILDIV(data_len * 7, 8), bit_offset);
+       dest[data_len] = 0;
        blobmsg_add_string_buffer(&status);
-
-       if (udh)
-               decode_udh(udh_start);
 }
 
 static char *pdu_add_semioctet(char *str, char val)
@@ -253,6 +249,42 @@ static void wms_decode_address(char *name, unsigned char *data, int len)
        blobmsg_add_string_buffer(&status);
 }
 
+static void blobmsg_add_hex(struct blob_buf *buf, const char *name, unsigned const char *data, int len)
+{
+       char* str = blobmsg_alloc_string_buffer(buf, name, len * 2 + 1);
+       for (int i = 0; i < len; i++) {
+               str += sprintf(str, "%02x", data[i]);
+       }
+       blobmsg_add_string_buffer(buf);
+}
+
+#define cmd_wms_delete_message_cb no_cb
+static enum qmi_cmd_result
+cmd_wms_delete_message_prepare(struct qmi_dev *qmi, struct qmi_request *req, struct qmi_msg *msg, char *arg)
+{
+       char *err;
+       int id;
+
+       id = strtoul(arg, &err, 10);
+       if (err && *err) {
+               uqmi_add_error("Invalid message ID");
+               return QMI_CMD_EXIT;
+       }
+
+       static struct qmi_wms_delete_request mreq = {
+               QMI_INIT(memory_storage, QMI_WMS_STORAGE_TYPE_UIM),
+               QMI_INIT(message_mode, QMI_WMS_MESSAGE_MODE_GSM_WCDMA),
+       };
+
+       mreq.set.memory_index = 1;
+       mreq.data.memory_index = id;
+
+       qmi_set_wms_delete_request(msg, &mreq);
+
+       return QMI_CMD_REQUEST;
+}
+
+
 static void cmd_wms_get_message_cb(struct qmi_dev *qmi, struct qmi_request *req, struct qmi_msg *msg)
 {
        struct qmi_wms_raw_read_response res;
@@ -305,10 +337,6 @@ static void cmd_wms_get_message_cb(struct qmi_dev *qmi, struct qmi_request *req,
        /* Data Encoding */
        dcs = *(data++);
 
-       /* only 7-bit encoding supported for now */
-       if (dcs & 0x0c)
-               goto error;
-
        if (dcs & 0x10)
                blobmsg_add_u32(&status, "class", (dcs & 3));
 
@@ -348,10 +376,42 @@ static void cmd_wms_get_message_cb(struct qmi_dev *qmi, struct qmi_request *req,
                data += 7;
        }
 
-       cur_len = *(data++);
-       decode_7bit_field("text", data, end - data, !!(first & 0x40));
-       blobmsg_close_table(&status, c);
+       int message_len = *(data++);
+       int udh_len = 0;
+       int bit_offset = 0;
 
+       /* User Data Header */
+       if (first & 0x40) {
+               udh_len = decode_udh(data);
+               data += udh_len;
+               bit_offset = udh_len % 7;
+               }
+
+       if (data >= end)
+               goto error;
+
+       switch(dcs & 0x0c) {
+               case 0x00:
+                       /* 7 bit GSM alphabet */
+                       message_len = message_len - CEILDIV(udh_len * 8, 7);
+                       message_len = MIN(message_len, CEILDIV((end - data) * 8, 7));
+                       decode_7bit_field("text", data, message_len, bit_offset);
+                       break;
+               case 0x04:
+                       /* 8 bit data */
+                       message_len = MIN(message_len - udh_len, end - data);
+                       blobmsg_add_hex(&status, "data", data, message_len);
+                       break;
+               case 0x08:
+                       /* 16 bit UCS-2 string */
+                       message_len = MIN(message_len - udh_len, end - data);
+                       blobmsg_add_hex(&status, "ucs-2", data, message_len);
+                       break;
+               default:
+                       goto error;
+               }
+
+       blobmsg_close_table(&status, c);
        return;
 
 error:
@@ -500,7 +560,7 @@ pdu_encode_number(unsigned char *dest, const char *str, bool smsc)
        }
 
        for (i = 0; str[i]; i++) {
-               if (str[i] >= '0' || str[i] <= '9')
+               if (str[i] >= '0' && str[i] <= '9')
                        continue;
 
                ascii = true;
@@ -531,7 +591,7 @@ pdu_encode_data(unsigned char *dest, const char *str)
 
        dest[len++] = 0;
        len += pdu_encode_7bit_str(&dest[len], str);
-       dest[0] = len - 1;
+       dest[0] = strlen(str);
 
        return len;
 }