| /* |
| * mc.c: vlib reliable sequenced multicast distributed applications |
| * |
| * Copyright (c) 2010 Cisco and/or its affiliates. |
| * Licensed under the Apache License, Version 2.0 (the "License"); |
| * you may not use this file except in compliance with the License. |
| * You may obtain a copy of the License at: |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, software |
| * distributed under the License is distributed on an "AS IS" BASIS, |
| * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| * See the License for the specific language governing permissions and |
| * limitations under the License. |
| */ |
| |
| #include <vlib/vlib.h> |
| |
| /* |
| * 1 to enable msg id training wheels, which are useful for tracking |
| * down catchup and/or partitioned network problems |
| */ |
| #define MSG_ID_DEBUG 0 |
| |
| static format_function_t format_mc_stream_state; |
| |
| static u32 |
| elog_id_for_peer_id (mc_main_t * m, u64 peer_id) |
| { |
| uword *p, r; |
| mhash_t *h = &m->elog_id_by_peer_id; |
| |
| if (!m->elog_id_by_peer_id.hash) |
| mhash_init (h, sizeof (uword), sizeof (mc_peer_id_t)); |
| |
| p = mhash_get (h, &peer_id); |
| if (p) |
| return p[0]; |
| r = elog_string (m->elog_main, "%U", m->transport.format_peer_id, peer_id); |
| mhash_set (h, &peer_id, r, /* old_value */ 0); |
| return r; |
| } |
| |
| static u32 |
| elog_id_for_msg_name (mc_main_t * m, char *msg_name) |
| { |
| uword *p, r; |
| uword *h = m->elog_id_by_msg_name; |
| u8 *name_copy; |
| |
| if (!h) |
| h = m->elog_id_by_msg_name = hash_create_string (0, sizeof (uword)); |
| |
| p = hash_get_mem (h, msg_name); |
| if (p) |
| return p[0]; |
| r = elog_string (m->elog_main, "%s", msg_name); |
| |
| name_copy = format (0, "%s%c", msg_name, 0); |
| |
| hash_set_mem (h, name_copy, r); |
| m->elog_id_by_msg_name = h; |
| |
| return r; |
| } |
| |
| static void |
| elog_tx_msg (mc_main_t * m, u32 stream_id, u32 local_sequence, |
| u32 retry_count) |
| { |
| if (MC_EVENT_LOGGING > 0) |
| { |
| /* *INDENT-OFF* */ |
| ELOG_TYPE_DECLARE (e) = |
| { |
| .format = "tx-msg: stream %d local seq %d attempt %d", |
| .format_args = "i4i4i4", |
| }; |
| /* *INDENT-ON* */ |
| struct |
| { |
| u32 stream_id, local_sequence, retry_count; |
| } *ed; |
| ed = ELOG_DATA (m->elog_main, e); |
| ed->stream_id = stream_id; |
| ed->local_sequence = local_sequence; |
| ed->retry_count = retry_count; |
| } |
| } |
| |
| /* |
| * seq_cmp |
| * correctly compare two unsigned sequence numbers. |
| * This function works so long as x and y are within 2**(n-1) of each |
| * other, where n = bits(x, y). |
| * |
| * Magic decoder ring: |
| * seq_cmp == 0 => x and y are equal |
| * seq_cmp < 0 => x is "in the past" with respect to y |
| * seq_cmp > 0 => x is "in the future" with respect to y |
| */ |
| always_inline i32 |
| mc_seq_cmp (u32 x, u32 y) |
| { |
| return (i32) x - (i32) y; |
| } |
| |
| void * |
| mc_get_vlib_buffer (vlib_main_t * vm, u32 n_bytes, u32 * bi_return) |
| { |
| u32 n_alloc, bi; |
| vlib_buffer_t *b; |
| |
| n_alloc = vlib_buffer_alloc (vm, &bi, 1); |
| ASSERT (n_alloc == 1); |
| |
| b = vlib_get_buffer (vm, bi); |
| b->current_length = n_bytes; |
| *bi_return = bi; |
| return (void *) b->data; |
| } |
| |
| static void |
| delete_peer_with_index (mc_main_t * mcm, mc_stream_t * s, |
| uword index, int notify_application) |
| { |
| mc_stream_peer_t *p = pool_elt_at_index (s->peers, index); |
| ASSERT (p != 0); |
| if (s->config.peer_died && notify_application) |
| s->config.peer_died (mcm, s, p->id); |
| |
| s->all_peer_bitmap = clib_bitmap_andnoti (s->all_peer_bitmap, p - s->peers); |
| |
| if (MC_EVENT_LOGGING > 0) |
| { |
| /* *INDENT-OFF* */ |
| ELOG_TYPE_DECLARE (e) = |
| { |
| .format = "delete peer %s from all_peer_bitmap", |
| .format_args = "T4", |
| }; |
| /* *INDENT-ON* */ |
| struct |
| { |
| u32 peer; |
| } *ed = 0; |
| |
| ed = ELOG_DATA (mcm->elog_main, e); |
| ed->peer = elog_id_for_peer_id (mcm, p->id.as_u64); |
| } |
| /* Do not delete the pool / hash table entries, or we lose sequence number state */ |
| } |
| |
| static mc_stream_peer_t * |
| get_or_create_peer_with_id (mc_main_t * mcm, |
| mc_stream_t * s, mc_peer_id_t id, int *created) |
| { |
| uword *q = mhash_get (&s->peer_index_by_id, &id); |
| mc_stream_peer_t *p; |
| |
| if (q) |
| { |
| p = pool_elt_at_index (s->peers, q[0]); |
| goto done; |
| } |
| |
| pool_get (s->peers, p); |
| memset (p, 0, sizeof (p[0])); |
| p->id = id; |
| p->last_sequence_received = ~0; |
| mhash_set (&s->peer_index_by_id, &id, p - s->peers, /* old_value */ 0); |
| if (created) |
| *created = 1; |
| |
| done: |
| if (MC_EVENT_LOGGING > 0) |
| { |
| /* *INDENT-OFF* */ |
| ELOG_TYPE_DECLARE (e) = |
| { |
| .format = "get_or_create %s peer %s stream %d seq %d", |
| .format_args = "t4T4i4i4", |
| .n_enum_strings = 2, |
| .enum_strings = { |
| "old", "new", |
| }, |
| }; |
| /* *INDENT-ON* */ |
| struct |
| { |
| u32 is_new, peer, stream_index, rx_sequence; |
| } *ed = 0; |
| |
| ed = ELOG_DATA (mcm->elog_main, e); |
| ed->is_new = q ? 0 : 1; |
| ed->peer = elog_id_for_peer_id (mcm, p->id.as_u64); |
| ed->stream_index = s->index; |
| ed->rx_sequence = p->last_sequence_received; |
| } |
| /* $$$$ Enable or reenable this peer */ |
| s->all_peer_bitmap = clib_bitmap_ori (s->all_peer_bitmap, p - s->peers); |
| return p; |
| } |
| |
| static void |
| maybe_send_window_open_event (vlib_main_t * vm, mc_stream_t * stream) |
| { |
| vlib_one_time_waiting_process_t *p; |
| |
| if (pool_elts (stream->retry_pool) >= stream->config.window_size) |
| return; |
| |
| vec_foreach (p, stream->procs_waiting_for_open_window) |
| vlib_signal_one_time_waiting_process (vm, p); |
| |
| if (stream->procs_waiting_for_open_window) |
| _vec_len (stream->procs_waiting_for_open_window) = 0; |
| } |
| |
| static void |
| mc_retry_free (mc_main_t * mcm, mc_stream_t * s, mc_retry_t * r) |
| { |
| mc_retry_t record, *retp; |
| |
| if (r->unacked_by_peer_bitmap) |
| _vec_len (r->unacked_by_peer_bitmap) = 0; |
| |
| if (clib_fifo_elts (s->retired_fifo) >= 2 * s->config.window_size) |
| { |
| clib_fifo_sub1 (s->retired_fifo, record); |
| vlib_buffer_free_one (mcm->vlib_main, record.buffer_index); |
| } |
| |
| clib_fifo_add2 (s->retired_fifo, retp); |
| |
| retp->buffer_index = r->buffer_index; |
| retp->local_sequence = r->local_sequence; |
| |
| r->buffer_index = ~0; /* poison buffer index in this retry */ |
| } |
| |
| static void |
| mc_resend_retired (mc_main_t * mcm, mc_stream_t * s, u32 local_sequence) |
| { |
| mc_retry_t *retry; |
| |
| if (MC_EVENT_LOGGING > 0) |
| { |
| /* *INDENT-OFF* */ |
| ELOG_TYPE_DECLARE (e) = |
| { |
| .format = "resend-retired: search for local seq %d", |
| .format_args = "i4", |
| }; |
| /* *INDENT-ON* */ |
| struct |
| { |
| u32 local_sequence; |
| } *ed; |
| ed = ELOG_DATA (mcm->elog_main, e); |
| ed->local_sequence = local_sequence; |
| } |
| |
| /* *INDENT-OFF* */ |
| clib_fifo_foreach (retry, s->retired_fifo, |
| ({ |
| if (retry->local_sequence == local_sequence) |
| { |
| elog_tx_msg (mcm, s->index, retry-> local_sequence, -13); |
| mcm->transport.tx_buffer (mcm->transport.opaque, |
| MC_TRANSPORT_USER_REQUEST_TO_RELAY, |
| retry->buffer_index); |
| return; |
| } |
| })); |
| /* *INDENT-ON* */ |
| |
| if (MC_EVENT_LOGGING > 0) |
| { |
| /* *INDENT-OFF* */ |
| ELOG_TYPE_DECLARE (e) = |
| { |
| .format = "resend-retired: FAILED search for local seq %d", |
| .format_args = "i4", |
| }; |
| /* *INDENT-ON* */ |
| struct |
| { |
| u32 local_sequence; |
| } *ed; |
| ed = ELOG_DATA (mcm->elog_main, e); |
| ed->local_sequence = local_sequence; |
| } |
| } |
| |
| static uword * |
| delete_retry_fifo_elt (mc_main_t * mcm, |
| mc_stream_t * stream, |
| mc_retry_t * r, uword * dead_peer_bitmap) |
| { |
| mc_stream_peer_t *p; |
| |
| /* *INDENT-OFF* */ |
| pool_foreach (p, stream->peers, ({ |
| uword pi = p - stream->peers; |
| uword is_alive = 0 == clib_bitmap_get (r->unacked_by_peer_bitmap, pi); |
| |
| if (! is_alive) |
| dead_peer_bitmap = clib_bitmap_ori (dead_peer_bitmap, pi); |
| |
| if (MC_EVENT_LOGGING > 0) |
| { |
| ELOG_TYPE_DECLARE (e) = { |
| .format = "delete_retry_fifo_elt: peer %s is %s", |
| .format_args = "T4t4", |
| .n_enum_strings = 2, |
| .enum_strings = { "alive", "dead", }, |
| }; |
| struct { u32 peer, is_alive; } * ed; |
| ed = ELOG_DATA (mcm->elog_main, e); |
| ed->peer = elog_id_for_peer_id (mcm, p->id.as_u64); |
| ed->is_alive = is_alive; |
| } |
| })); |
| /* *INDENT-ON* */ |
| |
| hash_unset (stream->retry_index_by_local_sequence, r->local_sequence); |
| mc_retry_free (mcm, stream, r); |
| |
| return dead_peer_bitmap; |
| } |
| |
| always_inline mc_retry_t * |
| prev_retry (mc_stream_t * s, mc_retry_t * r) |
| { |
| return (r->prev_index != ~0 |
| ? pool_elt_at_index (s->retry_pool, r->prev_index) : 0); |
| } |
| |
| always_inline mc_retry_t * |
| next_retry (mc_stream_t * s, mc_retry_t * r) |
| { |
| return (r->next_index != ~0 |
| ? pool_elt_at_index (s->retry_pool, r->next_index) : 0); |
| } |
| |
| always_inline void |
| remove_retry_from_pool (mc_stream_t * s, mc_retry_t * r) |
| { |
| mc_retry_t *p = prev_retry (s, r); |
| mc_retry_t *n = next_retry (s, r); |
| |
| if (p) |
| p->next_index = r->next_index; |
| else |
| s->retry_head_index = r->next_index; |
| if (n) |
| n->prev_index = r->prev_index; |
| else |
| s->retry_tail_index = r->prev_index; |
| |
| pool_put_index (s->retry_pool, r - s->retry_pool); |
| } |
| |
| static void |
| check_retry (mc_main_t * mcm, mc_stream_t * s) |
| { |
| mc_retry_t *r; |
| vlib_main_t *vm = mcm->vlib_main; |
| f64 now = vlib_time_now (vm); |
| uword *dead_peer_bitmap = 0; |
| u32 ri, ri_next; |
| |
| for (ri = s->retry_head_index; ri != ~0; ri = ri_next) |
| { |
| r = pool_elt_at_index (s->retry_pool, ri); |
| ri_next = r->next_index; |
| |
| if (now < r->sent_at + s->config.retry_interval) |
| continue; |
| |
| r->n_retries += 1; |
| if (r->n_retries > s->config.retry_limit) |
| { |
| dead_peer_bitmap = |
| delete_retry_fifo_elt (mcm, s, r, dead_peer_bitmap); |
| remove_retry_from_pool (s, r); |
| } |
| else |
| { |
| if (MC_EVENT_LOGGING > 0) |
| { |
| mc_stream_peer_t *p; |
| |
| /* *INDENT-OFF* */ |
| ELOG_TYPE_DECLARE (t) = |
| { |
| .format = "resend local seq %d attempt %d", |
| .format_args = "i4i4", |
| }; |
| /* *INDENT-ON* */ |
| |
| /* *INDENT-OFF* */ |
| pool_foreach (p, s->peers, ({ |
| if (clib_bitmap_get (r->unacked_by_peer_bitmap, p - s->peers)) |
| { |
| ELOG_TYPE_DECLARE (ev) = { |
| .format = "resend: needed by peer %s local seq %d", |
| .format_args = "T4i4", |
| }; |
| struct { u32 peer, rx_sequence; } * ed; |
| ed = ELOG_DATA (mcm->elog_main, ev); |
| ed->peer = elog_id_for_peer_id (mcm, p->id.as_u64); |
| ed->rx_sequence = r->local_sequence; |
| } |
| })); |
| /* *INDENT-ON* */ |
| |
| struct |
| { |
| u32 sequence; |
| u32 trail; |
| } *ed; |
| ed = ELOG_DATA (mcm->elog_main, t); |
| ed->sequence = r->local_sequence; |
| ed->trail = r->n_retries; |
| } |
| |
| r->sent_at = vlib_time_now (vm); |
| s->stats.n_retries += 1; |
| |
| elog_tx_msg (mcm, s->index, r->local_sequence, r->n_retries); |
| |
| mcm->transport.tx_buffer |
| (mcm->transport.opaque, |
| MC_TRANSPORT_USER_REQUEST_TO_RELAY, r->buffer_index); |
| } |
| } |
| |
| maybe_send_window_open_event (mcm->vlib_main, s); |
| |
| /* Delete any dead peers we've found. */ |
| if (!clib_bitmap_is_zero (dead_peer_bitmap)) |
| { |
| uword i; |
| |
| /* *INDENT-OFF* */ |
| clib_bitmap_foreach (i, dead_peer_bitmap, ({ |
| delete_peer_with_index (mcm, s, i, /* notify_application */ 1); |
| |
| /* Delete any references to just deleted peer in retry pool. */ |
| pool_foreach (r, s->retry_pool, ({ |
| r->unacked_by_peer_bitmap = |
| clib_bitmap_andnoti (r->unacked_by_peer_bitmap, i); |
| })); |
| })); |
| /* *INDENT-ON* */ |
| clib_bitmap_free (dead_peer_bitmap); |
| } |
| } |
| |
| always_inline mc_main_t * |
| mc_node_get_main (vlib_node_runtime_t * node) |
| { |
| mc_main_t **p = (void *) node->runtime_data; |
| return p[0]; |
| } |
| |
| static uword |
| mc_retry_process (vlib_main_t * vm, |
| vlib_node_runtime_t * node, vlib_frame_t * f) |
| { |
| mc_main_t *mcm = mc_node_get_main (node); |
| mc_stream_t *s; |
| |
| while (1) |
| { |
| vlib_process_suspend (vm, 1.0); |
| vec_foreach (s, mcm->stream_vector) |
| { |
| if (s->state != MC_STREAM_STATE_invalid) |
| check_retry (mcm, s); |
| } |
| } |
| return 0; /* not likely */ |
| } |
| |
| static void |
| send_join_or_leave_request (mc_main_t * mcm, u32 stream_index, u32 is_join) |
| { |
| vlib_main_t *vm = mcm->vlib_main; |
| mc_msg_join_or_leave_request_t *mp; |
| u32 bi; |
| |
| mp = mc_get_vlib_buffer (vm, sizeof (mp[0]), &bi); |
| memset (mp, 0, sizeof (*mp)); |
| mp->type = MC_MSG_TYPE_join_or_leave_request; |
| mp->peer_id = mcm->transport.our_ack_peer_id; |
| mp->stream_index = stream_index; |
| mp->is_join = is_join; |
| |
| mc_byte_swap_msg_join_or_leave_request (mp); |
| |
| /* |
| * These msgs are unnumbered, unordered so send on the from-relay |
| * channel. |
| */ |
| mcm->transport.tx_buffer (mcm->transport.opaque, MC_TRANSPORT_JOIN, bi); |
| } |
| |
| static uword |
| mc_join_ager_process (vlib_main_t * vm, |
| vlib_node_runtime_t * node, vlib_frame_t * f) |
| { |
| mc_main_t *mcm = mc_node_get_main (node); |
| |
| while (1) |
| { |
| if (mcm->joins_in_progress) |
| { |
| mc_stream_t *s; |
| vlib_one_time_waiting_process_t *p; |
| f64 now = vlib_time_now (vm); |
| |
| vec_foreach (s, mcm->stream_vector) |
| { |
| if (s->state != MC_STREAM_STATE_join_in_progress) |
| continue; |
| |
| if (now > s->join_timeout) |
| { |
| s->state = MC_STREAM_STATE_ready; |
| |
| if (MC_EVENT_LOGGING > 0) |
| { |
| /* *INDENT-OFF* */ |
| ELOG_TYPE_DECLARE (e) = |
| { |
| .format = "stream %d join timeout", |
| }; |
| /* *INDENT-ON* */ |
| ELOG (mcm->elog_main, e, s->index); |
| } |
| /* Make sure that this app instance exists as a stream peer, |
| or we may answer a catchup request with a NULL |
| all_peer_bitmap... */ |
| (void) get_or_create_peer_with_id |
| (mcm, s, mcm->transport.our_ack_peer_id, /* created */ 0); |
| |
| vec_foreach (p, s->procs_waiting_for_join_done) |
| vlib_signal_one_time_waiting_process (vm, p); |
| if (s->procs_waiting_for_join_done) |
| _vec_len (s->procs_waiting_for_join_done) = 0; |
| |
| mcm->joins_in_progress--; |
| ASSERT (mcm->joins_in_progress >= 0); |
| } |
| else |
| { |
| /* Resent join request which may have been lost. */ |
| send_join_or_leave_request (mcm, s->index, 1 /* is_join */ ); |
| |
| /* We're *not* alone, retry for as long as it takes */ |
| if (mcm->relay_state == MC_RELAY_STATE_SLAVE) |
| s->join_timeout = vlib_time_now (vm) + 2.0; |
| |
| |
| if (MC_EVENT_LOGGING > 0) |
| { |
| /* *INDENT-OFF* */ |
| ELOG_TYPE_DECLARE (e) = |
| { |
| .format = "stream %d resend join request", |
| }; |
| /* *INDENT-ON* */ |
| ELOG (mcm->elog_main, e, s->index); |
| } |
| } |
| } |
| } |
| |
| vlib_process_suspend (vm, .5); |
| } |
| |
| return 0; /* not likely */ |
| } |
| |
| static void |
| serialize_mc_register_stream_name (serialize_main_t * m, va_list * va) |
| { |
| char *name = va_arg (*va, char *); |
| serialize_cstring (m, name); |
| } |
| |
| static void |
| elog_stream_name (char *buf, int n_buf_bytes, char *v) |
| { |
| clib_memcpy (buf, v, clib_min (n_buf_bytes - 1, vec_len (v))); |
| buf[n_buf_bytes - 1] = 0; |
| } |
| |
| static void |
| unserialize_mc_register_stream_name (serialize_main_t * m, va_list * va) |
| { |
| mc_main_t *mcm = va_arg (*va, mc_main_t *); |
| char *name; |
| mc_stream_t *s; |
| uword *p; |
| |
| unserialize_cstring (m, &name); |
| |
| if ((p = hash_get_mem (mcm->stream_index_by_name, name))) |
| { |
| if (MC_EVENT_LOGGING > 0) |
| { |
| /* *INDENT-OFF* */ |
| ELOG_TYPE_DECLARE (e) = |
| { |
| .format = "stream index %d already named %s", |
| .format_args = "i4s16", |
| }; |
| /* *INDENT-ON* */ |
| struct |
| { |
| u32 stream_index; |
| char name[16]; |
| } *ed; |
| ed = ELOG_DATA (mcm->elog_main, e); |
| ed->stream_index = p[0]; |
| elog_stream_name (ed->name, sizeof (ed->name), name); |
| } |
| |
| vec_free (name); |
| return; |
| } |
| |
| vec_add2 (mcm->stream_vector, s, 1); |
| mc_stream_init (s); |
| s->state = MC_STREAM_STATE_name_known; |
| s->index = s - mcm->stream_vector; |
| s->config.name = name; |
| |
| if (MC_EVENT_LOGGING > 0) |
| { |
| /* *INDENT-OFF* */ |
| ELOG_TYPE_DECLARE (e) = |
| { |
| .format = "stream index %d named %s", |
| .format_args = "i4s16", |
| }; |
| /* *INDENT-ON* */ |
| struct |
| { |
| u32 stream_index; |
| char name[16]; |
| } *ed; |
| ed = ELOG_DATA (mcm->elog_main, e); |
| ed->stream_index = s->index; |
| elog_stream_name (ed->name, sizeof (ed->name), name); |
| } |
| |
| hash_set_mem (mcm->stream_index_by_name, name, s->index); |
| |
| p = hash_get (mcm->procs_waiting_for_stream_name_by_name, name); |
| if (p) |
| { |
| vlib_one_time_waiting_process_t *wp, **w; |
| w = pool_elt_at_index (mcm->procs_waiting_for_stream_name_pool, p[0]); |
| vec_foreach (wp, w[0]) |
| vlib_signal_one_time_waiting_process (mcm->vlib_main, wp); |
| pool_put (mcm->procs_waiting_for_stream_name_pool, w); |
| hash_unset_mem (mcm->procs_waiting_for_stream_name_by_name, name); |
| } |
| } |
| |
| /* *INDENT-OFF* */ |
| MC_SERIALIZE_MSG (mc_register_stream_name_msg, static) = |
| { |
| .name = "mc_register_stream_name", |
| .serialize = serialize_mc_register_stream_name, |
| .unserialize = unserialize_mc_register_stream_name, |
| }; |
| /* *INDENT-ON* */ |
| |
| void |
| mc_rx_buffer_unserialize (mc_main_t * mcm, |
| mc_stream_t * stream, |
| mc_peer_id_t peer_id, u32 buffer_index) |
| { |
| return mc_unserialize (mcm, stream, buffer_index); |
| } |
| |
| static u8 * |
| mc_internal_catchup_snapshot (mc_main_t * mcm, |
| u8 * data_vector, |
| u32 last_global_sequence_processed) |
| { |
| serialize_main_t m; |
| |
| /* Append serialized data to data vector. */ |
| serialize_open_vector (&m, data_vector); |
| m.stream.current_buffer_index = vec_len (data_vector); |
| |
| serialize (&m, serialize_mc_main, mcm); |
| return serialize_close_vector (&m); |
| } |
| |
| static void |
| mc_internal_catchup (mc_main_t * mcm, u8 * data, u32 n_data_bytes) |
| { |
| serialize_main_t s; |
| |
| unserialize_open_data (&s, data, n_data_bytes); |
| |
| unserialize (&s, unserialize_mc_main, mcm); |
| } |
| |
| /* Overridden from the application layer, not actually used here */ |
| void mc_stream_join_process_hold (void) __attribute__ ((weak)); |
| void |
| mc_stream_join_process_hold (void) |
| { |
| } |
| |
| static u32 |
| mc_stream_join_helper (mc_main_t * mcm, |
| mc_stream_config_t * config, u32 is_internal) |
| { |
| mc_stream_t *s; |
| vlib_main_t *vm = mcm->vlib_main; |
| |
| s = 0; |
| if (!is_internal) |
| { |
| uword *p; |
| |
| /* Already have a stream with given name? */ |
| if ((s = mc_stream_by_name (mcm, config->name))) |
| { |
| /* Already joined and ready? */ |
| if (s->state == MC_STREAM_STATE_ready) |
| return s->index; |
| } |
| |
| /* First join MC internal stream. */ |
| if (!mcm->stream_vector |
| || (mcm->stream_vector[MC_STREAM_INDEX_INTERNAL].state |
| == MC_STREAM_STATE_invalid)) |
| { |
| static mc_stream_config_t c = { |
| .name = "mc-internal", |
| .rx_buffer = mc_rx_buffer_unserialize, |
| .catchup = mc_internal_catchup, |
| .catchup_snapshot = mc_internal_catchup_snapshot, |
| }; |
| |
| c.save_snapshot = config->save_snapshot; |
| |
| mc_stream_join_helper (mcm, &c, /* is_internal */ 1); |
| } |
| |
| /* If stream is still unknown register this name and wait for |
| sequenced message to name stream. This way all peers agree |
| on stream name to index mappings. */ |
| s = mc_stream_by_name (mcm, config->name); |
| if (!s) |
| { |
| vlib_one_time_waiting_process_t *wp, **w; |
| u8 *name_copy = format (0, "%s", config->name); |
| |
| mc_serialize_stream (mcm, |
| MC_STREAM_INDEX_INTERNAL, |
| &mc_register_stream_name_msg, config->name); |
| |
| /* Wait for this stream to be named. */ |
| p = |
| hash_get_mem (mcm->procs_waiting_for_stream_name_by_name, |
| name_copy); |
| if (p) |
| w = |
| pool_elt_at_index (mcm->procs_waiting_for_stream_name_pool, |
| p[0]); |
| else |
| { |
| pool_get (mcm->procs_waiting_for_stream_name_pool, w); |
| if (!mcm->procs_waiting_for_stream_name_by_name) |
| mcm->procs_waiting_for_stream_name_by_name = hash_create_string ( /* elts */ 0, /* value size */ |
| sizeof |
| (uword)); |
| hash_set_mem (mcm->procs_waiting_for_stream_name_by_name, |
| name_copy, |
| w - mcm->procs_waiting_for_stream_name_pool); |
| w[0] = 0; |
| } |
| |
| vec_add2 (w[0], wp, 1); |
| vlib_current_process_wait_for_one_time_event (vm, wp); |
| vec_free (name_copy); |
| } |
| |
| /* Name should be known now. */ |
| s = mc_stream_by_name (mcm, config->name); |
| ASSERT (s != 0); |
| ASSERT (s->state == MC_STREAM_STATE_name_known); |
| } |
| |
| if (!s) |
| { |
| vec_add2 (mcm->stream_vector, s, 1); |
| mc_stream_init (s); |
| s->index = s - mcm->stream_vector; |
| } |
| |
| { |
| /* Save name since we could have already used it as hash key. */ |
| char *name_save = s->config.name; |
| |
| s->config = config[0]; |
| |
| if (name_save) |
| s->config.name = name_save; |
| } |
| |
| if (s->config.window_size == 0) |
| s->config.window_size = 8; |
| |
| if (s->config.retry_interval == 0.0) |
| s->config.retry_interval = 1.0; |
| |
| /* Sanity. */ |
| ASSERT (s->config.retry_interval < 30); |
| |
| if (s->config.retry_limit == 0) |
| s->config.retry_limit = 7; |
| |
| s->state = MC_STREAM_STATE_join_in_progress; |
| if (!s->peer_index_by_id.hash) |
| mhash_init (&s->peer_index_by_id, sizeof (uword), sizeof (mc_peer_id_t)); |
| |
| /* If we don't hear from someone in 5 seconds, we're alone */ |
| s->join_timeout = vlib_time_now (vm) + 5.0; |
| mcm->joins_in_progress++; |
| |
| if (MC_EVENT_LOGGING > 0) |
| { |
| /* *INDENT-OFF* */ |
| ELOG_TYPE_DECLARE (e) = |
| { |
| .format = "stream index %d join request %s", |
| .format_args = "i4s16", |
| }; |
| /* *INDENT-ON* */ |
| struct |
| { |
| u32 stream_index; |
| char name[16]; |
| } *ed; |
| ed = ELOG_DATA (mcm->elog_main, e); |
| ed->stream_index = s->index; |
| elog_stream_name (ed->name, sizeof (ed->name), s->config.name); |
| } |
| |
| send_join_or_leave_request (mcm, s->index, 1 /* join */ ); |
| |
| vlib_current_process_wait_for_one_time_event_vector |
| (vm, &s->procs_waiting_for_join_done); |
| |
| if (MC_EVENT_LOGGING) |
| { |
| ELOG_TYPE (e, "join complete stream %d"); |
| ELOG (mcm->elog_main, e, s->index); |
| } |
| |
| return s->index; |
| } |
| |
| u32 |
| mc_stream_join (mc_main_t * mcm, mc_stream_config_t * config) |
| { |
| return mc_stream_join_helper (mcm, config, /* is_internal */ 0); |
| } |
| |
| void |
| mc_stream_leave (mc_main_t * mcm, u32 stream_index) |
| { |
| mc_stream_t *s = mc_stream_by_index (mcm, stream_index); |
| |
| if (!s) |
| return; |
| |
| if (MC_EVENT_LOGGING) |
| { |
| /* *INDENT-OFF* */ |
| ELOG_TYPE_DECLARE (t) = |
| { |
| .format = "leave-stream: %d",.format_args = "i4", |
| }; |
| /* *INDENT-ON* */ |
| struct |
| { |
| u32 index; |
| } *ed; |
| ed = ELOG_DATA (mcm->elog_main, t); |
| ed->index = stream_index; |
| } |
| |
| send_join_or_leave_request (mcm, stream_index, 0 /* is_join */ ); |
| mc_stream_free (s); |
| s->state = MC_STREAM_STATE_name_known; |
| } |
| |
| void |
| mc_msg_join_or_leave_request_handler (mc_main_t * mcm, |
| mc_msg_join_or_leave_request_t * req, |
| u32 buffer_index) |
| { |
| mc_stream_t *s; |
| mc_msg_join_reply_t *rep; |
| u32 bi; |
| |
| mc_byte_swap_msg_join_or_leave_request (req); |
| |
| s = mc_stream_by_index (mcm, req->stream_index); |
| if (!s || s->state != MC_STREAM_STATE_ready) |
| return; |
| |
| /* If the peer is joining, create it */ |
| if (req->is_join) |
| { |
| mc_stream_t *this_s; |
| |
| /* We're not in a position to catch up a peer until all |
| stream joins are complete. */ |
| if (0) |
| { |
| /* XXX This is hard to test so we've. */ |
| vec_foreach (this_s, mcm->stream_vector) |
| { |
| if (this_s->state != MC_STREAM_STATE_ready |
| && this_s->state != MC_STREAM_STATE_name_known) |
| return; |
| } |
| } |
| else if (mcm->joins_in_progress > 0) |
| return; |
| |
| (void) get_or_create_peer_with_id (mcm, s, req->peer_id, |
| /* created */ 0); |
| |
| rep = mc_get_vlib_buffer (mcm->vlib_main, sizeof (rep[0]), &bi); |
| memset (rep, 0, sizeof (rep[0])); |
| rep->type = MC_MSG_TYPE_join_reply; |
| rep->stream_index = req->stream_index; |
| |
| mc_byte_swap_msg_join_reply (rep); |
| /* These two are already in network byte order... */ |
| rep->peer_id = mcm->transport.our_ack_peer_id; |
| rep->catchup_peer_id = mcm->transport.our_catchup_peer_id; |
| |
| mcm->transport.tx_buffer (mcm->transport.opaque, MC_TRANSPORT_JOIN, bi); |
| } |
| else |
| { |
| if (s->config.peer_died) |
| s->config.peer_died (mcm, s, req->peer_id); |
| } |
| } |
| |
| void |
| mc_msg_join_reply_handler (mc_main_t * mcm, |
| mc_msg_join_reply_t * mp, u32 buffer_index) |
| { |
| mc_stream_t *s; |
| |
| mc_byte_swap_msg_join_reply (mp); |
| |
| s = mc_stream_by_index (mcm, mp->stream_index); |
| |
| if (!s || s->state != MC_STREAM_STATE_join_in_progress) |
| return; |
| |
| /* Switch to catchup state; next join reply |
| for this stream will be ignored. */ |
| s->state = MC_STREAM_STATE_catchup; |
| |
| mcm->joins_in_progress--; |
| mcm->transport.catchup_request_fun (mcm->transport.opaque, |
| mp->stream_index, mp->catchup_peer_id); |
| } |
| |
| void |
| mc_wait_for_stream_ready (mc_main_t * m, char *stream_name) |
| { |
| mc_stream_t *s; |
| |
| while (1) |
| { |
| s = mc_stream_by_name (m, stream_name); |
| if (s) |
| break; |
| vlib_process_suspend (m->vlib_main, .1); |
| } |
| |
| /* It's OK to send a message in catchup and ready states. */ |
| if (s->state == MC_STREAM_STATE_catchup |
| || s->state == MC_STREAM_STATE_ready) |
| return; |
| |
| /* Otherwise we are waiting for a join to finish. */ |
| vlib_current_process_wait_for_one_time_event_vector |
| (m->vlib_main, &s->procs_waiting_for_join_done); |
| } |
| |
| u32 |
| mc_stream_send (mc_main_t * mcm, u32 stream_index, u32 buffer_index) |
| { |
| mc_stream_t *s = mc_stream_by_index (mcm, stream_index); |
| vlib_main_t *vm = mcm->vlib_main; |
| mc_retry_t *r; |
| mc_msg_user_request_t *mp; |
| vlib_buffer_t *b = vlib_get_buffer (vm, buffer_index); |
| u32 ri; |
| |
| if (!s) |
| return 0; |
| |
| if (s->state != MC_STREAM_STATE_ready) |
| vlib_current_process_wait_for_one_time_event_vector |
| (vm, &s->procs_waiting_for_join_done); |
| |
| while (pool_elts (s->retry_pool) >= s->config.window_size) |
| { |
| vlib_current_process_wait_for_one_time_event_vector |
| (vm, &s->procs_waiting_for_open_window); |
| } |
| |
| pool_get (s->retry_pool, r); |
| ri = r - s->retry_pool; |
| |
| r->prev_index = s->retry_tail_index; |
| r->next_index = ~0; |
| s->retry_tail_index = ri; |
| |
| if (r->prev_index == ~0) |
| s->retry_head_index = ri; |
| else |
| { |
| mc_retry_t *p = pool_elt_at_index (s->retry_pool, r->prev_index); |
| p->next_index = ri; |
| } |
| |
| vlib_buffer_advance (b, -sizeof (mp[0])); |
| mp = vlib_buffer_get_current (b); |
| |
| mp->peer_id = mcm->transport.our_ack_peer_id; |
| /* mp->transport.global_sequence set by relay agent. */ |
| mp->global_sequence = 0xdeadbeef; |
| mp->stream_index = s->index; |
| mp->local_sequence = s->our_local_sequence++; |
| mp->n_data_bytes = |
| vlib_buffer_index_length_in_chain (vm, buffer_index) - sizeof (mp[0]); |
| |
| r->buffer_index = buffer_index; |
| r->local_sequence = mp->local_sequence; |
| r->sent_at = vlib_time_now (vm); |
| r->n_retries = 0; |
| |
| /* Retry will be freed when all currently known peers have acked. */ |
| vec_validate (r->unacked_by_peer_bitmap, vec_len (s->all_peer_bitmap) - 1); |
| vec_copy (r->unacked_by_peer_bitmap, s->all_peer_bitmap); |
| |
| hash_set (s->retry_index_by_local_sequence, r->local_sequence, |
| r - s->retry_pool); |
| |
| elog_tx_msg (mcm, s->index, mp->local_sequence, r->n_retries); |
| |
| mc_byte_swap_msg_user_request (mp); |
| |
| mcm->transport.tx_buffer (mcm->transport.opaque, |
| MC_TRANSPORT_USER_REQUEST_TO_RELAY, buffer_index); |
| |
| s->user_requests_sent++; |
| |
| /* return amount of window remaining */ |
| return s->config.window_size - pool_elts (s->retry_pool); |
| } |
| |
| void |
| mc_msg_user_request_handler (mc_main_t * mcm, mc_msg_user_request_t * mp, |
| u32 buffer_index) |
| { |
| vlib_main_t *vm = mcm->vlib_main; |
| mc_stream_t *s; |
| mc_stream_peer_t *peer; |
| i32 seq_cmp_result; |
| static int once = 0; |
| |
| mc_byte_swap_msg_user_request (mp); |
| |
| s = mc_stream_by_index (mcm, mp->stream_index); |
| |
| /* Not signed up for this stream? Turf-o-matic */ |
| if (!s || s->state != MC_STREAM_STATE_ready) |
| { |
| vlib_buffer_free_one (vm, buffer_index); |
| return; |
| } |
| |
| /* Find peer, including ourselves. */ |
| peer = get_or_create_peer_with_id (mcm, s, mp->peer_id, |
| /* created */ 0); |
| |
| seq_cmp_result = mc_seq_cmp (mp->local_sequence, |
| peer->last_sequence_received + 1); |
| |
| if (MC_EVENT_LOGGING > 0) |
| { |
| /* *INDENT-OFF* */ |
| ELOG_TYPE_DECLARE (e) = |
| { |
| .format = "rx-msg: peer %s stream %d rx seq %d seq_cmp %d", |
| .format_args = "T4i4i4i4", |
| }; |
| /* *INDENT-ON* */ |
| struct |
| { |
| u32 peer, stream_index, rx_sequence; |
| i32 seq_cmp_result; |
| } *ed; |
| ed = ELOG_DATA (mcm->elog_main, e); |
| ed->peer = elog_id_for_peer_id (mcm, peer->id.as_u64); |
| ed->stream_index = mp->stream_index; |
| ed->rx_sequence = mp->local_sequence; |
| ed->seq_cmp_result = seq_cmp_result; |
| } |
| |
| if (0 && mp->stream_index == 1 && once == 0) |
| { |
| once = 1; |
| ELOG_TYPE (e, "FAKE lost msg on stream 1"); |
| ELOG (mcm->elog_main, e, 0); |
| return; |
| } |
| |
| peer->last_sequence_received += seq_cmp_result == 0; |
| s->user_requests_received++; |
| |
| if (seq_cmp_result > 0) |
| peer->stats.n_msgs_from_future += 1; |
| |
| /* Send ack even if msg from future */ |
| if (1) |
| { |
| mc_msg_user_ack_t *rp; |
| u32 bi; |
| |
| rp = mc_get_vlib_buffer (vm, sizeof (rp[0]), &bi); |
| rp->peer_id = mcm->transport.our_ack_peer_id; |
| rp->stream_index = s->index; |
| rp->local_sequence = mp->local_sequence; |
| rp->seq_cmp_result = seq_cmp_result; |
| |
| if (MC_EVENT_LOGGING > 0) |
| { |
| /* *INDENT-OFF* */ |
| ELOG_TYPE_DECLARE (e) = |
| { |
| .format = "tx-ack: stream %d local seq %d", |
| .format_args = "i4i4", |
| }; |
| /* *INDENT-ON* */ |
| struct |
| { |
| u32 stream_index; |
| u32 local_sequence; |
| } *ed; |
| ed = ELOG_DATA (mcm->elog_main, e); |
| ed->stream_index = rp->stream_index; |
| ed->local_sequence = rp->local_sequence; |
| } |
| |
| mc_byte_swap_msg_user_ack (rp); |
| |
| mcm->transport.tx_ack (mcm->transport.opaque, mp->peer_id, bi); |
| /* Msg from past? If so, free the buffer... */ |
| if (seq_cmp_result < 0) |
| { |
| vlib_buffer_free_one (vm, buffer_index); |
| peer->stats.n_msgs_from_past += 1; |
| } |
| } |
| |
| if (seq_cmp_result == 0) |
| { |
| vlib_buffer_t *b = vlib_get_buffer (vm, buffer_index); |
| switch (s->state) |
| { |
| case MC_STREAM_STATE_ready: |
| vlib_buffer_advance (b, sizeof (mp[0])); |
| s->config.rx_buffer (mcm, s, mp->peer_id, buffer_index); |
| |
| /* Stream vector can change address via rx callback for mc-internal |
| stream. */ |
| s = mc_stream_by_index (mcm, mp->stream_index); |
| ASSERT (s != 0); |
| s->last_global_sequence_processed = mp->global_sequence; |
| break; |
| |
| case MC_STREAM_STATE_catchup: |
| clib_fifo_add1 (s->catchup_fifo, buffer_index); |
| break; |
| |
| default: |
| clib_warning ("stream in unknown state %U", |
| format_mc_stream_state, s->state); |
| break; |
| } |
| } |
| } |
| |
| void |
| mc_msg_user_ack_handler (mc_main_t * mcm, mc_msg_user_ack_t * mp, |
| u32 buffer_index) |
| { |
| vlib_main_t *vm = mcm->vlib_main; |
| uword *p; |
| mc_stream_t *s; |
| mc_stream_peer_t *peer; |
| mc_retry_t *r; |
| int peer_created = 0; |
| |
| mc_byte_swap_msg_user_ack (mp); |
| |
| s = mc_stream_by_index (mcm, mp->stream_index); |
| |
| if (MC_EVENT_LOGGING > 0) |
| { |
| /* *INDENT-OFF* */ |
| ELOG_TYPE_DECLARE (t) = |
| { |
| .format = "rx-ack: local seq %d peer %s seq_cmp_result %d", |
| .format_args = "i4T4i4", |
| }; |
| /* *INDENT-ON* */ |
| |
| struct |
| { |
| u32 local_sequence; |
| u32 peer; |
| i32 seq_cmp_result; |
| } *ed; |
| ed = ELOG_DATA (mcm->elog_main, t); |
| ed->local_sequence = mp->local_sequence; |
| ed->peer = elog_id_for_peer_id (mcm, mp->peer_id.as_u64); |
| ed->seq_cmp_result = mp->seq_cmp_result; |
| } |
| |
| /* Unknown stream? */ |
| if (!s) |
| return; |
| |
| /* Find the peer which just ack'ed. */ |
| peer = get_or_create_peer_with_id (mcm, s, mp->peer_id, |
| /* created */ &peer_created); |
| |
| /* |
| * Peer reports message from the future. If it's not in the retry |
| * fifo, look for a retired message. |
| */ |
| if (mp->seq_cmp_result > 0) |
| { |
| p = hash_get (s->retry_index_by_local_sequence, mp->local_sequence - |
| mp->seq_cmp_result); |
| if (p == 0) |
| mc_resend_retired (mcm, s, mp->local_sequence - mp->seq_cmp_result); |
| |
| /* Normal retry should fix it... */ |
| return; |
| } |
| |
| /* |
| * Pointer to the indicated retry fifo entry. |
| * Worth hashing because we could use a window size of 100 or 1000. |
| */ |
| p = hash_get (s->retry_index_by_local_sequence, mp->local_sequence); |
| |
| /* |
| * Is this a duplicate ACK, received after we've retired the |
| * fifo entry. This can happen when learning about new |
| * peers. |
| */ |
| if (p == 0) |
| { |
| if (MC_EVENT_LOGGING > 0) |
| { |
| /* *INDENT-OFF* */ |
| ELOG_TYPE_DECLARE (t) = |
| { |
| .format = "ack: for seq %d from peer %s no fifo elt", |
| .format_args = "i4T4", |
| }; |
| /* *INDENT-ON* */ |
| |
| struct |
| { |
| u32 seq; |
| u32 peer; |
| } *ed; |
| ed = ELOG_DATA (mcm->elog_main, t); |
| ed->seq = mp->local_sequence; |
| ed->peer = elog_id_for_peer_id (mcm, mp->peer_id.as_u64); |
| } |
| |
| return; |
| } |
| |
| r = pool_elt_at_index (s->retry_pool, p[0]); |
| |
| /* Make sure that this new peer ACKs our msgs from now on */ |
| if (peer_created) |
| { |
| mc_retry_t *later_retry = next_retry (s, r); |
| |
| while (later_retry) |
| { |
| later_retry->unacked_by_peer_bitmap = |
| clib_bitmap_ori (later_retry->unacked_by_peer_bitmap, |
| peer - s->peers); |
| later_retry = next_retry (s, later_retry); |
| } |
| } |
| |
| ASSERT (mp->local_sequence == r->local_sequence); |
| |
| /* If we weren't expecting to hear from this peer */ |
| if (!peer_created && |
| !clib_bitmap_get (r->unacked_by_peer_bitmap, peer - s->peers)) |
| { |
| if (MC_EVENT_LOGGING > 0) |
| { |
| /* *INDENT-OFF* */ |
| ELOG_TYPE_DECLARE (t) = |
| { |
| .format = "dup-ack: for seq %d from peer %s", |
| .format_args = "i4T4", |
| }; |
| /* *INDENT-ON* */ |
| struct |
| { |
| u32 seq; |
| u32 peer; |
| } *ed; |
| ed = ELOG_DATA (mcm->elog_main, t); |
| ed->seq = r->local_sequence; |
| ed->peer = elog_id_for_peer_id (mcm, peer->id.as_u64); |
| } |
| if (!clib_bitmap_is_zero (r->unacked_by_peer_bitmap)) |
| return; |
| } |
| |
| if (MC_EVENT_LOGGING > 0) |
| { |
| /* *INDENT-OFF* */ |
| ELOG_TYPE_DECLARE (t) = |
| { |
| .format = "ack: for seq %d from peer %s", |
| .format_args = "i4T4", |
| }; |
| /* *INDENT-ON* */ |
| struct |
| { |
| u32 seq; |
| u32 peer; |
| } *ed; |
| ed = ELOG_DATA (mcm->elog_main, t); |
| ed->seq = mp->local_sequence; |
| ed->peer = elog_id_for_peer_id (mcm, peer->id.as_u64); |
| } |
| |
| r->unacked_by_peer_bitmap = |
| clib_bitmap_andnoti (r->unacked_by_peer_bitmap, peer - s->peers); |
| |
| /* Not all clients have ack'ed */ |
| if (!clib_bitmap_is_zero (r->unacked_by_peer_bitmap)) |
| { |
| return; |
| } |
| if (MC_EVENT_LOGGING > 0) |
| { |
| /* *INDENT-OFF* */ |
| ELOG_TYPE_DECLARE (t) = |
| { |
| .format = "ack: retire fifo elt loc seq %d after %d acks", |
| .format_args = "i4i4", |
| }; |
| /* *INDENT-ON* */ |
| struct |
| { |
| u32 seq; |
| u32 npeers; |
| } *ed; |
| ed = ELOG_DATA (mcm->elog_main, t); |
| ed->seq = r->local_sequence; |
| ed->npeers = pool_elts (s->peers); |
| } |
| |
| hash_unset (s->retry_index_by_local_sequence, mp->local_sequence); |
| mc_retry_free (mcm, s, r); |
| remove_retry_from_pool (s, r); |
| maybe_send_window_open_event (vm, s); |
| } |
| |
| #define EVENT_MC_SEND_CATCHUP_DATA 0 |
| |
| static uword |
| mc_catchup_process (vlib_main_t * vm, |
| vlib_node_runtime_t * node, vlib_frame_t * f) |
| { |
| mc_main_t *mcm = mc_node_get_main (node); |
| uword *event_data = 0; |
| mc_catchup_process_arg_t *args; |
| int i; |
| |
| while (1) |
| { |
| if (event_data) |
| _vec_len (event_data) = 0; |
| vlib_process_wait_for_event_with_type (vm, &event_data, |
| EVENT_MC_SEND_CATCHUP_DATA); |
| |
| for (i = 0; i < vec_len (event_data); i++) |
| { |
| args = pool_elt_at_index (mcm->catchup_process_args, event_data[i]); |
| |
| mcm->transport.catchup_send_fun (mcm->transport.opaque, |
| args->catchup_opaque, |
| args->catchup_snapshot); |
| |
| /* Send function will free snapshot data vector. */ |
| pool_put (mcm->catchup_process_args, args); |
| } |
| } |
| |
| return 0; /* not likely */ |
| } |
| |
| static void |
| serialize_mc_stream (serialize_main_t * m, va_list * va) |
| { |
| mc_stream_t *s = va_arg (*va, mc_stream_t *); |
| mc_stream_peer_t *p; |
| |
| serialize_integer (m, pool_elts (s->peers), sizeof (u32)); |
| /* *INDENT-OFF* */ |
| pool_foreach (p, s->peers, ({ |
| u8 * x = serialize_get (m, sizeof (p->id)); |
| clib_memcpy (x, p->id.as_u8, sizeof (p->id)); |
| serialize_integer (m, p->last_sequence_received, |
| sizeof (p->last_sequence_received)); |
| })); |
| /* *INDENT-ON* */ |
| serialize_bitmap (m, s->all_peer_bitmap); |
| } |
| |
| void |
| unserialize_mc_stream (serialize_main_t * m, va_list * va) |
| { |
| mc_stream_t *s = va_arg (*va, mc_stream_t *); |
| u32 i, n_peers; |
| mc_stream_peer_t *p; |
| |
| unserialize_integer (m, &n_peers, sizeof (u32)); |
| mhash_init (&s->peer_index_by_id, sizeof (uword), sizeof (mc_peer_id_t)); |
| for (i = 0; i < n_peers; i++) |
| { |
| u8 *x; |
| pool_get (s->peers, p); |
| x = unserialize_get (m, sizeof (p->id)); |
| clib_memcpy (p->id.as_u8, x, sizeof (p->id)); |
| unserialize_integer (m, &p->last_sequence_received, |
| sizeof (p->last_sequence_received)); |
| mhash_set (&s->peer_index_by_id, &p->id, p - s->peers, /* old_value */ |
| 0); |
| } |
| s->all_peer_bitmap = unserialize_bitmap (m); |
| |
| /* This is really bad. */ |
| if (!s->all_peer_bitmap) |
| clib_warning ("BUG: stream %s all_peer_bitmap NULL", s->config.name); |
| } |
| |
| void |
| mc_msg_catchup_request_handler (mc_main_t * mcm, |
| mc_msg_catchup_request_t * req, |
| u32 catchup_opaque) |
| { |
| vlib_main_t *vm = mcm->vlib_main; |
| mc_stream_t *s; |
| mc_catchup_process_arg_t *args; |
| |
| mc_byte_swap_msg_catchup_request (req); |
| |
| s = mc_stream_by_index (mcm, req->stream_index); |
| if (!s || s->state != MC_STREAM_STATE_ready) |
| return; |
| |
| if (MC_EVENT_LOGGING > 0) |
| { |
| /* *INDENT-OFF* */ |
| ELOG_TYPE_DECLARE (t) = |
| { |
| .format = "catchup-request: from %s stream %d", |
| .format_args = "T4i4", |
| }; |
| /* *INDENT-ON* */ |
| struct |
| { |
| u32 peer, stream; |
| } *ed; |
| ed = ELOG_DATA (mcm->elog_main, t); |
| ed->peer = elog_id_for_peer_id (mcm, req->peer_id.as_u64); |
| ed->stream = req->stream_index; |
| } |
| |
| /* |
| * The application has to snapshoot its data structures right |
| * here, right now. If we process any messages after |
| * noting the last global sequence we've processed, the client |
| * won't be able to accurately reconstruct our data structures. |
| * |
| * Once the data structures are e.g. vec_dup()'ed, we |
| * send the resulting messages from a separate process, to |
| * make sure that we don't cause a bunch of message retransmissions |
| */ |
| pool_get (mcm->catchup_process_args, args); |
| |
| args->stream_index = s - mcm->stream_vector; |
| args->catchup_opaque = catchup_opaque; |
| args->catchup_snapshot = 0; |
| |
| /* Construct catchup reply and snapshot state for stream to send as |
| catchup reply payload. */ |
| { |
| mc_msg_catchup_reply_t *rep; |
| serialize_main_t m; |
| |
| vec_resize (args->catchup_snapshot, sizeof (rep[0])); |
| |
| rep = (void *) args->catchup_snapshot; |
| |
| rep->peer_id = req->peer_id; |
| rep->stream_index = req->stream_index; |
| rep->last_global_sequence_included = s->last_global_sequence_processed; |
| |
| /* Setup for serialize to append to catchup snapshot. */ |
| serialize_open_vector (&m, args->catchup_snapshot); |
| m.stream.current_buffer_index = vec_len (m.stream.buffer); |
| |
| serialize (&m, serialize_mc_stream, s); |
| |
| args->catchup_snapshot = serialize_close_vector (&m); |
| |
| /* Actually copy internal state */ |
| args->catchup_snapshot = s->config.catchup_snapshot |
| (mcm, args->catchup_snapshot, rep->last_global_sequence_included); |
| |
| rep = (void *) args->catchup_snapshot; |
| rep->n_data_bytes = vec_len (args->catchup_snapshot) - sizeof (rep[0]); |
| |
| mc_byte_swap_msg_catchup_reply (rep); |
| } |
| |
| /* now go send it... */ |
| vlib_process_signal_event (vm, mcm->catchup_process, |
| EVENT_MC_SEND_CATCHUP_DATA, |
| args - mcm->catchup_process_args); |
| } |
| |
| #define EVENT_MC_UNSERIALIZE_BUFFER 0 |
| #define EVENT_MC_UNSERIALIZE_CATCHUP 1 |
| |
| void |
| mc_msg_catchup_reply_handler (mc_main_t * mcm, mc_msg_catchup_reply_t * mp, |
| u32 catchup_opaque) |
| { |
| vlib_process_signal_event (mcm->vlib_main, |
| mcm->unserialize_process, |
| EVENT_MC_UNSERIALIZE_CATCHUP, |
| pointer_to_uword (mp)); |
| } |
| |
| static void |
| perform_catchup (mc_main_t * mcm, mc_msg_catchup_reply_t * mp) |
| { |
| mc_stream_t *s; |
| i32 seq_cmp_result; |
| |
| mc_byte_swap_msg_catchup_reply (mp); |
| |
| s = mc_stream_by_index (mcm, mp->stream_index); |
| |
| /* Never heard of this stream or already caught up. */ |
| if (!s || s->state == MC_STREAM_STATE_ready) |
| return; |
| |
| { |
| serialize_main_t m; |
| mc_stream_peer_t *p; |
| u32 n_stream_bytes; |
| |
| /* For offline sim replay: save the entire catchup snapshot... */ |
| if (s->config.save_snapshot) |
| s->config.save_snapshot (mcm, /* is_catchup */ 1, mp->data, |
| mp->n_data_bytes); |
| |
| unserialize_open_data (&m, mp->data, mp->n_data_bytes); |
| unserialize (&m, unserialize_mc_stream, s); |
| |
| /* Make sure we start numbering our messages as expected */ |
| /* *INDENT-OFF* */ |
| pool_foreach (p, s->peers, ({ |
| if (p->id.as_u64 == mcm->transport.our_ack_peer_id.as_u64) |
| s->our_local_sequence = p->last_sequence_received + 1; |
| })); |
| /* *INDENT-ON* */ |
| |
| n_stream_bytes = m.stream.current_buffer_index; |
| |
| /* No need to unserialize close; nothing to free. */ |
| |
| /* After serialized stream is user's catchup data. */ |
| s->config.catchup (mcm, mp->data + n_stream_bytes, |
| mp->n_data_bytes - n_stream_bytes); |
| } |
| |
| /* Vector could have been moved by catchup. |
| This can only happen for mc-internal stream. */ |
| s = mc_stream_by_index (mcm, mp->stream_index); |
| |
| s->last_global_sequence_processed = mp->last_global_sequence_included; |
| |
| while (clib_fifo_elts (s->catchup_fifo)) |
| { |
| mc_msg_user_request_t *gp; |
| u32 bi; |
| vlib_buffer_t *b; |
| |
| clib_fifo_sub1 (s->catchup_fifo, bi); |
| |
| b = vlib_get_buffer (mcm->vlib_main, bi); |
| gp = vlib_buffer_get_current (b); |
| |
| /* Make sure we're replaying "new" news */ |
| seq_cmp_result = mc_seq_cmp (gp->global_sequence, |
| mp->last_global_sequence_included); |
| |
| if (seq_cmp_result > 0) |
| { |
| vlib_buffer_advance (b, sizeof (gp[0])); |
| s->config.rx_buffer (mcm, s, gp->peer_id, bi); |
| s->last_global_sequence_processed = gp->global_sequence; |
| |
| if (MC_EVENT_LOGGING) |
| { |
| /* *INDENT-OFF* */ |
| ELOG_TYPE_DECLARE (t) = |
| { |
| .format = "catchup replay local sequence 0x%x", |
| .format_args = "i4", |
| }; |
| /* *INDENT-ON* */ |
| struct |
| { |
| u32 local_sequence; |
| } *ed; |
| ed = ELOG_DATA (mcm->elog_main, t); |
| ed->local_sequence = gp->local_sequence; |
| } |
| } |
| else |
| { |
| if (MC_EVENT_LOGGING) |
| { |
| /* *INDENT-OFF* */ |
| ELOG_TYPE_DECLARE (t) = |
| { |
| .format = "catchup discard local sequence 0x%x", |
| .format_args = "i4", |
| }; |
| /* *INDENT-ON* */ |
| struct |
| { |
| u32 local_sequence; |
| } *ed; |
| ed = ELOG_DATA (mcm->elog_main, t); |
| ed->local_sequence = gp->local_sequence; |
| } |
| |
| vlib_buffer_free_one (mcm->vlib_main, bi); |
| } |
| } |
| |
| s->state = MC_STREAM_STATE_ready; |
| |
| /* Now that we are caught up wake up joining process. */ |
| { |
| vlib_one_time_waiting_process_t *wp; |
| vec_foreach (wp, s->procs_waiting_for_join_done) |
| vlib_signal_one_time_waiting_process (mcm->vlib_main, wp); |
| if (s->procs_waiting_for_join_done) |
| _vec_len (s->procs_waiting_for_join_done) = 0; |
| } |
| } |
| |
| static void |
| this_node_maybe_master (mc_main_t * mcm) |
| { |
| vlib_main_t *vm = mcm->vlib_main; |
| mc_msg_master_assert_t *mp; |
| uword event_type; |
| int timeouts = 0; |
| int is_master = mcm->relay_state == MC_RELAY_STATE_MASTER; |
| clib_error_t *error; |
| f64 now, time_last_master_assert = -1; |
| u32 bi; |
| |
| while (1) |
| { |
| if (!mcm->we_can_be_relay_master) |
| { |
| mcm->relay_state = MC_RELAY_STATE_SLAVE; |
| if (MC_EVENT_LOGGING) |
| { |
| ELOG_TYPE (e, "become slave (config)"); |
| ELOG (mcm->elog_main, e, 0); |
| } |
| return; |
| } |
| |
| now = vlib_time_now (vm); |
| if (now >= time_last_master_assert + 1) |
| { |
| time_last_master_assert = now; |
| mp = mc_get_vlib_buffer (mcm->vlib_main, sizeof (mp[0]), &bi); |
| |
| mp->peer_id = mcm->transport.our_ack_peer_id; |
| mp->global_sequence = mcm->relay_global_sequence; |
| |
| /* |
| * these messages clog the event log, set MC_EVENT_LOGGING higher |
| * if you want them |
| */ |
| if (MC_EVENT_LOGGING > 1) |
| { |
| /* *INDENT-OFF* */ |
| ELOG_TYPE_DECLARE (e) = |
| { |
| .format = "tx-massert: peer %s global seq %u", |
| .format_args = "T4i4", |
| }; |
| /* *INDENT-ON* */ |
| struct |
| { |
| u32 peer, global_sequence; |
| } *ed; |
| ed = ELOG_DATA (mcm->elog_main, e); |
| ed->peer = elog_id_for_peer_id (mcm, mp->peer_id.as_u64); |
| ed->global_sequence = mp->global_sequence; |
| } |
| |
| mc_byte_swap_msg_master_assert (mp); |
| |
| error = |
| mcm->transport.tx_buffer (mcm->transport.opaque, |
| MC_TRANSPORT_MASTERSHIP, bi); |
| if (error) |
| clib_error_report (error); |
| } |
| |
| vlib_process_wait_for_event_or_clock (vm, 1.0); |
| event_type = vlib_process_get_events (vm, /* no event data */ 0); |
| |
| switch (event_type) |
| { |
| case ~0: |
| if (!is_master && timeouts++ > 2) |
| { |
| mcm->relay_state = MC_RELAY_STATE_MASTER; |
| mcm->relay_master_peer_id = |
| mcm->transport.our_ack_peer_id.as_u64; |
| if (MC_EVENT_LOGGING) |
| { |
| ELOG_TYPE (e, "become master (was maybe_master)"); |
| ELOG (mcm->elog_main, e, 0); |
| } |
| return; |
| } |
| break; |
| |
| case MC_RELAY_STATE_SLAVE: |
| mcm->relay_state = MC_RELAY_STATE_SLAVE; |
| if (MC_EVENT_LOGGING && mcm->relay_state != MC_RELAY_STATE_SLAVE) |
| { |
| ELOG_TYPE (e, "become slave (was maybe_master)"); |
| ELOG (mcm->elog_main, e, 0); |
| } |
| return; |
| } |
| } |
| } |
| |
| static void |
| this_node_slave (mc_main_t * mcm) |
| { |
| vlib_main_t *vm = mcm->vlib_main; |
| uword event_type; |
| int timeouts = 0; |
| |
| if (MC_EVENT_LOGGING) |
| { |
| ELOG_TYPE (e, "become slave"); |
| ELOG (mcm->elog_main, e, 0); |
| } |
| |
| while (1) |
| { |
| vlib_process_wait_for_event_or_clock (vm, 1.0); |
| event_type = vlib_process_get_events (vm, /* no event data */ 0); |
| |
| switch (event_type) |
| { |
| case ~0: |
| if (timeouts++ > 2) |
| { |
| mcm->relay_state = MC_RELAY_STATE_NEGOTIATE; |
| mcm->relay_master_peer_id = ~0ULL; |
| if (MC_EVENT_LOGGING) |
| { |
| ELOG_TYPE (e, "timeouts; negoitate mastership"); |
| ELOG (mcm->elog_main, e, 0); |
| } |
| return; |
| } |
| break; |
| |
| case MC_RELAY_STATE_SLAVE: |
| mcm->relay_state = MC_RELAY_STATE_SLAVE; |
| timeouts = 0; |
| break; |
| } |
| } |
| } |
| |
| static uword |
| mc_mastership_process (vlib_main_t * vm, |
| vlib_node_runtime_t * node, vlib_frame_t * f) |
| { |
| mc_main_t *mcm = mc_node_get_main (node); |
| |
| while (1) |
| { |
| switch (mcm->relay_state) |
| { |
| case MC_RELAY_STATE_NEGOTIATE: |
| case MC_RELAY_STATE_MASTER: |
| this_node_maybe_master (mcm); |
| break; |
| |
| case MC_RELAY_STATE_SLAVE: |
| this_node_slave (mcm); |
| break; |
| } |
| } |
| return 0; /* not likely */ |
| } |
| |
| void |
| mc_enable_disable_mastership (mc_main_t * mcm, int we_can_be_master) |
| { |
| if (we_can_be_master != mcm->we_can_be_relay_master) |
| { |
| mcm->we_can_be_relay_master = we_can_be_master; |
| vlib_process_signal_event (mcm->vlib_main, |
| mcm->mastership_process, |
| MC_RELAY_STATE_NEGOTIATE, 0); |
| } |
| } |
| |
| void |
| mc_msg_master_assert_handler (mc_main_t * mcm, mc_msg_master_assert_t * mp, |
| u32 buffer_index) |
| { |
| mc_peer_id_t his_peer_id, our_peer_id; |
| i32 seq_cmp_result; |
| u8 signal_slave = 0; |
| u8 update_global_sequence = 0; |
| |
| mc_byte_swap_msg_master_assert (mp); |
| |
| his_peer_id = mp->peer_id; |
| our_peer_id = mcm->transport.our_ack_peer_id; |
| |
| /* compare the incoming global sequence with ours */ |
| seq_cmp_result = mc_seq_cmp (mp->global_sequence, |
| mcm->relay_global_sequence); |
| |
| /* If the sender has a lower peer id and the sender's sequence >= |
| our global sequence, we become a slave. Otherwise we are master. */ |
| if (mc_peer_id_compare (his_peer_id, our_peer_id) < 0 |
| && seq_cmp_result >= 0) |
| { |
| vlib_process_signal_event (mcm->vlib_main, |
| mcm->mastership_process, |
| MC_RELAY_STATE_SLAVE, 0); |
| signal_slave = 1; |
| } |
| |
| /* Update our global sequence. */ |
| if (seq_cmp_result > 0) |
| { |
| mcm->relay_global_sequence = mp->global_sequence; |
| update_global_sequence = 1; |
| } |
| |
| { |
| uword *q = mhash_get (&mcm->mastership_peer_index_by_id, &his_peer_id); |
| mc_mastership_peer_t *p; |
| |
| if (q) |
| p = vec_elt_at_index (mcm->mastership_peers, q[0]); |
| else |
| { |
| vec_add2 (mcm->mastership_peers, p, 1); |
| p->peer_id = his_peer_id; |
| mhash_set (&mcm->mastership_peer_index_by_id, &p->peer_id, |
| p - mcm->mastership_peers, |
| /* old_value */ 0); |
| } |
| p->time_last_master_assert_received = vlib_time_now (mcm->vlib_main); |
| } |
| |
| /* |
| * these messages clog the event log, set MC_EVENT_LOGGING higher |
| * if you want them. |
| */ |
| if (MC_EVENT_LOGGING > 1) |
| { |
| /* *INDENT-OFF* */ |
| ELOG_TYPE_DECLARE (e) = |
| { |
| .format = "rx-massert: peer %s global seq %u upd %d slave %d", |
| .format_args = "T4i4i1i1", |
| }; |
| /* *INDENT-ON* */ |
| |
| struct |
| { |
| u32 peer; |
| u32 global_sequence; |
| u8 update_sequence; |
| u8 slave; |
| } *ed; |
| ed = ELOG_DATA (mcm->elog_main, e); |
| ed->peer = elog_id_for_peer_id (mcm, his_peer_id.as_u64); |
| ed->global_sequence = mp->global_sequence; |
| ed->update_sequence = update_global_sequence; |
| ed->slave = signal_slave; |
| } |
| } |
| |
| static void |
| mc_serialize_init (mc_main_t * mcm) |
| { |
| mc_serialize_msg_t *m; |
| vlib_main_t *vm = vlib_get_main (); |
| |
| mcm->global_msg_index_by_name |
| = hash_create_string ( /* elts */ 0, sizeof (uword)); |
| |
| m = vm->mc_msg_registrations; |
| |
| while (m) |
| { |
| m->global_index = vec_len (mcm->global_msgs); |
| hash_set_mem (mcm->global_msg_index_by_name, m->name, m->global_index); |
| vec_add1 (mcm->global_msgs, m); |
| m = m->next_registration; |
| } |
| } |
| |
| clib_error_t * |
| mc_serialize_va (mc_main_t * mc, |
| u32 stream_index, |
| u32 multiple_messages_per_vlib_buffer, |
| mc_serialize_msg_t * msg, va_list * va) |
| { |
| mc_stream_t *s; |
| clib_error_t *error; |
| serialize_main_t *m = &mc->serialize_mains[VLIB_TX]; |
| vlib_serialize_buffer_main_t *sbm = &mc->serialize_buffer_mains[VLIB_TX]; |
| u32 bi, n_before, n_after, n_total, n_this_msg; |
| u32 si, gi; |
| |
| if (!sbm->vlib_main) |
| { |
| sbm->tx.max_n_data_bytes_per_chain = 4096; |
| sbm->tx.free_list_index = VLIB_BUFFER_DEFAULT_FREE_LIST_INDEX; |
| } |
| |
| if (sbm->first_buffer == 0) |
| serialize_open_vlib_buffer (m, mc->vlib_main, sbm); |
| |
| n_before = serialize_vlib_buffer_n_bytes (m); |
| |
| s = mc_stream_by_index (mc, stream_index); |
| gi = msg->global_index; |
| ASSERT (msg == vec_elt (mc->global_msgs, gi)); |
| |
| si = ~0; |
| if (gi < vec_len (s->stream_msg_index_by_global_index)) |
| si = s->stream_msg_index_by_global_index[gi]; |
| |
| serialize_likely_small_unsigned_integer (m, si); |
| |
| /* For first time message is sent, use name to identify message. */ |
| if (si == ~0 || MSG_ID_DEBUG) |
| serialize_cstring (m, msg->name); |
| |
| if (MSG_ID_DEBUG && MC_EVENT_LOGGING > 0) |
| { |
| /* *INDENT-OFF* */ |
| ELOG_TYPE_DECLARE (e) = |
| { |
| .format = "serialize-msg: %s index %d", |
| .format_args = "T4i4", |
| }; |
| /* *INDENT-ON* */ |
| struct |
| { |
| u32 c[2]; |
| } *ed; |
| ed = ELOG_DATA (mc->elog_main, e); |
| ed->c[0] = elog_id_for_msg_name (mc, msg->name); |
| ed->c[1] = si; |
| } |
| |
| error = va_serialize (m, va); |
| |
| n_after = serialize_vlib_buffer_n_bytes (m); |
| n_this_msg = n_after - n_before; |
| n_total = n_after + sizeof (mc_msg_user_request_t); |
| |
| /* For max message size ignore first message where string name is sent. */ |
| if (si != ~0) |
| msg->max_n_bytes_serialized = |
| clib_max (msg->max_n_bytes_serialized, n_this_msg); |
| |
| if (!multiple_messages_per_vlib_buffer |
| || si == ~0 |
| || n_total + msg->max_n_bytes_serialized > |
| mc->transport.max_packet_size) |
| { |
| bi = serialize_close_vlib_buffer (m); |
| sbm->first_buffer = 0; |
| if (!error) |
| mc_stream_send (mc, stream_index, bi); |
| else if (bi != ~0) |
| vlib_buffer_free_one (mc->vlib_main, bi); |
| } |
| |
| return error; |
| } |
| |
| clib_error_t * |
| mc_serialize_internal (mc_main_t * mc, |
| u32 stream_index, |
| u32 multiple_messages_per_vlib_buffer, |
| mc_serialize_msg_t * msg, ...) |
| { |
| vlib_main_t *vm = mc->vlib_main; |
| va_list va; |
| clib_error_t *error; |
| |
| if (stream_index == ~0) |
| { |
| if (vm->mc_main && vm->mc_stream_index == ~0) |
| vlib_current_process_wait_for_one_time_event_vector |
| (vm, &vm->procs_waiting_for_mc_stream_join); |
| stream_index = vm->mc_stream_index; |
| } |
| |
| va_start (va, msg); |
| error = mc_serialize_va (mc, stream_index, |
| multiple_messages_per_vlib_buffer, msg, &va); |
| va_end (va); |
| return error; |
| } |
| |
| uword |
| mc_unserialize_message (mc_main_t * mcm, |
| mc_stream_t * s, serialize_main_t * m) |
| { |
| mc_serialize_stream_msg_t *sm; |
| u32 gi, si; |
| |
| si = unserialize_likely_small_unsigned_integer (m); |
| |
| if (!(si == ~0 || MSG_ID_DEBUG)) |
| { |
| sm = vec_elt_at_index (s->stream_msgs, si); |
| gi = sm->global_index; |
| } |
| else |
| { |
| char *name; |
| |
| unserialize_cstring (m, &name); |
| |
| if (MSG_ID_DEBUG && MC_EVENT_LOGGING > 0) |
| { |
| /* *INDENT-OFF* */ |
| ELOG_TYPE_DECLARE (e) = |
| { |
| .format = "unserialize-msg: %s rx index %d", |
| .format_args = "T4i4", |
| }; |
| /* *INDENT-ON* */ |
| struct |
| { |
| u32 c[2]; |
| } *ed; |
| ed = ELOG_DATA (mcm->elog_main, e); |
| ed->c[0] = elog_id_for_msg_name (mcm, name); |
| ed->c[1] = si; |
| } |
| |
| { |
| uword *p = hash_get_mem (mcm->global_msg_index_by_name, name); |
| gi = p ? p[0] : ~0; |
| } |
| |
| /* Unknown message? */ |
| if (gi == ~0) |
| { |
| vec_free (name); |
| goto done; |
| } |
| |
| vec_validate_init_empty (s->stream_msg_index_by_global_index, gi, ~0); |
| si = s->stream_msg_index_by_global_index[gi]; |
| |
| /* Stream local index unknown? Create it. */ |
| if (si == ~0) |
| { |
| vec_add2 (s->stream_msgs, sm, 1); |
| |
| si = sm - s->stream_msgs; |
| sm->global_index = gi; |
| s->stream_msg_index_by_global_index[gi] = si; |
| |
| if (MC_EVENT_LOGGING > 0) |
| { |
| /* *INDENT-OFF* */ |
| ELOG_TYPE_DECLARE (e) = |
| { |
| .format = "msg-bind: stream %d %s to index %d", |
| .format_args = "i4T4i4", |
| }; |
| /* *INDENT-ON* */ |
| struct |
| { |
| u32 c[3]; |
| } *ed; |
| ed = ELOG_DATA (mcm->elog_main, e); |
| ed->c[0] = s->index; |
| ed->c[1] = elog_id_for_msg_name (mcm, name); |
| ed->c[2] = si; |
| } |
| } |
| else |
| { |
| sm = vec_elt_at_index (s->stream_msgs, si); |
| if (gi != sm->global_index && MC_EVENT_LOGGING > 0) |
| { |
| /* *INDENT-OFF* */ |
| ELOG_TYPE_DECLARE (e) = |
| { |
| .format = "msg-id-ERROR: %s index %d expected %d", |
| .format_args = "T4i4i4", |
| }; |
| /* *INDENT-ON* */ |
| struct |
| { |
| u32 c[3]; |
| } *ed; |
| ed = ELOG_DATA (mcm->elog_main, e); |
| ed->c[0] = elog_id_for_msg_name (mcm, name); |
| ed->c[1] = si; |
| ed->c[2] = ~0; |
| if (sm->global_index < |
| vec_len (s->stream_msg_index_by_global_index)) |
| ed->c[2] = |
| s->stream_msg_index_by_global_index[sm->global_index]; |
| } |
| } |
| |
| vec_free (name); |
| } |
| |
| if (gi != ~0) |
| { |
| mc_serialize_msg_t *msg; |
| msg = vec_elt (mcm->global_msgs, gi); |
| unserialize (m, msg->unserialize, mcm); |
| } |
| |
| done: |
| return gi != ~0; |
| } |
| |
| void |
| mc_unserialize_internal (mc_main_t * mcm, u32 stream_and_buffer_index) |
| { |
| vlib_main_t *vm = mcm->vlib_main; |
| serialize_main_t *m = &mcm->serialize_mains[VLIB_RX]; |
| vlib_serialize_buffer_main_t *sbm = &mcm->serialize_buffer_mains[VLIB_RX]; |
| mc_stream_and_buffer_t *sb; |
| mc_stream_t *stream; |
| u32 buffer_index; |
| |
| sb = |
| pool_elt_at_index (mcm->mc_unserialize_stream_and_buffers, |
| stream_and_buffer_index); |
| buffer_index = sb->buffer_index; |
| stream = vec_elt_at_index (mcm->stream_vector, sb->stream_index); |
| pool_put (mcm->mc_unserialize_stream_and_buffers, sb); |
| |
| if (stream->config.save_snapshot) |
| { |
| u32 n_bytes = vlib_buffer_index_length_in_chain (vm, buffer_index); |
| static u8 *contents; |
| vec_reset_length (contents); |
| vec_validate (contents, n_bytes - 1); |
| vlib_buffer_contents (vm, buffer_index, contents); |
| stream->config.save_snapshot (mcm, /* is_catchup */ 0, contents, |
| n_bytes); |
| } |
| |
| ASSERT (vlib_in_process_context (vm)); |
| |
| unserialize_open_vlib_buffer (m, vm, sbm); |
| |
| clib_fifo_add1 (sbm->rx.buffer_fifo, buffer_index); |
| |
| while (unserialize_vlib_buffer_n_bytes (m) > 0) |
| mc_unserialize_message (mcm, stream, m); |
| |
| /* Frees buffer. */ |
| unserialize_close_vlib_buffer (m); |
| } |
| |
| void |
| mc_unserialize (mc_main_t * mcm, mc_stream_t * s, u32 buffer_index) |
| { |
| vlib_main_t *vm = mcm->vlib_main; |
| mc_stream_and_buffer_t *sb; |
| pool_get (mcm->mc_unserialize_stream_and_buffers, sb); |
| sb->stream_index = s->index; |
| sb->buffer_index = buffer_index; |
| vlib_process_signal_event (vm, mcm->unserialize_process, |
| EVENT_MC_UNSERIALIZE_BUFFER, |
| sb - mcm->mc_unserialize_stream_and_buffers); |
| } |
| |
| static uword |
| mc_unserialize_process (vlib_main_t * vm, |
| vlib_node_runtime_t * node, vlib_frame_t * f) |
| { |
| mc_main_t *mcm = mc_node_get_main (node); |
| uword event_type, *event_data = 0; |
| int i; |
| |
| while (1) |
| { |
| if (event_data) |
| _vec_len (event_data) = 0; |
| |
| vlib_process_wait_for_event (vm); |
| event_type = vlib_process_get_events (vm, &event_data); |
| switch (event_type) |
| { |
| case EVENT_MC_UNSERIALIZE_BUFFER: |
| for (i = 0; i < vec_len (event_data); i++) |
| mc_unserialize_internal (mcm, event_data[i]); |
| break; |
| |
| case EVENT_MC_UNSERIALIZE_CATCHUP: |
| for (i = 0; i < vec_len (event_data); i++) |
| { |
| u8 *mp = uword_to_pointer (event_data[i], u8 *); |
| perform_catchup (mcm, (void *) mp); |
| vec_free (mp); |
| } |
| break; |
| |
| default: |
| break; |
| } |
| } |
| |
| return 0; /* not likely */ |
| } |
| |
| void |
| serialize_mc_main (serialize_main_t * m, va_list * va) |
| { |
| mc_main_t *mcm = va_arg (*va, mc_main_t *); |
| mc_stream_t *s; |
| mc_serialize_stream_msg_t *sm; |
| mc_serialize_msg_t *msg; |
| |
| serialize_integer (m, vec_len (mcm->stream_vector), sizeof (u32)); |
| vec_foreach (s, mcm->stream_vector) |
| { |
| /* Stream name. */ |
| serialize_cstring (m, s->config.name); |
| |
| /* Serialize global names for all sent messages. */ |
| serialize_integer (m, vec_len (s->stream_msgs), sizeof (u32)); |
| vec_foreach (sm, s->stream_msgs) |
| { |
| msg = vec_elt (mcm->global_msgs, sm->global_index); |
| serialize_cstring (m, msg->name); |
| } |
| } |
| } |
| |
| void |
| unserialize_mc_main (serialize_main_t * m, va_list * va) |
| { |
| mc_main_t *mcm = va_arg (*va, mc_main_t *); |
| u32 i, n_streams, n_stream_msgs; |
| char *name; |
| mc_stream_t *s; |
| mc_serialize_stream_msg_t *sm; |
| |
| unserialize_integer (m, &n_streams, sizeof (u32)); |
| for (i = 0; i < n_streams; i++) |
| { |
| unserialize_cstring (m, &name); |
| if (i != MC_STREAM_INDEX_INTERNAL && !mc_stream_by_name (mcm, name)) |
| { |
| vec_validate (mcm->stream_vector, i); |
| s = vec_elt_at_index (mcm->stream_vector, i); |
| mc_stream_init (s); |
| s->index = s - mcm->stream_vector; |
| s->config.name = name; |
| s->state = MC_STREAM_STATE_name_known; |
| hash_set_mem (mcm->stream_index_by_name, s->config.name, s->index); |
| } |
| else |
| vec_free (name); |
| |
| s = vec_elt_at_index (mcm->stream_vector, i); |
| |
| vec_free (s->stream_msgs); |
| vec_free (s->stream_msg_index_by_global_index); |
| |
| unserialize_integer (m, &n_stream_msgs, sizeof (u32)); |
| vec_resize (s->stream_msgs, n_stream_msgs); |
| vec_foreach (sm, s->stream_msgs) |
| { |
| uword *p; |
| u32 si, gi; |
| |
| unserialize_cstring (m, &name); |
| p = hash_get (mcm->global_msg_index_by_name, name); |
| gi = p ? p[0] : ~0; |
| si = sm - s->stream_msgs; |
| |
| if (MC_EVENT_LOGGING > 0) |
| { |
| /* *INDENT-OFF* */ |
| ELOG_TYPE_DECLARE (e) = |
| { |
| .format = "catchup-bind: %s to %d global index %d stream %d", |
| .format_args = "T4i4i4i4", |
| }; |
| /* *INDENT-ON* */ |
| |
| struct |
| { |
| u32 c[4]; |
| } *ed; |
| ed = ELOG_DATA (mcm->elog_main, e); |
| ed->c[0] = elog_id_for_msg_name (mcm, name); |
| ed->c[1] = si; |
| ed->c[2] = gi; |
| ed->c[3] = s->index; |
| } |
| |
| vec_free (name); |
| |
| sm->global_index = gi; |
| if (gi != ~0) |
| { |
| vec_validate_init_empty (s->stream_msg_index_by_global_index, |
| gi, ~0); |
| s->stream_msg_index_by_global_index[gi] = si; |
| } |
| } |
| } |
| } |
| |
| void |
| mc_main_init (mc_main_t * mcm, char *tag) |
| { |
| vlib_main_t *vm = vlib_get_main (); |
| |
| mcm->vlib_main = vm; |
| mcm->elog_main = &vm->elog_main; |
| |
| mcm->relay_master_peer_id = ~0ULL; |
| mcm->relay_state = MC_RELAY_STATE_NEGOTIATE; |
| |
| mcm->stream_index_by_name |
| = hash_create_string ( /* elts */ 0, /* value size */ sizeof (uword)); |
| |
| { |
| vlib_node_registration_t r; |
| |
| memset (&r, 0, sizeof (r)); |
| |
| r.type = VLIB_NODE_TYPE_PROCESS; |
| |
| /* Point runtime data to main instance. */ |
| r.runtime_data = &mcm; |
| r.runtime_data_bytes = sizeof (&mcm); |
| |
| r.name = (char *) format (0, "mc-mastership-%s", tag); |
| r.function = mc_mastership_process; |
| mcm->mastership_process = vlib_register_node (vm, &r); |
| |
| r.name = (char *) format (0, "mc-join-ager-%s", tag); |
| r.function = mc_join_ager_process; |
| mcm->join_ager_process = vlib_register_node (vm, &r); |
| |
| r.name = (char *) format (0, "mc-retry-%s", tag); |
| r.function = mc_retry_process; |
| mcm->retry_process = vlib_register_node (vm, &r); |
| |
| r.name = (char *) format (0, "mc-catchup-%s", tag); |
| r.function = mc_catchup_process; |
| mcm->catchup_process = vlib_register_node (vm, &r); |
| |
| r.name = (char *) format (0, "mc-unserialize-%s", tag); |
| r.function = mc_unserialize_process; |
| mcm->unserialize_process = vlib_register_node (vm, &r); |
| } |
| |
| if (MC_EVENT_LOGGING > 0) |
| mhash_init (&mcm->elog_id_by_peer_id, sizeof (uword), |
| sizeof (mc_peer_id_t)); |
| |
| mhash_init (&mcm->mastership_peer_index_by_id, sizeof (uword), |
| sizeof (mc_peer_id_t)); |
| mc_serialize_init (mcm); |
| } |
| |
| static u8 * |
| format_mc_relay_state (u8 * s, va_list * args) |
| { |
| mc_relay_state_t state = va_arg (*args, mc_relay_state_t); |
| char *t = 0; |
| switch (state) |
| { |
| case MC_RELAY_STATE_NEGOTIATE: |
| t = "negotiate"; |
| break; |
| case MC_RELAY_STATE_MASTER: |
| t = "master"; |
| break; |
| case MC_RELAY_STATE_SLAVE: |
| t = "slave"; |
| break; |
| default: |
| return format (s, "unknown 0x%x", state); |
| } |
| |
| return format (s, "%s", t); |
| } |
| |
| static u8 * |
| format_mc_stream_state (u8 * s, va_list * args) |
| { |
| mc_stream_state_t state = va_arg (*args, mc_stream_state_t); |
| char *t = 0; |
| switch (state) |
| { |
| #define _(f) case MC_STREAM_STATE_##f: t = #f; break; |
| foreach_mc_stream_state |
| #undef _ |
| default: |
| return format (s, "unknown 0x%x", state); |
| } |
| |
| return format (s, "%s", t); |
| } |
| |
| static int |
| mc_peer_comp (void *a1, void *a2) |
| { |
| mc_stream_peer_t *p1 = a1; |
| mc_stream_peer_t *p2 = a2; |
| |
| return mc_peer_id_compare (p1->id, p2->id); |
| } |
| |
| u8 * |
| format_mc_main (u8 * s, va_list * args) |
| { |
| mc_main_t *mcm = va_arg (*args, mc_main_t *); |
| mc_stream_t *t; |
| mc_stream_peer_t *p, *ps; |
| uword indent = format_get_indent (s); |
| |
| s = format (s, "MC state %U, %d streams joined, global sequence 0x%x", |
| format_mc_relay_state, mcm->relay_state, |
| vec_len (mcm->stream_vector), mcm->relay_global_sequence); |
| |
| { |
| mc_mastership_peer_t *mp; |
| f64 now = vlib_time_now (mcm->vlib_main); |
| s = format (s, "\n%UMost recent mastership peers:", |
| format_white_space, indent + 2); |
| vec_foreach (mp, mcm->mastership_peers) |
| { |
| s = format (s, "\n%U%-30U%.4e", |
| format_white_space, indent + 4, |
| mcm->transport.format_peer_id, mp->peer_id, |
| now - mp->time_last_master_assert_received); |
| } |
| } |
| |
| vec_foreach (t, mcm->stream_vector) |
| { |
| s = format (s, "\n%Ustream `%s' index %d", |
| format_white_space, indent + 2, t->config.name, t->index); |
| |
| s = format (s, "\n%Ustate %U", |
| format_white_space, indent + 4, |
| format_mc_stream_state, t->state); |
| |
| s = |
| format (s, |
| "\n%Uretries: interval %.0f sec, limit %d, pool elts %d, %Ld sent", |
| format_white_space, indent + 4, t->config.retry_interval, |
| t->config.retry_limit, pool_elts (t->retry_pool), |
| t->stats.n_retries - t->stats_last_clear.n_retries); |
| |
| s = format (s, "\n%U%Ld/%Ld user requests sent/received", |
| format_white_space, indent + 4, |
| t->user_requests_sent, t->user_requests_received); |
| |
| s = format (s, "\n%U%d peers, local/global sequence 0x%x/0x%x", |
| format_white_space, indent + 4, |
| pool_elts (t->peers), |
| t->our_local_sequence, t->last_global_sequence_processed); |
| |
| ps = 0; |
| /* *INDENT-OFF* */ |
| pool_foreach (p, t->peers, |
| ({ |
| if (clib_bitmap_get (t->all_peer_bitmap, p - t->peers)) |
| vec_add1 (ps, p[0]); |
| })); |
| /* *INDENT-ON* */ |
| vec_sort_with_function (ps, mc_peer_comp); |
| s = format (s, "\n%U%=30s%10s%16s%16s", |
| format_white_space, indent + 6, |
| "Peer", "Last seq", "Retries", "Future"); |
| |
| vec_foreach (p, ps) |
| { |
| s = format (s, "\n%U%-30U0x%08x%16Ld%16Ld%s", |
| format_white_space, indent + 6, |
| mcm->transport.format_peer_id, p->id.as_u64, |
| p->last_sequence_received, |
| p->stats.n_msgs_from_past - |
| p->stats_last_clear.n_msgs_from_past, |
| p->stats.n_msgs_from_future - |
| p->stats_last_clear.n_msgs_from_future, |
| (mcm->transport.our_ack_peer_id.as_u64 == |
| p->id.as_u64 ? " (self)" : "")); |
| } |
| vec_free (ps); |
| } |
| |
| return s; |
| } |
| |
| /* |
| * fd.io coding-style-patch-verification: ON |
| * |
| * Local Variables: |
| * eval: (c-set-style "gnu") |
| * End: |
| */ |