import PULS_20180308

[GitHub/mt8127/android_kernel_alcatel_ttab.git] / drivers / input / touchscreen / mediatek / GT9XX_2 / goodix_tool.c

diff --git a/drivers/input/touchscreen/mediatek/GT9XX_2/goodix_tool.c b/drivers/input/touchscreen/mediatek/GT9XX_2/goodix_tool.c
index d1faccce4287c8467d051670915c4b9bb485a21f..ab5ba2d04f6e0a2adc1faeb3d3860ef877db9814 100644 (file)
--- a/drivers/input/touchscreen/mediatek/GT9XX_2/goodix_tool.c
+++ b/drivers/input/touchscreen/mediatek/GT9XX_2/goodix_tool.c
@@ -503,6 +503,13 @@ static s32 goodix_tool_write(struct file *filp, const char __user *buff, unsigne
     {
         show_len = 0;
         total_len = 0;
+               if ((cmd_head.data == NULL)
+                   || (cmd_head.data_len >= DATA_LENGTH)
+                   || (cmd_head.data_len >= (len - CMD_HEAD_LENGTH))) {
+                       GTP_ERROR("copy_from_user data out of range.");
+                       return -EINVAL;
+               }
+
         memset(cmd_head.data, 0, cmd_head.data_len + 1);
         memcpy(cmd_head.data, &buff[CMD_HEAD_LENGTH], cmd_head.data_len);
         GTP_DEBUG("update firmware, filename: %s", cmd_head.data);