/*
 * Copyright (c) 2020-2022 The strace developers.
 * All rights reserved.
 *
 * SPDX-License-Identifier: GPL-2.0-or-later
 */

#include "tests.h"
#include "xmalloc.h"
#include <unistd.h>

char *secontext_full_fd(int) ATTRIBUTE_MALLOC;
char *secontext_full_file(const char *, bool) ATTRIBUTE_MALLOC;
char *secontext_full_pid(pid_t) ATTRIBUTE_MALLOC;

char *secontext_short_fd(int) ATTRIBUTE_MALLOC;
char *secontext_short_file(const char *, bool) ATTRIBUTE_MALLOC;
char *secontext_short_pid(pid_t) ATTRIBUTE_MALLOC;

enum secontext_field {
	SECONTEXT_USER,
	SECONTEXT_ROLE,
	SECONTEXT_TYPE
};

#if defined TEST_SECONTEXT && defined HAVE_SELINUX_RUNTIME

/**
 * Parse a SELinux context string and return a specified field, duplicated
 * in a separate string.  The caller is responsible for freeing the memory
 * pointed by the returned value.
 */
char *get_secontext_field(const char *full_context, enum secontext_field field);

char *get_secontext_field_fd(int fd, enum secontext_field field);
char *get_secontext_field_file(const char *file, enum secontext_field field);

void reset_secontext_file(const char *file);

void update_secontext_field(const char *file, enum secontext_field field,
			    const char *newvalue);

# ifdef PRINT_SECONTEXT_FULL

#  ifdef PRINT_SECONTEXT_MISMATCH
#   define SECONTEXT_FILE(filename)	secontext_full_file(filename, true)
#  else
#   define SECONTEXT_FILE(filename)	secontext_full_file(filename, false)
#  endif
#  define SECONTEXT_FD(fd)		secontext_full_fd(fd)
#  define SECONTEXT_PID(pid)		secontext_full_pid(pid)

# else

#  ifdef PRINT_SECONTEXT_MISMATCH
#   define SECONTEXT_FILE(filename)	secontext_short_file(filename, true)
#  else
#   define SECONTEXT_FILE(filename)	secontext_short_file(filename, false)
#  endif
#  define SECONTEXT_FD(fd)		secontext_short_fd(fd)
#  define SECONTEXT_PID(pid)		secontext_short_pid(pid)

# endif

#else

static inline char *
get_secontext_field(const char *ctx, enum secontext_field field)
{
	return NULL;
}
static inline char *
get_secontext_field_fd(int fd, enum secontext_field field)
{
	return NULL;
}

static inline char *
get_secontext_field_file(const char *file, enum secontext_field field)
{
	return NULL;
}

static inline void
reset_secontext_file(const char *file)
{
}

static inline void
update_secontext_field(const char *file, enum secontext_field field,
		       const char *newvalue)
{
}

# define SECONTEXT_FD(fd)			xstrdup("")
# define SECONTEXT_FILE(filename)		xstrdup("")
# define SECONTEXT_PID(pid)			xstrdup("")

#endif

#define SECONTEXT_PID_MY()		SECONTEXT_PID(getpid())
