You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
327 lines
9.4 KiB
327 lines
9.4 KiB
#include "webfuse/adapter/impl/client_protocol.h"
|
|
#include "webfuse/adapter/client_callback.h"
|
|
#include "webfuse/adapter/impl/credentials.h"
|
|
#include "webfuse/adapter/impl/filesystem.h"
|
|
#include "webfuse/adapter/impl/mountpoint.h"
|
|
#include "webfuse/core/protocol_names.h"
|
|
#include "webfuse/core/url.h"
|
|
#include "webfuse/core/util.h"
|
|
#include "webfuse/core/timer/manager.h"
|
|
#include "webfuse/core/jsonrpc/response.h"
|
|
#include "webfuse/core/jsonrpc/proxy.h"
|
|
|
|
#include "webfuse/core/message.h"
|
|
#include "webfuse/core/message_queue.h"
|
|
#include "webfuse/core/container_of.h"
|
|
|
|
|
|
#include <stddef.h>
|
|
#include <libwebsockets.h>
|
|
|
|
#define WF_DEFAULT_TIMEOUT (10 * 1000)
|
|
|
|
struct wf_impl_client_protocol_add_filesystem_context
|
|
{
|
|
struct wf_client_protocol * protocol;
|
|
char * local_path;
|
|
};
|
|
|
|
static void
|
|
wf_impl_client_protocol_process(
|
|
struct wf_client_protocol * protocol,
|
|
char const * data,
|
|
size_t length)
|
|
{
|
|
json_t * message = json_loadb(data, length, 0, NULL);
|
|
if (NULL != message)
|
|
{
|
|
if (wf_jsonrpc_is_response(message))
|
|
{
|
|
wf_jsonrpc_proxy_onresult(protocol->proxy, message);
|
|
}
|
|
|
|
json_decref(message);
|
|
}
|
|
}
|
|
|
|
static bool
|
|
wf_impl_client_protocol_send(
|
|
json_t * request,
|
|
void * user_data)
|
|
{
|
|
bool result = false;
|
|
struct wf_client_protocol * protocol = user_data;
|
|
|
|
if (NULL != protocol->wsi)
|
|
{
|
|
struct wf_message * message = wf_message_create(request);
|
|
if (NULL != message)
|
|
{
|
|
wf_slist_append(&protocol->messages, &message->item);
|
|
lws_callback_on_writable(protocol->wsi);
|
|
result = true;
|
|
}
|
|
}
|
|
|
|
return result;
|
|
}
|
|
|
|
|
|
static void
|
|
wf_impl_client_protocol_on_authenticate_finished(
|
|
void * user_data,
|
|
json_t const * result,
|
|
json_t const * WF_UNUSED_PARAM(error))
|
|
{
|
|
struct wf_client_protocol * protocol = user_data;
|
|
int const reason = (NULL != result) ? WF_CLIENT_AUTHENTICATED : WF_CLIENT_AUTHENTICATION_FAILED;
|
|
|
|
protocol->callback(protocol->user_data, reason, NULL);
|
|
}
|
|
|
|
static void
|
|
wf_impl_client_protocol_on_add_filesystem_finished(
|
|
void * user_data,
|
|
json_t const * result,
|
|
json_t const * WF_UNUSED_PARAM(error))
|
|
{
|
|
struct wf_impl_client_protocol_add_filesystem_context * context = user_data;
|
|
struct wf_client_protocol * protocol = context->protocol;
|
|
|
|
int reason = WF_CLIENT_FILESYSTEM_ADD_FAILED;
|
|
if (NULL == protocol->filesystem)
|
|
{
|
|
json_t * id = json_object_get(result, "id");
|
|
if (json_is_string(id))
|
|
{
|
|
char const * name = json_string_value(id);
|
|
struct wf_mountpoint * mountpoint = wf_mountpoint_create(context->local_path);
|
|
protocol->filesystem = wf_impl_filesystem_create(protocol->wsi,protocol->proxy, name, mountpoint);
|
|
if (NULL != protocol->filesystem)
|
|
{
|
|
reason = WF_CLIENT_FILESYSTEM_ADDED;
|
|
}
|
|
else
|
|
{
|
|
wf_mountpoint_dispose(mountpoint);
|
|
}
|
|
}
|
|
}
|
|
|
|
free(context->local_path);
|
|
free(context);
|
|
protocol->callback(protocol->user_data, reason, NULL);
|
|
}
|
|
|
|
static int wf_impl_client_protocol_lws_callback(
|
|
struct lws * wsi,
|
|
enum lws_callback_reasons reason,
|
|
void * WF_UNUSED_PARAM(user),
|
|
void * in,
|
|
size_t WF_UNUSED_PARAM(len))
|
|
{
|
|
int result = 0;
|
|
struct lws_protocols const * ws_protocol = lws_get_protocol(wsi);
|
|
struct wf_client_protocol * protocol = (NULL != ws_protocol) ? ws_protocol->user : NULL;
|
|
|
|
if (NULL != protocol)
|
|
{
|
|
wf_timer_manager_check(protocol->timer_manager);
|
|
|
|
switch (reason)
|
|
{
|
|
case LWS_CALLBACK_CLIENT_ESTABLISHED:
|
|
protocol->is_connected = true;
|
|
protocol->callback(protocol->user_data, WF_CLIENT_CONNECTED, NULL);
|
|
break;
|
|
case LWS_CALLBACK_CLIENT_CONNECTION_ERROR:
|
|
protocol->is_connected = false;
|
|
protocol->callback(protocol->user_data, WF_CLIENT_DISCONNECTED, NULL);
|
|
break;
|
|
case LWS_CALLBACK_CLIENT_CLOSED:
|
|
protocol->is_connected = false;
|
|
protocol->callback(protocol->user_data, WF_CLIENT_DISCONNECTED, NULL);
|
|
protocol->wsi = NULL;
|
|
break;
|
|
case LWS_CALLBACK_CLIENT_RECEIVE:
|
|
wf_impl_client_protocol_process(protocol, in, len);
|
|
case LWS_CALLBACK_SERVER_WRITEABLE:
|
|
// fall-through
|
|
case LWS_CALLBACK_CLIENT_WRITEABLE:
|
|
if (wsi == protocol->wsi)
|
|
{
|
|
if (protocol->is_shutdown_requested)
|
|
{
|
|
result = 1;
|
|
}
|
|
else if (!wf_slist_empty(&protocol->messages))
|
|
{
|
|
struct wf_slist_item * item = wf_slist_remove_first(&protocol->messages);
|
|
struct wf_message * message = wf_container_of(item, struct wf_message, item);
|
|
lws_write(wsi, (unsigned char*) message->data, message->length, LWS_WRITE_TEXT);
|
|
wf_message_dispose(message);
|
|
|
|
if (!wf_slist_empty(&protocol->messages))
|
|
{
|
|
lws_callback_on_writable(wsi);
|
|
}
|
|
}
|
|
}
|
|
default:
|
|
break;
|
|
}
|
|
}
|
|
|
|
return result;
|
|
}
|
|
|
|
void
|
|
wf_impl_client_protocol_init(
|
|
struct wf_client_protocol * protocol,
|
|
wf_client_protocol_callback_fn * callback,
|
|
void * user_data)
|
|
{
|
|
protocol->is_connected = false,
|
|
protocol->is_shutdown_requested = false;
|
|
protocol->wsi = NULL;
|
|
protocol->callback = callback;
|
|
protocol->user_data = user_data;
|
|
protocol->filesystem = NULL;
|
|
|
|
wf_slist_init(&protocol->messages);
|
|
protocol->timer_manager = wf_timer_manager_create();
|
|
protocol->proxy = wf_jsonrpc_proxy_create(protocol->timer_manager, WF_DEFAULT_TIMEOUT, &wf_impl_client_protocol_send, protocol);
|
|
|
|
protocol->callback(protocol->user_data, WF_CLIENT_INIT, NULL);
|
|
}
|
|
|
|
void
|
|
wf_impl_client_protocol_cleanup(
|
|
struct wf_client_protocol * protocol)
|
|
{
|
|
if (NULL != protocol->filesystem)
|
|
{
|
|
wf_impl_filesystem_dispose(protocol->filesystem);
|
|
}
|
|
|
|
protocol->callback(protocol->user_data, WF_CLIENT_CLEANUP, NULL);
|
|
wf_jsonrpc_proxy_dispose(protocol->proxy);
|
|
wf_timer_manager_dispose(protocol->timer_manager);
|
|
wf_message_queue_cleanup(&protocol->messages);
|
|
}
|
|
|
|
void
|
|
wf_impl_client_protocol_callback(
|
|
struct wf_client_protocol * protocol,
|
|
int reason,
|
|
void * arg)
|
|
{
|
|
protocol->callback(protocol->user_data, reason, arg);
|
|
}
|
|
|
|
void
|
|
wf_impl_client_protocol_init_lws(
|
|
struct wf_client_protocol * protocol,
|
|
struct lws_protocols * lws_protocol)
|
|
{
|
|
lws_protocol->name = WF_PROTOCOL_NAME_ADAPTER_CLIENT;
|
|
lws_protocol->callback = &wf_impl_client_protocol_lws_callback;
|
|
lws_protocol->per_session_data_size = 0;
|
|
lws_protocol->user = protocol;
|
|
}
|
|
|
|
void
|
|
wf_impl_client_protocol_connect(
|
|
struct wf_client_protocol * protocol,
|
|
struct lws_context * context,
|
|
char const * url)
|
|
{
|
|
struct wf_url url_data;
|
|
bool const success = wf_url_init(&url_data, url);
|
|
if (success)
|
|
{
|
|
struct lws_client_connect_info info;
|
|
memset(&info, 0 ,sizeof(struct lws_client_connect_info));
|
|
info.context = context;
|
|
info.port = url_data.port;
|
|
info.address = url_data.host;
|
|
info.path = url_data.path;
|
|
info.host = info.address;
|
|
info.origin = info.address;
|
|
info.ssl_connection = (url_data.use_tls) ? LCCSCF_USE_SSL : 0;
|
|
info.protocol = WF_PROTOCOL_NAME_PROVIDER_SERVER;
|
|
info.local_protocol_name = WF_PROTOCOL_NAME_ADAPTER_CLIENT;
|
|
info.pwsi = &protocol->wsi;
|
|
|
|
lws_client_connect_via_info(&info);
|
|
wf_url_cleanup(&url_data);
|
|
}
|
|
else
|
|
{
|
|
protocol->callback(protocol->user_data, WF_CLIENT_DISCONNECTED, NULL);
|
|
}
|
|
}
|
|
|
|
void
|
|
wf_impl_client_protocol_disconnect(
|
|
struct wf_client_protocol * protocol)
|
|
{
|
|
if (protocol->is_connected)
|
|
{
|
|
protocol->is_shutdown_requested = true;
|
|
lws_callback_on_writable(protocol->wsi);
|
|
}
|
|
else
|
|
{
|
|
protocol->callback(protocol->user_data, WF_CLIENT_DISCONNECTED, NULL);
|
|
}
|
|
|
|
}
|
|
|
|
void
|
|
wf_impl_client_protocol_authenticate(
|
|
struct wf_client_protocol * protocol)
|
|
{
|
|
struct wf_credentials creds;
|
|
wf_impl_credentials_init_default(&creds);
|
|
protocol->callback(protocol->user_data, WF_CLIENT_AUTHENTICATE_GET_CREDENTIALS, &creds);
|
|
|
|
json_incref(creds.data);
|
|
wf_jsonrpc_proxy_invoke(
|
|
protocol->proxy,
|
|
&wf_impl_client_protocol_on_authenticate_finished,
|
|
protocol,
|
|
"authenticate",
|
|
"sj",
|
|
creds.type, creds.data);
|
|
|
|
wf_impl_credentials_cleanup(&creds);
|
|
}
|
|
|
|
void
|
|
wf_impl_client_protocol_add_filesystem(
|
|
struct wf_client_protocol * protocol,
|
|
char const * local_path,
|
|
char const * name)
|
|
{
|
|
if (NULL == protocol->filesystem)
|
|
{
|
|
struct wf_impl_client_protocol_add_filesystem_context * context = malloc(sizeof(struct wf_impl_client_protocol_add_filesystem_context));
|
|
context->protocol = protocol;
|
|
context->local_path = strdup(local_path);
|
|
|
|
wf_jsonrpc_proxy_invoke(
|
|
protocol->proxy,
|
|
&wf_impl_client_protocol_on_add_filesystem_finished,
|
|
context,
|
|
"add_filesystem",
|
|
"s",
|
|
name);
|
|
}
|
|
else
|
|
{
|
|
protocol->callback(protocol->user_data, WF_CLIENT_FILESYSTEM_ADD_FAILED, NULL);
|
|
}
|
|
}
|
|
|