Telodendria/src/Routes/RouteRegister.c

301 lines
9 KiB
C
Raw Normal View History

2022-12-14 15:40:23 +00:00
/*
2022-12-26 15:52:52 +00:00
* Copyright (C) 2022-2023 Jordan Bancino <@jordan:bancino.net>
2022-12-14 15:40:23 +00:00
*
* Permission is hereby granted, free of charge, to any person
* obtaining a copy of this software and associated documentation files
* (the "Software"), to deal in the Software without restriction,
* including without limitation the rights to use, copy, modify, merge,
* publish, distribute, sublicense, and/or sell copies of the Software,
* and to permit persons to whom the Software is furnished to do so,
* subject to the following conditions:
*
* The above copyright notice and this permission notice shall be
* included in all copies or portions of the Software.
*
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
* EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
* MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
* NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS
* BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN
* ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
* CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
* SOFTWARE.
*/
#include <Routes.h>
#include <string.h>
#include <Json.h>
#include <HashMap.h>
#include <Str.h>
#include <Memory.h>
2023-01-07 15:51:56 +00:00
#include <User.h>
#include <Uia.h>
2022-12-14 15:40:23 +00:00
2023-03-14 00:37:24 +00:00
static Array *
RouteRegisterRegFlow(void)
{
Array *response = ArrayCreate();
if (!response)
{
return NULL;
}
ArrayAdd(response, UiaStageBuild("m.login.registration_token", NULL));
return response;
}
ROUTE_IMPL(RouteRegister, path, argp)
2022-12-14 15:40:23 +00:00
{
RouteArgs *args = argp;
2022-12-14 15:40:23 +00:00
HashMap *request = NULL;
HashMap *response = NULL;
JsonValue *val;
char *kind;
char *username = NULL;
char *password = NULL;
char *initialDeviceDisplayName = NULL;
int refreshToken = 0;
int inhibitLogin = 0;
2023-01-06 21:54:33 +00:00
char *deviceId = NULL;
char *fullUsername;
2023-01-07 15:51:56 +00:00
Db *db = args->matrixArgs->db;
2023-01-09 19:22:09 +00:00
User *user = NULL;
Array *uiaFlows = NULL;
int uiaResult;
if (ArraySize(path) == 0)
2022-12-14 15:40:23 +00:00
{
if (HttpRequestMethodGet(args->context) != HTTP_POST)
{
HttpResponseStatus(args->context, HTTP_BAD_REQUEST);
return MatrixErrorCreate(M_UNRECOGNIZED);
}
request = JsonDecode(HttpServerStream(args->context));
if (!request)
{
HttpResponseStatus(args->context, HTTP_BAD_REQUEST);
return MatrixErrorCreate(M_NOT_JSON);
}
val = HashMapGet(request, "username");
if (val)
{
if (JsonValueType(val) != JSON_STRING)
{
HttpResponseStatus(args->context, HTTP_BAD_REQUEST);
response = MatrixErrorCreate(M_BAD_JSON);
goto finish;
}
username = StrDuplicate(JsonValueAsString(val));
2023-01-07 15:51:56 +00:00
if (!UserValidate(username, args->matrixArgs->config->serverName))
{
HttpResponseStatus(args->context, HTTP_BAD_REQUEST);
response = MatrixErrorCreate(M_INVALID_USERNAME);
goto finish;
}
2023-01-07 15:51:56 +00:00
if (UserExists(db, username))
{
HttpResponseStatus(args->context, HTTP_BAD_REQUEST);
response = MatrixErrorCreate(M_USER_IN_USE);
goto finish;
}
}
uiaFlows = ArrayCreate();
2023-03-14 00:37:24 +00:00
ArrayAdd(uiaFlows, RouteRegisterRegFlow());
2023-03-22 17:17:30 +00:00
if (args->matrixArgs->config->flags & CONFIG_REGISTRATION)
2023-03-14 00:37:24 +00:00
{
ArrayAdd(uiaFlows, UiaDummyFlow());
}
uiaResult = UiaComplete(uiaFlows, args->context,
args->matrixArgs->db, request, &response,
args->matrixArgs->config);
if (uiaResult < 0)
{
HttpResponseStatus(args->context, HTTP_INTERNAL_SERVER_ERROR);
response = MatrixErrorCreate(M_UNKNOWN);
goto finish;
}
else if (!uiaResult)
{
/* UiaComplete() sets the response and status for us. */
goto finish;
}
kind = HashMapGet(HttpRequestParams(args->context), "kind");
/* We don't support guest accounts yet */
if (kind && strcmp(kind, "user") != 0)
{
HttpResponseStatus(args->context, HTTP_FORBIDDEN);
response = MatrixErrorCreate(M_INVALID_PARAM);
goto finish;
}
val = HashMapGet(request, "password");
if (!val)
{
HttpResponseStatus(args->context, HTTP_BAD_REQUEST);
response = MatrixErrorCreate(M_MISSING_PARAM);
goto finish;
}
if (JsonValueType(val) != JSON_STRING)
{
HttpResponseStatus(args->context, HTTP_BAD_REQUEST);
response = MatrixErrorCreate(M_BAD_JSON);
goto finish;
}
password = StrDuplicate(JsonValueAsString(val));
val = HashMapGet(request, "device_id");
if (val)
{
if (JsonValueType(val) != JSON_STRING)
{
HttpResponseStatus(args->context, HTTP_BAD_REQUEST);
response = MatrixErrorCreate(M_BAD_JSON);
goto finish;
}
deviceId = StrDuplicate(JsonValueAsString(val));
}
val = HashMapGet(request, "inhibit_login");
if (val)
{
if (JsonValueType(val) != JSON_BOOLEAN)
{
HttpResponseStatus(args->context, HTTP_BAD_REQUEST);
response = MatrixErrorCreate(M_BAD_JSON);
goto finish;
}
inhibitLogin = JsonValueAsBoolean(val);
}
val = HashMapGet(request, "initial_device_display_name");
if (val)
{
if (JsonValueType(val) != JSON_STRING)
{
HttpResponseStatus(args->context, HTTP_BAD_REQUEST);
response = MatrixErrorCreate(M_BAD_JSON);
goto finish;
}
initialDeviceDisplayName = StrDuplicate(JsonValueAsString(val));
}
val = HashMapGet(request, "refresh_token");
if (val)
{
if (JsonValueType(val) != JSON_BOOLEAN)
{
HttpResponseStatus(args->context, HTTP_BAD_REQUEST);
response = MatrixErrorCreate(M_BAD_JSON);
goto finish;
}
refreshToken = JsonValueAsBoolean(val);
}
2023-01-09 19:22:09 +00:00
user = UserCreate(db, username, password);
response = HashMapCreate();
fullUsername = StrConcat(4, "@", UserGetName(user), ":", args->matrixArgs->config->serverName);
HashMapSet(response, "user_id", JsonValueString(fullUsername));
Free(fullUsername);
2023-01-09 19:22:09 +00:00
HttpResponseStatus(args->context, HTTP_OK);
2023-01-07 15:51:56 +00:00
if (!inhibitLogin)
{
2023-01-16 21:17:44 +00:00
UserLoginInfo *loginInfo = UserLogin(user, password, deviceId,
initialDeviceDisplayName, refreshToken);
HashMapSet(response, "access_token",
2023-02-17 03:23:25 +00:00
JsonValueString(loginInfo->accessToken->string));
2023-01-16 21:17:44 +00:00
HashMapSet(response, "device_id",
2023-02-17 03:23:25 +00:00
JsonValueString(loginInfo->accessToken->deviceId));
2023-01-16 21:17:44 +00:00
if (refreshToken)
{
HashMapSet(response, "expires_in_ms",
2023-02-17 03:23:25 +00:00
JsonValueInteger(loginInfo->accessToken->lifetime));
2023-01-16 21:17:44 +00:00
HashMapSet(response, "refresh_token",
JsonValueString(loginInfo->refreshToken));
}
UserAccessTokenFree(loginInfo->accessToken);
Free(loginInfo->refreshToken);
2023-01-16 21:17:44 +00:00
Free(loginInfo);
2023-01-07 15:51:56 +00:00
}
Log(LOG_INFO, "Registered user '%s'", UserGetName(user));
2023-01-10 00:38:47 +00:00
UserUnlock(user);
finish:
UiaFlowsFree(uiaFlows);
2023-01-07 15:51:56 +00:00
Free(username);
Free(password);
Free(deviceId);
Free(initialDeviceDisplayName);
JsonFree(request);
}
else
2022-12-14 15:40:23 +00:00
{
if (HttpRequestMethodGet(args->context) == HTTP_GET &&
MATRIX_PATH_EQUALS(ArrayGet(path, 0), "available"))
{
username = HashMapGet(
HttpRequestParams(args->context), "username");
2022-12-14 21:23:20 +00:00
if (!username)
{
HttpResponseStatus(args->context, HTTP_BAD_REQUEST);
response = MatrixErrorCreate(M_MISSING_PARAM);
}
2023-01-07 15:51:56 +00:00
else if (!UserValidate(username, args->matrixArgs->config->serverName))
{
HttpResponseStatus(args->context, HTTP_BAD_REQUEST);
response = MatrixErrorCreate(M_INVALID_USERNAME);
}
2023-01-07 18:24:16 +00:00
else if (!UserExists(db, username))
2023-01-07 15:51:56 +00:00
{
response = HashMapCreate();
HashMapSet(response, "available", JsonValueBoolean(1));
}
else
{
HttpResponseStatus(args->context, HTTP_BAD_REQUEST);
response = MatrixErrorCreate(M_USER_IN_USE);
}
}
else
{
HttpResponseStatus(args->context, HTTP_NOT_FOUND);
response = MatrixErrorCreate(M_UNRECOGNIZED);
}
2022-12-14 15:40:23 +00:00
}
return response;
}