| xj | b04a402 | 2021-11-25 15:01:52 +0800 | [diff] [blame^] | 1 | /* | 
|  | 2 | * An implementation of key value pair (KVP) functionality for Linux. | 
|  | 3 | * | 
|  | 4 | * | 
|  | 5 | * Copyright (C) 2010, Novell, Inc. | 
|  | 6 | * Author : K. Y. Srinivasan <ksrinivasan@novell.com> | 
|  | 7 | * | 
|  | 8 | * This program is free software; you can redistribute it and/or modify it | 
|  | 9 | * under the terms of the GNU General Public License version 2 as published | 
|  | 10 | * by the Free Software Foundation. | 
|  | 11 | * | 
|  | 12 | * This program is distributed in the hope that it will be useful, but | 
|  | 13 | * WITHOUT ANY WARRANTY; without even the implied warranty of | 
|  | 14 | * MERCHANTABILITY OR FITNESS FOR A PARTICULAR PURPOSE, GOOD TITLE or | 
|  | 15 | * NON INFRINGEMENT.  See the GNU General Public License for more | 
|  | 16 | * details. | 
|  | 17 | * | 
|  | 18 | * You should have received a copy of the GNU General Public License | 
|  | 19 | * along with this program; if not, write to the Free Software | 
|  | 20 | * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA. | 
|  | 21 | * | 
|  | 22 | */ | 
|  | 23 | #define pr_fmt(fmt) KBUILD_MODNAME ": " fmt | 
|  | 24 |  | 
|  | 25 | #include <linux/net.h> | 
|  | 26 | #include <linux/nls.h> | 
|  | 27 | #include <linux/connector.h> | 
|  | 28 | #include <linux/workqueue.h> | 
|  | 29 | #include <linux/hyperv.h> | 
|  | 30 |  | 
|  | 31 | #include "hyperv_vmbus.h" | 
|  | 32 | #include "hv_utils_transport.h" | 
|  | 33 |  | 
|  | 34 | /* | 
|  | 35 | * Pre win8 version numbers used in ws2008 and ws 2008 r2 (win7) | 
|  | 36 | */ | 
|  | 37 | #define WS2008_SRV_MAJOR	1 | 
|  | 38 | #define WS2008_SRV_MINOR	0 | 
|  | 39 | #define WS2008_SRV_VERSION     (WS2008_SRV_MAJOR << 16 | WS2008_SRV_MINOR) | 
|  | 40 |  | 
|  | 41 | #define WIN7_SRV_MAJOR   3 | 
|  | 42 | #define WIN7_SRV_MINOR   0 | 
|  | 43 | #define WIN7_SRV_VERSION     (WIN7_SRV_MAJOR << 16 | WIN7_SRV_MINOR) | 
|  | 44 |  | 
|  | 45 | #define WIN8_SRV_MAJOR   4 | 
|  | 46 | #define WIN8_SRV_MINOR   0 | 
|  | 47 | #define WIN8_SRV_VERSION     (WIN8_SRV_MAJOR << 16 | WIN8_SRV_MINOR) | 
|  | 48 |  | 
|  | 49 | #define KVP_VER_COUNT 3 | 
|  | 50 | static const int kvp_versions[] = { | 
|  | 51 | WIN8_SRV_VERSION, | 
|  | 52 | WIN7_SRV_VERSION, | 
|  | 53 | WS2008_SRV_VERSION | 
|  | 54 | }; | 
|  | 55 |  | 
|  | 56 | #define FW_VER_COUNT 2 | 
|  | 57 | static const int fw_versions[] = { | 
|  | 58 | UTIL_FW_VERSION, | 
|  | 59 | UTIL_WS2K8_FW_VERSION | 
|  | 60 | }; | 
|  | 61 |  | 
|  | 62 | /* | 
|  | 63 | * Global state maintained for transaction that is being processed. For a class | 
|  | 64 | * of integration services, including the "KVP service", the specified protocol | 
|  | 65 | * is a "request/response" protocol which means that there can only be single | 
|  | 66 | * outstanding transaction from the host at any given point in time. We use | 
|  | 67 | * this to simplify memory management in this driver - we cache and process | 
|  | 68 | * only one message at a time. | 
|  | 69 | * | 
|  | 70 | * While the request/response protocol is guaranteed by the host, we further | 
|  | 71 | * ensure this by serializing packet processing in this driver - we do not | 
|  | 72 | * read additional packets from the VMBUS until the current packet is fully | 
|  | 73 | * handled. | 
|  | 74 | */ | 
|  | 75 |  | 
|  | 76 | static struct { | 
|  | 77 | int state;   /* hvutil_device_state */ | 
|  | 78 | int recv_len; /* number of bytes received. */ | 
|  | 79 | struct hv_kvp_msg  *kvp_msg; /* current message */ | 
|  | 80 | struct vmbus_channel *recv_channel; /* chn we got the request */ | 
|  | 81 | u64 recv_req_id; /* request ID. */ | 
|  | 82 | } kvp_transaction; | 
|  | 83 |  | 
|  | 84 | /* | 
|  | 85 | * This state maintains the version number registered by the daemon. | 
|  | 86 | */ | 
|  | 87 | static int dm_reg_value; | 
|  | 88 |  | 
|  | 89 | static void kvp_send_key(struct work_struct *dummy); | 
|  | 90 |  | 
|  | 91 |  | 
|  | 92 | static void kvp_respond_to_host(struct hv_kvp_msg *msg, int error); | 
|  | 93 | static void kvp_timeout_func(struct work_struct *dummy); | 
|  | 94 | static void kvp_host_handshake_func(struct work_struct *dummy); | 
|  | 95 | static void kvp_register(int); | 
|  | 96 |  | 
|  | 97 | static DECLARE_DELAYED_WORK(kvp_timeout_work, kvp_timeout_func); | 
|  | 98 | static DECLARE_DELAYED_WORK(kvp_host_handshake_work, kvp_host_handshake_func); | 
|  | 99 | static DECLARE_WORK(kvp_sendkey_work, kvp_send_key); | 
|  | 100 |  | 
|  | 101 | static const char kvp_devname[] = "vmbus/hv_kvp"; | 
|  | 102 | static u8 *recv_buffer; | 
|  | 103 | static struct hvutil_transport *hvt; | 
|  | 104 | /* | 
|  | 105 | * Register the kernel component with the user-level daemon. | 
|  | 106 | * As part of this registration, pass the LIC version number. | 
|  | 107 | * This number has no meaning, it satisfies the registration protocol. | 
|  | 108 | */ | 
|  | 109 | #define HV_DRV_VERSION           "3.1" | 
|  | 110 |  | 
|  | 111 | static void kvp_poll_wrapper(void *channel) | 
|  | 112 | { | 
|  | 113 | /* Transaction is finished, reset the state here to avoid races. */ | 
|  | 114 | kvp_transaction.state = HVUTIL_READY; | 
|  | 115 | tasklet_schedule(&((struct vmbus_channel *)channel)->callback_event); | 
|  | 116 | } | 
|  | 117 |  | 
|  | 118 | static void kvp_register_done(void) | 
|  | 119 | { | 
|  | 120 | /* | 
|  | 121 | * If we're still negotiating with the host cancel the timeout | 
|  | 122 | * work to not poll the channel twice. | 
|  | 123 | */ | 
|  | 124 | pr_debug("KVP: userspace daemon registered\n"); | 
|  | 125 | cancel_delayed_work_sync(&kvp_host_handshake_work); | 
|  | 126 | hv_poll_channel(kvp_transaction.recv_channel, kvp_poll_wrapper); | 
|  | 127 | } | 
|  | 128 |  | 
|  | 129 | static void | 
|  | 130 | kvp_register(int reg_value) | 
|  | 131 | { | 
|  | 132 |  | 
|  | 133 | struct hv_kvp_msg *kvp_msg; | 
|  | 134 | char *version; | 
|  | 135 |  | 
|  | 136 | kvp_msg = kzalloc(sizeof(*kvp_msg), GFP_KERNEL); | 
|  | 137 |  | 
|  | 138 | if (kvp_msg) { | 
|  | 139 | version = kvp_msg->body.kvp_register.version; | 
|  | 140 | kvp_msg->kvp_hdr.operation = reg_value; | 
|  | 141 | strcpy(version, HV_DRV_VERSION); | 
|  | 142 |  | 
|  | 143 | hvutil_transport_send(hvt, kvp_msg, sizeof(*kvp_msg), | 
|  | 144 | kvp_register_done); | 
|  | 145 | kfree(kvp_msg); | 
|  | 146 | } | 
|  | 147 | } | 
|  | 148 |  | 
|  | 149 | static void kvp_timeout_func(struct work_struct *dummy) | 
|  | 150 | { | 
|  | 151 | /* | 
|  | 152 | * If the timer fires, the user-mode component has not responded; | 
|  | 153 | * process the pending transaction. | 
|  | 154 | */ | 
|  | 155 | kvp_respond_to_host(NULL, HV_E_FAIL); | 
|  | 156 |  | 
|  | 157 | hv_poll_channel(kvp_transaction.recv_channel, kvp_poll_wrapper); | 
|  | 158 | } | 
|  | 159 |  | 
|  | 160 | static void kvp_host_handshake_func(struct work_struct *dummy) | 
|  | 161 | { | 
|  | 162 | tasklet_schedule(&kvp_transaction.recv_channel->callback_event); | 
|  | 163 | } | 
|  | 164 |  | 
|  | 165 | static int kvp_handle_handshake(struct hv_kvp_msg *msg) | 
|  | 166 | { | 
|  | 167 | switch (msg->kvp_hdr.operation) { | 
|  | 168 | case KVP_OP_REGISTER: | 
|  | 169 | dm_reg_value = KVP_OP_REGISTER; | 
|  | 170 | pr_info("KVP: IP injection functionality not available\n"); | 
|  | 171 | pr_info("KVP: Upgrade the KVP daemon\n"); | 
|  | 172 | break; | 
|  | 173 | case KVP_OP_REGISTER1: | 
|  | 174 | dm_reg_value = KVP_OP_REGISTER1; | 
|  | 175 | break; | 
|  | 176 | default: | 
|  | 177 | pr_info("KVP: incompatible daemon\n"); | 
|  | 178 | pr_info("KVP: KVP version: %d, Daemon version: %d\n", | 
|  | 179 | KVP_OP_REGISTER1, msg->kvp_hdr.operation); | 
|  | 180 | return -EINVAL; | 
|  | 181 | } | 
|  | 182 |  | 
|  | 183 | /* | 
|  | 184 | * We have a compatible daemon; complete the handshake. | 
|  | 185 | */ | 
|  | 186 | pr_debug("KVP: userspace daemon ver. %d connected\n", | 
|  | 187 | msg->kvp_hdr.operation); | 
|  | 188 | kvp_register(dm_reg_value); | 
|  | 189 |  | 
|  | 190 | return 0; | 
|  | 191 | } | 
|  | 192 |  | 
|  | 193 |  | 
|  | 194 | /* | 
|  | 195 | * Callback when data is received from user mode. | 
|  | 196 | */ | 
|  | 197 |  | 
|  | 198 | static int kvp_on_msg(void *msg, int len) | 
|  | 199 | { | 
|  | 200 | struct hv_kvp_msg *message = (struct hv_kvp_msg *)msg; | 
|  | 201 | struct hv_kvp_msg_enumerate *data; | 
|  | 202 | int	error = 0; | 
|  | 203 |  | 
|  | 204 | if (len < sizeof(*message)) | 
|  | 205 | return -EINVAL; | 
|  | 206 |  | 
|  | 207 | /* | 
|  | 208 | * If we are negotiating the version information | 
|  | 209 | * with the daemon; handle that first. | 
|  | 210 | */ | 
|  | 211 |  | 
|  | 212 | if (kvp_transaction.state < HVUTIL_READY) { | 
|  | 213 | return kvp_handle_handshake(message); | 
|  | 214 | } | 
|  | 215 |  | 
|  | 216 | /* We didn't send anything to userspace so the reply is spurious */ | 
|  | 217 | if (kvp_transaction.state < HVUTIL_USERSPACE_REQ) | 
|  | 218 | return -EINVAL; | 
|  | 219 |  | 
|  | 220 | kvp_transaction.state = HVUTIL_USERSPACE_RECV; | 
|  | 221 |  | 
|  | 222 | /* | 
|  | 223 | * Based on the version of the daemon, we propagate errors from the | 
|  | 224 | * daemon differently. | 
|  | 225 | */ | 
|  | 226 |  | 
|  | 227 | data = &message->body.kvp_enum_data; | 
|  | 228 |  | 
|  | 229 | switch (dm_reg_value) { | 
|  | 230 | case KVP_OP_REGISTER: | 
|  | 231 | /* | 
|  | 232 | * Null string is used to pass back error condition. | 
|  | 233 | */ | 
|  | 234 | if (data->data.key[0] == 0) | 
|  | 235 | error = HV_S_CONT; | 
|  | 236 | break; | 
|  | 237 |  | 
|  | 238 | case KVP_OP_REGISTER1: | 
|  | 239 | /* | 
|  | 240 | * We use the message header information from | 
|  | 241 | * the user level daemon to transmit errors. | 
|  | 242 | */ | 
|  | 243 | error = message->error; | 
|  | 244 | break; | 
|  | 245 | } | 
|  | 246 |  | 
|  | 247 | /* | 
|  | 248 | * Complete the transaction by forwarding the key value | 
|  | 249 | * to the host. But first, cancel the timeout. | 
|  | 250 | */ | 
|  | 251 | if (cancel_delayed_work_sync(&kvp_timeout_work)) { | 
|  | 252 | kvp_respond_to_host(message, error); | 
|  | 253 | hv_poll_channel(kvp_transaction.recv_channel, kvp_poll_wrapper); | 
|  | 254 | } | 
|  | 255 |  | 
|  | 256 | return 0; | 
|  | 257 | } | 
|  | 258 |  | 
|  | 259 |  | 
|  | 260 | static int process_ob_ipinfo(void *in_msg, void *out_msg, int op) | 
|  | 261 | { | 
|  | 262 | struct hv_kvp_msg *in = in_msg; | 
|  | 263 | struct hv_kvp_ip_msg *out = out_msg; | 
|  | 264 | int len; | 
|  | 265 |  | 
|  | 266 | switch (op) { | 
|  | 267 | case KVP_OP_GET_IP_INFO: | 
|  | 268 | /* | 
|  | 269 | * Transform all parameters into utf16 encoding. | 
|  | 270 | */ | 
|  | 271 | len = utf8s_to_utf16s((char *)in->body.kvp_ip_val.ip_addr, | 
|  | 272 | strlen((char *)in->body.kvp_ip_val.ip_addr), | 
|  | 273 | UTF16_HOST_ENDIAN, | 
|  | 274 | (wchar_t *)out->kvp_ip_val.ip_addr, | 
|  | 275 | MAX_IP_ADDR_SIZE); | 
|  | 276 | if (len < 0) | 
|  | 277 | return len; | 
|  | 278 |  | 
|  | 279 | len = utf8s_to_utf16s((char *)in->body.kvp_ip_val.sub_net, | 
|  | 280 | strlen((char *)in->body.kvp_ip_val.sub_net), | 
|  | 281 | UTF16_HOST_ENDIAN, | 
|  | 282 | (wchar_t *)out->kvp_ip_val.sub_net, | 
|  | 283 | MAX_IP_ADDR_SIZE); | 
|  | 284 | if (len < 0) | 
|  | 285 | return len; | 
|  | 286 |  | 
|  | 287 | len = utf8s_to_utf16s((char *)in->body.kvp_ip_val.gate_way, | 
|  | 288 | strlen((char *)in->body.kvp_ip_val.gate_way), | 
|  | 289 | UTF16_HOST_ENDIAN, | 
|  | 290 | (wchar_t *)out->kvp_ip_val.gate_way, | 
|  | 291 | MAX_GATEWAY_SIZE); | 
|  | 292 | if (len < 0) | 
|  | 293 | return len; | 
|  | 294 |  | 
|  | 295 | len = utf8s_to_utf16s((char *)in->body.kvp_ip_val.dns_addr, | 
|  | 296 | strlen((char *)in->body.kvp_ip_val.dns_addr), | 
|  | 297 | UTF16_HOST_ENDIAN, | 
|  | 298 | (wchar_t *)out->kvp_ip_val.dns_addr, | 
|  | 299 | MAX_IP_ADDR_SIZE); | 
|  | 300 | if (len < 0) | 
|  | 301 | return len; | 
|  | 302 |  | 
|  | 303 | len = utf8s_to_utf16s((char *)in->body.kvp_ip_val.adapter_id, | 
|  | 304 | strlen((char *)in->body.kvp_ip_val.adapter_id), | 
|  | 305 | UTF16_HOST_ENDIAN, | 
|  | 306 | (wchar_t *)out->kvp_ip_val.adapter_id, | 
|  | 307 | MAX_ADAPTER_ID_SIZE); | 
|  | 308 | if (len < 0) | 
|  | 309 | return len; | 
|  | 310 |  | 
|  | 311 | out->kvp_ip_val.dhcp_enabled = | 
|  | 312 | in->body.kvp_ip_val.dhcp_enabled; | 
|  | 313 | out->kvp_ip_val.addr_family = | 
|  | 314 | in->body.kvp_ip_val.addr_family; | 
|  | 315 | } | 
|  | 316 |  | 
|  | 317 | return 0; | 
|  | 318 | } | 
|  | 319 |  | 
|  | 320 | static void process_ib_ipinfo(void *in_msg, void *out_msg, int op) | 
|  | 321 | { | 
|  | 322 | struct hv_kvp_ip_msg *in = in_msg; | 
|  | 323 | struct hv_kvp_msg *out = out_msg; | 
|  | 324 |  | 
|  | 325 | switch (op) { | 
|  | 326 | case KVP_OP_SET_IP_INFO: | 
|  | 327 | /* | 
|  | 328 | * Transform all parameters into utf8 encoding. | 
|  | 329 | */ | 
|  | 330 | utf16s_to_utf8s((wchar_t *)in->kvp_ip_val.ip_addr, | 
|  | 331 | MAX_IP_ADDR_SIZE, | 
|  | 332 | UTF16_LITTLE_ENDIAN, | 
|  | 333 | (__u8 *)out->body.kvp_ip_val.ip_addr, | 
|  | 334 | MAX_IP_ADDR_SIZE); | 
|  | 335 |  | 
|  | 336 | utf16s_to_utf8s((wchar_t *)in->kvp_ip_val.sub_net, | 
|  | 337 | MAX_IP_ADDR_SIZE, | 
|  | 338 | UTF16_LITTLE_ENDIAN, | 
|  | 339 | (__u8 *)out->body.kvp_ip_val.sub_net, | 
|  | 340 | MAX_IP_ADDR_SIZE); | 
|  | 341 |  | 
|  | 342 | utf16s_to_utf8s((wchar_t *)in->kvp_ip_val.gate_way, | 
|  | 343 | MAX_GATEWAY_SIZE, | 
|  | 344 | UTF16_LITTLE_ENDIAN, | 
|  | 345 | (__u8 *)out->body.kvp_ip_val.gate_way, | 
|  | 346 | MAX_GATEWAY_SIZE); | 
|  | 347 |  | 
|  | 348 | utf16s_to_utf8s((wchar_t *)in->kvp_ip_val.dns_addr, | 
|  | 349 | MAX_IP_ADDR_SIZE, | 
|  | 350 | UTF16_LITTLE_ENDIAN, | 
|  | 351 | (__u8 *)out->body.kvp_ip_val.dns_addr, | 
|  | 352 | MAX_IP_ADDR_SIZE); | 
|  | 353 |  | 
|  | 354 | out->body.kvp_ip_val.dhcp_enabled = in->kvp_ip_val.dhcp_enabled; | 
|  | 355 |  | 
|  | 356 | /* fallthrough */ | 
|  | 357 |  | 
|  | 358 | case KVP_OP_GET_IP_INFO: | 
|  | 359 | utf16s_to_utf8s((wchar_t *)in->kvp_ip_val.adapter_id, | 
|  | 360 | MAX_ADAPTER_ID_SIZE, | 
|  | 361 | UTF16_LITTLE_ENDIAN, | 
|  | 362 | (__u8 *)out->body.kvp_ip_val.adapter_id, | 
|  | 363 | MAX_ADAPTER_ID_SIZE); | 
|  | 364 |  | 
|  | 365 | out->body.kvp_ip_val.addr_family = in->kvp_ip_val.addr_family; | 
|  | 366 | } | 
|  | 367 | } | 
|  | 368 |  | 
|  | 369 |  | 
|  | 370 |  | 
|  | 371 |  | 
|  | 372 | static void | 
|  | 373 | kvp_send_key(struct work_struct *dummy) | 
|  | 374 | { | 
|  | 375 | struct hv_kvp_msg *message; | 
|  | 376 | struct hv_kvp_msg *in_msg; | 
|  | 377 | __u8 operation = kvp_transaction.kvp_msg->kvp_hdr.operation; | 
|  | 378 | __u8 pool = kvp_transaction.kvp_msg->kvp_hdr.pool; | 
|  | 379 | __u32 val32; | 
|  | 380 | __u64 val64; | 
|  | 381 | int rc; | 
|  | 382 |  | 
|  | 383 | /* The transaction state is wrong. */ | 
|  | 384 | if (kvp_transaction.state != HVUTIL_HOSTMSG_RECEIVED) | 
|  | 385 | return; | 
|  | 386 |  | 
|  | 387 | message = kzalloc(sizeof(*message), GFP_KERNEL); | 
|  | 388 | if (!message) | 
|  | 389 | return; | 
|  | 390 |  | 
|  | 391 | message->kvp_hdr.operation = operation; | 
|  | 392 | message->kvp_hdr.pool = pool; | 
|  | 393 | in_msg = kvp_transaction.kvp_msg; | 
|  | 394 |  | 
|  | 395 | /* | 
|  | 396 | * The key/value strings sent from the host are encoded in | 
|  | 397 | * in utf16; convert it to utf8 strings. | 
|  | 398 | * The host assures us that the utf16 strings will not exceed | 
|  | 399 | * the max lengths specified. We will however, reserve room | 
|  | 400 | * for the string terminating character - in the utf16s_utf8s() | 
|  | 401 | * function we limit the size of the buffer where the converted | 
|  | 402 | * string is placed to HV_KVP_EXCHANGE_MAX_*_SIZE -1 to guarantee | 
|  | 403 | * that the strings can be properly terminated! | 
|  | 404 | */ | 
|  | 405 |  | 
|  | 406 | switch (message->kvp_hdr.operation) { | 
|  | 407 | case KVP_OP_SET_IP_INFO: | 
|  | 408 | process_ib_ipinfo(in_msg, message, KVP_OP_SET_IP_INFO); | 
|  | 409 | break; | 
|  | 410 | case KVP_OP_GET_IP_INFO: | 
|  | 411 | /* | 
|  | 412 | * We only need to pass on the info of operation, adapter_id | 
|  | 413 | * and addr_family to the userland kvp daemon. | 
|  | 414 | */ | 
|  | 415 | process_ib_ipinfo(in_msg, message, KVP_OP_GET_IP_INFO); | 
|  | 416 | break; | 
|  | 417 | case KVP_OP_SET: | 
|  | 418 | switch (in_msg->body.kvp_set.data.value_type) { | 
|  | 419 | case REG_SZ: | 
|  | 420 | /* | 
|  | 421 | * The value is a string - utf16 encoding. | 
|  | 422 | */ | 
|  | 423 | message->body.kvp_set.data.value_size = | 
|  | 424 | utf16s_to_utf8s( | 
|  | 425 | (wchar_t *)in_msg->body.kvp_set.data.value, | 
|  | 426 | in_msg->body.kvp_set.data.value_size, | 
|  | 427 | UTF16_LITTLE_ENDIAN, | 
|  | 428 | message->body.kvp_set.data.value, | 
|  | 429 | HV_KVP_EXCHANGE_MAX_VALUE_SIZE - 1) + 1; | 
|  | 430 | break; | 
|  | 431 |  | 
|  | 432 | case REG_U32: | 
|  | 433 | /* | 
|  | 434 | * The value is a 32 bit scalar. | 
|  | 435 | * We save this as a utf8 string. | 
|  | 436 | */ | 
|  | 437 | val32 = in_msg->body.kvp_set.data.value_u32; | 
|  | 438 | message->body.kvp_set.data.value_size = | 
|  | 439 | sprintf(message->body.kvp_set.data.value, | 
|  | 440 | "%d", val32) + 1; | 
|  | 441 | break; | 
|  | 442 |  | 
|  | 443 | case REG_U64: | 
|  | 444 | /* | 
|  | 445 | * The value is a 64 bit scalar. | 
|  | 446 | * We save this as a utf8 string. | 
|  | 447 | */ | 
|  | 448 | val64 = in_msg->body.kvp_set.data.value_u64; | 
|  | 449 | message->body.kvp_set.data.value_size = | 
|  | 450 | sprintf(message->body.kvp_set.data.value, | 
|  | 451 | "%llu", val64) + 1; | 
|  | 452 | break; | 
|  | 453 |  | 
|  | 454 | } | 
|  | 455 |  | 
|  | 456 | /* | 
|  | 457 | * The key is always a string - utf16 encoding. | 
|  | 458 | */ | 
|  | 459 | message->body.kvp_set.data.key_size = | 
|  | 460 | utf16s_to_utf8s( | 
|  | 461 | (wchar_t *)in_msg->body.kvp_set.data.key, | 
|  | 462 | in_msg->body.kvp_set.data.key_size, | 
|  | 463 | UTF16_LITTLE_ENDIAN, | 
|  | 464 | message->body.kvp_set.data.key, | 
|  | 465 | HV_KVP_EXCHANGE_MAX_KEY_SIZE - 1) + 1; | 
|  | 466 |  | 
|  | 467 | break; | 
|  | 468 |  | 
|  | 469 | case KVP_OP_GET: | 
|  | 470 | message->body.kvp_get.data.key_size = | 
|  | 471 | utf16s_to_utf8s( | 
|  | 472 | (wchar_t *)in_msg->body.kvp_get.data.key, | 
|  | 473 | in_msg->body.kvp_get.data.key_size, | 
|  | 474 | UTF16_LITTLE_ENDIAN, | 
|  | 475 | message->body.kvp_get.data.key, | 
|  | 476 | HV_KVP_EXCHANGE_MAX_KEY_SIZE - 1) + 1; | 
|  | 477 | break; | 
|  | 478 |  | 
|  | 479 | case KVP_OP_DELETE: | 
|  | 480 | message->body.kvp_delete.key_size = | 
|  | 481 | utf16s_to_utf8s( | 
|  | 482 | (wchar_t *)in_msg->body.kvp_delete.key, | 
|  | 483 | in_msg->body.kvp_delete.key_size, | 
|  | 484 | UTF16_LITTLE_ENDIAN, | 
|  | 485 | message->body.kvp_delete.key, | 
|  | 486 | HV_KVP_EXCHANGE_MAX_KEY_SIZE - 1) + 1; | 
|  | 487 | break; | 
|  | 488 |  | 
|  | 489 | case KVP_OP_ENUMERATE: | 
|  | 490 | message->body.kvp_enum_data.index = | 
|  | 491 | in_msg->body.kvp_enum_data.index; | 
|  | 492 | break; | 
|  | 493 | } | 
|  | 494 |  | 
|  | 495 | kvp_transaction.state = HVUTIL_USERSPACE_REQ; | 
|  | 496 | rc = hvutil_transport_send(hvt, message, sizeof(*message), NULL); | 
|  | 497 | if (rc) { | 
|  | 498 | pr_debug("KVP: failed to communicate to the daemon: %d\n", rc); | 
|  | 499 | if (cancel_delayed_work_sync(&kvp_timeout_work)) { | 
|  | 500 | kvp_respond_to_host(message, HV_E_FAIL); | 
|  | 501 | kvp_transaction.state = HVUTIL_READY; | 
|  | 502 | } | 
|  | 503 | } | 
|  | 504 |  | 
|  | 505 | kfree(message); | 
|  | 506 | } | 
|  | 507 |  | 
|  | 508 | /* | 
|  | 509 | * Send a response back to the host. | 
|  | 510 | */ | 
|  | 511 |  | 
|  | 512 | static void | 
|  | 513 | kvp_respond_to_host(struct hv_kvp_msg *msg_to_host, int error) | 
|  | 514 | { | 
|  | 515 | struct hv_kvp_msg  *kvp_msg; | 
|  | 516 | struct hv_kvp_exchg_msg_value  *kvp_data; | 
|  | 517 | char	*key_name; | 
|  | 518 | char	*value; | 
|  | 519 | struct icmsg_hdr *icmsghdrp; | 
|  | 520 | int	keylen = 0; | 
|  | 521 | int	valuelen = 0; | 
|  | 522 | u32	buf_len; | 
|  | 523 | struct vmbus_channel *channel; | 
|  | 524 | u64	req_id; | 
|  | 525 | int ret; | 
|  | 526 |  | 
|  | 527 | /* | 
|  | 528 | * Copy the global state for completing the transaction. Note that | 
|  | 529 | * only one transaction can be active at a time. | 
|  | 530 | */ | 
|  | 531 |  | 
|  | 532 | buf_len = kvp_transaction.recv_len; | 
|  | 533 | channel = kvp_transaction.recv_channel; | 
|  | 534 | req_id = kvp_transaction.recv_req_id; | 
|  | 535 |  | 
|  | 536 | icmsghdrp = (struct icmsg_hdr *) | 
|  | 537 | &recv_buffer[sizeof(struct vmbuspipe_hdr)]; | 
|  | 538 |  | 
|  | 539 | if (channel->onchannel_callback == NULL) | 
|  | 540 | /* | 
|  | 541 | * We have raced with util driver being unloaded; | 
|  | 542 | * silently return. | 
|  | 543 | */ | 
|  | 544 | return; | 
|  | 545 |  | 
|  | 546 | icmsghdrp->status = error; | 
|  | 547 |  | 
|  | 548 | /* | 
|  | 549 | * If the error parameter is set, terminate the host's enumeration | 
|  | 550 | * on this pool. | 
|  | 551 | */ | 
|  | 552 | if (error) { | 
|  | 553 | /* | 
|  | 554 | * Something failed or we have timed out; | 
|  | 555 | * terminate the current host-side iteration. | 
|  | 556 | */ | 
|  | 557 | goto response_done; | 
|  | 558 | } | 
|  | 559 |  | 
|  | 560 | kvp_msg = (struct hv_kvp_msg *) | 
|  | 561 | &recv_buffer[sizeof(struct vmbuspipe_hdr) + | 
|  | 562 | sizeof(struct icmsg_hdr)]; | 
|  | 563 |  | 
|  | 564 | switch (kvp_transaction.kvp_msg->kvp_hdr.operation) { | 
|  | 565 | case KVP_OP_GET_IP_INFO: | 
|  | 566 | ret = process_ob_ipinfo(msg_to_host, | 
|  | 567 | (struct hv_kvp_ip_msg *)kvp_msg, | 
|  | 568 | KVP_OP_GET_IP_INFO); | 
|  | 569 | if (ret < 0) | 
|  | 570 | icmsghdrp->status = HV_E_FAIL; | 
|  | 571 |  | 
|  | 572 | goto response_done; | 
|  | 573 | case KVP_OP_SET_IP_INFO: | 
|  | 574 | goto response_done; | 
|  | 575 | case KVP_OP_GET: | 
|  | 576 | kvp_data = &kvp_msg->body.kvp_get.data; | 
|  | 577 | goto copy_value; | 
|  | 578 |  | 
|  | 579 | case KVP_OP_SET: | 
|  | 580 | case KVP_OP_DELETE: | 
|  | 581 | goto response_done; | 
|  | 582 |  | 
|  | 583 | default: | 
|  | 584 | break; | 
|  | 585 | } | 
|  | 586 |  | 
|  | 587 | kvp_data = &kvp_msg->body.kvp_enum_data.data; | 
|  | 588 | key_name = msg_to_host->body.kvp_enum_data.data.key; | 
|  | 589 |  | 
|  | 590 | /* | 
|  | 591 | * The windows host expects the key/value pair to be encoded | 
|  | 592 | * in utf16. Ensure that the key/value size reported to the host | 
|  | 593 | * will be less than or equal to the MAX size (including the | 
|  | 594 | * terminating character). | 
|  | 595 | */ | 
|  | 596 | keylen = utf8s_to_utf16s(key_name, strlen(key_name), UTF16_HOST_ENDIAN, | 
|  | 597 | (wchar_t *) kvp_data->key, | 
|  | 598 | (HV_KVP_EXCHANGE_MAX_KEY_SIZE / 2) - 2); | 
|  | 599 | kvp_data->key_size = 2*(keylen + 1); /* utf16 encoding */ | 
|  | 600 |  | 
|  | 601 | copy_value: | 
|  | 602 | value = msg_to_host->body.kvp_enum_data.data.value; | 
|  | 603 | valuelen = utf8s_to_utf16s(value, strlen(value), UTF16_HOST_ENDIAN, | 
|  | 604 | (wchar_t *) kvp_data->value, | 
|  | 605 | (HV_KVP_EXCHANGE_MAX_VALUE_SIZE / 2) - 2); | 
|  | 606 | kvp_data->value_size = 2*(valuelen + 1); /* utf16 encoding */ | 
|  | 607 |  | 
|  | 608 | /* | 
|  | 609 | * If the utf8s to utf16s conversion failed; notify host | 
|  | 610 | * of the error. | 
|  | 611 | */ | 
|  | 612 | if ((keylen < 0) || (valuelen < 0)) | 
|  | 613 | icmsghdrp->status = HV_E_FAIL; | 
|  | 614 |  | 
|  | 615 | kvp_data->value_type = REG_SZ; /* all our values are strings */ | 
|  | 616 |  | 
|  | 617 | response_done: | 
|  | 618 | icmsghdrp->icflags = ICMSGHDRFLAG_TRANSACTION | ICMSGHDRFLAG_RESPONSE; | 
|  | 619 |  | 
|  | 620 | vmbus_sendpacket(channel, recv_buffer, buf_len, req_id, | 
|  | 621 | VM_PKT_DATA_INBAND, 0); | 
|  | 622 | } | 
|  | 623 |  | 
|  | 624 | /* | 
|  | 625 | * This callback is invoked when we get a KVP message from the host. | 
|  | 626 | * The host ensures that only one KVP transaction can be active at a time. | 
|  | 627 | * KVP implementation in Linux needs to forward the key to a user-mde | 
|  | 628 | * component to retrieve the corresponding value. Consequently, we cannot | 
|  | 629 | * respond to the host in the context of this callback. Since the host | 
|  | 630 | * guarantees that at most only one transaction can be active at a time, | 
|  | 631 | * we stash away the transaction state in a set of global variables. | 
|  | 632 | */ | 
|  | 633 |  | 
|  | 634 | void hv_kvp_onchannelcallback(void *context) | 
|  | 635 | { | 
|  | 636 | struct vmbus_channel *channel = context; | 
|  | 637 | u32 recvlen; | 
|  | 638 | u64 requestid; | 
|  | 639 |  | 
|  | 640 | struct hv_kvp_msg *kvp_msg; | 
|  | 641 |  | 
|  | 642 | struct icmsg_hdr *icmsghdrp; | 
|  | 643 | int kvp_srv_version; | 
|  | 644 | static enum {NEGO_NOT_STARTED, | 
|  | 645 | NEGO_IN_PROGRESS, | 
|  | 646 | NEGO_FINISHED} host_negotiatied = NEGO_NOT_STARTED; | 
|  | 647 |  | 
|  | 648 | if (kvp_transaction.state < HVUTIL_READY) { | 
|  | 649 | /* | 
|  | 650 | * If userspace daemon is not connected and host is asking | 
|  | 651 | * us to negotiate we need to delay to not lose messages. | 
|  | 652 | * This is important for Failover IP setting. | 
|  | 653 | */ | 
|  | 654 | if (host_negotiatied == NEGO_NOT_STARTED) { | 
|  | 655 | host_negotiatied = NEGO_IN_PROGRESS; | 
|  | 656 | schedule_delayed_work(&kvp_host_handshake_work, | 
|  | 657 | HV_UTIL_NEGO_TIMEOUT * HZ); | 
|  | 658 | } | 
|  | 659 | return; | 
|  | 660 | } | 
|  | 661 | if (kvp_transaction.state > HVUTIL_READY) | 
|  | 662 | return; | 
|  | 663 |  | 
|  | 664 | vmbus_recvpacket(channel, recv_buffer, PAGE_SIZE * 4, &recvlen, | 
|  | 665 | &requestid); | 
|  | 666 |  | 
|  | 667 | if (recvlen > 0) { | 
|  | 668 | icmsghdrp = (struct icmsg_hdr *)&recv_buffer[ | 
|  | 669 | sizeof(struct vmbuspipe_hdr)]; | 
|  | 670 |  | 
|  | 671 | if (icmsghdrp->icmsgtype == ICMSGTYPE_NEGOTIATE) { | 
|  | 672 | if (vmbus_prep_negotiate_resp(icmsghdrp, | 
|  | 673 | recv_buffer, fw_versions, FW_VER_COUNT, | 
|  | 674 | kvp_versions, KVP_VER_COUNT, | 
|  | 675 | NULL, &kvp_srv_version)) { | 
|  | 676 | pr_info("KVP IC version %d.%d\n", | 
|  | 677 | kvp_srv_version >> 16, | 
|  | 678 | kvp_srv_version & 0xFFFF); | 
|  | 679 | } | 
|  | 680 | } else { | 
|  | 681 | kvp_msg = (struct hv_kvp_msg *)&recv_buffer[ | 
|  | 682 | sizeof(struct vmbuspipe_hdr) + | 
|  | 683 | sizeof(struct icmsg_hdr)]; | 
|  | 684 |  | 
|  | 685 | /* | 
|  | 686 | * Stash away this global state for completing the | 
|  | 687 | * transaction; note transactions are serialized. | 
|  | 688 | */ | 
|  | 689 |  | 
|  | 690 | kvp_transaction.recv_len = recvlen; | 
|  | 691 | kvp_transaction.recv_req_id = requestid; | 
|  | 692 | kvp_transaction.kvp_msg = kvp_msg; | 
|  | 693 |  | 
|  | 694 | if (kvp_transaction.state < HVUTIL_READY) { | 
|  | 695 | /* Userspace is not registered yet */ | 
|  | 696 | kvp_respond_to_host(NULL, HV_E_FAIL); | 
|  | 697 | return; | 
|  | 698 | } | 
|  | 699 | kvp_transaction.state = HVUTIL_HOSTMSG_RECEIVED; | 
|  | 700 |  | 
|  | 701 | /* | 
|  | 702 | * Get the information from the | 
|  | 703 | * user-mode component. | 
|  | 704 | * component. This transaction will be | 
|  | 705 | * completed when we get the value from | 
|  | 706 | * the user-mode component. | 
|  | 707 | * Set a timeout to deal with | 
|  | 708 | * user-mode not responding. | 
|  | 709 | */ | 
|  | 710 | schedule_work(&kvp_sendkey_work); | 
|  | 711 | schedule_delayed_work(&kvp_timeout_work, | 
|  | 712 | HV_UTIL_TIMEOUT * HZ); | 
|  | 713 |  | 
|  | 714 | return; | 
|  | 715 |  | 
|  | 716 | } | 
|  | 717 |  | 
|  | 718 | icmsghdrp->icflags = ICMSGHDRFLAG_TRANSACTION | 
|  | 719 | | ICMSGHDRFLAG_RESPONSE; | 
|  | 720 |  | 
|  | 721 | vmbus_sendpacket(channel, recv_buffer, | 
|  | 722 | recvlen, requestid, | 
|  | 723 | VM_PKT_DATA_INBAND, 0); | 
|  | 724 |  | 
|  | 725 | host_negotiatied = NEGO_FINISHED; | 
|  | 726 | hv_poll_channel(kvp_transaction.recv_channel, kvp_poll_wrapper); | 
|  | 727 | } | 
|  | 728 |  | 
|  | 729 | } | 
|  | 730 |  | 
|  | 731 | static void kvp_on_reset(void) | 
|  | 732 | { | 
|  | 733 | if (cancel_delayed_work_sync(&kvp_timeout_work)) | 
|  | 734 | kvp_respond_to_host(NULL, HV_E_FAIL); | 
|  | 735 | kvp_transaction.state = HVUTIL_DEVICE_INIT; | 
|  | 736 | } | 
|  | 737 |  | 
|  | 738 | int | 
|  | 739 | hv_kvp_init(struct hv_util_service *srv) | 
|  | 740 | { | 
|  | 741 | recv_buffer = srv->recv_buffer; | 
|  | 742 | kvp_transaction.recv_channel = srv->channel; | 
|  | 743 |  | 
|  | 744 | /* | 
|  | 745 | * When this driver loads, the user level daemon that | 
|  | 746 | * processes the host requests may not yet be running. | 
|  | 747 | * Defer processing channel callbacks until the daemon | 
|  | 748 | * has registered. | 
|  | 749 | */ | 
|  | 750 | kvp_transaction.state = HVUTIL_DEVICE_INIT; | 
|  | 751 |  | 
|  | 752 | hvt = hvutil_transport_init(kvp_devname, CN_KVP_IDX, CN_KVP_VAL, | 
|  | 753 | kvp_on_msg, kvp_on_reset); | 
|  | 754 | if (!hvt) | 
|  | 755 | return -EFAULT; | 
|  | 756 |  | 
|  | 757 | return 0; | 
|  | 758 | } | 
|  | 759 |  | 
|  | 760 | void hv_kvp_deinit(void) | 
|  | 761 | { | 
|  | 762 | kvp_transaction.state = HVUTIL_DEVICE_DYING; | 
|  | 763 | cancel_delayed_work_sync(&kvp_host_handshake_work); | 
|  | 764 | cancel_delayed_work_sync(&kvp_timeout_work); | 
|  | 765 | cancel_work_sync(&kvp_sendkey_work); | 
|  | 766 | hvutil_transport_destroy(hvt); | 
|  | 767 | } |