vlan.c

嵌入式系统设计与实验教材二源码linux内核移植与编译

	case VLAN_NAME_TYPE_PLUS_VID_NO_PAD:
		/* Put our vlan.VID in the name.
		 * Name will look like:	 vlan5
		 */
		sprintf(new_dev->name, "vlan%i", VLAN_ID);
		break;
	case VLAN_NAME_TYPE_RAW_PLUS_VID_NO_PAD:
		/* Put our vlan.VID in the name.
		 * Name will look like:	 eth0.5
		 */
		sprintf(new_dev->name, "%s.%i", real_dev->name, VLAN_ID);
		break;
	case VLAN_NAME_TYPE_PLUS_VID:
		/* Put our vlan.VID in the name.
		 * Name will look like:	 vlan0005
		 */
	default:
		sprintf(new_dev->name, "vlan%.4i", VLAN_ID);
	};
		    
#ifdef VLAN_DEBUG
	printk(VLAN_DBG "Allocated new name -:%s:-\n", new_dev->name);
#endif
	/* set up method calls */
	new_dev->init = vlan_dev_init;
	new_dev->destructor = vlan_dev_destruct;
	new_dev->features |= NETIF_F_DYNALLOC ; 
	    
	/* new_dev->ifindex = 0;  it will be set when added to
	 * the global list.
	 * iflink is set as well.
	 */
	new_dev->get_stats = vlan_dev_get_stats;
	    
	/* IFF_BROADCAST|IFF_MULTICAST; ??? */
	new_dev->flags = real_dev->flags;
	new_dev->flags &= ~IFF_UP;

	/* Make this thing known as a VLAN device */
	new_dev->priv_flags |= IFF_802_1Q_VLAN;
				
	/* need 4 bytes for extra VLAN header info,
	 * hope the underlying device can handle it.
	 */
	new_dev->mtu = real_dev->mtu;
	new_dev->change_mtu = vlan_dev_change_mtu;

	/* TODO: maybe just assign it to be ETHERNET? */
	new_dev->type = real_dev->type;

	/* Regular ethernet + 4 bytes (18 total). */
	new_dev->hard_header_len = VLAN_HLEN + real_dev->hard_header_len;

	new_dev->priv = kmalloc(sizeof(struct vlan_dev_info),
				GFP_KERNEL);
	VLAN_MEM_DBG("new_dev->priv malloc, addr: %p  size: %i\n",
		     new_dev->priv,
		     sizeof(struct vlan_dev_info));
	    
	if (new_dev->priv == NULL) {
		kfree(new_dev);
		goto out_put_dev;
	}

	memset(new_dev->priv, 0, sizeof(struct vlan_dev_info));

	memcpy(new_dev->broadcast, real_dev->broadcast, real_dev->addr_len);
	memcpy(new_dev->dev_addr, real_dev->dev_addr, real_dev->addr_len);
	new_dev->addr_len = real_dev->addr_len;

	new_dev->open = vlan_dev_open;
	new_dev->stop = vlan_dev_stop;
	new_dev->hard_header = vlan_dev_hard_header;

	new_dev->hard_start_xmit = vlan_dev_hard_start_xmit;
	new_dev->rebuild_header = vlan_dev_rebuild_header;
	new_dev->hard_header_parse = real_dev->hard_header_parse;
	new_dev->set_mac_address = vlan_dev_set_mac_address;
	new_dev->set_multicast_list = vlan_dev_set_multicast_list;

	VLAN_DEV_INFO(new_dev)->vlan_id = VLAN_ID; /* 1 through 0xFFF */
	VLAN_DEV_INFO(new_dev)->real_dev = real_dev;
	VLAN_DEV_INFO(new_dev)->dent = NULL;
	VLAN_DEV_INFO(new_dev)->flags = vlan_default_dev_flags;

#ifdef VLAN_DEBUG
	printk(VLAN_DBG "About to go find the group for idx: %i\n",
	       real_dev->ifindex);
#endif
	    
	/* So, got the sucker initialized, now lets place
	 * it into our local structure.
	 */
	grp = vlan_find_group(real_dev->ifindex);
	if (!grp) { /* need to add a new group */
		grp = kmalloc(sizeof(struct vlan_group), GFP_KERNEL);
		VLAN_MEM_DBG("grp malloc, addr: %p  size: %i\n",
			     grp, sizeof(struct vlan_group));
		if (!grp) {
			kfree(new_dev->priv);
			VLAN_FMEM_DBG("new_dev->priv free, addr: %p\n",
				      new_dev->priv);
			kfree(new_dev);
			VLAN_FMEM_DBG("new_dev free, addr: %p\n", new_dev);

			goto out_put_dev;
		}
					
		printk(KERN_ALERT "VLAN REGISTER:  Allocated new group.\n");
		memset(grp, 0, sizeof(struct vlan_group));
		grp->real_dev_ifindex = real_dev->ifindex;

		br_write_lock_bh(BR_NETPROTO_LOCK);
		grp->next = p802_1Q_vlan_list;
		p802_1Q_vlan_list = grp;
		br_write_unlock_bh(BR_NETPROTO_LOCK);
	}
	    
	grp->vlan_devices[VLAN_ID] = new_dev;
	vlan_proc_add_dev(new_dev); /* create it's proc entry */

	/* TODO: Please check this: RTNL   --Ben */
	rtnl_lock();
	register_netdevice(new_dev);
	rtnl_unlock();
	    
	/* NOTE:  We have a reference to the real device,
	 * so hold on to the reference.
	 */
	MOD_INC_USE_COUNT; /* Add was a success!! */
#ifdef VLAN_DEBUG
	printk(VLAN_DBG "Allocated new device successfully, returning.\n");
#endif
	return new_dev;

out_put_dev:
	dev_put(real_dev);

out_ret_null:
	return NULL;
}

static int vlan_device_event(struct notifier_block *unused, unsigned long event, void *ptr)
{
	struct net_device *dev = (struct net_device *)(ptr);
	struct vlan_group *grp = NULL;
	int i = 0;
	struct net_device *vlandev = NULL;

	switch (event) {
	case NETDEV_CHANGEADDR:
		/* Ignore for now */
		break;

	case NETDEV_GOING_DOWN:
		/* Ignore for now */
		break;

	case NETDEV_DOWN:
		/* TODO:  Please review this code. */
		/* put all related VLANs in the down state too. */
		for (grp = p802_1Q_vlan_list; grp != NULL; grp = grp->next) {
			int flgs = 0;

			for (i = 0; i < VLAN_GROUP_ARRAY_LEN; i++) {
				vlandev = grp->vlan_devices[i];
				if (!vlandev ||
				    (VLAN_DEV_INFO(vlandev)->real_dev != dev) ||
				    (!(vlandev->flags & IFF_UP)))
					continue;

				flgs = vlandev->flags;
				flgs &= ~IFF_UP;
				dev_change_flags(vlandev, flgs);
			}
		}
		break;

	case NETDEV_UP:
		/* TODO:  Please review this code. */
		/* put all related VLANs in the down state too. */
		for (grp = p802_1Q_vlan_list; grp != NULL; grp = grp->next) {
			int flgs;

			for (i = 0; i < VLAN_GROUP_ARRAY_LEN; i++) {
				vlandev = grp->vlan_devices[i];
				if (!vlandev ||
				    (VLAN_DEV_INFO(vlandev)->real_dev != dev) ||
				    (vlandev->flags & IFF_UP))
					continue;
				
				flgs = vlandev->flags;
				flgs |= IFF_UP;
				dev_change_flags(vlandev, flgs);
			}
		}
		break;
		
	case NETDEV_UNREGISTER:
		/* TODO:  Please review this code. */
		/* delete all related VLANs. */
		for (grp = p802_1Q_vlan_list; grp != NULL; grp = grp->next) {
			for (i = 0; i < VLAN_GROUP_ARRAY_LEN; i++) {
				vlandev = grp->vlan_devices[i];
				if (!vlandev ||
				    (VLAN_DEV_INFO(vlandev)->real_dev != dev))
					continue;

				unregister_802_1Q_vlan_dev(
					VLAN_DEV_INFO(vlandev)->real_dev->ifindex,
					VLAN_DEV_INFO(vlandev)->vlan_id,
					0, 0);
				vlandev = NULL;
			}
		}
		break;
	};

	return NOTIFY_DONE;
}

/*
 *	VLAN IOCTL handler.
 *	o execute requested action or pass command to the device driver
 *   arg is really a void* to a vlan_ioctl_args structure.
 */
int vlan_ioctl_handler(unsigned long arg)
{
	int err = 0;
	struct vlan_ioctl_args args;

	/* everything here needs root permissions, except aguably the
	 * hack ioctls for sending packets.  However, I know _I_ don't
	 * want users running that on my network! --BLG
	 */
	if (!capable(CAP_NET_ADMIN))
		return -EPERM;

	if (copy_from_user(&args, (void*)arg,
                           sizeof(struct vlan_ioctl_args)))
		return -EFAULT;

	/* Null terminate this sucker, just in case. */
	args.device1[23] = 0;
	args.u.device2[23] = 0;

#ifdef VLAN_DEBUG
	printk(VLAN_DBG __FUNCTION__ ": args.cmd: %x\n", args.cmd);
#endif

	switch (args.cmd) {
	case SET_VLAN_INGRESS_PRIORITY_CMD:
		err = vlan_dev_set_ingress_priority(args.device1,
						    args.u.skb_priority,
						    args.vlan_qos);
		break;

	case SET_VLAN_EGRESS_PRIORITY_CMD:
		err = vlan_dev_set_egress_priority(args.device1,
						   args.u.skb_priority,
						   args.vlan_qos);
		break;

	case SET_VLAN_FLAG_CMD:
		err = vlan_dev_set_vlan_flag(args.device1,
					     args.u.flag,
					     args.vlan_qos);
		break;

	case SET_VLAN_NAME_TYPE_CMD:
		if ((args.u.name_type >= 0) &&
		    (args.u.name_type < VLAN_NAME_TYPE_HIGHEST)) {
			vlan_name_type = args.u.name_type;
			err = 0;
		} else {
			err = -EINVAL;
		}
		break;

		/* TODO:  Figure out how to pass info back...
		   case GET_VLAN_INGRESS_PRIORITY_IOCTL:
		   err = vlan_dev_get_ingress_priority(args);
		   break;

		   case GET_VLAN_EGRESS_PRIORITY_IOCTL:
		   err = vlan_dev_get_egress_priority(args);
		   break;
		*/

	case ADD_VLAN_CMD:
		/* we have been given the name of the Ethernet Device we want to
		 * talk to:  args.dev1	 We also have the
		 * VLAN ID:  args.u.VID
		 */
		if (register_802_1Q_vlan_device(args.device1, args.u.VID)) {
			err = 0;
		} else {
			err = -EINVAL;
		}
		break;

	case DEL_VLAN_CMD:
		/* Here, the args.dev1 is the actual VLAN we want
		 * to get rid of.
		 */
		err = unregister_802_1Q_vlan_device(args.device1);
		break;

	default:
		/* pass on to underlying device instead?? */
		printk(VLAN_DBG __FUNCTION__ ": Unknown VLAN CMD: %x \n",
		       args.cmd);
		return -EINVAL;
	};

	return err;
}