[PATCH 1/2] A network device tapset

[leitao at linux dot vnet dot ibm dot com]

A tapset that helps those who are working with network devices.
 This tapset tries to cover the most used functions related to
 these devices

Signed-off-by: Breno LeitÃo <leitao@linux.vnet.ibm.com>
---
 tapset/netdev.stp |  178 +++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 files changed, 178 insertions(+), 0 deletions(-)
 create mode 100644 tapset/netdev.stp

diff --git a/tapset/netdev.stp b/tapset/netdev.stp
new file mode 100644
index 0000000..dfd9c2b
--- /dev/null
+++ b/tapset/netdev.stp
@@ -0,0 +1,178 @@
+// Network devices tapset
+// Copyright (C) 2009, IBM Inc.
+// Author : Breno Leitao <leitao@linux.vnet.ibm.com>
+//
+// This file is free software.  You can redistribute it and/or modify it under
+// the terms of the GNU General Public License (GPL), version 2.
+//
+
+/* A function that returns the device name given the net_device struct */
+function get_netdev_name:string (addr:long) {
+	return kernel_string(@cast(addr, "net_device")->name)
+}
+
+/**
+ * probe netdev.change_mtu - Called when the netdev MTU is changed
+ * @dev: The device that will have the MTU changed
+ * @old_mtu: The current MTU
+ * @new_mtu: The new MTU
+ */
+probe netdev.change_mtu = kernel.function("dev_set_mtu"){
+	old_mtu = @cast($dev, "net_device")->mtu
+	new_mtu = $new_mtu
+	dev = get_netdev_name($dev)
+}
+
+/**
+ * probe netdev.open - Called when the device is opened
+ * @dev: The device that is going to be opened 
+ */
+probe netdev.open = kernel.function("dev_open") {
+	dev = get_netdev_name($dev)
+}
+
+/**
+ * probe netdev.close - Called when the device is closed
+ * @dev: The device that is going to be closed
+ */
+probe netdev.close = kernel.function("dev_close"){
+	dev = get_netdev_name($dev)
+}
+
+/**
+ * probe netdev.hard_tx - Called when the devices is going to TX (hard)
+ * @dev: The device scheduled to transmit
+ * @protocol: The protocol used in the transmission
+ */
+probe netdev.hard_tx = kernel.function("dev_hard_start_xmit"){
+	dev = get_netdev_name($dev)
+	protocol = @cast($skb, "sk_buff")->protocol
+}
+
+/**
+ * probe netdev.queue_tx - Called when the device is going to TX (queued)
+ * @dev: The device scheduled to transmit
+ * @protocol: The protocol used in the transmission
+ */
+probe netdev.queue_tx = kernel.function("dev_queue_xmit"){
+	netdev = @cast($skb, "sk_buff")->dev
+	dev = get_netdev_name(netdev)
+	protocol = @cast($skb, "sk_buff")->protocol
+}	
+
+/**
+ * probe netdev.rx - Called when the device is going to receive a packet
+ * @dev: The device received the packet
+ * @protocol: The packet protocol
+ */
+probe netdev.rx = kernel.function("netif_rx"){
+	netdev = @cast($skb, "sk_buff")->dev
+	dev = get_netdev_name(netdev)
+	protocol = @cast($skb, "sk_buff")->protocol
+}	
+
+/**
+ * probe netdev.rx_skb - Called when the SKB is going to be processed
+ * @dev: The device received the packet
+ * @protocol: The packet protocol
+ */
+probe netdev.rx_skb = kernel.function("netif_receive_skb"){
+	netdev = @cast($skb, "sk_buff")->dev
+	dev = get_netdev_name(netdev)
+	protocol = @cast($skb, "sk_buff")->protocol
+}	
+
+/**
+ * probe netdev.change_rx_flag - Called when the device RX flag will be changed
+ * @dev: The device that will be changed
+ * @flags: The new flags
+ */
+probe netdev.change_rx_flag = kernel.function("dev_change_rx_flags"){
+	dev = get_netdev_name($dev)
+	flags = $flags
+}	
+
+/**
+ * probe netdev.set_promiscuity - Called when the device enters/leaves promiscuity
+ * @dev: The device that is entering/leaving promiscuity mode
+ * @enable: If the device is entering promiscuity mode
+ * @disable: If the device is leaving promiscuity mode
+ * @inc: Count the number of promiscuity openers
+ */
+probe netdev.set_promiscuity = kernel.function("dev_set_promiscuity"){
+	dev = get_netdev_name($dev)
+	if ($inc){
+		enable = 1
+	} else {
+		disable = 1
+	}
+	inc = $inc
+}
+
+/**
+ * probe netdev.ioctl - Called when the device suffers an IOCTL
+ * @cmd: The IOCTL request
+ * @arg: The IOCTL argument (usually the netdev interface)
+ */
+probe netdev.ioctl = kernel.function("dev_ioctl"){
+	cmd = $cmd
+	arg = user_string($arg)
+}
+
+/**
+ * probe netdev.register - Called when the device is registered
+ * @dev: The device that is going to be registered
+ */
+probe netdev.register = kernel.function("register_netdevice"), 
+			kernel.function("register_netdev"){
+	dev = get_netdev_name($dev)
+}
+
+/**
+ * probe netdev.unregister - Called when the device is being unregistered
+ * @dev: The device that is going to be unregistered
+ */
+probe netdev.unregister = kernel.function("unregister_netdev"){ 
+	dev = get_netdev_name($dev)
+}
+
+/**
+ * probe netdev.get_stats - Called when someone asks the device statistics
+ * @dev: The device that is going to provide the statistics 
+ */
+probe netdev.get_stats = kernel.function("dev_get_stats"){
+	dev = get_netdev_name($dev)
+}
+
+/**
+ * probe netdev.change_mac - Called when the netdev has the MAC changed
+ * @dev: The device that will have the MTU changed
+ * @mac_len: The MAC length
+ * @old_mac: The current MAC address
+ * @new_mac: The new MAC address
+ */
+probe netdev.change_mac = kernel.function("dev_set_mac_address"){
+	dev = get_netdev_name($dev)	
+	mac_len = @cast($dev, "net_device")->addr_len
+
+	// Old MAC Address
+	zero = @cast($dev, "net_device")->dev_addr[0]
+	one = @cast($dev, "net_device")->dev_addr[1]
+	two = @cast($dev, "net_device")->dev_addr[2]
+	three = @cast($dev, "net_device")->dev_addr[3]
+	four = @cast($dev, "net_device")->dev_addr[4]
+	five = @cast($dev, "net_device")->dev_addr[5]
+	new_mac = sprintf("%x:%x:%x:%x:%x:%x", zero, one, two, three,
+						four, five)
+
+	// New MAC Address
+	zero = @cast($dev, "sockaddr")->sa_data[0]
+	one  = @cast($dev, "sockaddr")->sa_data[1]
+	two  = @cast($dev, "sockaddr")->sa_data[2]
+	three = @cast($dev, "sockaddr")->sa_data[3]
+	four  = @cast($dev, "sockaddr")->sa_data[4]
+	five = @cast($dev, "sockaddr")->sa_data[5]
+	old_mac = sprintf("%x:%x:%x:%x:%x:%x", zero, one, two, three,
+						four, five)
+}
+
-- 
1.6.0.2