355 lines
8.4 KiB
C
355 lines
8.4 KiB
C
/*
|
|
* Copyright (C) 2007-2010 B.A.T.M.A.N. contributors:
|
|
*
|
|
* Marek Lindner
|
|
*
|
|
* This program is free software; you can redistribute it and/or
|
|
* modify it under the terms of version 2 of the GNU General Public
|
|
* License as published by the Free Software Foundation.
|
|
*
|
|
* This program is distributed in the hope that it will be useful, but
|
|
* WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
|
* General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with this program; if not, write to the Free Software
|
|
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
|
|
* 02110-1301, USA
|
|
*
|
|
*/
|
|
|
|
#include <linux/device.h>
|
|
#include <linux/slab.h>
|
|
#include "main.h"
|
|
#include "device.h"
|
|
#include "send.h"
|
|
#include "types.h"
|
|
#include "hash.h"
|
|
#include "hard-interface.h"
|
|
|
|
static struct class *batman_class;
|
|
|
|
static int Major; /* Major number assigned to our device driver */
|
|
|
|
static const struct file_operations fops = {
|
|
.open = bat_device_open,
|
|
.release = bat_device_release,
|
|
.read = bat_device_read,
|
|
.write = bat_device_write,
|
|
.poll = bat_device_poll,
|
|
};
|
|
|
|
static struct device_client *device_client_hash[256];
|
|
|
|
void bat_device_init(void)
|
|
{
|
|
memset(device_client_hash, 0, sizeof(device_client_hash));
|
|
}
|
|
|
|
int bat_device_setup(void)
|
|
{
|
|
int tmp_major;
|
|
|
|
if (Major)
|
|
return 1;
|
|
|
|
/* register our device - kernel assigns a free major number */
|
|
tmp_major = register_chrdev(0, DRIVER_DEVICE, &fops);
|
|
if (tmp_major < 0) {
|
|
printk(KERN_ERR "batman-adv:"
|
|
"Registering the character device failed with %d\n",
|
|
tmp_major);
|
|
return 0;
|
|
}
|
|
|
|
batman_class = class_create(THIS_MODULE, "batman-adv");
|
|
|
|
if (IS_ERR(batman_class)) {
|
|
printk(KERN_ERR "batman-adv:"
|
|
"Could not register class 'batman-adv'\n");
|
|
return 0;
|
|
}
|
|
|
|
device_create(batman_class, NULL, MKDEV(tmp_major, 0), NULL,
|
|
"batman-adv");
|
|
|
|
Major = tmp_major;
|
|
return 1;
|
|
}
|
|
|
|
void bat_device_destroy(void)
|
|
{
|
|
if (!Major)
|
|
return;
|
|
|
|
device_destroy(batman_class, MKDEV(Major, 0));
|
|
class_destroy(batman_class);
|
|
|
|
/* Unregister the device */
|
|
unregister_chrdev(Major, DRIVER_DEVICE);
|
|
|
|
Major = 0;
|
|
}
|
|
|
|
int bat_device_open(struct inode *inode, struct file *file)
|
|
{
|
|
unsigned int i;
|
|
struct device_client *device_client;
|
|
|
|
device_client = kmalloc(sizeof(struct device_client), GFP_KERNEL);
|
|
|
|
if (!device_client)
|
|
return -ENOMEM;
|
|
|
|
for (i = 0; i < ARRAY_SIZE(device_client_hash); i++) {
|
|
if (!device_client_hash[i]) {
|
|
device_client_hash[i] = device_client;
|
|
break;
|
|
}
|
|
}
|
|
|
|
if (i == ARRAY_SIZE(device_client_hash)) {
|
|
printk(KERN_ERR "batman-adv:"
|
|
"Error - can't add another packet client: "
|
|
"maximum number of clients reached\n");
|
|
kfree(device_client);
|
|
return -EXFULL;
|
|
}
|
|
|
|
INIT_LIST_HEAD(&device_client->queue_list);
|
|
device_client->queue_len = 0;
|
|
device_client->index = i;
|
|
spin_lock_init(&device_client->lock);
|
|
init_waitqueue_head(&device_client->queue_wait);
|
|
|
|
file->private_data = device_client;
|
|
|
|
inc_module_count();
|
|
return 0;
|
|
}
|
|
|
|
int bat_device_release(struct inode *inode, struct file *file)
|
|
{
|
|
struct device_client *device_client =
|
|
(struct device_client *)file->private_data;
|
|
struct device_packet *device_packet;
|
|
struct list_head *list_pos, *list_pos_tmp;
|
|
unsigned long flags;
|
|
|
|
spin_lock_irqsave(&device_client->lock, flags);
|
|
|
|
/* for all packets in the queue ... */
|
|
list_for_each_safe(list_pos, list_pos_tmp, &device_client->queue_list) {
|
|
device_packet = list_entry(list_pos,
|
|
struct device_packet, list);
|
|
|
|
list_del(list_pos);
|
|
kfree(device_packet);
|
|
}
|
|
|
|
device_client_hash[device_client->index] = NULL;
|
|
spin_unlock_irqrestore(&device_client->lock, flags);
|
|
|
|
kfree(device_client);
|
|
dec_module_count();
|
|
|
|
return 0;
|
|
}
|
|
|
|
ssize_t bat_device_read(struct file *file, char __user *buf, size_t count,
|
|
loff_t *ppos)
|
|
{
|
|
struct device_client *device_client =
|
|
(struct device_client *)file->private_data;
|
|
struct device_packet *device_packet;
|
|
int error;
|
|
unsigned long flags;
|
|
|
|
if ((file->f_flags & O_NONBLOCK) && (device_client->queue_len == 0))
|
|
return -EAGAIN;
|
|
|
|
if ((!buf) || (count < sizeof(struct icmp_packet)))
|
|
return -EINVAL;
|
|
|
|
if (!access_ok(VERIFY_WRITE, buf, count))
|
|
return -EFAULT;
|
|
|
|
error = wait_event_interruptible(device_client->queue_wait,
|
|
device_client->queue_len);
|
|
|
|
if (error)
|
|
return error;
|
|
|
|
spin_lock_irqsave(&device_client->lock, flags);
|
|
|
|
device_packet = list_first_entry(&device_client->queue_list,
|
|
struct device_packet, list);
|
|
list_del(&device_packet->list);
|
|
device_client->queue_len--;
|
|
|
|
spin_unlock_irqrestore(&device_client->lock, flags);
|
|
|
|
error = __copy_to_user(buf, &device_packet->icmp_packet,
|
|
sizeof(struct icmp_packet));
|
|
|
|
kfree(device_packet);
|
|
|
|
if (error)
|
|
return -EFAULT;
|
|
|
|
return sizeof(struct icmp_packet);
|
|
}
|
|
|
|
ssize_t bat_device_write(struct file *file, const char __user *buff,
|
|
size_t len, loff_t *off)
|
|
{
|
|
struct device_client *device_client =
|
|
(struct device_client *)file->private_data;
|
|
struct icmp_packet icmp_packet;
|
|
struct orig_node *orig_node;
|
|
struct batman_if *batman_if;
|
|
uint8_t dstaddr[ETH_ALEN];
|
|
unsigned long flags;
|
|
|
|
if (len < sizeof(struct icmp_packet)) {
|
|
bat_dbg(DBG_BATMAN, "batman-adv:"
|
|
"Error - can't send packet from char device: "
|
|
"invalid packet size\n");
|
|
return -EINVAL;
|
|
}
|
|
|
|
if (!access_ok(VERIFY_READ, buff, sizeof(struct icmp_packet)))
|
|
return -EFAULT;
|
|
|
|
if (__copy_from_user(&icmp_packet, buff, sizeof(icmp_packet)))
|
|
return -EFAULT;
|
|
|
|
if (icmp_packet.packet_type != BAT_ICMP) {
|
|
bat_dbg(DBG_BATMAN, "batman-adv:"
|
|
"Error - can't send packet from char device: "
|
|
"got bogus packet type (expected: BAT_ICMP)\n");
|
|
return -EINVAL;
|
|
}
|
|
|
|
if (icmp_packet.msg_type != ECHO_REQUEST) {
|
|
bat_dbg(DBG_BATMAN, "batman-adv:"
|
|
"Error - can't send packet from char device: "
|
|
"got bogus message type (expected: ECHO_REQUEST)\n");
|
|
return -EINVAL;
|
|
}
|
|
|
|
icmp_packet.uid = device_client->index;
|
|
|
|
if (icmp_packet.version != COMPAT_VERSION) {
|
|
icmp_packet.msg_type = PARAMETER_PROBLEM;
|
|
icmp_packet.ttl = COMPAT_VERSION;
|
|
bat_device_add_packet(device_client, &icmp_packet);
|
|
goto out;
|
|
}
|
|
|
|
if (atomic_read(&module_state) != MODULE_ACTIVE)
|
|
goto dst_unreach;
|
|
|
|
spin_lock_irqsave(&orig_hash_lock, flags);
|
|
orig_node = ((struct orig_node *)hash_find(orig_hash, icmp_packet.dst));
|
|
|
|
if (!orig_node)
|
|
goto unlock;
|
|
|
|
if (!orig_node->router)
|
|
goto unlock;
|
|
|
|
batman_if = orig_node->router->if_incoming;
|
|
memcpy(dstaddr, orig_node->router->addr, ETH_ALEN);
|
|
|
|
spin_unlock_irqrestore(&orig_hash_lock, flags);
|
|
|
|
if (!batman_if)
|
|
goto dst_unreach;
|
|
|
|
if (batman_if->if_status != IF_ACTIVE)
|
|
goto dst_unreach;
|
|
|
|
memcpy(icmp_packet.orig,
|
|
batman_if->net_dev->dev_addr,
|
|
ETH_ALEN);
|
|
|
|
send_raw_packet((unsigned char *)&icmp_packet,
|
|
sizeof(struct icmp_packet),
|
|
batman_if, dstaddr);
|
|
|
|
goto out;
|
|
|
|
unlock:
|
|
spin_unlock_irqrestore(&orig_hash_lock, flags);
|
|
dst_unreach:
|
|
icmp_packet.msg_type = DESTINATION_UNREACHABLE;
|
|
bat_device_add_packet(device_client, &icmp_packet);
|
|
out:
|
|
return len;
|
|
}
|
|
|
|
unsigned int bat_device_poll(struct file *file, poll_table *wait)
|
|
{
|
|
struct device_client *device_client =
|
|
(struct device_client *)file->private_data;
|
|
|
|
poll_wait(file, &device_client->queue_wait, wait);
|
|
|
|
if (device_client->queue_len > 0)
|
|
return POLLIN | POLLRDNORM;
|
|
|
|
return 0;
|
|
}
|
|
|
|
void bat_device_add_packet(struct device_client *device_client,
|
|
struct icmp_packet *icmp_packet)
|
|
{
|
|
struct device_packet *device_packet;
|
|
unsigned long flags;
|
|
|
|
device_packet = kmalloc(sizeof(struct device_packet), GFP_ATOMIC);
|
|
|
|
if (!device_packet)
|
|
return;
|
|
|
|
INIT_LIST_HEAD(&device_packet->list);
|
|
memcpy(&device_packet->icmp_packet, icmp_packet,
|
|
sizeof(struct icmp_packet));
|
|
|
|
spin_lock_irqsave(&device_client->lock, flags);
|
|
|
|
/* while waiting for the lock the device_client could have been
|
|
* deleted */
|
|
if (!device_client_hash[icmp_packet->uid]) {
|
|
spin_unlock_irqrestore(&device_client->lock, flags);
|
|
kfree(device_packet);
|
|
return;
|
|
}
|
|
|
|
list_add_tail(&device_packet->list, &device_client->queue_list);
|
|
device_client->queue_len++;
|
|
|
|
if (device_client->queue_len > 100) {
|
|
device_packet = list_first_entry(&device_client->queue_list,
|
|
struct device_packet, list);
|
|
|
|
list_del(&device_packet->list);
|
|
kfree(device_packet);
|
|
device_client->queue_len--;
|
|
}
|
|
|
|
spin_unlock_irqrestore(&device_client->lock, flags);
|
|
|
|
wake_up(&device_client->queue_wait);
|
|
}
|
|
|
|
void bat_device_receive_packet(struct icmp_packet *icmp_packet)
|
|
{
|
|
struct device_client *hash = device_client_hash[icmp_packet->uid];
|
|
|
|
if (hash)
|
|
bat_device_add_packet(hash, icmp_packet);
|
|
}
|