On 2014-11-03 at 21:42:52 +0100, Stephanie Wallick <stephanie.s.wall...@intel.com> wrote: > This is where we handle media specific packets and transport. The MS driver > interfaces with a media agnostic (MA) driver via a series of transfer pairs. > Transfer pairs consist of a set of functions to pass MA USB packets back > and forth between MA and MS drivers. There is one transfer pair per device > endpoint and one transfer pair for control/management traffic. When the MA > driver needs to send an MA USB packet, it hands the packet off to the MS > layer where the packet is converted into an MS form and sent via TCP over > the underlying ethernet or wireless medium. When the MS driver receives a > packet, it converts it into an MA USB packet and hands it off the the MA > driver for handling. > > In addition, the MS driver provides an interface to inititate connection > events. > Because there are no physical MA USB ports in an MA USB host, the host must be > notified via software when a device is connected. > > Lastly, the MS driver contains a number of ioctl functions that are used by a > utility to adjust medium-related driver parameters and connect or disconnect > the > MA USB host and device drivers. > > Signed-off-by: Sean O. Stalley <sean.stal...@intel.com> > Signed-off-by: Stephanie Wallick <stephanie.s.wall...@intel.com> > --- > drivers/staging/mausb/drivers/mausb_ioctl.c | 373 +++++++++++++++++++ > drivers/staging/mausb/drivers/mausb_ioctl.h | 99 +++++ > drivers/staging/mausb/drivers/mausb_msapi.c | 110 ++++++ > drivers/staging/mausb/drivers/mausb_msapi.h | 232 ++++++++++++ > drivers/staging/mausb/drivers/mausb_tcp-device.c | 147 ++++++++ > drivers/staging/mausb/drivers/mausb_tcp-host.c | 144 ++++++++ > drivers/staging/mausb/drivers/mausb_tcp.c | 446 > +++++++++++++++++++++++ > drivers/staging/mausb/drivers/mausb_tcp.h | 129 +++++++ > 8 files changed, 1680 insertions(+) > create mode 100644 drivers/staging/mausb/drivers/mausb_ioctl.c > create mode 100644 drivers/staging/mausb/drivers/mausb_ioctl.h > create mode 100644 drivers/staging/mausb/drivers/mausb_msapi.c > create mode 100644 drivers/staging/mausb/drivers/mausb_msapi.h > create mode 100644 drivers/staging/mausb/drivers/mausb_tcp-device.c > create mode 100644 drivers/staging/mausb/drivers/mausb_tcp-host.c > create mode 100644 drivers/staging/mausb/drivers/mausb_tcp.c > create mode 100644 drivers/staging/mausb/drivers/mausb_tcp.h > > diff --git a/drivers/staging/mausb/drivers/mausb_ioctl.c > b/drivers/staging/mausb/drivers/mausb_ioctl.c > new file mode 100644 > index 0000000..0c6c6bd > --- /dev/null > +++ b/drivers/staging/mausb/drivers/mausb_ioctl.c
[...] > +/** > + * This function is used to send a message to the user, in other words, the > + * calling process. It basically copies the message one byte at a time. > + * > + * @msg: The message to be sent to the user. > + * @buffer: The buffer in which to put the message. This buffer was given to > + * us to fill. > + */ > +void to_user(char *msg, long unsigned int buffer) > +{ > + int length = (int)strlen(msg); > + int bytes = 0; > + > + while (length && *msg) { > + put_user(*(msg++), (char *)buffer++); > + length--; > + bytes++; > + } Any reason not to use copy_to_user here? That way, access_ok would only need to be executed once for the whole range. In any case, the return value of put_user/copy_to_user will need to be checked. > + > + put_user('\0', (char *)buffer + bytes); > +} [...] > +/** > + * This function is used to read from the device file. From the perspective > of > + * the device, the user is reading information from us. This is one of the > + * entry points to this module. > + * > + * @file: The device file. We don't use it directly, but it's passed in. > + * @buffer: The buffer to put the message into. > + * @length: The max length to be read. > + * @offset: File offset, which we don't use but it is passed in nontheless. > + */ > +static ssize_t mausb_read(struct file *file, char __user *buffer, > + size_t length, loff_t *offset) > +{ > + int bytes_read = 0; > + > + if (*message_point == 0) > + return 0; > + while (length && *message_point) { > + put_user(*(message_point++), buffer++); > + length--; > + bytes_read++; > + } See comment for to_user above. Why not use copy_to_user? > + > + return bytes_read; > +} > + > +/** > + * This function is used to write to the device file. From the perspective of > + * the device, the user is writing information to us. This is one of the > + * entry points to this module. > + * > + * @file: The device file. We don't use it directly, but it's passed in. > + * @buffer: The buffer that holds the message. > + * @length: The length of the message to be written. > + * @offset: File offset, which we don't use but it is passed in nontheless. > + */ > +static ssize_t mausb_write(struct file *file, const char __user *buffer, > + size_t length, loff_t *offset) > +{ > + int i; > + > + for (i = 0; i < length && i < BUFFER; i++) > + get_user(message[i], buffer + i); copy_from_user? In any case, check the return value here as well. > + message_point = message; > + > + return i; > +} > + > +/** > + * This function is used to execute ioctl commands, determined by ioctl_func. > + * > + * @file: The device file. We don't use it directly, but it's passed in. > + * @ioctl_func: This value determines which ioctl function will be > used. > + * @ioctl_buffer: This buffer is used to transfer data to/from the device. > + */ > +long mausb_ioctl(struct file *file, unsigned int ioctl_func, > + unsigned long ioctl_buffer) > +{ This entire function needs return value checks for put_user/get_user. > + int bytes = 0; > + char *msg, *ip_addr; > + char chr; > + int ret, value; > + unsigned long int long_ret; > + > + switch (ioctl_func) { > + case IOCTL_SET_MSG: > + msg = (char *)ioctl_buffer; > + get_user(chr, msg); > + for (bytes = 0; chr && bytes < BUFFER; bytes++, msg++) > + get_user(chr, msg); > + mausb_write(file, (char *)ioctl_buffer, bytes, 0); > + break; > + case IOCTL_GET_MSG: > + bytes = mausb_read(file, (char *)ioctl_buffer, 99, 0); > + put_user('\0', (char *)ioctl_buffer + bytes); > + break; > + case IOCTL_GET_VRSN: > + to_user(DRIVER_VERSION, ioctl_buffer); > + break; > + case IOCTL_GET_NAME: > + to_user(MAUSB_NAME, ioctl_buffer); > + break; > + case IOCTL_GADGET_C: > + ret = gadget_connection(1); > + if (ret >= 0) > + to_user("g_zero connect process complete", > ioctl_buffer); > + else > + to_user("g_zero connect process failed", ioctl_buffer); > + break; > + case IOCTL_GADGET_D: > + ret = gadget_connection(0); > + if (ret >= 0) > + to_user("g_zero disconnect process complete", > + ioctl_buffer); > + else > + to_user("g_zero disconnect process failed", > + ioctl_buffer); > + break; > + case IOCTL_MED_DELAY: > + msg = (char *)ioctl_buffer; > + get_user(chr, msg); > + for (bytes = 0; chr && bytes < BUFFER; bytes++, msg++) > + get_user(chr, msg); > + mausb_write(file, (char *)ioctl_buffer, bytes, 0); > + if (kstrtoint((const char *)message_point, 0, &value) != 0) { > + /* TODO: handle error */ > + } > + ret = set_medium_delay(value); > + sprintf(message_point, "DELAY VALUE: ms: %d, jiffies: %d\n", > + value, ret); > + to_user(message_point, ioctl_buffer); > + break; > + case IOCTL_SET_IP: > + msg = (char *)ioctl_buffer; > + get_user(chr, msg); > + for (bytes = 0; chr && bytes < BUFFER; bytes++, msg++) > + get_user(chr, msg); > + mausb_write(file, (char *)ioctl_buffer, bytes, 0); > + ip_addr = kmalloc(strlen(message_point)+1, GFP_KERNEL); > + if (!ip_addr) { > + printk(KERN_ALERT "Memory allocation failed!\n"); No need to print an error message for memory allocation failures, kmalloc will take care of printing a more extensive message. > + break; > + } > + strcpy(ip_addr, message_point); > + sprintf(message_point, "Connecting to ...\nIP Address: %s\n", > + ip_addr); > + to_user(message_point, ioctl_buffer); > + kfree(ip_addr); > + break; > + case IOCTL_SET_PORT: > + msg = (char *)ioctl_buffer; > + get_user(chr, msg); > + for (bytes = 0; chr && bytes < BUFFER; bytes++, msg++) > + get_user(chr, msg); > + mausb_write(file, (char *)ioctl_buffer, bytes, 0); > + if (kstrtoint((const char *)message_point, 0, &value) != 0) { > + /* TODO: handle error */ > + } > + ret = set_port_no(value); > + sprintf(message_point, "PORT NUMBER:%d, Returned %i\n", value, > + ret); > + to_user(message_point, ioctl_buffer); > + break; > + case IOCTL_SET_IP_DECIMAL: > + msg = (char *)ioctl_buffer; > + get_user(chr, msg); > + for (bytes = 0; chr && bytes < BUFFER; bytes++, msg++) > + get_user(chr, msg); > + mausb_write(file, (char *)ioctl_buffer, bytes, 0); > + if (kstrtoul((const char *)message_point, 0, &long_ret) != 0) { > + /* TODO: handle error */ > + } > + > + ret = set_ip_addr(long_ret); > + sprintf(message_point, "\nDecimal Value:%lx returned %i\n", > + long_ret, ret); > + to_user(message_point, ioctl_buffer); > + break; > + case IOCTL_SET_MAC: > + { > + u8 *mac = kmalloc(6, GFP_KERNEL); > + u8 *buf = (u8 __user *)ioctl_buffer; > + int i, ret; > + if (!mac) { > + pr_err("Memory allocation failed!\n"); See comment above. Since this is only a 6 byte buffer, it's probably easier to just allocate it on the stack. > + break; > + } > + ret = copy_from_user(mac, buf, 6); > + if (ret) { > + pr_err("copy_from_user failed\n"); > + kfree(mac); > + break; > + } > + for (i = 0; i < ETH_ALEN; i++) > + pr_info("mac[%d]=0x%x\n", i, mac[i]); > + ret = set_mac_addr(mac); > + if (ret) > + pr_err("unable to set MAC addr\n"); > + kfree(mac); > + break; > + } > + } > + > + return 0; You probably want to return an error here in case anything went wrong above or if the ioctl number is invalid. > +} [...] > diff --git a/drivers/staging/mausb/drivers/mausb_msapi.c > b/drivers/staging/mausb/drivers/mausb_msapi.c > new file mode 100644 > index 0000000..9dd8fa5 > --- /dev/null > +++ b/drivers/staging/mausb/drivers/mausb_msapi.c [...] > +/** > + * Frees the given ms_pkt and associated buffers. This function is not > + * necessary to use the API, but could be useful on both sides of the > interface. > + */ > +void mausb_free_ms_pkt(struct ms_pkt *pkt) > +{ > + int i; > + void *current_buf; > + > + for (i = 0; i < pkt->nents; ++i) { > + current_buf = pkt[i].kvec->iov_base; > + if (NULL != current_buf) { > + kfree(current_buf); > + } else { > + printk(KERN_DEBUG "%s: cannot find buffer for " > + "kvec #%i in ms_pkt at %p\n", > + __func__, i, pkt->kvec); pr_debug() > + } > + } > + > + kfree(pkt); > + > + return; > +} > +EXPORT_SYMBOL(mausb_free_ms_pkt); > + > +/** > + * Calculates the total length of the data in a ms_pkt. Returns the total > + * length of the data in the ms_pkt, or a negative errno. > + */ > +int mausb_ms_pkt_length(struct ms_pkt *pkt) > +{ > + int i; > + int total_length = 0; > + > + for (i = 0; i < pkt->nents; ++i) > + total_length += pkt[i].kvec->iov_len; > + > + printk(KERN_DEBUG "%s: total *kvec length: %i\n", __func__, > + total_length); pr_debug() > + > + return total_length; > +} [...] > diff --git a/drivers/staging/mausb/drivers/mausb_tcp-device.c > b/drivers/staging/mausb/drivers/mausb_tcp-device.c > new file mode 100644 > index 0000000..28978a0 > --- /dev/null > +++ b/drivers/staging/mausb/drivers/mausb_tcp-device.c [...] > +static int mausb_tcp_device_connect(int on) > +{ > + int ret; > + > + if (on && dev_tcp_medium->socket == NULL) { > + ret = sock_create(PF_INET, SOCK_STREAM, IPPROTO_TCP, > + &dev_tcp_medium->socket); > + > + if (0 > ret) /* TODO: real errorchecking */ > + return ret; > + > + do { > + ret = kernel_connect(dev_tcp_medium->socket, > + &dev_tcp_medium->addr, > + sizeof(dev_tcp_medium->addr_in), O_RDWR); > + printk(KERN_DEBUG "%s:kernel_connect returned %i\n", > + __func__, ret); pr_debug(), here any in several other places across the file... > + > + if (0 > ret) { > + /* poll until we can connect sucessfully */ > + msleep(MAUSB_TCP_DEV_CONNECT_POLL_MS); > + } > + > + > + } while (0 > ret); > + > + /*spawn off a listening thread */ > + dev_tcp_medium->recv_task = kthread_run(mausb_tcp_device_thread, > + NULL, "mausb_tcp_device_thread"); kthread_run might return an ERR_PTR which needs to be handled here. > + } > + > + ret = dev_tcp_medium->ma_driver->device_connect(on); > + > + return ret; > +} [...] > diff --git a/drivers/staging/mausb/drivers/mausb_tcp-host.c > b/drivers/staging/mausb/drivers/mausb_tcp-host.c > new file mode 100644 > index 0000000..0302031 > --- /dev/null > +++ b/drivers/staging/mausb/drivers/mausb_tcp-host.c [...] > +static int mausb_tcp_host_connect(int on) > +{ > + int ret; > + > + if (on) { > + ret = kernel_bind(host_tcp_medium->setup_socket, > + &host_tcp_medium->addr, > + sizeof(host_tcp_medium->addr_in)); Missing error handling. > + > + ret = kernel_listen(host_tcp_medium->setup_socket, > + MAUSB_TCP_MAX_NUM_CHANNELS); Missing error handling. > + printk(KERN_DEBUG "%s: kernel_listen returned %i\n", > + __func__, ret); > + > + ret = kernel_accept(host_tcp_medium->setup_socket, > + &host_tcp_medium->socket, 0); > + printk(KERN_DEBUG "%s:kernel_accept returned %i\n", > + __func__, ret); > + > + if (0 > ret) > + return ret; kernel_accept might return negative values in case of an error, which needs to be handled properly here. > + > + if (NULL == host_tcp_medium->recv_task) { > + host_tcp_medium->recv_task = kthread_run( > + mausb_tcp_host_thread, NULL, > + "mausb_tcp_host_thread"); > + } > + } > + > + ret = host_tcp_medium->ma_driver->device_connect(on); > + > + return ret; > +} [...] > diff --git a/drivers/staging/mausb/drivers/mausb_tcp.c > b/drivers/staging/mausb/drivers/mausb_tcp.c > new file mode 100644 > index 0000000..291139e > --- /dev/null > +++ b/drivers/staging/mausb/drivers/mausb_tcp.c [...] > +int mausb_tcp_receive_loop(struct mausb_tcp_medium *tcp_medium) > +{ > + struct msghdr msg; > + struct ms_pkt *pkt; > + int data_rcvd = 0; > + > + mausb_tcp_init_msg(&msg); > + > + while (!kthread_should_stop()) { > + > + pkt = kzalloc(sizeof(struct ms_pkt), GFP_KERNEL); Missing return value check. > + > + printk(KERN_DEBUG "%s: preparing to receive data\n", > + __func__); > + > + data_rcvd = mausb_tcp_receive_packet(tcp_medium, &msg, pkt); > + > + if (0 >= data_rcvd) { > + printk(KERN_DEBUG "%s: received no data (err %i)\n", > + __func__, data_rcvd); > + > + sock_release(tcp_medium->socket); > + return data_rcvd; > + > + } else { > + printk(KERN_DEBUG "%s: received %i bytes\n", > + __func__, data_rcvd); > + } > + > + if (data_rcvd > 0) { > + mausb_transfer_packet(pkt, > + &tcp_medium->ma_driver->pkt_dmux); > + } > + > + data_rcvd = 0; > + } > + > + sock_release(tcp_medium->socket); > + > + return 0; > +} > +EXPORT_SYMBOL(mausb_tcp_receive_loop); [...] > +/** > + * initialization function > + */ > +struct mausb_tcp_medium *alloc_init_mausb_tcp_medium( > + enum mausb_tcp_module_type type) > +{ > + struct mausb_tcp_medium *medium; > + int ret; > + > + printk(KERN_DEBUG "%s\n", __func__); > + > + medium = kzalloc(sizeof(struct mausb_tcp_medium), GFP_KERNEL); > + if (NULL == medium) { > + printk(KERN_DEBUG "%s: memory allocation failed\n", __func__); No error message needed, kmalloc will take care of it. > + return NULL; > + } > + > + ret = sock_create(PF_INET, SOCK_STREAM, IPPROTO_TCP, > + &medium->setup_socket); Error handling is missing. > + > + medium->addr_in.sin_family = AF_INET; > + medium->addr_in.sin_port = htons(MAUSB_TCP_PORT_HOST); > + > + spin_lock_init(&medium->lock); > + > + tcp_medium[type] = medium; > + > + return medium; > +} > +EXPORT_SYMBOL(alloc_init_mausb_tcp_medium); _______________________________________________ devel mailing list de...@linuxdriverproject.org http://driverdev.linuxdriverproject.org/mailman/listinfo/driverdev-devel