dsfield.h 1.1 KB
Newer Older
L
Linus Torvalds 已提交
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22
/* include/net/dsfield.h - Manipulation of the Differentiated Services field */

/* Written 1998-2000 by Werner Almesberger, EPFL ICA */


#ifndef __NET_DSFIELD_H
#define __NET_DSFIELD_H

#include <linux/types.h>
#include <linux/ip.h>
#include <linux/ipv6.h>
#include <asm/byteorder.h>


static inline __u8 ipv4_get_dsfield(struct iphdr *iph)
{
	return iph->tos;
}


static inline __u8 ipv6_get_dsfield(struct ipv6hdr *ipv6h)
{
A
Al Viro 已提交
23
	return ntohs(*(__be16 *) ipv6h) >> 4;
L
Linus Torvalds 已提交
24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47
}


static inline void ipv4_change_dsfield(struct iphdr *iph,__u8 mask,
    __u8 value)
{
        __u32 check = ntohs(iph->check);
	__u8 dsfield;

	dsfield = (iph->tos & mask) | value;
	check += iph->tos;
	if ((check+1) >> 16) check = (check+1) & 0xffff;
	check -= dsfield;
	check += check >> 16; /* adjust carry */
	iph->check = htons(check);
	iph->tos = dsfield;
}


static inline void ipv6_change_dsfield(struct ipv6hdr *ipv6h,__u8 mask,
    __u8 value)
{
        __u16 tmp;

A
Al Viro 已提交
48
	tmp = ntohs(*(__be16 *) ipv6h);
L
Linus Torvalds 已提交
49
	tmp = (tmp & ((mask << 4) | 0xf00f)) | (value << 4);
A
Al Viro 已提交
50
	*(__be16 *) ipv6h = htons(tmp);
L
Linus Torvalds 已提交
51 52 53 54
}


#endif