return ret;
 }
 
-static int cros_ec_host_command_proto_query_v2(struct cros_ec_device *ec_dev)
+static int cros_ec_get_proto_info_legacy(struct cros_ec_device *ec_dev)
 {
        struct cros_ec_command *msg;
-       struct ec_params_hello *hello_params;
-       struct ec_response_hello *hello_response;
+       struct ec_params_hello *params;
+       struct ec_response_hello *response;
        int ret;
-       int len = max(sizeof(*hello_params), sizeof(*hello_response));
 
-       msg = kmalloc(sizeof(*msg) + len, GFP_KERNEL);
+       ec_dev->proto_version = 2;
+
+       msg = kzalloc(sizeof(*msg) + max(sizeof(*params), sizeof(*response)), GFP_KERNEL);
        if (!msg)
                return -ENOMEM;
 
-       msg->version = 0;
        msg->command = EC_CMD_HELLO;
-       hello_params = (struct ec_params_hello *)msg->data;
-       msg->outsize = sizeof(*hello_params);
-       hello_response = (struct ec_response_hello *)msg->data;
-       msg->insize = sizeof(*hello_response);
+       msg->insize = sizeof(*response);
+       msg->outsize = sizeof(*params);
 
-       hello_params->in_data = 0xa0b0c0d0;
+       params = (struct ec_params_hello *)msg->data;
+       params->in_data = 0xa0b0c0d0;
 
        ret = send_command(ec_dev, msg);
-
        if (ret < 0) {
-               dev_dbg(ec_dev->dev,
-                       "EC failed to respond to v2 hello: %d\n",
-                       ret);
+               dev_dbg(ec_dev->dev, "EC failed to respond to v2 hello: %d\n", ret);
                goto exit;
-       } else if (msg->result != EC_RES_SUCCESS) {
-               dev_err(ec_dev->dev,
-                       "EC responded to v2 hello with error: %d\n",
-                       msg->result);
-               ret = msg->result;
+       }
+
+       ret = cros_ec_map_error(msg->result);
+       if (ret) {
+               dev_err(ec_dev->dev, "EC responded to v2 hello with error: %d\n", msg->result);
                goto exit;
-       } else if (hello_response->out_data != 0xa1b2c3d4) {
+       }
+
+       response = (struct ec_response_hello *)msg->data;
+       if (response->out_data != 0xa1b2c3d4) {
                dev_err(ec_dev->dev,
                        "EC responded to v2 hello with bad result: %u\n",
-                       hello_response->out_data);
+                       response->out_data);
                ret = -EBADMSG;
                goto exit;
        }
 
-       ret = 0;
+       ec_dev->max_request = EC_PROTO2_MAX_PARAM_SIZE;
+       ec_dev->max_response = EC_PROTO2_MAX_PARAM_SIZE;
+       ec_dev->max_passthru = 0;
+       ec_dev->pkt_xfer = NULL;
+       ec_dev->din_size = EC_PROTO2_MSG_BYTES;
+       ec_dev->dout_size = EC_PROTO2_MSG_BYTES;
 
- exit:
+       dev_dbg(ec_dev->dev, "falling back to proto v2\n");
+       ret = 0;
+exit:
        kfree(msg);
        return ret;
 }
                cros_ec_get_proto_info(ec_dev, CROS_EC_DEV_PD_INDEX);
        } else {
                /* Try querying with a v2 hello message. */
-               ec_dev->proto_version = 2;
-               ret = cros_ec_host_command_proto_query_v2(ec_dev);
-
-               if (ret == 0) {
-                       /* V2 hello succeeded. */
-                       dev_dbg(ec_dev->dev, "falling back to proto v2\n");
-
-                       ec_dev->max_request = EC_PROTO2_MAX_PARAM_SIZE;
-                       ec_dev->max_response = EC_PROTO2_MAX_PARAM_SIZE;
-                       ec_dev->max_passthru = 0;
-                       ec_dev->pkt_xfer = NULL;
-                       ec_dev->din_size = EC_PROTO2_MSG_BYTES;
-                       ec_dev->dout_size = EC_PROTO2_MSG_BYTES;
-               } else {
+               ret = cros_ec_get_proto_info_legacy(ec_dev);
+               if (ret) {
                        /*
                         * It's possible for a test to occur too early when
                         * the EC isn't listening. If this happens, we'll
                         */
                        ec_dev->proto_version = EC_PROTO_VERSION_UNKNOWN;
                        dev_dbg(ec_dev->dev, "EC query failed: %d\n", ret);
-                       goto exit;
+                       return ret;
                }
        }
 
 
                KUNIT_ASSERT_PTR_NE(test, mock, NULL);
        }
 
-       /* For cros_ec_host_command_proto_query_v2(). */
+       /* For cros_ec_get_proto_info_legacy(). */
        {
                struct ec_response_hello *data;
 
                KUNIT_EXPECT_EQ(test, mock->msg.outsize, 0);
        }
 
-       /* For cros_ec_host_command_proto_query_v2(). */
+       /* For cros_ec_get_proto_info_legacy(). */
        {
                struct ec_params_hello *data;
 
                KUNIT_ASSERT_PTR_NE(test, mock, NULL);
        }
 
-       /* For cros_ec_host_command_proto_query_v2(). */
+       /* For cros_ec_get_proto_info_legacy(). */
        {
                struct ec_response_hello *data;
 
                KUNIT_EXPECT_EQ(test, mock->msg.outsize, 0);
        }
 
-       /* For cros_ec_host_command_proto_query_v2(). */
+       /* For cros_ec_get_proto_info_legacy(). */
        {
                struct ec_params_hello *data;
 
                KUNIT_ASSERT_PTR_NE(test, mock, NULL);
        }
 
-       /* For cros_ec_host_command_proto_query_v2(). */
+       /* For cros_ec_get_proto_info_legacy(). */
        {
                mock = cros_kunit_ec_xfer_mock_addx(test, -EIO, EC_RES_SUCCESS, 0);
                KUNIT_ASSERT_PTR_NE(test, mock, NULL);
                KUNIT_EXPECT_EQ(test, mock->msg.outsize, 0);
        }
 
-       /* For cros_ec_host_command_proto_query_v2(). */
+       /* For cros_ec_get_proto_info_legacy(). */
        {
                mock = cros_kunit_ec_xfer_mock_next();
                KUNIT_EXPECT_PTR_NE(test, mock, NULL);
                KUNIT_ASSERT_PTR_NE(test, mock, NULL);
        }
 
-       /* For cros_ec_host_command_proto_query_v2(). */
+       /* For cros_ec_get_proto_info_legacy(). */
        {
                mock = cros_kunit_ec_xfer_mock_addx(test, 0, EC_RES_INVALID_COMMAND, 0);
                KUNIT_ASSERT_PTR_NE(test, mock, NULL);
 
        cros_ec_proto_test_query_all_pretest(test);
        ret = cros_ec_query_all(ec_dev);
-       KUNIT_EXPECT_EQ(test, ret, EC_RES_INVALID_COMMAND);
+       KUNIT_EXPECT_EQ(test, ret, -EOPNOTSUPP);
        KUNIT_EXPECT_EQ(test, ec_dev->proto_version, EC_PROTO_VERSION_UNKNOWN);
 
        /* For cros_ec_get_proto_info() without passthru. */
                KUNIT_EXPECT_EQ(test, mock->msg.outsize, 0);
        }
 
-       /* For cros_ec_host_command_proto_query_v2(). */
+       /* For cros_ec_get_proto_info_legacy(). */
        {
                mock = cros_kunit_ec_xfer_mock_next();
                KUNIT_EXPECT_PTR_NE(test, mock, NULL);
                KUNIT_ASSERT_PTR_NE(test, mock, NULL);
        }
 
-       /* For cros_ec_host_command_proto_query_v2(). */
+       /* For cros_ec_get_proto_info_legacy(). */
        {
                struct ec_response_hello *data;
 
                KUNIT_EXPECT_EQ(test, mock->msg.outsize, 0);
        }
 
-       /* For cros_ec_host_command_proto_query_v2(). */
+       /* For cros_ec_get_proto_info_legacy(). */
        {
                mock = cros_kunit_ec_xfer_mock_next();
                KUNIT_EXPECT_PTR_NE(test, mock, NULL);