/* * This file is part of the Palacios Virtual Machine Monitor developed * by the V3VEE Project with funding from the United States National * Science Foundation and the Department of Energy. * * The V3VEE Project is a joint project between Northwestern University * and the University of New Mexico. You can find out more at * http://www.v3vee.org * * Copyright (c) 2008, Jack Lange * Copyright (c) 2008, The V3VEE Project * All rights reserved. * * Author: Jack Lange * * This is free software. You are permitted to use, * redistribute, and modify it as specified in the file "V3VEE_LICENSE". */ #include #include #include #include #ifndef CONFIG_DEBUG_IDE #undef PrintDebug #define PrintDebug(fmt, args...) #endif #define NBD_READ_CMD 0x1 #define NBD_WRITE_CMD 0x2 #define NBD_CAPACITY_CMD 0x3 #define NBD_STATUS_OK 0x00 #define NBD_STATUS_ERR 0xff struct cd_state { uint64_t capacity; // in bytes int socket; uint32_t ip_addr; uint16_t port; char disk_name[32]; struct vm_device * ide; uint_t bus; uint_t drive; }; static int send_all(int socket, char * buf, int length) { int bytes_sent = 0; PrintDebug("Sending %d bytes\n", length - bytes_sent); while (bytes_sent < length) { int tmp_bytes = V3_Send(socket, buf + bytes_sent, length - bytes_sent); PrintDebug("Sent %d bytes\n", tmp_bytes); if (tmp_bytes == 0) { PrintError("Connection Closed unexpectedly\n"); return -1; } bytes_sent += tmp_bytes; } return 0; } static int recv_all(int socket, char * buf, int length) { int bytes_read = 0; PrintDebug("Reading %d bytes\n", length - bytes_read); while (bytes_read < length) { int tmp_bytes = V3_Recv(socket, buf + bytes_read, length - bytes_read); PrintDebug("Received %d bytes\n", tmp_bytes); if (tmp_bytes == 0) { PrintError("Connection Closed unexpectedly\n"); return -1; } bytes_read += tmp_bytes; } return 0; } // CDs always read 2048 byte blocks... ? static int cd_read(uint8_t * buf, int block_count, uint64_t lba, void * private_data) { struct vm_device * cd_dev = (struct vm_device *)private_data; struct cd_state * cd = (struct cd_state *)(cd_dev->private_data); uint64_t offset = lba * ATAPI_BLOCK_SIZE; int length = block_count * ATAPI_BLOCK_SIZE; uint8_t status; uint32_t ret_len = 0; char nbd_cmd[4] = {0,0,0,0}; nbd_cmd[0] = NBD_READ_CMD; if (send_all(cd->socket, nbd_cmd, 4) == -1) { PrintError("Error sending capacity command\n"); return -1; } if (send_all(cd->socket, (char *)&offset, 8) == -1) { PrintError("Error sending read offset\n"); return -1; } if (send_all(cd->socket, (char *)&length, 4) == -1) { PrintError("Error sending read length\n"); return -1; } if (recv_all(cd->socket, (char *)&status, 1) == -1) { PrintError("Error receiving status\n"); return -1; } if (status != NBD_STATUS_OK) { PrintError("NBD Error....\n"); return -1; } PrintDebug("Reading Data Ret Length\n"); if (recv_all(cd->socket, (char *)&ret_len, 4) == -1) { PrintError("Error receiving Return read length\n"); return -1; } if (ret_len != length) { PrintError("Read length mismatch (req=%d) (result=%d)\n", length, ret_len); return -1; } PrintDebug("Reading Data (%d bytes)\n", ret_len); if (recv_all(cd->socket, (char *)buf, ret_len) == -1) { PrintError("Read Data Error\n"); return -1; } return 0; } static uint32_t cd_get_capacity(void * private_data) { struct vm_device * cd_dev = (struct vm_device *)private_data; struct cd_state * cd = (struct cd_state *)(cd_dev->private_data); return cd->capacity / ATAPI_BLOCK_SIZE; } static struct v3_cd_ops cd_ops = { .read = cd_read, .get_capacity = cd_get_capacity, }; static int cd_free(struct vm_device * dev) { return 0; } static struct v3_device_ops dev_ops = { .free = cd_free, .reset = NULL, .start = NULL, .stop = NULL, }; static int socket_init(struct cd_state * cd) { char header[64]; PrintDebug("Intializing Net CD\n"); cd->socket = V3_Create_TCP_Socket(); PrintDebug("CD socket: %d\n", cd->socket); PrintDebug("Connecting to: %s:%d\n", v3_inet_ntoa(cd->ip_addr), cd->port); V3_Connect_To_IP(cd->socket, v3_ntohl(cd->ip_addr), cd->port); PrintDebug("Connected to NBD server\n"); //snprintf(header, 64, "V3_NBD_1 %s\n", cd->disk_name); strcpy(header, "V3_NBD_1 "); strncat(header, cd->disk_name, 32); strncat(header, "\n", 1); if (send_all(cd->socket, header, strlen(header)) == -1) { PrintError("Error connecting to Network Block Device: %s\n", cd->disk_name); return -1; } // Cache Capacity { char nbd_cmd[4] = {0,0,0,0}; nbd_cmd[0] = NBD_CAPACITY_CMD; if (send_all(cd->socket, nbd_cmd, 4) == -1) { PrintError("Error sending capacity command\n"); return -1; } if (recv_all(cd->socket, (char *)&(cd->capacity), 8) == -1) { PrintError("Error Receiving Capacity\n"); return -1; } PrintDebug("Capacity: %p\n", (void *)(cd->capacity)); } return 0; } static int cd_init(struct guest_info * vm, void * cfg_data) { struct net_cd_cfg * cfg = (struct net_cd_cfg *)cfg_data; struct cd_state * cd = (struct cd_state *)V3_Malloc(sizeof(struct cd_state)); PrintDebug("Registering Net CD at %s:%d disk=%s\n", cfg->ip_str, cfg->port, cfg->disk_tag); strncpy(cd->disk_name, cfg->disk_tag, sizeof(cd->disk_name)); cd->ip_addr = v3_inet_addr(cfg->ip_str); cd->port = cfg->port; cd->ide = (struct vm_device * )v3_find_dev(vm, cfg->ide); if (cd->ide == 0) { PrintError("Could not find backend %s\n", cfg->ide); return -1; } cd->bus = cfg->bus; cd->drive = cfg->drive; struct vm_device * dev = v3_allocate_device("NET-CD", &dev_ops, cd); if (v3_attach_device(vm, dev) == -1) { PrintError("Could not attach device %s\n", "NET-CD"); return -1; } if (socket_init(cd) == -1) { PrintError("Could not initialize socket connection\n"); return -1; } PrintDebug("Registering CD\n"); if (v3_ide_register_cdrom(cd->ide, cd->bus, cd->drive, "NET-CD", &cd_ops, dev) == -1) { return -1; } PrintDebug("intialization done\n"); return 0; } device_register("NET-CD", cd_init)