| /* |
| * Licensed to the Apache Software Foundation (ASF) under one |
| * or more contributor license agreements. See the NOTICE file |
| * distributed with this work for additional information |
| * regarding copyright ownership. The ASF licenses this file |
| * to you 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 <qpid/dispatch/ctools.h> |
| #include "agent_config_address.h" |
| #include <stdio.h> |
| |
| #define QDR_CONFIG_ADDRESS_NAME 0 |
| #define QDR_CONFIG_ADDRESS_IDENTITY 1 |
| #define QDR_CONFIG_ADDRESS_TYPE 2 |
| #define QDR_CONFIG_ADDRESS_PREFIX 3 |
| #define QDR_CONFIG_ADDRESS_DISTRIBUTION 4 |
| #define QDR_CONFIG_ADDRESS_WAYPOINT 5 |
| #define QDR_CONFIG_ADDRESS_IN_PHASE 6 |
| #define QDR_CONFIG_ADDRESS_OUT_PHASE 7 |
| |
| const char *qdr_config_address_columns[] = |
| {"name", |
| "identity", |
| "type", |
| "prefix", |
| "distribution", |
| "waypoint", |
| "ingressPhase", |
| "egressPhase", |
| 0}; |
| |
| |
| static void qdr_config_address_insert_column_CT(qdr_address_config_t *addr, int col, qd_composed_field_t *body, bool as_map) |
| { |
| const char *text = 0; |
| const char *key; |
| |
| if (as_map) |
| qd_compose_insert_string(body, qdr_config_address_columns[col]); |
| |
| switch(col) { |
| case QDR_CONFIG_ADDRESS_NAME: |
| if (addr->name) |
| qd_compose_insert_string(body, addr->name); |
| else |
| qd_compose_insert_null(body); |
| break; |
| |
| case QDR_CONFIG_ADDRESS_IDENTITY: { |
| char id_str[100]; |
| snprintf(id_str, 100, "%ld", addr->identity); |
| qd_compose_insert_string(body, id_str); |
| break; |
| } |
| |
| case QDR_CONFIG_ADDRESS_TYPE: |
| qd_compose_insert_string(body, "org.apache.qpid.dispatch.config.address"); |
| break; |
| |
| case QDR_CONFIG_ADDRESS_PREFIX: |
| key = (const char*) qd_hash_key_by_handle(addr->hash_handle); |
| if (key && key[0] == 'Z') |
| qd_compose_insert_string(body, &key[1]); |
| else |
| qd_compose_insert_null(body); |
| break; |
| |
| case QDR_CONFIG_ADDRESS_DISTRIBUTION: |
| switch (addr->treatment) { |
| case QD_TREATMENT_MULTICAST_FLOOD: |
| case QD_TREATMENT_MULTICAST_ONCE: text = "multicast"; break; |
| case QD_TREATMENT_ANYCAST_CLOSEST: text = "closest"; break; |
| case QD_TREATMENT_ANYCAST_BALANCED: text = "balanced"; break; |
| default: |
| text = 0; |
| } |
| |
| if (text) |
| qd_compose_insert_string(body, text); |
| else |
| qd_compose_insert_null(body); |
| |
| break; |
| |
| case QDR_CONFIG_ADDRESS_WAYPOINT: |
| qd_compose_insert_bool(body, addr->in_phase == 0 && addr->out_phase == 1); |
| break; |
| |
| case QDR_CONFIG_ADDRESS_IN_PHASE: |
| qd_compose_insert_int(body, addr->in_phase); |
| break; |
| |
| case QDR_CONFIG_ADDRESS_OUT_PHASE: |
| qd_compose_insert_int(body, addr->out_phase); |
| break; |
| } |
| } |
| |
| |
| static void qdr_agent_write_config_address_CT(qdr_query_t *query, qdr_address_config_t *addr) |
| { |
| qd_composed_field_t *body = query->body; |
| |
| qd_compose_start_list(body); |
| int i = 0; |
| while (query->columns[i] >= 0) { |
| qdr_config_address_insert_column_CT(addr, query->columns[i], body, false); |
| i++; |
| } |
| qd_compose_end_list(body); |
| } |
| |
| |
| static void qdr_manage_advance_config_address_CT(qdr_query_t *query, qdr_address_config_t *addr) |
| { |
| query->next_offset++; |
| addr = DEQ_NEXT(addr); |
| query->more = !!addr; |
| } |
| |
| |
| void qdra_config_address_get_first_CT(qdr_core_t *core, qdr_query_t *query, int offset) |
| { |
| // |
| // Queries that get this far will always succeed. |
| // |
| query->status = QD_AMQP_OK; |
| |
| // |
| // If the offset goes beyond the set of objects, end the query now. |
| // |
| if (offset >= DEQ_SIZE(core->addr_config)) { |
| query->more = false; |
| qdr_agent_enqueue_response_CT(core, query); |
| return; |
| } |
| |
| // |
| // Run to the object at the offset. |
| // |
| qdr_address_config_t *addr = DEQ_HEAD(core->addr_config); |
| for (int i = 0; i < offset && addr; i++) |
| addr = DEQ_NEXT(addr); |
| assert(addr); |
| |
| // |
| // Write the columns of the object into the response body. |
| // |
| qdr_agent_write_config_address_CT(query, addr); |
| |
| // |
| // Advance to the next address |
| // |
| query->next_offset = offset; |
| qdr_manage_advance_config_address_CT(query, addr); |
| |
| // |
| // Enqueue the response. |
| // |
| qdr_agent_enqueue_response_CT(core, query); |
| } |
| |
| |
| void qdra_config_address_get_next_CT(qdr_core_t *core, qdr_query_t *query) |
| { |
| qdr_address_config_t *addr = 0; |
| |
| if (query->next_offset < DEQ_SIZE(core->addr_config)) { |
| addr = DEQ_HEAD(core->addr_config); |
| for (int i = 0; i < query->next_offset && addr; i++) |
| addr = DEQ_NEXT(addr); |
| } |
| |
| if (addr) { |
| // |
| // Write the columns of the addr entity into the response body. |
| // |
| qdr_agent_write_config_address_CT(query, addr); |
| |
| // |
| // Advance to the next object |
| // |
| qdr_manage_advance_config_address_CT(query, addr); |
| } else |
| query->more = false; |
| |
| // |
| // Enqueue the response. |
| // |
| qdr_agent_enqueue_response_CT(core, query); |
| } |
| |
| |
| static qd_address_treatment_t qdra_address_treatment_CT(qd_parsed_field_t *field) |
| { |
| if (field) { |
| qd_field_iterator_t *iter = qd_parse_raw(field); |
| if (qd_field_iterator_equal(iter, (unsigned char*) "multicast")) return QD_TREATMENT_MULTICAST_ONCE; |
| if (qd_field_iterator_equal(iter, (unsigned char*) "closest")) return QD_TREATMENT_ANYCAST_CLOSEST; |
| if (qd_field_iterator_equal(iter, (unsigned char*) "balanced")) return QD_TREATMENT_ANYCAST_BALANCED; |
| } |
| return QD_TREATMENT_ANYCAST_BALANCED; |
| } |
| |
| |
| static qdr_address_config_t *qdr_address_config_find_by_identity_CT(qdr_core_t *core, qd_field_iterator_t *identity) |
| { |
| if (!identity) |
| return 0; |
| |
| qdr_address_config_t *rc = DEQ_HEAD(core->addr_config); |
| while (rc) { |
| // Convert the passed in identity to a char* |
| char id[100]; |
| snprintf(id, 100, "%ld", rc->identity); |
| if (qd_field_iterator_equal(identity, (const unsigned char*) id)) |
| break; |
| rc = DEQ_NEXT(rc); |
| } |
| |
| return rc; |
| |
| } |
| |
| |
| static qdr_address_config_t *qdr_address_config_find_by_name_CT(qdr_core_t *core, qd_field_iterator_t *name) |
| { |
| if (!name) |
| return 0; |
| |
| qdr_address_config_t *rc = DEQ_HEAD(core->addr_config); |
| while (rc) { // Sometimes the name can be null |
| if (rc->name && qd_field_iterator_equal(name, (const unsigned char*) rc->name)) |
| break; |
| rc = DEQ_NEXT(rc); |
| } |
| |
| return rc; |
| } |
| |
| |
| void qdra_config_address_delete_CT(qdr_core_t *core, |
| qdr_query_t *query, |
| qd_field_iterator_t *name, |
| qd_field_iterator_t *identity) |
| { |
| qdr_address_config_t *addr = 0; |
| |
| if (!name && !identity) |
| query->status = QD_AMQP_BAD_REQUEST; |
| else { |
| if (identity) |
| addr = qdr_address_config_find_by_identity_CT(core, identity); |
| else if (name) |
| addr = qdr_address_config_find_by_name_CT(core, name); |
| |
| if (addr) { |
| // |
| // Remove the address from the list and the hash index. |
| // |
| qd_hash_remove_by_handle(core->addr_hash, addr->hash_handle); |
| DEQ_REMOVE(core->addr_config, addr); |
| |
| // |
| // Free resources associated with this address. |
| // |
| if (addr->name) |
| free(addr->name); |
| free_qdr_address_config_t(addr); |
| |
| query->status = QD_AMQP_NO_CONTENT; |
| } else |
| query->status = QD_AMQP_NOT_FOUND; |
| } |
| |
| // |
| // Enqueue the response. |
| // |
| qdr_agent_enqueue_response_CT(core, query); |
| } |
| |
| void qdra_config_address_create_CT(qdr_core_t *core, |
| qd_field_iterator_t *name, |
| qdr_query_t *query, |
| qd_parsed_field_t *in_body) |
| { |
| while (true) { |
| // |
| // Ensure there isn't a duplicate name and that the body is a map |
| // |
| qdr_address_config_t *addr = DEQ_HEAD(core->addr_config); |
| while (addr) { |
| if (name && addr->name && qd_field_iterator_equal(name, (const unsigned char*) addr->name)) |
| break; |
| addr = DEQ_NEXT(addr); |
| } |
| |
| if (!!addr) { |
| query->status = QD_AMQP_BAD_REQUEST; |
| query->status.description = "Name conflicts with an existing entity"; |
| break; |
| } |
| |
| if (!qd_parse_is_map(in_body)) { |
| query->status = QD_AMQP_BAD_REQUEST; |
| break; |
| } |
| |
| // |
| // Extract the fields from the request |
| // |
| qd_parsed_field_t *prefix_field = qd_parse_value_by_key(in_body, qdr_config_address_columns[QDR_CONFIG_ADDRESS_PREFIX]); |
| qd_parsed_field_t *distrib_field = qd_parse_value_by_key(in_body, qdr_config_address_columns[QDR_CONFIG_ADDRESS_DISTRIBUTION]); |
| qd_parsed_field_t *waypoint_field = qd_parse_value_by_key(in_body, qdr_config_address_columns[QDR_CONFIG_ADDRESS_WAYPOINT]); |
| qd_parsed_field_t *in_phase_field = qd_parse_value_by_key(in_body, qdr_config_address_columns[QDR_CONFIG_ADDRESS_IN_PHASE]); |
| qd_parsed_field_t *out_phase_field = qd_parse_value_by_key(in_body, qdr_config_address_columns[QDR_CONFIG_ADDRESS_OUT_PHASE]); |
| |
| // |
| // Prefix field is mandatory. Fail if it is not here. |
| // |
| if (!prefix_field) { |
| query->status = QD_AMQP_BAD_REQUEST; |
| break; |
| } |
| |
| // |
| // Ensure that there isn't another configured address with the same prefix |
| // |
| qd_field_iterator_t *iter = qd_parse_raw(prefix_field); |
| qd_address_iterator_reset_view(iter, ITER_VIEW_ADDRESS_HASH); |
| qd_address_iterator_override_prefix(iter, 'Z'); |
| addr = 0; |
| qd_hash_retrieve(core->addr_hash, iter, (void**) &addr); |
| if (!!addr) { |
| query->status = QD_AMQP_BAD_REQUEST; |
| query->status.description = "Address prefix conflicts with an existing entity"; |
| break; |
| } |
| |
| bool waypoint = waypoint_field ? qd_parse_as_bool(waypoint_field) : false; |
| int in_phase = in_phase_field ? qd_parse_as_int(in_phase_field) : -1; |
| int out_phase = out_phase_field ? qd_parse_as_int(out_phase_field) : -1; |
| |
| // |
| // Handle the address-phasing logic. If the phases are provided, use them. Otherwise |
| // use the waypoint flag to set the most common defaults. |
| // |
| if (in_phase == -1 && out_phase == -1) { |
| in_phase = 0; |
| out_phase = waypoint ? 1 : 0; |
| } |
| |
| // |
| // Validate the phase values |
| // |
| if (in_phase < 0 || in_phase > 9 || out_phase < 0 || out_phase > 9) { |
| query->status = QD_AMQP_BAD_REQUEST; |
| query->status.description = "Phase values must be between 0 and 9"; |
| break; |
| } |
| |
| // |
| // The request is good. Create the entity and insert it into the hash index and list. |
| // |
| addr = new_qdr_address_config_t(); |
| DEQ_ITEM_INIT(addr); |
| addr->name = name ? (char*) qd_field_iterator_copy(name) : 0; |
| addr->identity = qdr_identifier(core); |
| addr->treatment = qdra_address_treatment_CT(distrib_field); |
| addr->in_phase = in_phase; |
| addr->out_phase = out_phase; |
| |
| qd_hash_insert(core->addr_hash, iter, addr, &addr->hash_handle); |
| DEQ_INSERT_TAIL(core->addr_config, addr); |
| |
| // |
| // Compose the result map for the response. |
| // |
| if (query->body) { |
| qd_compose_start_map(query->body); |
| for (int col = 0; col < QDR_CONFIG_ADDRESS_COLUMN_COUNT; col++) |
| qdr_config_address_insert_column_CT(addr, col, query->body, true); |
| qd_compose_end_map(query->body); |
| } |
| |
| query->status = QD_AMQP_CREATED; |
| break; |
| } |
| |
| // |
| // Enqueue the response if there is a body. If there is no body, this is a management |
| // operation created internally by the configuration file parser. |
| // |
| if (query->body) { |
| // |
| // If there was an error in processing the create, insert a NULL value into the body. |
| // |
| if (query->status.status / 100 > 2) |
| qd_compose_insert_null(query->body); |
| qdr_agent_enqueue_response_CT(core, query); |
| } else |
| qdr_query_free(query); |
| } |
