s3:net: Check return value of data_blob_talloc()

Signed-off-by: Joseph Sutton <josephsutton@catalyst.net.nz>
Reviewed-by: Andrew Bartlett <abartlet@samba.org>

diff --git a/source3/utils/net_offlinejoin.c b/source3/utils/net_offlinejoin.c
index 5294fd98ad760b42ebd91ce9344b0295225f384d..3ec5c97adcaefc2d783bd66a900a637a8b75ed5c 100644 (file)
--- a/source3/utils/net_offlinejoin.c
+++ b/source3/utils/net_offlinejoin.c
@@ -210,6 +210,11 @@ int net_offlinejoin_provision(struct net_context *c,
 
                /* Add the unicode BOM mark */
                blob = data_blob_talloc(c, NULL, ucs2_blob.length + 2);
+               if (blob.data == NULL) {
+                       d_printf("Failed to allocate blob: %s\n",
+                                strerror(errno));
+                       return -1;
+               }
 
                blob.data[0] = 0xff;
                blob.data[1] = 0xfe;
@@ -568,6 +573,11 @@ int net_offlinejoin_composeodj(struct net_context *c,
 
                /* Add the unicode BOM mark */
                blob = data_blob_talloc(c, NULL, ucs2_blob.length + 2);
+               if (blob.data == NULL) {
+                       d_printf("Failed to allocate blob: %s\n",
+                                strerror(errno));
+                       return -1;
+               }
 
                blob.data[0] = 0xff;
                blob.data[1] = 0xfe;