aws
diff --git a/‎tools/code-generation/generator/src/main/resources/com/amazonaws/util/awsclientgenerator/velocity/cpp/s3/s3-crt/S3CrtSpecificOperations.vm‎
Lines changed: 5 additions & 0 deletions b/‎tools/code-generation/generator/src/main/resources/com/amazonaws/util/awsclientgenerator/velocity/cpp/s3/s3-crt/S3CrtSpecificOperations.vm‎
Lines changed: 5 additions & 0 deletions
diff --git a/‎tools/code-generation/generator/src/main/resources/com/amazonaws/util/awsclientgenerator/velocity/cpp/s3/s3-crt/SmithyS3CrtSpecificOperations.vm‎
Lines changed: 5 additions & 0 deletions b/‎tools/code-generation/generator/src/main/resources/com/amazonaws/util/awsclientgenerator/velocity/cpp/s3/s3-crt/SmithyS3CrtSpecificOperations.vm‎
Lines changed: 5 additions & 0 deletions
@@ -57,6 +57,11 @@ int S3CrtClient::S3CrtRequestGetBodyCallback(struct aws_s3_meta_request *meta_re
   {
     bodyStream.flush();
   }
+  if (bodyStream.fail()) {
+    AWS_LOGSTREAM_ERROR(ALLOCATION_TAG, "Failed to write download data");
+    aws_raise_error(AWS_ERROR_FILE_WRITE_FAILURE);
+    return AWS_OP_ERR;
+  }
 
   // Replenish flow-control window (no-op if enable_read_backpressure is not set):
   aws_s3_meta_request_increment_read_window(meta_request, body->len);
 
@@ -57,6 +57,11 @@ int S3CrtClient::S3CrtRequestGetBodyCallback(struct aws_s3_meta_request *meta_re
   {
     bodyStream.flush();
   }
+  if (bodyStream.fail()) {
+    AWS_LOGSTREAM_ERROR(ALLOCATION_TAG, "Failed to write download data");
+    aws_raise_error(AWS_ERROR_FILE_WRITE_FAILURE);
+    return AWS_OP_ERR;
+  }
 
   // Replenish flow-control window (no-op if enable_read_backpressure is not set):
   aws_s3_meta_request_increment_read_window(meta_request, body->len);
Original file line number	Diff line number	Diff line change
`@@ -57,6 +57,11 @@ int S3CrtClient::S3CrtRequestGetBodyCallback(struct aws_s3_meta_request *meta_re`
`57`	`57`	`{`
`58`	`58`	`bodyStream.flush();`
`59`	`59`	`}`
	`60`	`+ if (bodyStream.fail()) {`
	`61`	`+ AWS_LOGSTREAM_ERROR(ALLOCATION_TAG, "Failed to write download data");`
	`62`	`+ aws_raise_error(AWS_ERROR_FILE_WRITE_FAILURE);`
	`63`	`+ return AWS_OP_ERR;`
	`64`	`+ }`
`60`	`65`
`61`	`66`	`// Replenish flow-control window (no-op if enable_read_backpressure is not set):`
`62`	`67`	`aws_s3_meta_request_increment_read_window(meta_request, body->len);`