Add a skeleton keystore_server.
It will handle the BKeyStore messages but is yet relatively empty. It only returns an error to two messages right now.
This commit is contained in:
Michael Lotz
Ryan Leavengood
parent
005a15bbcd
commit
8d9bc9e0ee
@ -6,6 +6,7 @@ SubInclude HAIKU_TOP src servers cddb_daemon ;
|
||||
SubInclude HAIKU_TOP src servers debug ;
|
||||
SubInclude HAIKU_TOP src servers index ;
|
||||
SubInclude HAIKU_TOP src servers input ;
|
||||
SubInclude HAIKU_TOP src servers keystore ;
|
||||
SubInclude HAIKU_TOP src servers mail ;
|
||||
SubInclude HAIKU_TOP src servers media ;
|
||||
SubInclude HAIKU_TOP src servers media_addon ;
|
||||
|
||||
9
src/servers/keystore/Jamfile
Normal file
9
src/servers/keystore/Jamfile
Normal file
@ -0,0 +1,9 @@
|
||||
SubDir HAIKU_TOP src servers keystore ;
|
||||
|
||||
UsePrivateHeaders app ;
|
||||
|
||||
Server keystore_server :
|
||||
KeyStoreServer.cpp
|
||||
: be $(TARGET_LIBSTDC++)
|
||||
: keystore_server.rdef
|
||||
;
|
||||
379
src/servers/keystore/KeyStoreServer.cpp
Normal file
379
src/servers/keystore/KeyStoreServer.cpp
Normal file
@ -0,0 +1,379 @@
|
||||
/*
|
||||
* Copyright 2012, Michael Lotz, mmlr@mlotz.ch. All Rights Reserved.
|
||||
* Distributed under the terms of the MIT License.
|
||||
*/
|
||||
|
||||
|
||||
#include "KeyStoreServer.h"
|
||||
|
||||
#include <KeyStoreDefs.h>
|
||||
|
||||
#include <new>
|
||||
|
||||
#include <stdio.h>
|
||||
|
||||
|
||||
using namespace BPrivate;
|
||||
|
||||
|
||||
KeyStoreServer::KeyStoreServer()
|
||||
:
|
||||
BApplication(kKeyStoreServerSignature)
|
||||
{
|
||||
_InitKeyStoreDatabase();
|
||||
}
|
||||
|
||||
|
||||
KeyStoreServer::~KeyStoreServer()
|
||||
{
|
||||
}
|
||||
|
||||
|
||||
void
|
||||
KeyStoreServer::MessageReceived(BMessage* message)
|
||||
{
|
||||
switch (message->what) {
|
||||
case KEY_STORE_GET_KEY:
|
||||
{
|
||||
BMessage reply(KEY_STORE_RESULT);
|
||||
reply.AddInt32("result", B_NOT_ALLOWED);
|
||||
message->SendReply(&reply);
|
||||
break;
|
||||
}
|
||||
|
||||
case KEY_STORE_GET_NEXT_KEY:
|
||||
{
|
||||
BMessage reply(KEY_STORE_RESULT);
|
||||
reply.AddInt32("result", B_ENTRY_NOT_FOUND);
|
||||
message->SendReply(&reply);
|
||||
break;
|
||||
}
|
||||
|
||||
default:
|
||||
{
|
||||
printf("unknown message received: %" B_PRIu32 " \"%.4s\"\n",
|
||||
message->what, (const char*)&message->what);
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
#if 0
|
||||
// Get thread info that contains our team ID for replying.
|
||||
thread_info threadInfo;
|
||||
status_t result = get_thread_info(find_thread(NULL), &threadInfo);
|
||||
if (result != B_OK)
|
||||
return result;
|
||||
|
||||
while (true) {
|
||||
KMessage message;
|
||||
port_message_info messageInfo;
|
||||
status_t error = message.ReceiveFrom(fRequestPort, -1, &messageInfo);
|
||||
if (error != B_OK)
|
||||
return B_OK;
|
||||
|
||||
bool isRoot = (messageInfo.sender == 0);
|
||||
|
||||
switch (message.What()) {
|
||||
case B_REG_GET_PASSWD_DB:
|
||||
{
|
||||
// lazily build the reply
|
||||
try {
|
||||
if (fPasswdDBReply->What() == 1) {
|
||||
FlatStore store;
|
||||
int32 count = fUserDB->WriteFlatPasswdDB(store);
|
||||
if (fPasswdDBReply->AddInt32("count", count) != B_OK
|
||||
|| fPasswdDBReply->AddData("entries", B_RAW_TYPE,
|
||||
store.Buffer(), store.BufferLength(),
|
||||
false) != B_OK) {
|
||||
error = B_NO_MEMORY;
|
||||
}
|
||||
|
||||
fPasswdDBReply->SetWhat(0);
|
||||
}
|
||||
} catch (...) {
|
||||
error = B_NO_MEMORY;
|
||||
}
|
||||
|
||||
if (error == B_OK) {
|
||||
message.SendReply(fPasswdDBReply, -1, -1, 0, registrarTeam);
|
||||
} else {
|
||||
fPasswdDBReply->SetTo(1);
|
||||
KMessage reply(error);
|
||||
message.SendReply(&reply, -1, -1, 0, registrarTeam);
|
||||
}
|
||||
|
||||
break;
|
||||
}
|
||||
|
||||
case B_REG_GET_GROUP_DB:
|
||||
{
|
||||
// lazily build the reply
|
||||
try {
|
||||
if (fGroupDBReply->What() == 1) {
|
||||
FlatStore store;
|
||||
int32 count = fGroupDB->WriteFlatGroupDB(store);
|
||||
if (fGroupDBReply->AddInt32("count", count) != B_OK
|
||||
|| fGroupDBReply->AddData("entries", B_RAW_TYPE,
|
||||
store.Buffer(), store.BufferLength(),
|
||||
false) != B_OK) {
|
||||
error = B_NO_MEMORY;
|
||||
}
|
||||
|
||||
fGroupDBReply->SetWhat(0);
|
||||
}
|
||||
} catch (...) {
|
||||
error = B_NO_MEMORY;
|
||||
}
|
||||
|
||||
if (error == B_OK) {
|
||||
message.SendReply(fGroupDBReply, -1, -1, 0, registrarTeam);
|
||||
} else {
|
||||
fGroupDBReply->SetTo(1);
|
||||
KMessage reply(error);
|
||||
message.SendReply(&reply, -1, -1, 0, registrarTeam);
|
||||
}
|
||||
|
||||
break;
|
||||
}
|
||||
|
||||
|
||||
case B_REG_GET_SHADOW_PASSWD_DB:
|
||||
{
|
||||
// only root may see the shadow passwd
|
||||
if (!isRoot)
|
||||
error = EPERM;
|
||||
|
||||
// lazily build the reply
|
||||
try {
|
||||
if (error == B_OK && fShadowPwdDBReply->What() == 1) {
|
||||
FlatStore store;
|
||||
int32 count = fUserDB->WriteFlatShadowDB(store);
|
||||
if (fShadowPwdDBReply->AddInt32("count", count) != B_OK
|
||||
|| fShadowPwdDBReply->AddData("entries", B_RAW_TYPE,
|
||||
store.Buffer(), store.BufferLength(),
|
||||
false) != B_OK) {
|
||||
error = B_NO_MEMORY;
|
||||
}
|
||||
|
||||
fShadowPwdDBReply->SetWhat(0);
|
||||
}
|
||||
} catch (...) {
|
||||
error = B_NO_MEMORY;
|
||||
}
|
||||
|
||||
if (error == B_OK) {
|
||||
message.SendReply(fShadowPwdDBReply, -1, -1, 0,
|
||||
registrarTeam);
|
||||
} else {
|
||||
fShadowPwdDBReply->SetTo(1);
|
||||
KMessage reply(error);
|
||||
message.SendReply(&reply, -1, -1, 0, registrarTeam);
|
||||
}
|
||||
|
||||
break;
|
||||
}
|
||||
|
||||
case B_REG_GET_USER:
|
||||
{
|
||||
User* user = NULL;
|
||||
int32 uid;
|
||||
const char* name;
|
||||
|
||||
// find user
|
||||
if (message.FindInt32("uid", &uid) == B_OK) {
|
||||
user = fUserDB->UserByID(uid);
|
||||
} else if (message.FindString("name", &name) == B_OK) {
|
||||
user = fUserDB->UserByName(name);
|
||||
} else {
|
||||
error = B_BAD_VALUE;
|
||||
}
|
||||
|
||||
if (error == B_OK && user == NULL)
|
||||
error = ENOENT;
|
||||
|
||||
bool getShadowPwd = message.GetBool("shadow", false);
|
||||
|
||||
// only root may see the shadow passwd
|
||||
if (error == B_OK && getShadowPwd && !isRoot)
|
||||
error = EPERM;
|
||||
|
||||
// add user to message
|
||||
KMessage reply;
|
||||
if (error == B_OK)
|
||||
error = user->WriteToMessage(reply, getShadowPwd);
|
||||
|
||||
// send reply
|
||||
reply.SetWhat(error);
|
||||
message.SendReply(&reply, -1, -1, 0, registrarTeam);
|
||||
|
||||
break;
|
||||
}
|
||||
|
||||
case B_REG_GET_GROUP:
|
||||
{
|
||||
Group* group = NULL;
|
||||
int32 gid;
|
||||
const char* name;
|
||||
|
||||
// find group
|
||||
if (message.FindInt32("gid", &gid) == B_OK) {
|
||||
group = fGroupDB->GroupByID(gid);
|
||||
} else if (message.FindString("name", &name) == B_OK) {
|
||||
group = fGroupDB->GroupByName(name);
|
||||
} else {
|
||||
error = B_BAD_VALUE;
|
||||
}
|
||||
|
||||
if (error == B_OK && group == NULL)
|
||||
error = ENOENT;
|
||||
|
||||
// add group to message
|
||||
KMessage reply;
|
||||
if (error == B_OK)
|
||||
error = group->WriteToMessage(reply);
|
||||
|
||||
// send reply
|
||||
reply.SetWhat(error);
|
||||
message.SendReply(&reply, -1, -1, 0, registrarTeam);
|
||||
|
||||
break;
|
||||
}
|
||||
|
||||
case B_REG_GET_USER_GROUPS:
|
||||
{
|
||||
// get user name
|
||||
const char* name;
|
||||
int32 maxCount;
|
||||
if (message.FindString("name", &name) != B_OK
|
||||
|| message.FindInt32("max count", &maxCount) != B_OK
|
||||
|| maxCount <= 0) {
|
||||
error = B_BAD_VALUE;
|
||||
}
|
||||
|
||||
// get groups
|
||||
gid_t groups[NGROUPS_MAX + 1];
|
||||
int32 count = 0;
|
||||
if (error == B_OK) {
|
||||
maxCount = min_c(maxCount, NGROUPS_MAX + 1);
|
||||
count = fGroupDB->GetUserGroups(name, groups, maxCount);
|
||||
}
|
||||
|
||||
// add groups to message
|
||||
KMessage reply;
|
||||
if (error == B_OK) {
|
||||
if (reply.AddInt32("count", count) != B_OK
|
||||
|| reply.AddData("groups", B_INT32_TYPE,
|
||||
groups, min_c(maxCount, count) * sizeof(gid_t),
|
||||
false) != B_OK) {
|
||||
error = B_NO_MEMORY;
|
||||
}
|
||||
}
|
||||
|
||||
// send reply
|
||||
reply.SetWhat(error);
|
||||
message.SendReply(&reply, -1, -1, 0, registrarTeam);
|
||||
|
||||
break;
|
||||
}
|
||||
|
||||
case B_REG_UPDATE_USER:
|
||||
{
|
||||
// find user
|
||||
User* user = NULL;
|
||||
int32 uid;
|
||||
const char* name;
|
||||
|
||||
if (message.FindInt32("uid", &uid) == B_OK) {
|
||||
user = fUserDB->UserByID(uid);
|
||||
} else if (message.FindString("name", &name) == B_OK) {
|
||||
user = fUserDB->UserByName(name);
|
||||
} else {
|
||||
error = B_BAD_VALUE;
|
||||
}
|
||||
|
||||
// only can change anything
|
||||
if (error == B_OK && !isRoot)
|
||||
error = EPERM;
|
||||
|
||||
// check addUser vs. existing user
|
||||
bool addUser = message.GetBool("add user", false);
|
||||
if (error == B_OK) {
|
||||
if (addUser) {
|
||||
if (user != NULL)
|
||||
error = EEXIST;
|
||||
} else if (user == NULL)
|
||||
error = ENOENT;
|
||||
}
|
||||
|
||||
// apply all changes
|
||||
if (error == B_OK) {
|
||||
// clone the user object and update it from the message
|
||||
User* oldUser = user;
|
||||
user = NULL;
|
||||
try {
|
||||
user = (oldUser != NULL ? new User(*oldUser)
|
||||
: new User);
|
||||
user->UpdateFromMessage(message);
|
||||
|
||||
// uid and name should remain the same
|
||||
if (oldUser != NULL) {
|
||||
if (oldUser->UID() != user->UID()
|
||||
|| oldUser->Name() != user->Name()) {
|
||||
error = B_BAD_VALUE;
|
||||
}
|
||||
}
|
||||
|
||||
// replace the old user and write DBs to disk
|
||||
if (error == B_OK) {
|
||||
fUserDB->AddUser(user);
|
||||
fUserDB->WriteToDisk();
|
||||
fPasswdDBReply->SetTo(1);
|
||||
fShadowPwdDBReply->SetTo(1);
|
||||
}
|
||||
} catch (...) {
|
||||
error = B_NO_MEMORY;
|
||||
}
|
||||
|
||||
if (error == B_OK)
|
||||
delete oldUser;
|
||||
else
|
||||
delete user;
|
||||
}
|
||||
|
||||
// send reply
|
||||
KMessage reply;
|
||||
reply.SetWhat(error);
|
||||
message.SendReply(&reply, -1, -1, 0, registrarTeam);
|
||||
|
||||
break;
|
||||
}
|
||||
case B_REG_UPDATE_GROUP:
|
||||
debug_printf("B_REG_UPDATE_GROUP done: currently unsupported!\n");
|
||||
break;
|
||||
default:
|
||||
debug_printf("REG: invalid message: %lu\n", message.What());
|
||||
|
||||
}
|
||||
}
|
||||
#endif
|
||||
}
|
||||
|
||||
|
||||
status_t
|
||||
KeyStoreServer::_InitKeyStoreDatabase()
|
||||
{
|
||||
return B_ERROR;
|
||||
}
|
||||
|
||||
|
||||
int
|
||||
main(int argc, char* argv[])
|
||||
{
|
||||
KeyStoreServer* app = new(std::nothrow) KeyStoreServer();
|
||||
if (app == NULL)
|
||||
return 1;
|
||||
|
||||
app->Run();
|
||||
delete app;
|
||||
return 0;
|
||||
}
|
||||
24
src/servers/keystore/KeyStoreServer.h
Normal file
24
src/servers/keystore/KeyStoreServer.h
Normal file
@ -0,0 +1,24 @@
|
||||
/*
|
||||
* Copyright 2012, Michael Lotz, mmlr@mlotz.ch. All Rights Reserved.
|
||||
* Distributed under the terms of the MIT License.
|
||||
*/
|
||||
#ifndef _KEY_STORE_SERVER_H
|
||||
#define _KEY_STORE_SERVER_H
|
||||
|
||||
|
||||
#include <Application.h>
|
||||
|
||||
|
||||
class KeyStoreServer : public BApplication {
|
||||
public:
|
||||
KeyStoreServer();
|
||||
virtual ~KeyStoreServer();
|
||||
|
||||
virtual void MessageReceived(BMessage* message);
|
||||
|
||||
private:
|
||||
status_t _InitKeyStoreDatabase();
|
||||
};
|
||||
|
||||
|
||||
#endif // _KEY_STORE_SERVER_H
|
||||
13
src/servers/keystore/keystore_server.rdef
Normal file
13
src/servers/keystore/keystore_server.rdef
Normal file
@ -0,0 +1,13 @@
|
||||
resource app_signature "application/x-vnd.Haiku-keystore_server";
|
||||
|
||||
resource app_flags B_EXCLUSIVE_LAUNCH;
|
||||
|
||||
resource app_version {
|
||||
major = 1,
|
||||
middle = 0,
|
||||
minor = 0,
|
||||
variety = B_APPV_ALPHA,
|
||||
internal = 0,
|
||||
short_info = "keystore_server",
|
||||
long_info = "keystore_server ©2012 Haiku, Inc."
|
||||
};
|
||||
Loading…
Reference in New Issue
Block a user