[go: up one dir, main page]

File: glib-helper.c

package info (click to toggle)
s390-tools 2.35.0-1
  • links: PTS
  • area: main
  • in suites: trixie
  • size: 12,220 kB
  • sloc: ansic: 184,236; sh: 12,152; cpp: 4,954; makefile: 2,763; perl: 2,519; asm: 1,085; python: 697; xml: 29
file content (181 lines) | stat: -rw-r--r-- 3,981 bytes parent folder | download | duplicates (2)
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
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
/*
 * Glib convenience functions
 *
 * Copyright IBM Corp. 2022
 *
 * s390-tools is free software; you can redistribute it and/or modify
 * it under the terms of the MIT license. See LICENSE for details.
 */
/* Must be included before any other header */
#include "config.h"

#include <errno.h>
#include <openssl/crypto.h>

#include "libpv/glib-helper.h"

struct __data {
	void *data;
	size_t size;
	GFreeFunc free_func;
};

static void __data_clear_and_free(void *p)
{
	struct __data *ptr = p;

	if (!ptr)
		return;

	if (ptr->data) {
		OPENSSL_cleanse(ptr->data, ptr->size);
		ptr->free_func(ptr->data);
	}
	g_free(ptr);
}

static GBytes *pv_sec_gbytes_new_take_func(void *data, size_t size, GFreeFunc free_func)
{
	struct __data *tmp = g_new(struct __data, 1);

	tmp->data = data;
	tmp->size = size;
	tmp->free_func = free_func;

	return g_bytes_new_with_free_func(data, size, __data_clear_and_free, tmp);
}

GBytes *pv_sec_gbytes_new_take(void *data, size_t size)
{
	return pv_sec_gbytes_new_take_func(data, size, g_free);
}

GBytes *pv_sec_gbytes_new(const void *data, size_t size)
{
	g_autofree void *tmp_data = NULL;

	g_return_val_if_fail(data || size != 0, NULL);

	tmp_data = g_malloc(size);
	memcpy(tmp_data, data, size);
	return pv_sec_gbytes_new_take(g_steal_pointer(&tmp_data), size);
}

int pv_file_seek(FILE *file, long offset, int whence, GError **error)
{
	int cached_errno;
	int ret = fseek(file, offset, whence);

	if (ret) {
		cached_errno = errno;
		g_set_error(error, PV_GLIB_HELPER_ERROR, PV_GLIB_HELPER_FILE_ERROR,
			    "Cannot seek: %s", g_strerror(cached_errno));
	}
	return ret;
}

size_t pv_file_write(FILE *file, const void *ptr, size_t size, GError **error)
{
	int cached_errno;
	size_t n = fwrite(ptr, 1, size, file);

	if (n != size) {
		cached_errno = errno;
		g_set_error(error, PV_GLIB_HELPER_ERROR, PV_GLIB_HELPER_FILE_ERROR,
			    "Cannot write: %s", g_strerror(cached_errno));
	}
	return n;
}

long pv_file_close(FILE *file, GError **error)
{
	int cached_errno;
	int ret = fclose(file);

	if (ret) {
		cached_errno = errno;
		g_set_error(error, PV_GLIB_HELPER_ERROR, PV_GLIB_HELPER_FILE_ERROR,
			    "Cannot close: %s", g_strerror(cached_errno));
	}
	return ret;
}

void pv_auto_close_file(FILE *file)
{
	if (!file)
		return;

	(void)pv_file_close(file, NULL);
}

long pv_file_tell(FILE *file, GError **error)
{
	int cached_errno;
	long n = ftell(file);

	if (n < 0) {
		cached_errno = errno;
		g_set_error(error, PV_GLIB_HELPER_ERROR, PV_GLIB_HELPER_FILE_ERROR,
			    "Cannot tell: %s", g_strerror(cached_errno));
	}
	return n;
}

FILE *pv_file_open(const char *filename, const char *mode, GError **error)
{
	FILE *file = fopen(filename, mode);
	int cached_errno;

	if (!file) {
		cached_errno = errno;
		g_set_error(error, PV_GLIB_HELPER_ERROR, PV_GLIB_HELPER_FILE_ERROR,
			    "Cannot open '%s'. %s", filename, g_strerror(cached_errno));
		return NULL;
	}
	return file;
}

GBytes *pv_file_get_content_as_g_bytes(const char *filename, GError **error)
{
	g_autofree char *data = NULL;
	size_t data_size;

	if (!g_file_get_contents(filename, &data, &data_size, error))
		return NULL;

	return g_bytes_new_take(g_steal_pointer(&data), data_size);
}

GBytes *pv_file_get_content_as_secure_bytes(const char *filename)
{
	g_autoptr(FILE) f = fopen(filename, "rb");
	g_autofree char *data = NULL;
	ssize_t file_size;
	size_t data_size;

	if (!f)
		return NULL;

	fseek(f, 0, SEEK_END);
	file_size = ftell(f);
	if (file_size < 0)
		return NULL;
	data_size = (size_t)file_size;
	fseek(f, 0, SEEK_SET);
	data = g_malloc0(data_size);
	if (data_size != fread(data, 1, data_size, f))
		return NULL;
	return pv_sec_gbytes_new_take(g_steal_pointer(&data), data_size);
}

void *pv_gbytes_memcpy(void *dst, size_t dst_size, GBytes *src, size_t *copied)
{
	size_t src_size;
	const void *src_data = g_bytes_get_data(src, &src_size);

	if (dst_size < src_size)
		return NULL;
	if (copied)
		*copied = src_size;
	return memcpy(dst, src_data, src_size);
}