0bc298c3ad
The fields of ROUTE message were not read from the input buffer, thus the input buffer got corrupted. Sending a new message to the server would result reading the remaining fields from the buffer, which means reading some values instead of message signature. Because of this unmet expectation, Memgraph closed the connection. With this fix, the fields of the ROUTE message are properly read and ignored.
81 lines
3.0 KiB
C++
81 lines
3.0 KiB
C++
// Copyright 2022 Memgraph Ltd.
|
|
//
|
|
// Use of this software is governed by the Business Source License
|
|
// included in the file licenses/BSL.txt; by using this file, you agree to be bound by the terms of the Business Source
|
|
// License, and you may not use this file except in compliance with the Business Source License.
|
|
//
|
|
// As of the Change Date specified in that file, in accordance with
|
|
// the Business Source License, use of this software will be governed
|
|
// by the Apache License, Version 2.0, included in the file
|
|
// licenses/APL.txt.
|
|
|
|
#include "mg_procedure.h"
|
|
|
|
static void TryToWrite(struct mgp_list *args, mgp_func_context *ctx, mgp_func_result *result,
|
|
struct mgp_memory *memory) {
|
|
mgp_value *value{nullptr};
|
|
mgp_vertex *vertex{nullptr};
|
|
static_cast<void>(mgp_list_at(args, 0, &value));
|
|
static_cast<void>(mgp_value_get_vertex(value, &vertex));
|
|
|
|
const char *name;
|
|
static_cast<void>(mgp_list_at(args, 1, &value));
|
|
static_cast<void>(mgp_value_get_string(value, &name));
|
|
|
|
static_cast<void>(mgp_list_at(args, 2, &value));
|
|
|
|
// Setting a property should set an error
|
|
auto err_code = mgp_vertex_set_property(vertex, name, value);
|
|
if (err_code != mgp_error::MGP_ERROR_NO_ERROR) {
|
|
static_cast<void>(mgp_func_result_set_error_msg(result, "Cannot set property in the function!", memory));
|
|
return;
|
|
}
|
|
|
|
err_code = mgp_func_result_set_value(result, value, memory);
|
|
if (err_code != mgp_error::MGP_ERROR_NO_ERROR) {
|
|
static_cast<void>(mgp_func_result_set_error_msg(result, "Failed to construct return value!", memory));
|
|
return;
|
|
}
|
|
}
|
|
|
|
// Each module needs to define mgp_init_module function.
|
|
// Here you can register multiple functions/procedures your module supports.
|
|
extern "C" int mgp_init_module(struct mgp_module *module, struct mgp_memory *memory) {
|
|
{
|
|
mgp_func *func{nullptr};
|
|
auto err_code = mgp_module_add_function(module, "try_to_write", TryToWrite, &func);
|
|
if (err_code != mgp_error::MGP_ERROR_NO_ERROR) {
|
|
return 1;
|
|
}
|
|
|
|
mgp_type *type_vertex{nullptr};
|
|
static_cast<void>(mgp_type_node(&type_vertex));
|
|
err_code = mgp_func_add_arg(func, "argument", type_vertex);
|
|
if (err_code != mgp_error::MGP_ERROR_NO_ERROR) {
|
|
return 1;
|
|
}
|
|
|
|
mgp_type *type_string{nullptr};
|
|
static_cast<void>(mgp_type_string(&type_string));
|
|
err_code = mgp_func_add_arg(func, "name", type_string);
|
|
if (err_code != mgp_error::MGP_ERROR_NO_ERROR) {
|
|
return 1;
|
|
}
|
|
|
|
mgp_type *any_type{nullptr};
|
|
static_cast<void>(mgp_type_any(&any_type));
|
|
mgp_type *nullable_type{nullptr};
|
|
static_cast<void>(mgp_type_nullable(any_type, &nullable_type));
|
|
err_code = mgp_func_add_arg(func, "value", nullable_type);
|
|
if (err_code != mgp_error::MGP_ERROR_NO_ERROR) {
|
|
return 1;
|
|
}
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
// This is an optional function if you need to release any resources before the
|
|
// module is unloaded. You will probably need this if you acquired some
|
|
// resources in mgp_init_module.
|
|
extern "C" int mgp_shutdown_module() { return 0; }
|