summaryrefslogtreecommitdiff
path: root/src/lib/grok/shm.cc
blob: 938b61834a9114d85c1310f339e193bfa8177c43 (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
/*
    Copyright (C) 2023 Grok Image Compression Inc.

    This file is part of DCP-o-matic.

    DCP-o-matic is free software; you can redistribute it and/or modify
    it under the terms of the GNU General Public License as published by
    the Free Software Foundation; either version 2 of the License, or
    (at your option) any later version.

    DCP-o-matic is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    GNU General Public License for more details.

    You should have received a copy of the GNU General Public License
    along with DCP-o-matic.  If not, see <http://www.gnu.org/licenses/>.

*/


#include "context.h"
#include "messenger.h"
#include "shm.h"


bool
grk_plugin::init_shm(char const* name, size_t len, int* shm_fd, char** buffer)
{
	*shm_fd = shm_open(name, O_CREAT | O_RDWR, 0666);

	if (*shm_fd < 0) {
		getGrokLogger()->error("Error opening shared memory: %s", strerror(errno));
		return false;
	}

	int rc = ftruncate(*shm_fd, sizeof(char) * len);

	if (rc) {
		getGrokLogger()->error("Error truncating shared memory: %s", strerror(errno));
		if (close(*shm_fd)) {
			getGrokLogger()->error("Error closing shared memory: %s", strerror(errno));
		}
		rc = shm_unlink(name);
		if (rc && errno != ENOENT) {
			getGrokLogger()->error("Error unlinking shared memory: %s", strerror(errno));
		}
		return false;
	}

	*buffer = static_cast<char*>(mmap(0, len, PROT_WRITE, MAP_SHARED, *shm_fd, 0));

	if (!*buffer) {
		getGrokLogger()->error("Error mapping shared memory: %s", strerror(errno));
		if (close(*shm_fd)) {
			getGrokLogger()->error("Error closing shared memory: %s", strerror(errno));
		}
		rc = shm_unlink(name);
		if (rc && errno != ENOENT) {
			getGrokLogger()->error("Error unlinking shared memory: %s", strerror(errno));
		}
	}

	return *buffer != nullptr;
}


bool
grk_plugin::deinit_shm(char const* name, size_t len, int &shm_fd, char** buffer)
{
	if (!*buffer || !shm_fd) {
		return true;
	}

	int rc = munmap(*buffer, len);
	*buffer = nullptr;

	if (rc) {
		getGrokLogger()->error("Error unmapping shared memory %s: %s", name, strerror(errno));
	}

	rc = close(shm_fd);
	shm_fd = 0;

	if (rc) {
		getGrokLogger()->error("Error closing shared memory %s: %s", name, strerror(errno));
	}

	rc = shm_unlink(name);

	if (rc && errno != ENOENT) {
		getGrokLogger()->error("Error unlinking shared memory %s : %s", name, strerror(errno));
	}

	return true;
}