Merge branch 'locking'
[openvswitch] / datapath / chain.c
1 /*
2  * Distributed under the terms of the GNU GPL version 2.
3  * Copyright (c) 2007, 2008 The Board of Trustees of The Leland 
4  * Stanford Junior University
5  */
6
7 #include "chain.h"
8 #include "flow.h"
9 #include "table.h"
10 #include <linux/module.h>
11 #include <linux/rcupdate.h>
12 #include <linux/slab.h>
13 #include <linux/spinlock.h>
14
15 static struct sw_table *(*create_hw_table_hook)(void);
16 static struct module *hw_table_owner;
17 static DEFINE_SPINLOCK(hook_lock);
18
19 /* Attempts to append 'table' to the set of tables in 'chain'.  Returns 0 or
20  * negative error.  If 'table' is null it is assumed that table creation failed
21  * due to out-of-memory. */
22 static int add_table(struct sw_chain *chain, struct sw_table *table)
23 {
24         if (table == NULL)
25                 return -ENOMEM;
26         if (chain->n_tables >= CHAIN_MAX_TABLES) {
27                 printk("too many tables in chain\n");
28                 table->destroy(table);
29                 return -ENOBUFS;
30         }
31         chain->tables[chain->n_tables++] = table;
32         return 0;
33 }
34
35 /* Creates and returns a new chain associated with 'dp'.  Returns NULL if the
36  * chain cannot be created. */
37 struct sw_chain *chain_create(struct datapath *dp)
38 {
39         struct sw_chain *chain = kzalloc(sizeof *chain, GFP_KERNEL);
40         if (chain == NULL)
41                 goto error;
42         chain->dp = dp;
43         chain->owner = try_module_get(hw_table_owner) ? hw_table_owner : NULL;
44         if (chain->owner && create_hw_table_hook) {
45                 struct sw_table *hwtable = create_hw_table_hook();
46                 if (!hwtable || add_table(chain, hwtable))
47                         goto error;
48         }
49
50         if (add_table(chain, table_hash2_create(0x1EDC6F41, TABLE_HASH_MAX_FLOWS,
51                                                 0x741B8CD7, TABLE_HASH_MAX_FLOWS))
52             || add_table(chain, table_linear_create(TABLE_LINEAR_MAX_FLOWS)))
53                 goto error;
54         return chain;
55
56 error:
57         if (chain)
58                 chain_destroy(chain);
59         return NULL;
60 }
61
62 /* Searches 'chain' for a flow matching 'key', which must not have any wildcard
63  * fields.  Returns the flow if successful, otherwise a null pointer.
64  *
65  * Caller must hold rcu_read_lock or dp_mutex. */
66 struct sw_flow *chain_lookup(struct sw_chain *chain,
67                          const struct sw_flow_key *key)
68 {
69         int i;
70
71         BUG_ON(key->wildcards);
72         for (i = 0; i < chain->n_tables; i++) {
73                 struct sw_table *t = chain->tables[i];
74                 struct sw_flow *flow = t->lookup(t, key);
75                 if (flow)
76                         return flow;
77         }
78         return NULL;
79 }
80
81 /* Inserts 'flow' into 'chain', replacing any duplicate flow.  Returns 0 if
82  * successful or a negative error.
83  *
84  * If successful, 'flow' becomes owned by the chain, otherwise it is retained
85  * by the caller.
86  *
87  * Caller must hold dp_mutex. */
88 int chain_insert(struct sw_chain *chain, struct sw_flow *flow)
89 {
90         int i;
91
92         might_sleep();
93         for (i = 0; i < chain->n_tables; i++) {
94                 struct sw_table *t = chain->tables[i];
95                 if (t->insert(t, flow))
96                         return 0;
97         }
98
99         return -ENOBUFS;
100 }
101
102 /* Deletes from 'chain' any and all flows that match 'key'.  Returns the number
103  * of flows that were deleted.
104  *
105  * Expensive in the general case as currently implemented, since it requires
106  * iterating through the entire contents of each table for keys that contain
107  * wildcards.  Relatively cheap for fully specified keys.
108  *
109  * Caller must hold dp_mutex. */
110 int chain_delete(struct sw_chain *chain, const struct sw_flow_key *key, 
111                 uint16_t priority, int strict)
112 {
113         int count = 0;
114         int i;
115
116         might_sleep();
117         for (i = 0; i < chain->n_tables; i++) {
118                 struct sw_table *t = chain->tables[i];
119                 count += t->delete(t, key, priority, strict);
120         }
121
122         return count;
123 }
124
125 /* Performs timeout processing on all the tables in 'chain'.  Returns the
126  * number of flow entries deleted through expiration.
127  *
128  * Expensive as currently implemented, since it iterates through the entire
129  * contents of each table.
130  *
131  * Caller must not hold dp_mutex, because individual tables take and release it
132  * as necessary. */
133 int chain_timeout(struct sw_chain *chain)
134 {
135         int count = 0;
136         int i;
137
138         might_sleep();
139         for (i = 0; i < chain->n_tables; i++) {
140                 struct sw_table *t = chain->tables[i];
141                 count += t->timeout(chain->dp, t);
142         }
143         return count;
144 }
145
146 /* Destroys 'chain', which must not have any users. */
147 void chain_destroy(struct sw_chain *chain)
148 {
149         int i;
150
151         synchronize_rcu();
152         for (i = 0; i < chain->n_tables; i++) {
153                 struct sw_table *t = chain->tables[i];
154                 if (t->destroy)
155                         t->destroy(t);
156         }
157         module_put(chain->owner);
158         kfree(chain);
159 }
160
161 /* Prints statistics for each of the tables in 'chain'. */
162 void chain_print_stats(struct sw_chain *chain)
163 {
164         int i;
165
166         printk("\n");
167         for (i = 0; i < chain->n_tables; i++) {
168                 struct sw_table *t = chain->tables[i];
169                 struct sw_table_stats stats;
170                 t->stats(t, &stats);
171                 printk("%s: %lu/%lu flows\n",
172                                         stats.name, stats.n_flows, stats.max_flows);
173         }
174 }
175
176
177 int chain_set_hw_hook(struct sw_table *(*create_hw_table)(void),
178                       struct module *owner)
179 {
180         int retval = -EBUSY;
181
182         spin_lock(&hook_lock);
183         if (!create_hw_table_hook) {
184                 create_hw_table_hook = create_hw_table;
185                 hw_table_owner = owner;
186                 retval = 0;
187         }
188         spin_unlock(&hook_lock);
189
190         return retval;
191 }
192 EXPORT_SYMBOL(chain_set_hw_hook);
193
194 void chain_clear_hw_hook(void)
195 {
196         create_hw_table_hook = NULL;
197         hw_table_owner = NULL;
198 }
199 EXPORT_SYMBOL(chain_clear_hw_hook);