diff options
Diffstat (limited to 'fluent-bit/lib/monkey/mk_server/mk_user.c')
| -rw-r--r-- | fluent-bit/lib/monkey/mk_server/mk_user.c | 175 |
1 files changed, 175 insertions, 0 deletions
diff --git a/fluent-bit/lib/monkey/mk_server/mk_user.c b/fluent-bit/lib/monkey/mk_server/mk_user.c new file mode 100644 index 00000000..7200ff08 --- /dev/null +++ b/fluent-bit/lib/monkey/mk_server/mk_user.c @@ -0,0 +1,175 @@ +/* -*- Mode: C; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 4 -*- */ + +/* Monkey HTTP Server + * ================== + * Copyright 2001-2017 Eduardo Silva <eduardo@monkey.io> + * + * 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 <monkey/monkey.h> +#include <monkey/mk_user.h> +#include <monkey/mk_http.h> +#include <monkey/mk_http_status.h> +#include <monkey/mk_core.h> +#include <monkey/mk_utils.h> +#include <monkey/mk_config.h> + +#ifndef _WIN32 +#include <pwd.h> +#include <sys/resource.h> +#include <sys/types.h> +#include <grp.h> + +int mk_user_init(struct mk_http_session *cs, struct mk_http_request *sr, + struct mk_server *server) +{ + int limit; + const int offset = 2; /* The user is defined after the '/~' string, so offset = 2 */ + const int user_len = 255; + char user[/*user_len*/ 255]; /* VC++ Doesn't allow for this to be a const int*/ + char *user_uri; + struct passwd *s_user; + + if (sr->uri_processed.len <= 2) { + return -1; + } + + limit = mk_string_char_search(sr->uri_processed.data + offset, '/', + sr->uri_processed.len); + + if (limit == -1) { + limit = (sr->uri_processed.len) - offset; + } + + if (limit + offset >= (user_len)) { + return -1; + } + + memcpy(user, sr->uri_processed.data + offset, limit); + user[limit] = '\0'; + + MK_TRACE("user: '%s'", user); + + /* Check system user */ + if ((s_user = getpwnam(user)) == NULL) { + mk_http_error(MK_CLIENT_NOT_FOUND, cs, sr, server); + return -1; + } + + if (sr->uri_processed.len > (unsigned int) (offset+limit)) { + user_uri = mk_mem_alloc(sr->uri_processed.len); + if (!user_uri) { + return -1; + } + + memcpy(user_uri, + sr->uri_processed.data + (offset + limit), + sr->uri_processed.len - offset - limit); + user_uri[sr->uri_processed.len - offset - limit] = '\0'; + + mk_string_build(&sr->real_path.data, &sr->real_path.len, + "%s/%s%s", + s_user->pw_dir, server->conf_user_pub, user_uri); + mk_mem_free(user_uri); + } + else { + mk_string_build(&sr->real_path.data, &sr->real_path.len, + "%s/%s", s_user->pw_dir, server->conf_user_pub); + } + + sr->user_home = MK_TRUE; + return 0; +} + +/* Change process user */ +int mk_user_set_uidgid(struct mk_server *server) +{ + struct passwd *usr; + + /* Launched by root ? */ + if (geteuid() == 0 && server->user) { + struct rlimit rl; + + if (getrlimit(RLIMIT_NOFILE, &rl)) { + mk_warn("cannot get resource limits"); + } + + /* Check if user exists */ + if ((usr = getpwnam(server->user)) == NULL) { + mk_err("Invalid user '%s'", server->user); + goto out; + } + + if (initgroups(server->user, usr->pw_gid) != 0) { + mk_err("Initgroups() failed"); + } + + /* Change process UID and GID */ + if (setegid(usr->pw_gid) == -1) { + mk_err("I cannot change the GID to %u", usr->pw_gid); + } + + if (seteuid(usr->pw_uid) == -1) { + mk_err("I cannot change the UID to %u", usr->pw_uid); + } + + server->is_seteuid = MK_TRUE; + } + + out: + /* Variables set for run checks on file permission */ + //FIXME + //EUID = geteuid(); + //EGID = getegid(); + + return 0; +} + +/* Return process to the original user */ +int mk_user_undo_uidgid(struct mk_server *server) +{ + if (server->is_seteuid == MK_TRUE) { + if (setegid(0) < 0) { + mk_err("Can't restore effective GID"); + } + if (seteuid(0) < 0) { + mk_err("Can't restore effective UID"); + } + } + return 0; +} + +#else +/* + None of these functionalities are going to be available in windows at the moment +*/ + +int mk_user_init(struct mk_http_session* cs, struct mk_http_request* sr, + struct mk_server* server) +{ + return -1; +} + +int mk_user_set_uidgid(struct mk_server* server) +{ + mk_err("Cannot impersonate users in windows"); + + return 0; +} + +int mk_user_undo_uidgid(struct mk_server* server) +{ + return 0; +} +#endif |