minor fixes

Author: azkrishpy

source/h1_connection.c

@@ -1030,6 +1030,17 @@ static void s_on_channel_write_complete(
      * to run again instead of calling the function directly.
      * This way, if the message completes synchronously,
      * we're not hogging the network by writing message after message in a tight loop */
+
+    /* If encoder is waiting for async body read, don't reschedule - the async callback will do it */
+    if (connection->thread_data.encoder.state == AWS_H1_ENCODER_STATE_ASYNC_WAITING) {
+        AWS_LOGF_TRACE(
+            AWS_LS_HTTP_CONNECTION,
+            "id=%p: Encoder waiting for async body, not rescheduling task.",
+            (void *)&connection->base);
+        connection->thread_data.is_outgoing_stream_task_active = false;
+        return;
+    }
+
     aws_channel_schedule_task_now(channel, &connection->outgoing_stream_task);
 }
 
@@ -1139,7 +1150,7 @@ static void s_write_outgoing_stream(struct aws_h1_connection *connection, bool f
 
             goto error;
         }
-    } else if(connection->thread_data.encoder.message->async_body) {
+    } else if(connection->thread_data.encoder.message && connection->thread_data.encoder.message->async_body) {
         AWS_LOGF_TRACE(
             AWS_LS_HTTP_CONNECTION,
             "id=%p: Outgoing async stream task is either complete or waiting on future. Never reschedule task.",

source/h1_encoder.c

@@ -33,6 +33,9 @@ static int s_scan_outgoing_headers(
 
     size_t total = 0;
     bool has_body_stream = aws_http_message_get_body_stream(message);
+    if (!has_body_stream) {
+        has_body_stream = aws_http_message_get_async_body_stream(message);
+    }
     bool has_content_length_header = false;
     bool has_transfer_encoding_header = false;
 
@@ -1087,6 +1090,12 @@ int aws_h1_encoder_process(struct aws_h1_encoder *encoder, struct aws_byte_buf *
     enum aws_h1_encoder_state prev_state;
     do {
         prev_state = encoder->state;
+
+        /* Exit if waiting for async read - callback will resume encoding */
+        if (encoder->state == AWS_H1_ENCODER_STATE_ASYNC_WAITING) {
+            return AWS_OP_SUCCESS;
+        }
+
         if (s_encoder_states[encoder->state].fn(encoder, out_buf)) {
             return AWS_OP_ERR;
         }