erpcgen: Fix templates for malloc errors.

If during the execution of the code the malloc fails to allocate
memory appropriate checks are needed for safe execution.
This patch introduces new checks to ensure safe execution.

Signed-off-by: Adithya Baglody <[email protected]>

Author: AdithyaBaglody

erpcgen/src/templates/c_coders.template

@@ -71,12 +71,14 @@ else
 {% enddef ------------------------------- BinaryType %}
 
 {% def decodeEnumType(info) ---------------- EnumType %}
-codec->read(&_tmp_local);
+if (!codec->getStatus()) {
+    codec->read(&_tmp_local);
 {% if source == "client" && info.pointerScalarTypes %}
-*{$info.name} = static_cast<{$info.enumName}>(_tmp_local);
+    *{$info.name} = static_cast<{$info.enumName}>(_tmp_local);
 {% else %}
-{$info.name} = static_cast<{$info.enumName}>(_tmp_local);
+    {$info.name} = static_cast<{$info.enumName}>(_tmp_local);
 {% endif -- pointerScalarTypes %}
+}
 {% enddef ---------------------------------- EnumType %}
 
 {% def decodeListType(info) ------------------- ListType %}
@@ -125,14 +127,24 @@ else
 {% else >%}
 for (uint32_t {$info.forLoopCount} = 0; {$info.forLoopCount} < {$info.sizeTemp}; ++{$info.forLoopCount})
 {
-{$addIndent("    ", info.protoNext.decode(info.protoNext))}
+{% if source == "server"%}
+    if (codec->getStatus()) {
+        break;
+    }
+{% endif >%}
+    if({$info.name})
+    {
+        {$addIndent("    ", info.protoNext.decode(info.protoNext))}
+    }
 }
 {% endif >%}
 {% enddef ----------------------------------- ArrayType %}
 
 {% def decodeStructType(info) --------------- StructType %}
 {% if info.inDataContainer %}
-read_{$info.typeName}_struct(codec, &({$info.name}));
+if (!codec->getStatus()) {
+    read_{$info.typeName}_struct(codec, &({$info.name}));
+}
 {% else -- not inDataContainer %}
 read_{$info.typeName}_struct(codec, {$info.name});
 {% endif -- inDataContainer %}
@@ -269,6 +281,11 @@ codec->startWriteList({% if source == "client" && info.pointerScalarTypes %}*{%
 {% else >%}
 for (uint32_t {$info.forLoopCount} = 0; {$info.forLoopCount} < {% if source == "client" && info.pointerScalarTypes %}*{% endif %}{$info.size}; ++{$info.forLoopCount})
 {
+{% if source == "server" %}
+    if (codec->getStatus()) {
+        break;
+    }
+{% endif >%}
 {$addIndent("    ", info.protoNext.encode(info.protoNext))}
 }
 {% endif >%}

erpcgen/src/templates/c_common_functions.template

@@ -144,6 +144,9 @@ static void read_{$struct.name}_struct_shared(erpc::{$codecClass} * codec, {$str
 static void read_{$struct.name}_struct(erpc::{$codecClass} * codec, {$struct.name} * data)
 {%   endif %}
 {
+    if(NULL == data) {
+        return;
+    }
 {%   if struct.hasNullableMember %}
     bool isNull;
 {%   endif -- hasNullableMember %}
@@ -185,6 +188,9 @@ static void write_{$struct.name}_struct_shared(erpc::{$codecClass} * codec, cons
 static void write_{$struct.name}_struct(erpc::{$codecClass} * codec, const {$struct.name} * data)
 {%   endif %}
 {
+    if(NULL == data) {
+        return;
+    }
 {%   for mem in struct.members if (shared == "def" && not mem.serializedViaMember) || (mem.noSharedMem && shared == "noSharedMem") %}
 {%    if mem.isNullable %}
     if ({$mem.coderCall.name}{$mem.structElements} == NULL)
@@ -271,6 +277,9 @@ static void read_{$union.name}_union_shared(erpc::{$codecClass} * codec, int32_t
 static void read_{$union.name}_union(erpc::{$codecClass} * codec, int32_t * discriminator, {$union.name} * data)
 {%   endif %}
 {
+    if(NULL == data) {
+        return;
+    }
 {%   if union.needTempVariable %}
     int32_t _tmp_local;
 
@@ -387,7 +396,10 @@ if ({$info.name})
 {% def freeArray(info) %}
 for (uint32_t {$info.forLoopCount} = 0; {$info.forLoopCount} < {$info.size}; ++{$info.forLoopCount})
 {
-{$addIndent("    ", info.protoNext.freeingCall(info.protoNext))}
+    if ({$info.name})
+    {
+        {$addIndent("    ", info.protoNext.freeingCall(info.protoNext))}
+    }
 }
 {% enddef ------------------------------- freeArray %}