⛏️ index : haiku.git

/*
 * Copyright 2012 Haiku, Inc. All rights reserved.
 * Distributed under the terms of the MIT License.
 *
 * Authors:
 *		Paweł Dziepak, pdziepak@quarnos.org
 */


#include "RPCAuth.h"

#include <string.h>
#include <stdlib.h>
#include <unistd.h>

#include <AutoDeleter.h>
#include <SupportDefs.h>
#include <util/kernel_cpp.h>

#include "RPCDefs.h"


using namespace RPC;


Auth::Auth()
{
}


const Auth*
Auth::CreateNone()
{
	Auth* auth = new(std::nothrow) Auth;
	if (auth == NULL)
		return NULL;

	auth->fStream.AddInt(AUTH_NONE);
	auth->fStream.AddOpaque(NULL, 0);
	if (auth->fStream.Error() != B_OK) {
		delete auth;
		return NULL;
	}

	return auth;
}


const Auth*
Auth::CreateSys(uid_t uid, gid_t gid)
{
	Auth* auth = new(std::nothrow) Auth;
	if (auth == NULL)
		return NULL;
	ObjectDeleter<Auth> authDeleter(auth);

	XDR::WriteStream xdr;
	xdr.AddUInt(time(NULL));

	char hostname[255];
	if (gethostname(hostname, 255) < 0)
		strcpy(hostname, "unknown");
	xdr.AddString(hostname, 255);

	xdr.AddUInt(uid);
	xdr.AddUInt(gid);

	int count = getgroups(0, NULL);
	if (count < B_OK)
		return NULL;
	gid_t* groups = (gid_t*)malloc(count * sizeof(gid_t));
	if (groups == NULL)
		return NULL;

	int len = getgroups(count, groups);
	if (len > 0) {
		len = min_c(len, 16);
		xdr.AddUInt(len);
		for (int i = 0; i < len; i++)
			xdr.AddUInt((uint32)groups[i]);
	} else
		xdr.AddUInt(0);
	free(groups);
	if (xdr.Error() != B_OK)
		return NULL;

	auth->fStream.AddInt(AUTH_SYS);
	auth->fStream.AddOpaque(xdr);
	if (auth->fStream.Error() != B_OK)
		return NULL;

	authDeleter.Detach();
	return auth;
}