f129d38be45f85b38adaf5138ea7dc7034140374
[openvswitch] / ofproto / ofproto-sflow.c
1 /*
2  * Copyright (c) 2009, 2010 InMon Corp.
3  * Copyright (c) 2009 Nicira Networks.
4  *
5  * Licensed under the Apache License, Version 2.0 (the "License");
6  * you may not use this file except in compliance with the License.
7  * You may obtain a copy of the License at:
8  *
9  *     http://www.apache.org/licenses/LICENSE-2.0
10  *
11  * Unless required by applicable law or agreed to in writing, software
12  * distributed under the License is distributed on an "AS IS" BASIS,
13  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14  * See the License for the specific language governing permissions and
15  * limitations under the License.
16  */
17
18 #include <config.h>
19 #include "ofproto-sflow.h"
20 #include <inttypes.h>
21 #include <stdlib.h>
22 #include "collectors.h"
23 #include "dpif.h"
24 #include "compiler.h"
25 #include "hash.h"
26 #include "hmap.h"
27 #include "netdev.h"
28 #include "ofpbuf.h"
29 #include "ofproto.h"
30 #include "poll-loop.h"
31 #include "sflow_api.h"
32 #include "socket-util.h"
33 #include "timeval.h"
34 #include "vlog.h"
35
36 VLOG_DEFINE_THIS_MODULE(sflow)
37
38 struct ofproto_sflow_port {
39     struct hmap_node hmap_node; /* In struct ofproto_sflow's "ports" hmap. */
40     struct netdev *netdev;      /* Underlying network device, for stats. */
41     SFLDataSource_instance dsi; /* sFlow library's notion of port number. */
42     uint16_t odp_port;          /* ODP port number. */
43 };
44
45 struct ofproto_sflow {
46     struct ofproto *ofproto;
47     struct collectors *collectors;
48     SFLAgent *sflow_agent;
49     struct ofproto_sflow_options *options;
50     struct dpif *dpif;
51     time_t next_tick;
52     size_t n_flood, n_all;
53     struct hmap ports;          /* Contains "struct ofproto_sflow_port"s. */
54 };
55
56 static void ofproto_sflow_del_port__(struct ofproto_sflow *,
57                                      struct ofproto_sflow_port *);
58
59 #define RECEIVER_INDEX 1
60
61 static struct vlog_rate_limit rl = VLOG_RATE_LIMIT_INIT(1, 5);
62
63 static bool
64 nullable_string_is_equal(const char *a, const char *b)
65 {
66     return a ? b && !strcmp(a, b) : !b;
67 }
68
69 static bool
70 ofproto_sflow_options_equal(const struct ofproto_sflow_options *a,
71                             const struct ofproto_sflow_options *b)
72 {
73     return (svec_equal(&a->targets, &b->targets)
74             && a->sampling_rate == b->sampling_rate
75             && a->polling_interval == b->polling_interval
76             && a->header_len == b->header_len
77             && a->sub_id == b->sub_id
78             && nullable_string_is_equal(a->agent_device, b->agent_device)
79             && nullable_string_is_equal(a->control_ip, b->control_ip));
80 }
81
82 static struct ofproto_sflow_options *
83 ofproto_sflow_options_clone(const struct ofproto_sflow_options *old)
84 {
85     struct ofproto_sflow_options *new = xmemdup(old, sizeof *old);
86     svec_clone(&new->targets, &old->targets);
87     new->agent_device = old->agent_device ? xstrdup(old->agent_device) : NULL;
88     new->control_ip = old->control_ip ? xstrdup(old->control_ip) : NULL;
89     return new;
90 }
91
92 static void
93 ofproto_sflow_options_destroy(struct ofproto_sflow_options *options)
94 {
95     if (options) {
96         svec_destroy(&options->targets);
97         free(options->agent_device);
98         free(options->control_ip);
99         free(options);
100     }
101 }
102
103 /* sFlow library callback to allocate memory. */
104 static void *
105 sflow_agent_alloc_cb(void *magic OVS_UNUSED, SFLAgent *agent OVS_UNUSED,
106                      size_t bytes)
107 {
108     return calloc(1, bytes);
109 }
110
111 /* sFlow library callback to free memory. */
112 static int
113 sflow_agent_free_cb(void *magic OVS_UNUSED, SFLAgent *agent OVS_UNUSED,
114                     void *obj)
115 {
116     free(obj);
117     return 0;
118 }
119
120 /* sFlow library callback to report error. */
121 static void
122 sflow_agent_error_cb(void *magic OVS_UNUSED, SFLAgent *agent OVS_UNUSED,
123                      char *msg)
124 {
125     VLOG_WARN("sFlow agent error: %s", msg);
126 }
127
128 /* sFlow library callback to send datagram. */
129 static void
130 sflow_agent_send_packet_cb(void *os_, SFLAgent *agent OVS_UNUSED,
131                            SFLReceiver *receiver OVS_UNUSED, u_char *pkt,
132                            uint32_t pktLen)
133 {
134     struct ofproto_sflow *os = os_;
135     collectors_send(os->collectors, pkt, pktLen);
136 }
137
138 static struct ofproto_sflow_port *
139 ofproto_sflow_find_port(const struct ofproto_sflow *os, uint16_t odp_port)
140 {
141     struct ofproto_sflow_port *osp;
142
143     HMAP_FOR_EACH_IN_BUCKET (osp, hmap_node,
144                              hash_int(odp_port, 0), &os->ports) {
145         if (osp->odp_port == odp_port) {
146             return osp;
147         }
148     }
149     return NULL;
150 }
151
152 static void
153 sflow_agent_get_counters(void *os_, SFLPoller *poller,
154                          SFL_COUNTERS_SAMPLE_TYPE *cs)
155 {
156     struct ofproto_sflow *os = os_;
157     SFLCounters_sample_element elem;
158     struct ofproto_sflow_port *osp;
159     SFLIf_counters *counters;
160     struct netdev_stats stats;
161     enum netdev_flags flags;
162     uint32_t current;
163
164     osp = ofproto_sflow_find_port(os, poller->bridgePort);
165     if (!osp) {
166         return;
167     }
168
169     elem.tag = SFLCOUNTERS_GENERIC;
170     counters = &elem.counterBlock.generic;
171     counters->ifIndex = SFL_DS_INDEX(poller->dsi);
172     counters->ifType = 6;
173     if (!netdev_get_features(osp->netdev, &current, NULL, NULL, NULL)) {
174       /* The values of ifDirection come from MAU MIB (RFC 2668): 0 = unknown,
175          1 = full-duplex, 2 = half-duplex, 3 = in, 4=out */
176         counters->ifSpeed = netdev_features_to_bps(current);
177         counters->ifDirection = (netdev_features_is_full_duplex(current)
178                                  ? 1 : 2);
179     } else {
180         counters->ifSpeed = 100000000;
181         counters->ifDirection = 0;
182     }
183     if (!netdev_get_flags(osp->netdev, &flags) && flags & NETDEV_UP) {
184         bool carrier;
185
186         counters->ifStatus = 1; /* ifAdminStatus up. */
187         if (!netdev_get_carrier(osp->netdev, &carrier) && carrier) {
188             counters->ifStatus |= 2; /* ifOperStatus us. */
189         }
190     } else {
191         counters->ifStatus = 0;  /* Down. */
192     }
193
194     /* XXX
195        1. Is the multicast counter filled in?
196        2. Does the multicast counter include broadcasts?
197        3. Does the rx_packets counter include multicasts/broadcasts?
198     */
199     netdev_get_stats(osp->netdev, &stats);
200     counters->ifInOctets = stats.rx_bytes;
201     counters->ifInUcastPkts = stats.rx_packets;
202     counters->ifInMulticastPkts = stats.multicast;
203     counters->ifInBroadcastPkts = -1;
204     counters->ifInDiscards = stats.rx_dropped;
205     counters->ifInErrors = stats.rx_errors;
206     counters->ifInUnknownProtos = -1;
207     counters->ifOutOctets = stats.tx_bytes;
208     counters->ifOutUcastPkts = stats.tx_packets;
209     counters->ifOutMulticastPkts = -1;
210     counters->ifOutBroadcastPkts = -1;
211     counters->ifOutDiscards = stats.tx_dropped;
212     counters->ifOutErrors = stats.tx_errors;
213     counters->ifPromiscuousMode = 0;
214
215     SFLADD_ELEMENT(cs, &elem);
216     sfl_poller_writeCountersSample(poller, cs);
217 }
218
219 /* Obtains an address to use for the local sFlow agent and stores it into
220  * '*agent_addr'.  Returns true if successful, false on failure.
221  *
222  * The sFlow agent address should be a local IP address that is persistent and
223  * reachable over the network, if possible.  The IP address associated with
224  * 'agent_device' is used if it has one, and otherwise 'control_ip', the IP
225  * address used to talk to the controller. */
226 static bool
227 sflow_choose_agent_address(const char *agent_device, const char *control_ip,
228                            SFLAddress *agent_addr)
229 {
230     struct in_addr in4;
231
232     memset(agent_addr, 0, sizeof *agent_addr);
233     agent_addr->type = SFLADDRESSTYPE_IP_V4;
234
235     if (agent_device) {
236         struct netdev *netdev;
237
238         if (!netdev_open_default(agent_device, &netdev)) {
239             int error = netdev_get_in4(netdev, &in4, NULL);
240             netdev_close(netdev);
241             if (!error) {
242                 goto success;
243             }
244         }
245     }
246
247     if (control_ip && !lookup_ip(control_ip, &in4)) {
248         goto success;
249     }
250
251     VLOG_ERR("could not determine IP address for sFlow agent");
252     return false;
253
254 success:
255     agent_addr->address.ip_v4.addr = in4.s_addr;
256     return true;
257 }
258
259 void
260 ofproto_sflow_clear(struct ofproto_sflow *os)
261 {
262     if (os->sflow_agent) {
263         sfl_agent_release(os->sflow_agent);
264         os->sflow_agent = NULL;
265     }
266     collectors_destroy(os->collectors);
267     os->collectors = NULL;
268     ofproto_sflow_options_destroy(os->options);
269     os->options = NULL;
270
271     /* Turn off sampling to save CPU cycles. */
272     dpif_set_sflow_probability(os->dpif, 0);
273 }
274
275 bool
276 ofproto_sflow_is_enabled(const struct ofproto_sflow *os)
277 {
278     return os->collectors != NULL;
279 }
280
281 struct ofproto_sflow *
282 ofproto_sflow_create(struct dpif *dpif)
283 {
284     struct ofproto_sflow *os;
285
286     os = xcalloc(1, sizeof *os);
287     os->dpif = dpif;
288     os->next_tick = time_now() + 1;
289     hmap_init(&os->ports);
290     return os;
291 }
292
293 void
294 ofproto_sflow_destroy(struct ofproto_sflow *os)
295 {
296     if (os) {
297         struct ofproto_sflow_port *osp, *next;
298
299         ofproto_sflow_clear(os);
300         HMAP_FOR_EACH_SAFE (osp, next, hmap_node, &os->ports) {
301             ofproto_sflow_del_port__(os, osp);
302         }
303         hmap_destroy(&os->ports);
304         free(os);
305     }
306 }
307
308 static void
309 ofproto_sflow_add_poller(struct ofproto_sflow *os,
310                          struct ofproto_sflow_port *osp, uint16_t odp_port)
311 {
312     SFLPoller *poller = sfl_agent_addPoller(os->sflow_agent, &osp->dsi, os,
313                                             sflow_agent_get_counters);
314     sfl_poller_set_sFlowCpInterval(poller, os->options->polling_interval);
315     sfl_poller_set_sFlowCpReceiver(poller, RECEIVER_INDEX);
316     sfl_poller_set_bridgePort(poller, odp_port);
317 }
318
319 static void
320 ofproto_sflow_add_sampler(struct ofproto_sflow *os,
321                           struct ofproto_sflow_port *osp)
322 {
323     SFLSampler *sampler = sfl_agent_addSampler(os->sflow_agent, &osp->dsi);
324     sfl_sampler_set_sFlowFsPacketSamplingRate(sampler, os->options->sampling_rate);
325     sfl_sampler_set_sFlowFsMaximumHeaderSize(sampler, os->options->header_len);
326     sfl_sampler_set_sFlowFsReceiver(sampler, RECEIVER_INDEX);
327 }
328
329 void
330 ofproto_sflow_add_port(struct ofproto_sflow *os, uint16_t odp_port,
331                        const char *netdev_name)
332 {
333     struct ofproto_sflow_port *osp;
334     struct netdev *netdev;
335     uint32_t ifindex;
336     int error;
337
338     ofproto_sflow_del_port(os, odp_port);
339
340     /* Open network device. */
341     error = netdev_open_default(netdev_name, &netdev);
342     if (error) {
343         VLOG_WARN_RL(&rl, "failed to open network device \"%s\": %s",
344                      netdev_name, strerror(error));
345         return;
346     }
347
348     /* Add to table of ports. */
349     osp = xmalloc(sizeof *osp);
350     osp->netdev = netdev;
351     ifindex = netdev_get_ifindex(netdev);
352     if (ifindex <= 0) {
353         ifindex = (os->sflow_agent->subId << 16) + odp_port;
354     }
355     SFL_DS_SET(osp->dsi, 0, ifindex, 0);
356     osp->odp_port = odp_port;
357     hmap_insert(&os->ports, &osp->hmap_node, hash_int(odp_port, 0));
358
359     /* Add poller and sampler. */
360     if (os->sflow_agent) {
361         ofproto_sflow_add_poller(os, osp, odp_port);
362         ofproto_sflow_add_sampler(os, osp);
363     }
364 }
365
366 static void
367 ofproto_sflow_del_port__(struct ofproto_sflow *os,
368                          struct ofproto_sflow_port *osp)
369 {
370     if (os->sflow_agent) {
371         sfl_agent_removePoller(os->sflow_agent, &osp->dsi);
372         sfl_agent_removeSampler(os->sflow_agent, &osp->dsi);
373     }
374     netdev_close(osp->netdev);
375     hmap_remove(&os->ports, &osp->hmap_node);
376     free(osp);
377 }
378
379 void
380 ofproto_sflow_del_port(struct ofproto_sflow *os, uint16_t odp_port)
381 {
382     struct ofproto_sflow_port *osp = ofproto_sflow_find_port(os, odp_port);
383     if (osp) {
384         ofproto_sflow_del_port__(os, osp);
385     }
386 }
387
388 void
389 ofproto_sflow_set_options(struct ofproto_sflow *os,
390                           const struct ofproto_sflow_options *options)
391 {
392     struct ofproto_sflow_port *osp;
393     bool options_changed;
394     SFLReceiver *receiver;
395     SFLAddress agentIP;
396     time_t now;
397
398     if (!options->targets.n || !options->sampling_rate) {
399         /* No point in doing any work if there are no targets or nothing to
400          * sample. */
401         ofproto_sflow_clear(os);
402         return;
403     }
404
405     options_changed = (!os->options
406                        || !ofproto_sflow_options_equal(options, os->options));
407
408     /* Configure collectors if options have changed or if we're shortchanged in
409      * collectors (which indicates that opening one or more of the configured
410      * collectors failed, so that we should retry). */
411     if (options_changed
412         || collectors_count(os->collectors) < options->targets.n) {
413         collectors_destroy(os->collectors);
414         collectors_create(&options->targets, SFL_DEFAULT_COLLECTOR_PORT,
415                           &os->collectors);
416         if (os->collectors == NULL) {
417             VLOG_WARN_RL(&rl, "no collectors could be initialized, "
418                          "sFlow disabled");
419             ofproto_sflow_clear(os);
420             return;
421         }
422     }
423
424     /* Avoid reconfiguring if options didn't change. */
425     if (!options_changed) {
426         return;
427     }
428     ofproto_sflow_options_destroy(os->options);
429     os->options = ofproto_sflow_options_clone(options);
430
431     /* Choose agent IP address. */
432     if (!sflow_choose_agent_address(options->agent_device,
433                                     options->control_ip, &agentIP)) {
434         ofproto_sflow_clear(os);
435         return;
436     }
437
438     /* Create agent. */
439     VLOG_INFO("creating sFlow agent %d", options->sub_id);
440     if (os->sflow_agent) {
441         sfl_agent_release(os->sflow_agent);
442     }
443     os->sflow_agent = xcalloc(1, sizeof *os->sflow_agent);
444     now = time_wall();
445     sfl_agent_init(os->sflow_agent,
446                    &agentIP,
447                    options->sub_id,
448                    now,         /* Boot time. */
449                    now,         /* Current time. */
450                    os,          /* Pointer supplied to callbacks. */
451                    sflow_agent_alloc_cb,
452                    sflow_agent_free_cb,
453                    sflow_agent_error_cb,
454                    sflow_agent_send_packet_cb);
455
456     receiver = sfl_agent_addReceiver(os->sflow_agent);
457     sfl_receiver_set_sFlowRcvrOwner(receiver, "Open vSwitch sFlow");
458     sfl_receiver_set_sFlowRcvrTimeout(receiver, 0xffffffff);
459
460     /* Set the sampling_rate down in the datapath. */
461     dpif_set_sflow_probability(os->dpif,
462                                MAX(1, UINT32_MAX / options->sampling_rate));
463
464     /* Add samplers and pollers for the currently known ports. */
465     HMAP_FOR_EACH (osp, hmap_node, &os->ports) {
466         ofproto_sflow_add_poller(os, osp, osp->odp_port);
467         ofproto_sflow_add_sampler(os, osp);
468     }
469 }
470
471 static int
472 ofproto_sflow_odp_port_to_ifindex(const struct ofproto_sflow *os,
473                                   uint16_t odp_port)
474 {
475     struct ofproto_sflow_port *osp = ofproto_sflow_find_port(os, odp_port);
476     return osp ? SFL_DS_INDEX(osp->dsi) : 0;
477 }
478
479 void
480 ofproto_sflow_received(struct ofproto_sflow *os, struct odp_msg *msg)
481 {
482     SFL_FLOW_SAMPLE_TYPE fs;
483     SFLFlow_sample_element hdrElem;
484     SFLSampled_header *header;
485     SFLFlow_sample_element switchElem;
486     SFLSampler *sampler;
487     const struct odp_sflow_sample_header *hdr;
488     const union odp_action *actions;
489     struct ofpbuf payload;
490     size_t n_actions, n_outputs;
491     size_t min_size;
492     flow_t flow;
493     size_t i;
494
495     /* Get odp_sflow_sample_header. */
496     min_size = sizeof *msg + sizeof *hdr;
497     if (min_size > msg->length) {
498         VLOG_WARN_RL(&rl, "sFlow packet too small (%"PRIu32" < %zu)",
499                      msg->length, min_size);
500         return;
501     }
502     hdr = (const struct odp_sflow_sample_header *) (msg + 1);
503
504     /* Get actions. */
505     n_actions = hdr->n_actions;
506     if (n_actions > 65536 / sizeof *actions) {
507         VLOG_WARN_RL(&rl, "too many actions in sFlow packet (%zu > %zu)",
508                      65536 / sizeof *actions, n_actions);
509         return;
510     }
511     min_size += n_actions * sizeof *actions;
512     if (min_size > msg->length) {
513         VLOG_WARN_RL(&rl, "sFlow packet with %zu actions too small "
514                      "(%"PRIu32" < %zu)",
515                      n_actions, msg->length, min_size);
516         return;
517     }
518     actions = (const union odp_action *) (hdr + 1);
519
520     /* Get packet payload and extract flow. */
521     payload.data = (union odp_action *) (actions + n_actions);
522     payload.size = msg->length - min_size;
523     flow_extract(&payload, 0, msg->port, &flow);
524
525     /* Build a flow sample */
526     memset(&fs, 0, sizeof fs);
527     fs.input = ofproto_sflow_odp_port_to_ifindex(os, msg->port);
528     fs.output = 0;              /* Filled in correctly below. */
529     fs.sample_pool = hdr->sample_pool;
530
531     /* We are going to give it to the sampler that represents this input port.
532      * By implementing "ingress-only" sampling like this we ensure that we
533      * never have to offer the same sample to more than one sampler. */
534     sampler = sfl_agent_getSamplerByIfIndex(os->sflow_agent, fs.input);
535     if (!sampler) {
536         VLOG_WARN_RL(&rl, "no sampler for input ifIndex (%"PRIu32")",
537                      fs.input);
538         return;
539     }
540
541     /* Sampled header. */
542     memset(&hdrElem, 0, sizeof hdrElem);
543     hdrElem.tag = SFLFLOW_HEADER;
544     header = &hdrElem.flowType.header;
545     header->header_protocol = SFLHEADER_ETHERNET_ISO8023;
546     /* The frame_length should include the Ethernet FCS (4 bytes),
547        but it has already been stripped,  so we need to add 4 here. */
548     header->frame_length = payload.size + 4;
549     /* Ethernet FCS stripped off. */
550     header->stripped = 4;
551     header->header_length = MIN(payload.size,
552                                 sampler->sFlowFsMaximumHeaderSize);
553     header->header_bytes = payload.data;
554
555     /* Add extended switch element. */
556     memset(&switchElem, 0, sizeof(switchElem));
557     switchElem.tag = SFLFLOW_EX_SWITCH;
558     switchElem.flowType.sw.src_vlan = ntohs(flow.dl_vlan);
559     switchElem.flowType.sw.src_priority = -1; /* XXX */
560      /* Initialize the output VLAN and priority to be the same as the input,
561         but these fields can be overriden below if affected by an action. */
562     switchElem.flowType.sw.dst_vlan = switchElem.flowType.sw.src_vlan;
563     switchElem.flowType.sw.dst_priority = switchElem.flowType.sw.src_priority;
564
565     /* Figure out the output ports. */
566     n_outputs = 0;
567     for (i = 0; i < n_actions; i++) {
568         const union odp_action *a = &actions[i];
569
570         switch (a->type) {
571         case ODPAT_OUTPUT:
572             fs.output = ofproto_sflow_odp_port_to_ifindex(os, a->output.port);
573             n_outputs++;
574             break;
575
576         case ODPAT_OUTPUT_GROUP:
577             n_outputs += (a->output_group.group == DP_GROUP_FLOOD ? os->n_flood
578                           : a->output_group.group == DP_GROUP_ALL ? os->n_all
579                           : 0);
580             break;
581
582         case ODPAT_SET_VLAN_VID:
583             switchElem.flowType.sw.dst_vlan = ntohs(a->vlan_vid.vlan_vid);
584             break;
585
586         case ODPAT_SET_VLAN_PCP:
587             switchElem.flowType.sw.dst_priority = a->vlan_pcp.vlan_pcp;
588             break;
589
590         default:
591             break;
592         }
593     }
594
595     /* Set output port, as defined by http://www.sflow.org/sflow_version_5.txt
596        (search for "Input/output port information"). */
597     if (!n_outputs) {
598         /* This value indicates that the packet was dropped for an unknown
599          * reason. */
600         fs.output = 0x40000000 | 256;
601     } else if (n_outputs > 1 || !fs.output) {
602         /* Setting the high bit means "multiple output ports". */
603         fs.output = 0x80000000 | n_outputs;
604     }
605
606     /* Submit the flow sample to be encoded into the next datagram. */
607     SFLADD_ELEMENT(&fs, &hdrElem);
608     SFLADD_ELEMENT(&fs, &switchElem);
609     sfl_sampler_writeFlowSample(sampler, &fs);
610 }
611
612 void
613 ofproto_sflow_set_group_sizes(struct ofproto_sflow *os,
614                               size_t n_flood, size_t n_all)
615 {
616     os->n_flood = n_flood;
617     os->n_all = n_all;
618 }
619
620 void
621 ofproto_sflow_run(struct ofproto_sflow *os)
622 {
623     if (ofproto_sflow_is_enabled(os)) {
624         time_t now = time_now();
625         if (now >= os->next_tick) {
626             sfl_agent_tick(os->sflow_agent, time_wall());
627             os->next_tick = now + 1;
628         }
629     }
630 }
631
632 void
633 ofproto_sflow_wait(struct ofproto_sflow *os)
634 {
635     if (ofproto_sflow_is_enabled(os)) {
636         poll_timer_wait_until(os->next_tick * 1000LL);
637     }
638 }