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