Kernel/Userspace: Add basic init process

This process parses /etc/passwd and promps login screen.
When an username is entered, it will launch that users shell
This commit is contained in:
Bananymous 2023-06-11 21:01:02 +03:00
parent cce2f3e19a
commit 78536f9678
8 changed files with 173 additions and 15 deletions

2
base/etc/passwd Normal file
View File

@ -0,0 +1,2 @@
root:x:0:0::/root:/bin/Shell
user:x:1000:1000::/home/user:/bin/Shell

View File

@ -300,8 +300,6 @@ namespace Kernel
delete range;
m_mapped_ranges.clear();
m_open_files.clear();
load_elf_to_memory(*elf);
m_userspace_info.entry = elf->file_header_native().e_entry;

View File

@ -180,7 +180,7 @@ static void init2(void* tty1)
((TTY*)tty1)->initialize_device();
MUST(Process::create_userspace({ 0, 0, 0, 0 }, "/usr/bin/Shell"sv));
MUST(Process::create_userspace({ 0, 0, 0, 0 }, "/usr/bin/init"sv));
return;
Process::create_kernel(

View File

@ -27,15 +27,6 @@ FILE* stdin = &s_files[0];
FILE* stdout = &s_files[1];
FILE* stderr = &s_files[2];
extern "C" void _init_stdio()
{
char tty[L_ctermid];
ctermid(tty);
if (open(tty, O_RDONLY) != 0) _exit(1);
if (open(tty, O_WRONLY) != 1) _exit(1);
if (open(tty, O_WRONLY) != 2) _exit(1);
}
void clearerr(FILE* file)
{
file->eof = false;

View File

@ -11,12 +11,9 @@
char** environ;
extern "C" void _init_stdio();
extern "C" void _init_libc(char** _environ)
{
environ = _environ;
_init_stdio();
}
void _exit(int status)

View File

@ -5,6 +5,7 @@ project(userspace CXX)
set(USERSPACE_PROJECTS
cat
echo
init
ls
Shell
test

View File

@ -0,0 +1,16 @@
cmake_minimum_required(VERSION 3.26)
project(init CXX)
set(SOURCES
main.cpp
)
add_executable(init ${SOURCES})
target_compile_options(init PUBLIC -O2 -g)
target_link_libraries(init PUBLIC libc ban)
add_custom_target(init-install
COMMAND cp ${CMAKE_CURRENT_BINARY_DIR}/init ${BANAN_BIN}/
DEPENDS init
)

153
userspace/init/main.cpp Normal file
View File

@ -0,0 +1,153 @@
#include <BAN/String.h>
#include <BAN/Optional.h>
#include <BAN/Vector.h>
#include <ctype.h>
#include <fcntl.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
struct User
{
BAN::String name;
uid_t uid;
gid_t gid;
BAN::String home;
BAN::String shell;
};
BAN::Optional<id_t> parse_id(BAN::StringView value)
{
// NOTE: we only allow 10^9 uids
if (value.size() < 1 || value.size() > 9)
return {};
id_t id { 0 };
for (char c : value)
{
if (!isdigit(c))
return {};
id = (id * 10) + (c - '0');
}
return id;
}
BAN::Optional<User> parse_user(BAN::StringView line)
{
auto parts = MUST(line.split(':', true));
if (parts.size() != 7)
return {};
User user;
user.name = parts[0];
user.uid = ({ auto id = parse_id(parts[2]); if (!id.has_value()) return {}; id.value(); });
user.gid = ({ auto id = parse_id(parts[3]); if (!id.has_value()) return {}; id.value(); });
user.home = parts[5];
user.shell = parts[6];
return user;
}
BAN::Vector<User> parse_users()
{
FILE* fp = fopen("/etc/passwd", "r");
if (fp == nullptr)
{
fprintf(stderr, "could not open /etc/passwd\n");
perror("fopen");
}
BAN::Vector<User> users;
char buffer[1024];
while (fgets(buffer, sizeof(buffer), fp))
{
*strchrnul(buffer, '\n') = '\0';
auto user = parse_user(buffer);
if (user.has_value())
MUST(users.push_back(user.release_value()));
}
if (ferror(fp))
{
perror("fread");
fclose(fp);
return {};
}
fclose(fp);
return users;
}
void initialize_stdio()
{
char tty[L_ctermid];
ctermid(tty);
if (open(tty, O_RDONLY) != 0) _exit(1);
if (open(tty, O_WRONLY) != 1) _exit(1);
if (open(tty, O_WRONLY) != 2) _exit(1);
}
int main()
{
initialize_stdio();
while (true)
{
auto users = parse_users();
char name_buffer[128];
BAN::StringView name;
while (true)
{
printf("username: ");
fflush(stdout);
size_t nread = fread(name_buffer, 1, sizeof(name_buffer) - 1, stdin);
if (nread == 0)
{
if (ferror(stdin))
{
fprintf(stderr, "Could not read from stdin\n");
return 1;
}
continue;
}
if (nread == 1)
continue;
name = BAN::StringView(name_buffer, nread - 1);
break;
}
for (const User& user : users)
{
if (user.name == name)
{
pid_t pid = fork();
if (pid == 0)
{
printf("Welcome back %s!\n", user.name.data());
execl(user.shell.data(), user.shell.data(), nullptr);
perror("execl");
exit(1);
}
if (pid == -1)
{
perror("fork");
break;
}
int status;
waitpid(pid, &status, 0);
}
}
}
}