video: driver: Fix to move enc driver state to DRAIN_LAST_FLAG

For encoder, driver state is not moved from DRAIN to
DRAIN_LAST_FLAG after receiving LAST_FLAG from firmware.
This causes RESUME command failure for v4l2 client. Fix the
same.

Change-Id: I2c64e9fa85e970e1d0feafdbd865dad61a6d21e9
Signed-off-by: Akshata Sahukar <[email protected]>

driver/vidc/src/venus_hfi_response.c

@@ -1668,12 +1668,14 @@ static int handle_session_response(struct msm_vidc_core *core,
 				type = RESP_WORK_OUTPUT_PSC;
 			}
 		} else if (packet->type == HFI_CMD_BUFFER &&
-			packet->port == HFI_PORT_RAW) {
+			vidc_port_from_hfi(inst, packet->port) ==
+				OUTPUT_PORT) {
 			if (check_last_flag(inst, packet)) {
 				offload = true;
 				type = RESP_WORK_LAST_FLAG;
 			}
 		}
+
 		if (offload)
 			break;