summaryrefslogtreecommitdiff
path: root/src/helper/mdso_create_output.c
blob: bb1cfe967d0caf22165ad1aeae500a0d59485c0f (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
/****************************************************************/
/*  mdso: midipix dso scavenger                                 */
/*  Copyright (C) 2015--2021  SysDeer Technologies, LLC         */
/*  Released under GPLv2 and GPLv3; see COPYING.MDSO.           */
/****************************************************************/

#include <stdint.h>
#include <stdlib.h>
#include <stdio.h>
#include <stddef.h>
#include <unistd.h>
#include <fcntl.h>
#include <sys/mman.h>

#include <mdso/mdso.h>
#include "mdso_driver_impl.h"
#include "mdso_errinfo_impl.h"

static int mdso_create_output(
	const struct mdso_driver_ctx *	dctx,
	const char *			name)
{
	int	fddst;
	int	fdout;

	fddst = mdso_driver_fddst(dctx);

	if ((fdout = openat(fddst,name,
			O_CREAT|O_TRUNC|O_RDWR|O_NOCTTY|O_NOFOLLOW,
			S_IRUSR|S_IWUSR|S_IRGRP|S_IROTH)) < 0)
		return MDSO_SYSTEM_ERROR(dctx);

	return fdout;
}

static int mdso_map_output(
	const struct mdso_driver_ctx *	dctx,
	struct mdso_object *		obj,
	int				fd)
{
	void * addr;

	if (ftruncate(fd,obj->size)) {
		close(fd);
		return MDSO_SYSTEM_ERROR(dctx);
	}

	addr = mmap(
		0,obj->size,
		PROT_WRITE,MAP_SHARED,
		fd,0);

	close(fd);

	if (addr == MAP_FAILED)
		return MDSO_SYSTEM_ERROR(dctx);

	obj->addr = addr;

	return 0;
}

static int mdso_create_mapped_output(
	const struct mdso_driver_ctx *	dctx,
	struct mdso_object *		obj)
{
	int fd;

	if ((fd = mdso_create_output(dctx,obj->name)) < 0)
		return MDSO_NESTED_ERROR(dctx);

	if (mdso_map_output(dctx,obj,fd) < 0)
		return MDSO_NESTED_ERROR(dctx);

	return 0;
}

int mdso_create_asmsrc(
	const struct mdso_driver_ctx *	dctx,
	const char *			asmname)
{
	if (dctx->cctx->dstdir)
		return mdso_create_output(dctx,asmname);

	else if (dctx->cctx->drvflags & MDSO_DRIVER_GENERATE_OBJECTS)
		return mdso_create_output(dctx,asmname);

	else
		return mdso_driver_fdout(dctx);
}

int mdso_create_object(
	const struct mdso_driver_ctx *	dctx,
	struct mdso_object *		obj)
{
	return mdso_create_mapped_output(dctx,obj);
}

int mdso_create_archive(
	const struct mdso_driver_ctx *	dctx,
	struct mdso_object *		obj)
{
	return mdso_create_mapped_output(dctx,obj);
}