Bart De Schuymer | 1abc55d | 2002-06-01 19:23:47 +0000 | [diff] [blame] | 1 | /* |
Bart De Schuymer | 0cff9e9 | 2002-07-25 12:29:50 +0000 | [diff] [blame] | 2 | * communication.c, v2.0 July 2002 |
Bart De Schuymer | 1abc55d | 2002-06-01 19:23:47 +0000 | [diff] [blame] | 3 | * |
| 4 | * Author: Bart De Schuymer |
| 5 | * |
| 6 | */ |
| 7 | |
Bart De Schuymer | 9895a8e | 2003-01-11 10:14:24 +0000 | [diff] [blame] | 8 | /* |
| 9 | * All the userspace/kernel communication is in this file. |
| 10 | * The other code should not have to know anything about the way the |
| 11 | * kernel likes the structure of the table data. |
| 12 | * The other code works with linked lists, lots of linked lists. |
| 13 | * So, the translation is done here. |
| 14 | */ |
Bart De Schuymer | 1abc55d | 2002-06-01 19:23:47 +0000 | [diff] [blame] | 15 | |
| 16 | #include <getopt.h> |
| 17 | #include <string.h> |
| 18 | #include <errno.h> |
| 19 | #include <stdio.h> |
| 20 | #include <stdlib.h> |
| 21 | #include <sys/socket.h> |
Bart De Schuymer | 1abc55d | 2002-06-01 19:23:47 +0000 | [diff] [blame] | 22 | #include "include/ebtables_u.h" |
| 23 | |
| 24 | extern char* hooknames[NF_BR_NUMHOOKS]; |
| 25 | |
Bart De Schuymer | f81c270 | 2003-07-23 21:07:04 +0000 | [diff] [blame] | 26 | #ifdef KERNEL_64_USERSPACE_32 |
| 27 | #define sparc_cast (uint64_t) |
| 28 | #else |
| 29 | #define sparc_cast |
| 30 | #endif |
| 31 | |
Bart De Schuymer | 1abc55d | 2002-06-01 19:23:47 +0000 | [diff] [blame] | 32 | int sockfd = -1; |
| 33 | |
Bart De Schuymer | 6622a01 | 2005-01-19 21:09:05 +0000 | [diff] [blame] | 34 | static int get_sockfd() |
Bart De Schuymer | 1abc55d | 2002-06-01 19:23:47 +0000 | [diff] [blame] | 35 | { |
Bart De Schuymer | 6622a01 | 2005-01-19 21:09:05 +0000 | [diff] [blame] | 36 | int ret = 0; |
Bart De Schuymer | 1abc55d | 2002-06-01 19:23:47 +0000 | [diff] [blame] | 37 | if (sockfd == -1) { |
| 38 | sockfd = socket(AF_INET, SOCK_RAW, PF_INET); |
Bart De Schuymer | 6622a01 | 2005-01-19 21:09:05 +0000 | [diff] [blame] | 39 | if (sockfd < 0) { |
Bart De Schuymer | 64182a3 | 2004-01-21 20:39:54 +0000 | [diff] [blame] | 40 | ebt_print_error("Problem getting a socket, " |
| 41 | "you probably don't have the right " |
| 42 | "permissions"); |
Bart De Schuymer | 6622a01 | 2005-01-19 21:09:05 +0000 | [diff] [blame] | 43 | ret = -1; |
| 44 | } |
Bart De Schuymer | 1abc55d | 2002-06-01 19:23:47 +0000 | [diff] [blame] | 45 | } |
Bart De Schuymer | 6622a01 | 2005-01-19 21:09:05 +0000 | [diff] [blame] | 46 | return ret; |
Bart De Schuymer | 1abc55d | 2002-06-01 19:23:47 +0000 | [diff] [blame] | 47 | } |
| 48 | |
Bart De Schuymer | 9bfcfd8 | 2005-08-27 16:52:19 +0000 | [diff] [blame] | 49 | static struct ebt_replace *translate_user2kernel(struct ebt_u_replace *u_repl) |
Bart De Schuymer | 1abc55d | 2002-06-01 19:23:47 +0000 | [diff] [blame] | 50 | { |
| 51 | struct ebt_replace *new; |
| 52 | struct ebt_u_entry *e; |
| 53 | struct ebt_u_match_list *m_l; |
| 54 | struct ebt_u_watcher_list *w_l; |
Bart De Schuymer | 60332e0 | 2002-06-23 08:01:47 +0000 | [diff] [blame] | 55 | struct ebt_u_entries *entries; |
Bart De Schuymer | 1abc55d | 2002-06-01 19:23:47 +0000 | [diff] [blame] | 56 | char *p, *base; |
| 57 | int i, j; |
Bart De Schuymer | 60332e0 | 2002-06-23 08:01:47 +0000 | [diff] [blame] | 58 | unsigned int entries_size = 0, *chain_offsets; |
Bart De Schuymer | 1abc55d | 2002-06-01 19:23:47 +0000 | [diff] [blame] | 59 | |
| 60 | new = (struct ebt_replace *)malloc(sizeof(struct ebt_replace)); |
| 61 | if (!new) |
Bart De Schuymer | 64182a3 | 2004-01-21 20:39:54 +0000 | [diff] [blame] | 62 | ebt_print_memory(); |
Bart De Schuymer | 1abc55d | 2002-06-01 19:23:47 +0000 | [diff] [blame] | 63 | new->valid_hooks = u_repl->valid_hooks; |
Bart De Schuymer | 0cff9e9 | 2002-07-25 12:29:50 +0000 | [diff] [blame] | 64 | strcpy(new->name, u_repl->name); |
Bart De Schuymer | 1abc55d | 2002-06-01 19:23:47 +0000 | [diff] [blame] | 65 | new->nentries = u_repl->nentries; |
| 66 | new->num_counters = u_repl->num_counters; |
Bart De Schuymer | f81c270 | 2003-07-23 21:07:04 +0000 | [diff] [blame] | 67 | new->counters = sparc_cast u_repl->counters; |
Bart De Schuymer | 9bfcfd8 | 2005-08-27 16:52:19 +0000 | [diff] [blame] | 68 | chain_offsets = (unsigned int *)malloc(u_repl->num_chains * sizeof(unsigned int)); |
Bart De Schuymer | 6622a01 | 2005-01-19 21:09:05 +0000 | [diff] [blame] | 69 | /* Determine size */ |
Bart De Schuymer | 9bfcfd8 | 2005-08-27 16:52:19 +0000 | [diff] [blame] | 70 | for (i = 0; i < u_repl->num_chains; i++) { |
| 71 | if (!(entries = u_repl->chains[i])) |
| 72 | continue; |
Bart De Schuymer | 60332e0 | 2002-06-23 08:01:47 +0000 | [diff] [blame] | 73 | chain_offsets[i] = entries_size; |
Bart De Schuymer | 1abc55d | 2002-06-01 19:23:47 +0000 | [diff] [blame] | 74 | entries_size += sizeof(struct ebt_entries); |
| 75 | j = 0; |
Bart De Schuymer | e94eaf7 | 2005-08-28 16:06:22 +0000 | [diff] [blame] | 76 | e = entries->entries->next; |
| 77 | while (e != entries->entries) { |
Bart De Schuymer | 1abc55d | 2002-06-01 19:23:47 +0000 | [diff] [blame] | 78 | j++; |
| 79 | entries_size += sizeof(struct ebt_entry); |
| 80 | m_l = e->m_list; |
| 81 | while (m_l) { |
| 82 | entries_size += m_l->m->match_size + |
| 83 | sizeof(struct ebt_entry_match); |
| 84 | m_l = m_l->next; |
| 85 | } |
| 86 | w_l = e->w_list; |
| 87 | while (w_l) { |
| 88 | entries_size += w_l->w->watcher_size + |
| 89 | sizeof(struct ebt_entry_watcher); |
| 90 | w_l = w_l->next; |
| 91 | } |
| 92 | entries_size += e->t->target_size + |
| 93 | sizeof(struct ebt_entry_target); |
| 94 | e = e->next; |
| 95 | } |
Bart De Schuymer | 6622a01 | 2005-01-19 21:09:05 +0000 | [diff] [blame] | 96 | /* A little sanity check */ |
Bart De Schuymer | 60332e0 | 2002-06-23 08:01:47 +0000 | [diff] [blame] | 97 | if (j != entries->nentries) |
Bart De Schuymer | 64182a3 | 2004-01-21 20:39:54 +0000 | [diff] [blame] | 98 | ebt_print_bug("Wrong nentries: %d != %d, hook = %s", j, |
Bart De Schuymer | 60332e0 | 2002-06-23 08:01:47 +0000 | [diff] [blame] | 99 | entries->nentries, entries->name); |
Bart De Schuymer | 1abc55d | 2002-06-01 19:23:47 +0000 | [diff] [blame] | 100 | } |
| 101 | |
| 102 | new->entries_size = entries_size; |
Bart De Schuymer | f81c270 | 2003-07-23 21:07:04 +0000 | [diff] [blame] | 103 | p = (char *)malloc(entries_size); |
| 104 | if (!p) |
Bart De Schuymer | 64182a3 | 2004-01-21 20:39:54 +0000 | [diff] [blame] | 105 | ebt_print_memory(); |
Bart De Schuymer | 1abc55d | 2002-06-01 19:23:47 +0000 | [diff] [blame] | 106 | |
Bart De Schuymer | 6622a01 | 2005-01-19 21:09:05 +0000 | [diff] [blame] | 107 | /* Put everything in one block */ |
Bart De Schuymer | f81c270 | 2003-07-23 21:07:04 +0000 | [diff] [blame] | 108 | new->entries = sparc_cast p; |
Bart De Schuymer | 9bfcfd8 | 2005-08-27 16:52:19 +0000 | [diff] [blame] | 109 | for (i = 0; i < u_repl->num_chains; i++) { |
Bart De Schuymer | 1abc55d | 2002-06-01 19:23:47 +0000 | [diff] [blame] | 110 | struct ebt_entries *hlp; |
| 111 | |
Bart De Schuymer | 1abc55d | 2002-06-01 19:23:47 +0000 | [diff] [blame] | 112 | hlp = (struct ebt_entries *)p; |
Bart De Schuymer | 9bfcfd8 | 2005-08-27 16:52:19 +0000 | [diff] [blame] | 113 | if (!(entries = u_repl->chains[i])) |
| 114 | continue; |
| 115 | if (i < NF_BR_NUMHOOKS) |
Bart De Schuymer | f81c270 | 2003-07-23 21:07:04 +0000 | [diff] [blame] | 116 | new->hook_entry[i] = sparc_cast hlp; |
Bart De Schuymer | 60332e0 | 2002-06-23 08:01:47 +0000 | [diff] [blame] | 117 | hlp->nentries = entries->nentries; |
| 118 | hlp->policy = entries->policy; |
| 119 | strcpy(hlp->name, entries->name); |
| 120 | hlp->counter_offset = entries->counter_offset; |
Bart De Schuymer | 6622a01 | 2005-01-19 21:09:05 +0000 | [diff] [blame] | 121 | hlp->distinguisher = 0; /* Make the kernel see the light */ |
Bart De Schuymer | 1abc55d | 2002-06-01 19:23:47 +0000 | [diff] [blame] | 122 | p += sizeof(struct ebt_entries); |
Bart De Schuymer | e94eaf7 | 2005-08-28 16:06:22 +0000 | [diff] [blame] | 123 | e = entries->entries->next; |
| 124 | while (e != entries->entries) { |
Bart De Schuymer | 1abc55d | 2002-06-01 19:23:47 +0000 | [diff] [blame] | 125 | struct ebt_entry *tmp = (struct ebt_entry *)p; |
| 126 | |
| 127 | tmp->bitmask = e->bitmask | EBT_ENTRY_OR_ENTRIES; |
| 128 | tmp->invflags = e->invflags; |
| 129 | tmp->ethproto = e->ethproto; |
Bart De Schuymer | 0cff9e9 | 2002-07-25 12:29:50 +0000 | [diff] [blame] | 130 | strcpy(tmp->in, e->in); |
| 131 | strcpy(tmp->out, e->out); |
| 132 | strcpy(tmp->logical_in, e->logical_in); |
| 133 | strcpy(tmp->logical_out, e->logical_out); |
Bart De Schuymer | 1abc55d | 2002-06-01 19:23:47 +0000 | [diff] [blame] | 134 | memcpy(tmp->sourcemac, e->sourcemac, |
| 135 | sizeof(tmp->sourcemac)); |
| 136 | memcpy(tmp->sourcemsk, e->sourcemsk, |
| 137 | sizeof(tmp->sourcemsk)); |
| 138 | memcpy(tmp->destmac, e->destmac, sizeof(tmp->destmac)); |
| 139 | memcpy(tmp->destmsk, e->destmsk, sizeof(tmp->destmsk)); |
| 140 | |
| 141 | base = p; |
| 142 | p += sizeof(struct ebt_entry); |
| 143 | m_l = e->m_list; |
| 144 | while (m_l) { |
| 145 | memcpy(p, m_l->m, m_l->m->match_size + |
| 146 | sizeof(struct ebt_entry_match)); |
| 147 | p += m_l->m->match_size + |
| 148 | sizeof(struct ebt_entry_match); |
| 149 | m_l = m_l->next; |
| 150 | } |
| 151 | tmp->watchers_offset = p - base; |
| 152 | w_l = e->w_list; |
| 153 | while (w_l) { |
| 154 | memcpy(p, w_l->w, w_l->w->watcher_size + |
| 155 | sizeof(struct ebt_entry_watcher)); |
| 156 | p += w_l->w->watcher_size + |
| 157 | sizeof(struct ebt_entry_watcher); |
| 158 | w_l = w_l->next; |
| 159 | } |
| 160 | tmp->target_offset = p - base; |
| 161 | memcpy(p, e->t, e->t->target_size + |
| 162 | sizeof(struct ebt_entry_target)); |
Bart De Schuymer | 60332e0 | 2002-06-23 08:01:47 +0000 | [diff] [blame] | 163 | if (!strcmp(e->t->u.name, EBT_STANDARD_TARGET)) { |
| 164 | struct ebt_standard_target *st = |
| 165 | (struct ebt_standard_target *)p; |
Bart De Schuymer | 6622a01 | 2005-01-19 21:09:05 +0000 | [diff] [blame] | 166 | /* Translate the jump to a udc */ |
Bart De Schuymer | 60332e0 | 2002-06-23 08:01:47 +0000 | [diff] [blame] | 167 | if (st->verdict >= 0) |
Bart De Schuymer | 0cff9e9 | 2002-07-25 12:29:50 +0000 | [diff] [blame] | 168 | st->verdict = chain_offsets |
| 169 | [st->verdict + NF_BR_NUMHOOKS]; |
Bart De Schuymer | 60332e0 | 2002-06-23 08:01:47 +0000 | [diff] [blame] | 170 | } |
Bart De Schuymer | 1abc55d | 2002-06-01 19:23:47 +0000 | [diff] [blame] | 171 | p += e->t->target_size + |
| 172 | sizeof(struct ebt_entry_target); |
| 173 | tmp->next_offset = p - base; |
| 174 | e = e->next; |
| 175 | } |
| 176 | } |
| 177 | |
Bart De Schuymer | 6622a01 | 2005-01-19 21:09:05 +0000 | [diff] [blame] | 178 | /* Sanity check */ |
Bart De Schuymer | f81c270 | 2003-07-23 21:07:04 +0000 | [diff] [blame] | 179 | if (p - (char *)new->entries != new->entries_size) |
Bart De Schuymer | 64182a3 | 2004-01-21 20:39:54 +0000 | [diff] [blame] | 180 | ebt_print_bug("Entries_size bug"); |
Bart De Schuymer | 60332e0 | 2002-06-23 08:01:47 +0000 | [diff] [blame] | 181 | free(chain_offsets); |
Bart De Schuymer | 1abc55d | 2002-06-01 19:23:47 +0000 | [diff] [blame] | 182 | return new; |
| 183 | } |
| 184 | |
Bart De Schuymer | 6242374 | 2002-07-14 19:06:20 +0000 | [diff] [blame] | 185 | static void store_table_in_file(char *filename, struct ebt_replace *repl) |
| 186 | { |
| 187 | char *command, *data; |
| 188 | int size; |
| 189 | FILE *file; |
| 190 | |
Bart De Schuymer | 6622a01 | 2005-01-19 21:09:05 +0000 | [diff] [blame] | 191 | /* Start from an empty file with right priviliges */ |
Bart De Schuymer | 6242374 | 2002-07-14 19:06:20 +0000 | [diff] [blame] | 192 | command = (char *)malloc(strlen(filename) + 15); |
| 193 | if (!command) |
Bart De Schuymer | 64182a3 | 2004-01-21 20:39:54 +0000 | [diff] [blame] | 194 | ebt_print_memory(); |
Bart De Schuymer | 6242374 | 2002-07-14 19:06:20 +0000 | [diff] [blame] | 195 | strcpy(command, "cat /dev/null>"); |
| 196 | strcpy(command + 14, filename); |
Bart De Schuymer | 6622a01 | 2005-01-19 21:09:05 +0000 | [diff] [blame] | 197 | if (system(command)) { |
Bart De Schuymer | 64182a3 | 2004-01-21 20:39:54 +0000 | [diff] [blame] | 198 | ebt_print_error("Couldn't create file %s", filename); |
Bart De Schuymer | 6622a01 | 2005-01-19 21:09:05 +0000 | [diff] [blame] | 199 | goto free_command; |
| 200 | } |
Bart De Schuymer | 6242374 | 2002-07-14 19:06:20 +0000 | [diff] [blame] | 201 | strcpy(command, "chmod 600 "); |
| 202 | strcpy(command + 10, filename); |
Bart De Schuymer | 6622a01 | 2005-01-19 21:09:05 +0000 | [diff] [blame] | 203 | if (system(command)) { |
Bart De Schuymer | 64182a3 | 2004-01-21 20:39:54 +0000 | [diff] [blame] | 204 | ebt_print_error("Couldn't chmod file %s", filename); |
Bart De Schuymer | 6622a01 | 2005-01-19 21:09:05 +0000 | [diff] [blame] | 205 | goto free_command; |
| 206 | } |
Bart De Schuymer | 6242374 | 2002-07-14 19:06:20 +0000 | [diff] [blame] | 207 | |
| 208 | size = sizeof(struct ebt_replace) + repl->entries_size + |
| 209 | repl->nentries * sizeof(struct ebt_counter); |
| 210 | data = (char *)malloc(size); |
| 211 | if (!data) |
Bart De Schuymer | 64182a3 | 2004-01-21 20:39:54 +0000 | [diff] [blame] | 212 | ebt_print_memory(); |
Bart De Schuymer | 6242374 | 2002-07-14 19:06:20 +0000 | [diff] [blame] | 213 | memcpy(data, repl, sizeof(struct ebt_replace)); |
Bart De Schuymer | f81c270 | 2003-07-23 21:07:04 +0000 | [diff] [blame] | 214 | memcpy(data + sizeof(struct ebt_replace), (char *)repl->entries, |
Bart De Schuymer | 6242374 | 2002-07-14 19:06:20 +0000 | [diff] [blame] | 215 | repl->entries_size); |
Bart De Schuymer | 6622a01 | 2005-01-19 21:09:05 +0000 | [diff] [blame] | 216 | /* Initialize counters to zero, deliver_counters() can update them */ |
Bart De Schuymer | 6242374 | 2002-07-14 19:06:20 +0000 | [diff] [blame] | 217 | memset(data + sizeof(struct ebt_replace) + repl->entries_size, |
| 218 | 0, repl->nentries * sizeof(struct ebt_counter)); |
Bart De Schuymer | 6622a01 | 2005-01-19 21:09:05 +0000 | [diff] [blame] | 219 | if (!(file = fopen(filename, "wb"))) { |
Bart De Schuymer | 64182a3 | 2004-01-21 20:39:54 +0000 | [diff] [blame] | 220 | ebt_print_error("Couldn't open file %s", filename); |
Bart De Schuymer | 6622a01 | 2005-01-19 21:09:05 +0000 | [diff] [blame] | 221 | goto free_data; |
| 222 | } else if (fwrite(data, sizeof(char), size, file) != size) |
Bart De Schuymer | 64182a3 | 2004-01-21 20:39:54 +0000 | [diff] [blame] | 223 | ebt_print_error("Couldn't write everything to file %s", |
| 224 | filename); |
Bart De Schuymer | 6242374 | 2002-07-14 19:06:20 +0000 | [diff] [blame] | 225 | fclose(file); |
Bart De Schuymer | 6622a01 | 2005-01-19 21:09:05 +0000 | [diff] [blame] | 226 | free_data: |
Bart De Schuymer | 6242374 | 2002-07-14 19:06:20 +0000 | [diff] [blame] | 227 | free(data); |
Bart De Schuymer | 6622a01 | 2005-01-19 21:09:05 +0000 | [diff] [blame] | 228 | free_command: |
| 229 | free(command); |
Bart De Schuymer | 6242374 | 2002-07-14 19:06:20 +0000 | [diff] [blame] | 230 | } |
| 231 | |
Bart De Schuymer | 8339ff1 | 2004-01-14 20:05:27 +0000 | [diff] [blame] | 232 | void ebt_deliver_table(struct ebt_u_replace *u_repl) |
Bart De Schuymer | 1abc55d | 2002-06-01 19:23:47 +0000 | [diff] [blame] | 233 | { |
| 234 | socklen_t optlen; |
| 235 | struct ebt_replace *repl; |
| 236 | |
Bart De Schuymer | 6622a01 | 2005-01-19 21:09:05 +0000 | [diff] [blame] | 237 | /* Translate the struct ebt_u_replace to a struct ebt_replace */ |
Bart De Schuymer | 1abc55d | 2002-06-01 19:23:47 +0000 | [diff] [blame] | 238 | repl = translate_user2kernel(u_repl); |
Bart De Schuymer | 6242374 | 2002-07-14 19:06:20 +0000 | [diff] [blame] | 239 | if (u_repl->filename != NULL) { |
| 240 | store_table_in_file(u_repl->filename, repl); |
| 241 | return; |
| 242 | } |
Bart De Schuymer | 6622a01 | 2005-01-19 21:09:05 +0000 | [diff] [blame] | 243 | /* Give the data to the kernel */ |
Bart De Schuymer | 0cff9e9 | 2002-07-25 12:29:50 +0000 | [diff] [blame] | 244 | optlen = sizeof(struct ebt_replace) + repl->entries_size; |
Bart De Schuymer | 6622a01 | 2005-01-19 21:09:05 +0000 | [diff] [blame] | 245 | if (get_sockfd()) |
| 246 | return; |
Bart De Schuymer | 0cb0179 | 2003-05-04 16:52:04 +0000 | [diff] [blame] | 247 | if (!setsockopt(sockfd, IPPROTO_IP, EBT_SO_SET_ENTRIES, repl, optlen)) |
| 248 | return; |
Bart De Schuymer | 6622a01 | 2005-01-19 21:09:05 +0000 | [diff] [blame] | 249 | if (u_repl->command == 8) { /* The ebtables module may not |
| 250 | * yet be loaded with --atomic-commit */ |
Bart De Schuymer | 0cb0179 | 2003-05-04 16:52:04 +0000 | [diff] [blame] | 251 | ebtables_insmod("ebtables"); |
| 252 | if (!setsockopt(sockfd, IPPROTO_IP, EBT_SO_SET_ENTRIES, |
| 253 | repl, optlen)) |
| 254 | return; |
| 255 | } |
| 256 | |
Bart De Schuymer | 64182a3 | 2004-01-21 20:39:54 +0000 | [diff] [blame] | 257 | ebt_print_error("The kernel doesn't support a certain ebtables" |
Bart De Schuymer | 0cb0179 | 2003-05-04 16:52:04 +0000 | [diff] [blame] | 258 | " extension, consider recompiling your kernel or insmod" |
| 259 | " the extension"); |
Bart De Schuymer | 1abc55d | 2002-06-01 19:23:47 +0000 | [diff] [blame] | 260 | } |
| 261 | |
Bart De Schuymer | 6622a01 | 2005-01-19 21:09:05 +0000 | [diff] [blame] | 262 | static int store_counters_in_file(char *filename, struct ebt_u_replace *repl) |
Bart De Schuymer | 6242374 | 2002-07-14 19:06:20 +0000 | [diff] [blame] | 263 | { |
Bart De Schuymer | 6622a01 | 2005-01-19 21:09:05 +0000 | [diff] [blame] | 264 | int size = repl->nentries * sizeof(struct ebt_counter), ret = 0; |
Bart De Schuymer | 0cff9e9 | 2002-07-25 12:29:50 +0000 | [diff] [blame] | 265 | unsigned int entries_size; |
Bart De Schuymer | 6242374 | 2002-07-14 19:06:20 +0000 | [diff] [blame] | 266 | struct ebt_replace hlp; |
| 267 | FILE *file; |
| 268 | |
Bart De Schuymer | 6622a01 | 2005-01-19 21:09:05 +0000 | [diff] [blame] | 269 | if (!(file = fopen(filename, "r+b"))) { |
Bart De Schuymer | 64182a3 | 2004-01-21 20:39:54 +0000 | [diff] [blame] | 270 | ebt_print_error("Could not open file %s", filename); |
Bart De Schuymer | 6622a01 | 2005-01-19 21:09:05 +0000 | [diff] [blame] | 271 | return -1; |
| 272 | } |
| 273 | /* Find out entries_size and then set the file pointer to the |
| 274 | * counters */ |
Bart De Schuymer | 6242374 | 2002-07-14 19:06:20 +0000 | [diff] [blame] | 275 | if (fseek(file, (char *)(&hlp.entries_size) - (char *)(&hlp), SEEK_SET) |
| 276 | || fread(&entries_size, sizeof(char), sizeof(unsigned int), file) != |
| 277 | sizeof(unsigned int) || |
| 278 | fseek(file, entries_size + sizeof(struct ebt_replace), SEEK_SET)) { |
Bart De Schuymer | 64182a3 | 2004-01-21 20:39:54 +0000 | [diff] [blame] | 279 | ebt_print_error("File %s is corrupt", filename); |
Bart De Schuymer | 6622a01 | 2005-01-19 21:09:05 +0000 | [diff] [blame] | 280 | ret = -1; |
| 281 | goto close_file; |
Bart De Schuymer | 6242374 | 2002-07-14 19:06:20 +0000 | [diff] [blame] | 282 | } |
| 283 | if (fwrite(repl->counters, sizeof(char), size, file) != size) { |
Bart De Schuymer | 64182a3 | 2004-01-21 20:39:54 +0000 | [diff] [blame] | 284 | ebt_print_error("Could not write everything to file %s", |
| 285 | filename); |
Bart De Schuymer | 6622a01 | 2005-01-19 21:09:05 +0000 | [diff] [blame] | 286 | ret = -1; |
Bart De Schuymer | 6242374 | 2002-07-14 19:06:20 +0000 | [diff] [blame] | 287 | } |
Bart De Schuymer | 6622a01 | 2005-01-19 21:09:05 +0000 | [diff] [blame] | 288 | close_file: |
Bart De Schuymer | 6242374 | 2002-07-14 19:06:20 +0000 | [diff] [blame] | 289 | fclose(file); |
Bart De Schuymer | 6622a01 | 2005-01-19 21:09:05 +0000 | [diff] [blame] | 290 | return 0; |
Bart De Schuymer | 6242374 | 2002-07-14 19:06:20 +0000 | [diff] [blame] | 291 | } |
| 292 | |
Bart De Schuymer | 6622a01 | 2005-01-19 21:09:05 +0000 | [diff] [blame] | 293 | /* Gets executed after ebt_deliver_table. Delivers the counters to the kernel |
| 294 | * and resets the counterchanges to CNT_NORM */ |
| 295 | void ebt_deliver_counters(struct ebt_u_replace *u_repl, int exec_style) |
Bart De Schuymer | 1abc55d | 2002-06-01 19:23:47 +0000 | [diff] [blame] | 296 | { |
Bart De Schuymer | 1abc55d | 2002-06-01 19:23:47 +0000 | [diff] [blame] | 297 | struct ebt_counter *old, *new, *newcounters; |
| 298 | socklen_t optlen; |
| 299 | struct ebt_replace repl; |
Bart De Schuymer | 482fbaf | 2005-08-28 13:16:25 +0000 | [diff] [blame] | 300 | struct ebt_cntchanges *cc = u_repl->cc->next, *cc2; |
Bart De Schuymer | d2ced82 | 2005-01-23 19:19:00 +0000 | [diff] [blame] | 301 | struct ebt_u_entries *entries; |
| 302 | struct ebt_u_entry *next = NULL; |
| 303 | int i, chainnr = 0; |
Bart De Schuymer | 1abc55d | 2002-06-01 19:23:47 +0000 | [diff] [blame] | 304 | |
| 305 | if (u_repl->nentries == 0) |
| 306 | return; |
| 307 | |
| 308 | newcounters = (struct ebt_counter *) |
| 309 | malloc(u_repl->nentries * sizeof(struct ebt_counter)); |
| 310 | if (!newcounters) |
Bart De Schuymer | 64182a3 | 2004-01-21 20:39:54 +0000 | [diff] [blame] | 311 | ebt_print_memory(); |
Bart De Schuymer | 1abc55d | 2002-06-01 19:23:47 +0000 | [diff] [blame] | 312 | memset(newcounters, 0, u_repl->nentries * sizeof(struct ebt_counter)); |
| 313 | old = u_repl->counters; |
| 314 | new = newcounters; |
Bart De Schuymer | 482fbaf | 2005-08-28 13:16:25 +0000 | [diff] [blame] | 315 | while (cc != u_repl->cc) { |
Bart De Schuymer | e94eaf7 | 2005-08-28 16:06:22 +0000 | [diff] [blame] | 316 | if (!next || next == entries->entries) { |
| 317 | while (chainnr < u_repl->num_chains && (!(entries = u_repl->chains[chainnr++]) || |
| 318 | (next = entries->entries->next) == entries->entries)); |
Bart De Schuymer | 482fbaf | 2005-08-28 13:16:25 +0000 | [diff] [blame] | 319 | if (chainnr == u_repl->num_chains) |
| 320 | break; |
Bart De Schuymer | d2ced82 | 2005-01-23 19:19:00 +0000 | [diff] [blame] | 321 | } |
Bart De Schuymer | 8339ff1 | 2004-01-14 20:05:27 +0000 | [diff] [blame] | 322 | if (cc->type == CNT_NORM) { |
Bart De Schuymer | 6622a01 | 2005-01-19 21:09:05 +0000 | [diff] [blame] | 323 | /* 'Normal' rule, meaning we didn't do anything to it |
| 324 | * So, we just copy */ |
Bart De Schuymer | d2ced82 | 2005-01-23 19:19:00 +0000 | [diff] [blame] | 325 | *new = *old; |
Bart De Schuymer | ab611e2 | 2005-02-14 20:20:03 +0000 | [diff] [blame] | 326 | next->cnt = *new; |
| 327 | next->cnt_surplus.pcnt = next->cnt_surplus.bcnt = 0; |
Bart De Schuymer | 482fbaf | 2005-08-28 13:16:25 +0000 | [diff] [blame] | 328 | old++; /* We've used an old counter */ |
| 329 | new++; /* We've set a new counter */ |
Bart De Schuymer | ff58720 | 2005-02-08 20:02:28 +0000 | [diff] [blame] | 330 | next = next->next; |
Bart De Schuymer | 8339ff1 | 2004-01-14 20:05:27 +0000 | [diff] [blame] | 331 | } else if (cc->type == CNT_DEL) { |
Bart De Schuymer | 482fbaf | 2005-08-28 13:16:25 +0000 | [diff] [blame] | 332 | old++; /* Don't use this old counter */ |
Bart De Schuymer | 1abc55d | 2002-06-01 19:23:47 +0000 | [diff] [blame] | 333 | } else { |
Bart De Schuymer | 0436eda | 2005-03-28 20:29:37 +0000 | [diff] [blame] | 334 | if (cc->type == CNT_CHANGE) { |
Bart De Schuymer | 0436eda | 2005-03-28 20:29:37 +0000 | [diff] [blame] | 335 | if (cc->change % 3 == 1) |
| 336 | new->pcnt = old->pcnt + next->cnt_surplus.pcnt; |
| 337 | else if (cc->change % 3 == 2) |
| 338 | new->pcnt = old->pcnt - next->cnt_surplus.pcnt; |
| 339 | else |
| 340 | new->pcnt = next->cnt.pcnt; |
| 341 | if (cc->change / 3 == 1) |
| 342 | new->bcnt = old->bcnt + next->cnt_surplus.bcnt; |
| 343 | else if (cc->change / 3 == 2) |
| 344 | new->bcnt = old->bcnt - next->cnt_surplus.bcnt; |
| 345 | else |
| 346 | new->bcnt = next->cnt.bcnt; |
Bart De Schuymer | ab611e2 | 2005-02-14 20:20:03 +0000 | [diff] [blame] | 347 | } else |
| 348 | *new = next->cnt; |
| 349 | next->cnt = *new; |
| 350 | next->cnt_surplus.pcnt = next->cnt_surplus.bcnt = 0; |
Bart De Schuymer | d2ced82 | 2005-01-23 19:19:00 +0000 | [diff] [blame] | 351 | if (cc->type == CNT_ADD) |
| 352 | new++; |
| 353 | else { |
| 354 | old++; |
| 355 | new++; |
| 356 | } |
Bart De Schuymer | ff58720 | 2005-02-08 20:02:28 +0000 | [diff] [blame] | 357 | next = next->next; |
Bart De Schuymer | 1abc55d | 2002-06-01 19:23:47 +0000 | [diff] [blame] | 358 | } |
Bart De Schuymer | 8339ff1 | 2004-01-14 20:05:27 +0000 | [diff] [blame] | 359 | cc = cc->next; |
Bart De Schuymer | 1abc55d | 2002-06-01 19:23:47 +0000 | [diff] [blame] | 360 | } |
| 361 | |
| 362 | free(u_repl->counters); |
| 363 | u_repl->counters = newcounters; |
| 364 | u_repl->num_counters = u_repl->nentries; |
Bart De Schuymer | 482fbaf | 2005-08-28 13:16:25 +0000 | [diff] [blame] | 365 | /* Reset the counterchanges to CNT_NORM and delete the unused cc */ |
| 366 | i = 0; |
| 367 | cc = u_repl->cc->next; |
| 368 | while (cc != u_repl->cc) { |
| 369 | if (cc->type == CNT_DEL) { |
| 370 | cc->prev->next = cc->next; |
| 371 | cc->next->prev = cc->prev; |
| 372 | cc2 = cc->next; |
| 373 | free(cc); |
| 374 | cc = cc2; |
| 375 | } else { |
| 376 | cc->type = CNT_NORM; |
| 377 | cc->change = 0; |
| 378 | i++; |
| 379 | cc = cc->next; |
| 380 | } |
| 381 | } |
| 382 | if (i != u_repl->nentries) |
| 383 | ebt_print_bug("i != u_repl->nentries"); |
Bart De Schuymer | 6242374 | 2002-07-14 19:06:20 +0000 | [diff] [blame] | 384 | if (u_repl->filename != NULL) { |
| 385 | store_counters_in_file(u_repl->filename, u_repl); |
| 386 | return; |
| 387 | } |
Bart De Schuymer | 1abc55d | 2002-06-01 19:23:47 +0000 | [diff] [blame] | 388 | optlen = u_repl->nentries * sizeof(struct ebt_counter) + |
| 389 | sizeof(struct ebt_replace); |
Bart De Schuymer | 6622a01 | 2005-01-19 21:09:05 +0000 | [diff] [blame] | 390 | /* Now put the stuff in the kernel's struct ebt_replace */ |
Bart De Schuymer | f81c270 | 2003-07-23 21:07:04 +0000 | [diff] [blame] | 391 | repl.counters = sparc_cast u_repl->counters; |
Bart De Schuymer | 1abc55d | 2002-06-01 19:23:47 +0000 | [diff] [blame] | 392 | repl.num_counters = u_repl->num_counters; |
| 393 | memcpy(repl.name, u_repl->name, sizeof(repl.name)); |
| 394 | |
Bart De Schuymer | 6622a01 | 2005-01-19 21:09:05 +0000 | [diff] [blame] | 395 | if (get_sockfd()) |
| 396 | return; |
Bart De Schuymer | 1abc55d | 2002-06-01 19:23:47 +0000 | [diff] [blame] | 397 | if (setsockopt(sockfd, IPPROTO_IP, EBT_SO_SET_COUNTERS, &repl, optlen)) |
Bart De Schuymer | 64182a3 | 2004-01-21 20:39:54 +0000 | [diff] [blame] | 398 | ebt_print_bug("Couldn't update kernel counters"); |
Bart De Schuymer | 6622a01 | 2005-01-19 21:09:05 +0000 | [diff] [blame] | 399 | |
| 400 | if (exec_style != EXEC_STYLE_DAEMON) |
| 401 | return; |
Bart De Schuymer | 1abc55d | 2002-06-01 19:23:47 +0000 | [diff] [blame] | 402 | } |
| 403 | |
| 404 | static int |
| 405 | ebt_translate_match(struct ebt_entry_match *m, struct ebt_u_match_list ***l) |
| 406 | { |
| 407 | struct ebt_u_match_list *new; |
Bart De Schuymer | 6622a01 | 2005-01-19 21:09:05 +0000 | [diff] [blame] | 408 | int ret = 0; |
Bart De Schuymer | 1abc55d | 2002-06-01 19:23:47 +0000 | [diff] [blame] | 409 | |
| 410 | new = (struct ebt_u_match_list *) |
| 411 | malloc(sizeof(struct ebt_u_match_list)); |
| 412 | if (!new) |
Bart De Schuymer | 64182a3 | 2004-01-21 20:39:54 +0000 | [diff] [blame] | 413 | ebt_print_memory(); |
Bart De Schuymer | 1abc55d | 2002-06-01 19:23:47 +0000 | [diff] [blame] | 414 | new->m = (struct ebt_entry_match *) |
| 415 | malloc(m->match_size + sizeof(struct ebt_entry_match)); |
| 416 | if (!new->m) |
Bart De Schuymer | 64182a3 | 2004-01-21 20:39:54 +0000 | [diff] [blame] | 417 | ebt_print_memory(); |
Bart De Schuymer | 1abc55d | 2002-06-01 19:23:47 +0000 | [diff] [blame] | 418 | memcpy(new->m, m, m->match_size + sizeof(struct ebt_entry_match)); |
| 419 | new->next = NULL; |
| 420 | **l = new; |
| 421 | *l = &new->next; |
Bart De Schuymer | 6622a01 | 2005-01-19 21:09:05 +0000 | [diff] [blame] | 422 | if (ebt_find_match(new->m->u.name) == NULL) { |
Bart De Schuymer | 64182a3 | 2004-01-21 20:39:54 +0000 | [diff] [blame] | 423 | ebt_print_error("Kernel match %s unsupported by userspace tool", |
| 424 | new->m->u.name); |
Bart De Schuymer | 6622a01 | 2005-01-19 21:09:05 +0000 | [diff] [blame] | 425 | ret = -1; |
| 426 | } |
| 427 | return ret; |
Bart De Schuymer | 1abc55d | 2002-06-01 19:23:47 +0000 | [diff] [blame] | 428 | } |
| 429 | |
| 430 | static int |
| 431 | ebt_translate_watcher(struct ebt_entry_watcher *w, |
| 432 | struct ebt_u_watcher_list ***l) |
| 433 | { |
| 434 | struct ebt_u_watcher_list *new; |
Bart De Schuymer | 6622a01 | 2005-01-19 21:09:05 +0000 | [diff] [blame] | 435 | int ret = 0; |
Bart De Schuymer | 1abc55d | 2002-06-01 19:23:47 +0000 | [diff] [blame] | 436 | |
| 437 | new = (struct ebt_u_watcher_list *) |
| 438 | malloc(sizeof(struct ebt_u_watcher_list)); |
| 439 | if (!new) |
Bart De Schuymer | 64182a3 | 2004-01-21 20:39:54 +0000 | [diff] [blame] | 440 | ebt_print_memory(); |
Bart De Schuymer | 1abc55d | 2002-06-01 19:23:47 +0000 | [diff] [blame] | 441 | new->w = (struct ebt_entry_watcher *) |
| 442 | malloc(w->watcher_size + sizeof(struct ebt_entry_watcher)); |
| 443 | if (!new->w) |
Bart De Schuymer | 64182a3 | 2004-01-21 20:39:54 +0000 | [diff] [blame] | 444 | ebt_print_memory(); |
Bart De Schuymer | 1abc55d | 2002-06-01 19:23:47 +0000 | [diff] [blame] | 445 | memcpy(new->w, w, w->watcher_size + sizeof(struct ebt_entry_watcher)); |
| 446 | new->next = NULL; |
| 447 | **l = new; |
| 448 | *l = &new->next; |
Bart De Schuymer | 6622a01 | 2005-01-19 21:09:05 +0000 | [diff] [blame] | 449 | if (ebt_find_watcher(new->w->u.name) == NULL) { |
Bart De Schuymer | 64182a3 | 2004-01-21 20:39:54 +0000 | [diff] [blame] | 450 | ebt_print_error("Kernel watcher %s unsupported by userspace " |
| 451 | "tool", new->w->u.name); |
Bart De Schuymer | 6622a01 | 2005-01-19 21:09:05 +0000 | [diff] [blame] | 452 | ret = -1; |
| 453 | } |
| 454 | return ret; |
Bart De Schuymer | 1abc55d | 2002-06-01 19:23:47 +0000 | [diff] [blame] | 455 | } |
| 456 | |
| 457 | static int |
| 458 | ebt_translate_entry(struct ebt_entry *e, unsigned int *hook, int *n, int *cnt, |
Bart De Schuymer | e94eaf7 | 2005-08-28 16:06:22 +0000 | [diff] [blame] | 459 | int *totalcnt, struct ebt_u_entry **u_e, struct ebt_u_replace *u_repl, |
Bart De Schuymer | 482fbaf | 2005-08-28 13:16:25 +0000 | [diff] [blame] | 460 | unsigned int valid_hooks, char *base, struct ebt_cntchanges **cc) |
Bart De Schuymer | 1abc55d | 2002-06-01 19:23:47 +0000 | [diff] [blame] | 461 | { |
Bart De Schuymer | 6622a01 | 2005-01-19 21:09:05 +0000 | [diff] [blame] | 462 | /* An entry */ |
Bart De Schuymer | 1abc55d | 2002-06-01 19:23:47 +0000 | [diff] [blame] | 463 | if (e->bitmask & EBT_ENTRY_OR_ENTRIES) { |
| 464 | struct ebt_u_entry *new; |
| 465 | struct ebt_u_match_list **m_l; |
| 466 | struct ebt_u_watcher_list **w_l; |
| 467 | struct ebt_entry_target *t; |
| 468 | |
| 469 | new = (struct ebt_u_entry *)malloc(sizeof(struct ebt_u_entry)); |
| 470 | if (!new) |
Bart De Schuymer | 64182a3 | 2004-01-21 20:39:54 +0000 | [diff] [blame] | 471 | ebt_print_memory(); |
Bart De Schuymer | 1abc55d | 2002-06-01 19:23:47 +0000 | [diff] [blame] | 472 | new->bitmask = e->bitmask; |
Bart De Schuymer | 9895a8e | 2003-01-11 10:14:24 +0000 | [diff] [blame] | 473 | /* |
Bart De Schuymer | 6622a01 | 2005-01-19 21:09:05 +0000 | [diff] [blame] | 474 | * Plain userspace code doesn't know about |
Bart De Schuymer | 9895a8e | 2003-01-11 10:14:24 +0000 | [diff] [blame] | 475 | * EBT_ENTRY_OR_ENTRIES |
| 476 | */ |
Bart De Schuymer | 1abc55d | 2002-06-01 19:23:47 +0000 | [diff] [blame] | 477 | new->bitmask &= ~EBT_ENTRY_OR_ENTRIES; |
| 478 | new->invflags = e->invflags; |
| 479 | new->ethproto = e->ethproto; |
Bart De Schuymer | e3cceb7 | 2002-07-26 12:47:33 +0000 | [diff] [blame] | 480 | strcpy(new->in, e->in); |
| 481 | strcpy(new->out, e->out); |
| 482 | strcpy(new->logical_in, e->logical_in); |
| 483 | strcpy(new->logical_out, e->logical_out); |
Bart De Schuymer | 1abc55d | 2002-06-01 19:23:47 +0000 | [diff] [blame] | 484 | memcpy(new->sourcemac, e->sourcemac, sizeof(new->sourcemac)); |
| 485 | memcpy(new->sourcemsk, e->sourcemsk, sizeof(new->sourcemsk)); |
| 486 | memcpy(new->destmac, e->destmac, sizeof(new->destmac)); |
| 487 | memcpy(new->destmsk, e->destmsk, sizeof(new->destmsk)); |
Bart De Schuymer | d2ced82 | 2005-01-23 19:19:00 +0000 | [diff] [blame] | 488 | if (*totalcnt >= u_repl->nentries) |
| 489 | ebt_print_bug("*totalcnt >= u_repl->nentries"); |
| 490 | new->cnt = u_repl->counters[*totalcnt]; |
Bart De Schuymer | 0436eda | 2005-03-28 20:29:37 +0000 | [diff] [blame] | 491 | new->cnt_surplus.pcnt = new->cnt_surplus.bcnt = 0; |
Bart De Schuymer | 482fbaf | 2005-08-28 13:16:25 +0000 | [diff] [blame] | 492 | new->cc = *cc; |
| 493 | *cc = (*cc)->next; |
Bart De Schuymer | 1abc55d | 2002-06-01 19:23:47 +0000 | [diff] [blame] | 494 | new->m_list = NULL; |
| 495 | new->w_list = NULL; |
Bart De Schuymer | e94eaf7 | 2005-08-28 16:06:22 +0000 | [diff] [blame] | 496 | new->next = (*u_e)->next; |
| 497 | new->next->prev = new; |
| 498 | (*u_e)->next = new; |
| 499 | new->prev = *u_e; |
| 500 | *u_e = new; |
Bart De Schuymer | 1abc55d | 2002-06-01 19:23:47 +0000 | [diff] [blame] | 501 | m_l = &new->m_list; |
| 502 | EBT_MATCH_ITERATE(e, ebt_translate_match, &m_l); |
| 503 | w_l = &new->w_list; |
| 504 | EBT_WATCHER_ITERATE(e, ebt_translate_watcher, &w_l); |
| 505 | |
| 506 | t = (struct ebt_entry_target *)(((char *)e) + e->target_offset); |
| 507 | new->t = (struct ebt_entry_target *) |
| 508 | malloc(t->target_size + sizeof(struct ebt_entry_target)); |
| 509 | if (!new->t) |
Bart De Schuymer | 64182a3 | 2004-01-21 20:39:54 +0000 | [diff] [blame] | 510 | ebt_print_memory(); |
Bart De Schuymer | 6622a01 | 2005-01-19 21:09:05 +0000 | [diff] [blame] | 511 | if (ebt_find_target(t->u.name) == NULL) { |
Bart De Schuymer | 64182a3 | 2004-01-21 20:39:54 +0000 | [diff] [blame] | 512 | ebt_print_error("Kernel target %s unsupported by " |
| 513 | "userspace tool", t->u.name); |
Bart De Schuymer | 6622a01 | 2005-01-19 21:09:05 +0000 | [diff] [blame] | 514 | return -1; |
| 515 | } |
Bart De Schuymer | 1abc55d | 2002-06-01 19:23:47 +0000 | [diff] [blame] | 516 | memcpy(new->t, t, t->target_size + |
| 517 | sizeof(struct ebt_entry_target)); |
Bart De Schuymer | 6622a01 | 2005-01-19 21:09:05 +0000 | [diff] [blame] | 518 | /* Deal with jumps to udc */ |
Bart De Schuymer | 60332e0 | 2002-06-23 08:01:47 +0000 | [diff] [blame] | 519 | if (!strcmp(t->u.name, EBT_STANDARD_TARGET)) { |
| 520 | char *tmp = base; |
| 521 | int verdict = ((struct ebt_standard_target *)t)->verdict; |
| 522 | int i; |
Bart De Schuymer | 60332e0 | 2002-06-23 08:01:47 +0000 | [diff] [blame] | 523 | |
| 524 | if (verdict >= 0) { |
| 525 | tmp += verdict; |
Bart De Schuymer | 9bfcfd8 | 2005-08-27 16:52:19 +0000 | [diff] [blame] | 526 | for (i = NF_BR_NUMHOOKS; i < u_repl->num_chains; i++) |
| 527 | if (u_repl->chains[i]->kernel_start == tmp) |
| 528 | break; |
| 529 | if (i == u_repl->num_chains) |
| 530 | ebt_print_bug("Can't find udc for jump"); |
Bart De Schuymer | 60332e0 | 2002-06-23 08:01:47 +0000 | [diff] [blame] | 531 | ((struct ebt_standard_target *)new->t)->verdict = i; |
| 532 | } |
| 533 | } |
Bart De Schuymer | 1abc55d | 2002-06-01 19:23:47 +0000 | [diff] [blame] | 534 | |
Bart De Schuymer | 1abc55d | 2002-06-01 19:23:47 +0000 | [diff] [blame] | 535 | (*cnt)++; |
| 536 | (*totalcnt)++; |
| 537 | return 0; |
Bart De Schuymer | 6622a01 | 2005-01-19 21:09:05 +0000 | [diff] [blame] | 538 | } else { /* A new chain */ |
Bart De Schuymer | 1abc55d | 2002-06-01 19:23:47 +0000 | [diff] [blame] | 539 | int i; |
| 540 | struct ebt_entries *entries = (struct ebt_entries *)e; |
Bart De Schuymer | 1abc55d | 2002-06-01 19:23:47 +0000 | [diff] [blame] | 541 | |
Bart De Schuymer | 1abc55d | 2002-06-01 19:23:47 +0000 | [diff] [blame] | 542 | if (*n != *cnt) |
Bart De Schuymer | 64182a3 | 2004-01-21 20:39:54 +0000 | [diff] [blame] | 543 | ebt_print_bug("Nr of entries in the chain is wrong"); |
Bart De Schuymer | 1abc55d | 2002-06-01 19:23:47 +0000 | [diff] [blame] | 544 | *n = entries->nentries; |
| 545 | *cnt = 0; |
Bart De Schuymer | 60332e0 | 2002-06-23 08:01:47 +0000 | [diff] [blame] | 546 | for (i = *hook + 1; i < NF_BR_NUMHOOKS; i++) |
| 547 | if (valid_hooks & (1 << i)) |
| 548 | break; |
| 549 | *hook = i; |
Bart De Schuymer | e94eaf7 | 2005-08-28 16:06:22 +0000 | [diff] [blame] | 550 | *u_e = u_repl->chains[*hook]->entries; |
Bart De Schuymer | 60332e0 | 2002-06-23 08:01:47 +0000 | [diff] [blame] | 551 | return 0; |
| 552 | } |
| 553 | } |
| 554 | |
Bart De Schuymer | 6622a01 | 2005-01-19 21:09:05 +0000 | [diff] [blame] | 555 | /* Initialize all chain headers */ |
Bart De Schuymer | 60332e0 | 2002-06-23 08:01:47 +0000 | [diff] [blame] | 556 | static int |
| 557 | ebt_translate_chains(struct ebt_entry *e, unsigned int *hook, |
| 558 | struct ebt_u_replace *u_repl, unsigned int valid_hooks) |
| 559 | { |
| 560 | int i; |
| 561 | struct ebt_entries *entries = (struct ebt_entries *)e; |
| 562 | struct ebt_u_entries *new; |
Bart De Schuymer | 60332e0 | 2002-06-23 08:01:47 +0000 | [diff] [blame] | 563 | |
| 564 | if (!(e->bitmask & EBT_ENTRY_OR_ENTRIES)) { |
| 565 | for (i = *hook + 1; i < NF_BR_NUMHOOKS; i++) |
| 566 | if (valid_hooks & (1 << i)) |
| 567 | break; |
Bart De Schuymer | 9bfcfd8 | 2005-08-27 16:52:19 +0000 | [diff] [blame] | 568 | new = (struct ebt_u_entries *)malloc(sizeof(struct ebt_u_entries)); |
| 569 | if (!new) |
| 570 | ebt_print_memory(); |
| 571 | if (i == u_repl->max_chains) |
| 572 | ebt_double_chains(u_repl); |
| 573 | u_repl->chains[i] = new; |
| 574 | if (i >= NF_BR_NUMHOOKS) |
| 575 | new->kernel_start = (char *)e; |
| 576 | *hook = i; |
Bart De Schuymer | 1abc55d | 2002-06-01 19:23:47 +0000 | [diff] [blame] | 577 | new->nentries = entries->nentries; |
| 578 | new->policy = entries->policy; |
Bart De Schuymer | e94eaf7 | 2005-08-28 16:06:22 +0000 | [diff] [blame] | 579 | new->entries = (struct ebt_u_entry *)malloc(sizeof(struct ebt_u_entry)); |
| 580 | if (!new->entries) |
| 581 | ebt_print_memory(); |
| 582 | new->entries->next = new->entries->prev = new->entries; |
Bart De Schuymer | 60332e0 | 2002-06-23 08:01:47 +0000 | [diff] [blame] | 583 | new->counter_offset = entries->counter_offset; |
| 584 | strcpy(new->name, entries->name); |
Bart De Schuymer | 1abc55d | 2002-06-01 19:23:47 +0000 | [diff] [blame] | 585 | } |
Bart De Schuymer | 60332e0 | 2002-06-23 08:01:47 +0000 | [diff] [blame] | 586 | return 0; |
Bart De Schuymer | 1abc55d | 2002-06-01 19:23:47 +0000 | [diff] [blame] | 587 | } |
| 588 | |
Bart De Schuymer | 6622a01 | 2005-01-19 21:09:05 +0000 | [diff] [blame] | 589 | static int retrieve_from_file(char *filename, struct ebt_replace *repl, |
Bart De Schuymer | 6242374 | 2002-07-14 19:06:20 +0000 | [diff] [blame] | 590 | char command) |
| 591 | { |
| 592 | FILE *file; |
Bart De Schuymer | f81c270 | 2003-07-23 21:07:04 +0000 | [diff] [blame] | 593 | char *hlp = NULL, *entries; |
| 594 | struct ebt_counter *counters; |
Bart De Schuymer | 6622a01 | 2005-01-19 21:09:05 +0000 | [diff] [blame] | 595 | int size, ret = 0; |
Bart De Schuymer | 6242374 | 2002-07-14 19:06:20 +0000 | [diff] [blame] | 596 | |
Bart De Schuymer | 6622a01 | 2005-01-19 21:09:05 +0000 | [diff] [blame] | 597 | if (!(file = fopen(filename, "r+b"))) { |
Bart De Schuymer | 64182a3 | 2004-01-21 20:39:54 +0000 | [diff] [blame] | 598 | ebt_print_error("Could not open file %s", filename); |
Bart De Schuymer | 6622a01 | 2005-01-19 21:09:05 +0000 | [diff] [blame] | 599 | return -1; |
| 600 | } |
| 601 | /* Make sure table name is right if command isn't -L or --atomic-commit */ |
Bart De Schuymer | 6242374 | 2002-07-14 19:06:20 +0000 | [diff] [blame] | 602 | if (command != 'L' && command != 8) { |
Bart De Schuymer | 0cff9e9 | 2002-07-25 12:29:50 +0000 | [diff] [blame] | 603 | hlp = (char *)malloc(strlen(repl->name) + 1); |
Bart De Schuymer | 6242374 | 2002-07-14 19:06:20 +0000 | [diff] [blame] | 604 | if (!hlp) |
Bart De Schuymer | 64182a3 | 2004-01-21 20:39:54 +0000 | [diff] [blame] | 605 | ebt_print_memory(); |
Bart De Schuymer | 6242374 | 2002-07-14 19:06:20 +0000 | [diff] [blame] | 606 | strcpy(hlp, repl->name); |
Bart De Schuymer | e8890ff | 2002-07-15 20:29:04 +0000 | [diff] [blame] | 607 | } |
Bart De Schuymer | 6242374 | 2002-07-14 19:06:20 +0000 | [diff] [blame] | 608 | if (fread(repl, sizeof(char), sizeof(struct ebt_replace), file) |
Bart De Schuymer | 6622a01 | 2005-01-19 21:09:05 +0000 | [diff] [blame] | 609 | != sizeof(struct ebt_replace)) { |
Bart De Schuymer | 64182a3 | 2004-01-21 20:39:54 +0000 | [diff] [blame] | 610 | ebt_print_error("File %s is corrupt", filename); |
Bart De Schuymer | 6622a01 | 2005-01-19 21:09:05 +0000 | [diff] [blame] | 611 | ret = -1; |
| 612 | goto close_file; |
| 613 | } |
Bart De Schuymer | e8890ff | 2002-07-15 20:29:04 +0000 | [diff] [blame] | 614 | if (command != 'L' && command != 8 && strcmp(hlp, repl->name)) { |
Bart De Schuymer | 64182a3 | 2004-01-21 20:39:54 +0000 | [diff] [blame] | 615 | ebt_print_error("File %s contains wrong table name or is " |
| 616 | "corrupt", filename); |
Bart De Schuymer | 6622a01 | 2005-01-19 21:09:05 +0000 | [diff] [blame] | 617 | ret = -1; |
| 618 | goto close_file; |
Bart De Schuymer | 8339ff1 | 2004-01-14 20:05:27 +0000 | [diff] [blame] | 619 | } else if (!ebt_find_table(repl->name)) { |
Bart De Schuymer | 64182a3 | 2004-01-21 20:39:54 +0000 | [diff] [blame] | 620 | ebt_print_error("File %s contains invalid table name", |
| 621 | filename); |
Bart De Schuymer | 6622a01 | 2005-01-19 21:09:05 +0000 | [diff] [blame] | 622 | ret = -1; |
| 623 | goto close_file; |
Bart De Schuymer | 0cff9e9 | 2002-07-25 12:29:50 +0000 | [diff] [blame] | 624 | } |
Bart De Schuymer | e8890ff | 2002-07-15 20:29:04 +0000 | [diff] [blame] | 625 | |
Bart De Schuymer | 6242374 | 2002-07-14 19:06:20 +0000 | [diff] [blame] | 626 | size = sizeof(struct ebt_replace) + |
| 627 | repl->nentries * sizeof(struct ebt_counter) + repl->entries_size; |
| 628 | fseek(file, 0, SEEK_END); |
Bart De Schuymer | e8890ff | 2002-07-15 20:29:04 +0000 | [diff] [blame] | 629 | if (size != ftell(file)) { |
Bart De Schuymer | 64182a3 | 2004-01-21 20:39:54 +0000 | [diff] [blame] | 630 | ebt_print_error("File %s has wrong size", filename); |
Bart De Schuymer | 2f7e8d1 | 2005-01-19 21:23:02 +0000 | [diff] [blame] | 631 | ret = -1; |
Bart De Schuymer | 6622a01 | 2005-01-19 21:09:05 +0000 | [diff] [blame] | 632 | goto close_file; |
Bart De Schuymer | e8890ff | 2002-07-15 20:29:04 +0000 | [diff] [blame] | 633 | } |
Bart De Schuymer | f81c270 | 2003-07-23 21:07:04 +0000 | [diff] [blame] | 634 | entries = (char *)malloc(repl->entries_size); |
| 635 | if (!entries) |
Bart De Schuymer | 64182a3 | 2004-01-21 20:39:54 +0000 | [diff] [blame] | 636 | ebt_print_memory(); |
Bart De Schuymer | f81c270 | 2003-07-23 21:07:04 +0000 | [diff] [blame] | 637 | repl->entries = sparc_cast entries; |
Bart De Schuymer | 6242374 | 2002-07-14 19:06:20 +0000 | [diff] [blame] | 638 | if (repl->nentries) { |
Bart De Schuymer | f81c270 | 2003-07-23 21:07:04 +0000 | [diff] [blame] | 639 | counters = (struct ebt_counter *) |
Bart De Schuymer | 6242374 | 2002-07-14 19:06:20 +0000 | [diff] [blame] | 640 | malloc(repl->nentries * sizeof(struct ebt_counter)); |
Bart De Schuymer | 8e96bee | 2003-08-27 16:59:39 +0000 | [diff] [blame] | 641 | repl->counters = sparc_cast counters; |
Bart De Schuymer | 6242374 | 2002-07-14 19:06:20 +0000 | [diff] [blame] | 642 | if (!repl->counters) |
Bart De Schuymer | 64182a3 | 2004-01-21 20:39:54 +0000 | [diff] [blame] | 643 | ebt_print_memory(); |
Bart De Schuymer | 6242374 | 2002-07-14 19:06:20 +0000 | [diff] [blame] | 644 | } else |
Bart De Schuymer | f81c270 | 2003-07-23 21:07:04 +0000 | [diff] [blame] | 645 | repl->counters = sparc_cast NULL; |
Bart De Schuymer | 6622a01 | 2005-01-19 21:09:05 +0000 | [diff] [blame] | 646 | /* Copy entries and counters */ |
Bart De Schuymer | 6242374 | 2002-07-14 19:06:20 +0000 | [diff] [blame] | 647 | if (fseek(file, sizeof(struct ebt_replace), SEEK_SET) || |
Bart De Schuymer | f81c270 | 2003-07-23 21:07:04 +0000 | [diff] [blame] | 648 | fread((char *)repl->entries, sizeof(char), repl->entries_size, file) |
Bart De Schuymer | 6242374 | 2002-07-14 19:06:20 +0000 | [diff] [blame] | 649 | != repl->entries_size || |
Bart De Schuymer | 64182a3 | 2004-01-21 20:39:54 +0000 | [diff] [blame] | 650 | fseek(file, sizeof(struct ebt_replace) + repl->entries_size, |
| 651 | SEEK_SET) |
Bart De Schuymer | f81c270 | 2003-07-23 21:07:04 +0000 | [diff] [blame] | 652 | || fread((char *)repl->counters, sizeof(char), |
Bart De Schuymer | 6242374 | 2002-07-14 19:06:20 +0000 | [diff] [blame] | 653 | repl->nentries * sizeof(struct ebt_counter), file) |
Bart De Schuymer | e8890ff | 2002-07-15 20:29:04 +0000 | [diff] [blame] | 654 | != repl->nentries * sizeof(struct ebt_counter)) { |
Bart De Schuymer | 64182a3 | 2004-01-21 20:39:54 +0000 | [diff] [blame] | 655 | ebt_print_error("File %s is corrupt", filename); |
Bart De Schuymer | 6622a01 | 2005-01-19 21:09:05 +0000 | [diff] [blame] | 656 | free(entries); |
| 657 | repl->entries = NULL; |
| 658 | ret = -1; |
Bart De Schuymer | e8890ff | 2002-07-15 20:29:04 +0000 | [diff] [blame] | 659 | } |
Bart De Schuymer | 6622a01 | 2005-01-19 21:09:05 +0000 | [diff] [blame] | 660 | close_file: |
Bart De Schuymer | 6242374 | 2002-07-14 19:06:20 +0000 | [diff] [blame] | 661 | fclose(file); |
Bart De Schuymer | 6622a01 | 2005-01-19 21:09:05 +0000 | [diff] [blame] | 662 | free(hlp); |
| 663 | return ret; |
Bart De Schuymer | 6242374 | 2002-07-14 19:06:20 +0000 | [diff] [blame] | 664 | } |
| 665 | |
Bart De Schuymer | 64182a3 | 2004-01-21 20:39:54 +0000 | [diff] [blame] | 666 | static int retrieve_from_kernel(struct ebt_replace *repl, char command, |
| 667 | int init) |
Bart De Schuymer | 6242374 | 2002-07-14 19:06:20 +0000 | [diff] [blame] | 668 | { |
| 669 | socklen_t optlen; |
| 670 | int optname; |
Bart De Schuymer | f81c270 | 2003-07-23 21:07:04 +0000 | [diff] [blame] | 671 | char *entries; |
Bart De Schuymer | 6242374 | 2002-07-14 19:06:20 +0000 | [diff] [blame] | 672 | |
| 673 | optlen = sizeof(struct ebt_replace); |
Bart De Schuymer | 6622a01 | 2005-01-19 21:09:05 +0000 | [diff] [blame] | 674 | if (get_sockfd()) |
| 675 | return -1; |
Bart De Schuymer | 9895a8e | 2003-01-11 10:14:24 +0000 | [diff] [blame] | 676 | /* --atomic-init || --init-table */ |
Bart De Schuymer | 64182a3 | 2004-01-21 20:39:54 +0000 | [diff] [blame] | 677 | if (init) |
Bart De Schuymer | 6242374 | 2002-07-14 19:06:20 +0000 | [diff] [blame] | 678 | optname = EBT_SO_GET_INIT_INFO; |
| 679 | else |
| 680 | optname = EBT_SO_GET_INFO; |
| 681 | if (getsockopt(sockfd, IPPROTO_IP, optname, repl, &optlen)) |
| 682 | return -1; |
| 683 | |
Bart De Schuymer | f81c270 | 2003-07-23 21:07:04 +0000 | [diff] [blame] | 684 | if ( !(entries = (char *)malloc(repl->entries_size)) ) |
Bart De Schuymer | 64182a3 | 2004-01-21 20:39:54 +0000 | [diff] [blame] | 685 | ebt_print_memory(); |
Bart De Schuymer | f81c270 | 2003-07-23 21:07:04 +0000 | [diff] [blame] | 686 | repl->entries = sparc_cast entries; |
Bart De Schuymer | 6242374 | 2002-07-14 19:06:20 +0000 | [diff] [blame] | 687 | if (repl->nentries) { |
Bart De Schuymer | f81c270 | 2003-07-23 21:07:04 +0000 | [diff] [blame] | 688 | struct ebt_counter *counters; |
| 689 | |
| 690 | if (!(counters = (struct ebt_counter *) |
Bart De Schuymer | 6242374 | 2002-07-14 19:06:20 +0000 | [diff] [blame] | 691 | malloc(repl->nentries * sizeof(struct ebt_counter))) ) |
Bart De Schuymer | 64182a3 | 2004-01-21 20:39:54 +0000 | [diff] [blame] | 692 | ebt_print_memory(); |
Bart De Schuymer | f81c270 | 2003-07-23 21:07:04 +0000 | [diff] [blame] | 693 | repl->counters = sparc_cast counters; |
Bart De Schuymer | 6242374 | 2002-07-14 19:06:20 +0000 | [diff] [blame] | 694 | } |
| 695 | else |
Bart De Schuymer | f81c270 | 2003-07-23 21:07:04 +0000 | [diff] [blame] | 696 | repl->counters = sparc_cast NULL; |
Bart De Schuymer | 6242374 | 2002-07-14 19:06:20 +0000 | [diff] [blame] | 697 | |
Bart De Schuymer | 6622a01 | 2005-01-19 21:09:05 +0000 | [diff] [blame] | 698 | /* We want to receive the counters */ |
Bart De Schuymer | 6242374 | 2002-07-14 19:06:20 +0000 | [diff] [blame] | 699 | repl->num_counters = repl->nentries; |
| 700 | optlen += repl->entries_size + repl->num_counters * |
| 701 | sizeof(struct ebt_counter); |
Bart De Schuymer | 64182a3 | 2004-01-21 20:39:54 +0000 | [diff] [blame] | 702 | if (init) |
Bart De Schuymer | 6242374 | 2002-07-14 19:06:20 +0000 | [diff] [blame] | 703 | optname = EBT_SO_GET_INIT_ENTRIES; |
| 704 | else |
| 705 | optname = EBT_SO_GET_ENTRIES; |
| 706 | if (getsockopt(sockfd, IPPROTO_IP, optname, repl, &optlen)) |
Bart De Schuymer | 6622a01 | 2005-01-19 21:09:05 +0000 | [diff] [blame] | 707 | ebt_print_bug("Hmm, what is wrong??? bug#1"); |
Bart De Schuymer | 6242374 | 2002-07-14 19:06:20 +0000 | [diff] [blame] | 708 | |
| 709 | return 0; |
| 710 | } |
| 711 | |
Bart De Schuymer | 64182a3 | 2004-01-21 20:39:54 +0000 | [diff] [blame] | 712 | int ebt_get_table(struct ebt_u_replace *u_repl, int init) |
Bart De Schuymer | 1abc55d | 2002-06-01 19:23:47 +0000 | [diff] [blame] | 713 | { |
| 714 | int i, j, k, hook; |
Bart De Schuymer | 1abc55d | 2002-06-01 19:23:47 +0000 | [diff] [blame] | 715 | struct ebt_replace repl; |
Bart De Schuymer | e94eaf7 | 2005-08-28 16:06:22 +0000 | [diff] [blame] | 716 | struct ebt_u_entry *u_e; |
Bart De Schuymer | 482fbaf | 2005-08-28 13:16:25 +0000 | [diff] [blame] | 717 | struct ebt_cntchanges *new_cc, *cc; |
Bart De Schuymer | 1abc55d | 2002-06-01 19:23:47 +0000 | [diff] [blame] | 718 | |
Bart De Schuymer | 1abc55d | 2002-06-01 19:23:47 +0000 | [diff] [blame] | 719 | strcpy(repl.name, u_repl->name); |
Bart De Schuymer | 62a7fdb | 2002-08-24 21:01:21 +0000 | [diff] [blame] | 720 | if (u_repl->filename != NULL) { |
Bart De Schuymer | 64182a3 | 2004-01-21 20:39:54 +0000 | [diff] [blame] | 721 | if (init) |
Bart De Schuymer | d2ced82 | 2005-01-23 19:19:00 +0000 | [diff] [blame] | 722 | ebt_print_bug("Getting initial table data from a file is impossible"); |
Bart De Schuymer | 6622a01 | 2005-01-19 21:09:05 +0000 | [diff] [blame] | 723 | if (retrieve_from_file(u_repl->filename, &repl, u_repl->command)) |
| 724 | return -1; |
| 725 | /* -L with a wrong table name should be dealt with silently */ |
Bart De Schuymer | 62a7fdb | 2002-08-24 21:01:21 +0000 | [diff] [blame] | 726 | strcpy(u_repl->name, repl.name); |
Bart De Schuymer | 6622a01 | 2005-01-19 21:09:05 +0000 | [diff] [blame] | 727 | } else if (retrieve_from_kernel(&repl, u_repl->command, init)) |
Bart De Schuymer | 0cff9e9 | 2002-07-25 12:29:50 +0000 | [diff] [blame] | 728 | return -1; |
Bart De Schuymer | 1abc55d | 2002-06-01 19:23:47 +0000 | [diff] [blame] | 729 | |
Bart De Schuymer | 6622a01 | 2005-01-19 21:09:05 +0000 | [diff] [blame] | 730 | /* Translate the struct ebt_replace to a struct ebt_u_replace */ |
Bart De Schuymer | 1abc55d | 2002-06-01 19:23:47 +0000 | [diff] [blame] | 731 | u_repl->valid_hooks = repl.valid_hooks; |
| 732 | u_repl->nentries = repl.nentries; |
| 733 | u_repl->num_counters = repl.num_counters; |
Bart De Schuymer | d2ced82 | 2005-01-23 19:19:00 +0000 | [diff] [blame] | 734 | u_repl->counters = repl.counters; |
Bart De Schuymer | 482fbaf | 2005-08-28 13:16:25 +0000 | [diff] [blame] | 735 | u_repl->cc = (struct ebt_cntchanges *)malloc(sizeof(struct ebt_cntchanges)); |
| 736 | if (!u_repl->cc) |
| 737 | ebt_print_memory(); |
| 738 | u_repl->cc->next = u_repl->cc->prev = u_repl->cc; |
| 739 | cc = u_repl->cc; |
Bart De Schuymer | 8339ff1 | 2004-01-14 20:05:27 +0000 | [diff] [blame] | 740 | for (i = 0; i < repl.nentries; i++) { |
Bart De Schuymer | 482fbaf | 2005-08-28 13:16:25 +0000 | [diff] [blame] | 741 | new_cc = (struct ebt_cntchanges *)malloc(sizeof(struct ebt_cntchanges)); |
Bart De Schuymer | 8339ff1 | 2004-01-14 20:05:27 +0000 | [diff] [blame] | 742 | if (!new_cc) |
Bart De Schuymer | 64182a3 | 2004-01-21 20:39:54 +0000 | [diff] [blame] | 743 | ebt_print_memory(); |
Bart De Schuymer | 8339ff1 | 2004-01-14 20:05:27 +0000 | [diff] [blame] | 744 | new_cc->type = CNT_NORM; |
Bart De Schuymer | 0436eda | 2005-03-28 20:29:37 +0000 | [diff] [blame] | 745 | new_cc->change = 0; |
Bart De Schuymer | 482fbaf | 2005-08-28 13:16:25 +0000 | [diff] [blame] | 746 | new_cc->prev = cc; |
| 747 | cc->next = new_cc; |
| 748 | cc = new_cc; |
| 749 | } |
| 750 | if (repl.nentries) { |
| 751 | new_cc->next = u_repl->cc; |
| 752 | u_repl->cc->prev = new_cc; |
Bart De Schuymer | 8339ff1 | 2004-01-14 20:05:27 +0000 | [diff] [blame] | 753 | } |
Bart De Schuymer | 9bfcfd8 | 2005-08-27 16:52:19 +0000 | [diff] [blame] | 754 | u_repl->chains = (struct ebt_u_entries **)calloc(EBT_ORI_MAX_CHAINS, sizeof(void *)); |
| 755 | u_repl->max_chains = EBT_ORI_MAX_CHAINS; |
Bart De Schuymer | 1abc55d | 2002-06-01 19:23:47 +0000 | [diff] [blame] | 756 | hook = -1; |
Bart De Schuymer | 6622a01 | 2005-01-19 21:09:05 +0000 | [diff] [blame] | 757 | /* FIXME: Clean up when an error is encountered */ |
Bart De Schuymer | 60332e0 | 2002-06-23 08:01:47 +0000 | [diff] [blame] | 758 | EBT_ENTRY_ITERATE(repl.entries, repl.entries_size, ebt_translate_chains, |
| 759 | &hook, u_repl, u_repl->valid_hooks); |
Bart De Schuymer | 9bfcfd8 | 2005-08-27 16:52:19 +0000 | [diff] [blame] | 760 | if (hook >= NF_BR_NUMHOOKS) |
| 761 | u_repl->num_chains = hook + 1; |
| 762 | else |
| 763 | u_repl->num_chains = NF_BR_NUMHOOKS; |
Bart De Schuymer | 6622a01 | 2005-01-19 21:09:05 +0000 | [diff] [blame] | 764 | i = 0; /* Holds the expected nr. of entries for the chain */ |
| 765 | j = 0; /* Holds the up to now counted entries for the chain */ |
Bart De Schuymer | d2ced82 | 2005-01-23 19:19:00 +0000 | [diff] [blame] | 766 | k = 0; /* Holds the total nr. of entries, should equal u_repl->nentries afterwards */ |
Bart De Schuymer | 482fbaf | 2005-08-28 13:16:25 +0000 | [diff] [blame] | 767 | cc = u_repl->cc->next; |
Bart De Schuymer | 60332e0 | 2002-06-23 08:01:47 +0000 | [diff] [blame] | 768 | hook = -1; |
Bart De Schuymer | 64182a3 | 2004-01-21 20:39:54 +0000 | [diff] [blame] | 769 | EBT_ENTRY_ITERATE((char *)repl.entries, repl.entries_size, |
| 770 | ebt_translate_entry, &hook, &i, &j, &k, &u_e, u_repl, |
Bart De Schuymer | 482fbaf | 2005-08-28 13:16:25 +0000 | [diff] [blame] | 771 | u_repl->valid_hooks, (char *)repl.entries, &cc); |
Bart De Schuymer | 1abc55d | 2002-06-01 19:23:47 +0000 | [diff] [blame] | 772 | if (k != u_repl->nentries) |
Bart De Schuymer | 64182a3 | 2004-01-21 20:39:54 +0000 | [diff] [blame] | 773 | ebt_print_bug("Wrong total nentries"); |
Bart De Schuymer | 9ce6ee9 | 2002-06-14 21:56:35 +0000 | [diff] [blame] | 774 | return 0; |
Bart De Schuymer | 1abc55d | 2002-06-01 19:23:47 +0000 | [diff] [blame] | 775 | } |