aboutsummaryrefslogtreecommitdiff
path: root/src/extlib.c
blob: 4ee5744291e1cfbae8a529693a4d5dd6c34ca9eb (plain) (blame)
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
/* extlib.c
 *
 * Copyright 2024 axtlos <axtlos@disroot.org>
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Lesser General Public License as published by
 * the Free Software Foundation, version 3.
 *
 * This program 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 Lesser General Public License
 * along with this program. If not, see <https://www.gnu.org/licenses/>.
 *
 * SPDX-License-Identifier: BSD-3-Clause
 */


#define _XOPEN_SOURCE 500
#define USE_SECURE_MEM
#define __STDC_WANT_LIB_EXT1__ 1
#include <ftw.h>

#include "extlib.h"
#include <stdlib.h>
#include <stdio.h>
#include <string.h>

#undef free
#undef malloc

void
free_secure(void **__ptr, size_t ptrlen)
{
  memset_s (*__ptr, ptrlen+1, 0, ptrlen);
  free (*__ptr);
  *__ptr = NULL;
  return;
}

void *
malloc_secure (size_t len)
{
  void *mem = calloc (1, len);
  return mem;
}

int
memvcmp (void *str,
	 char val,
	 size_t n)
{
  char str2[n];
  memset (str2, val, n);
  return memcmp (str, str2, n);
}

size_t
fcopy (FILE *src, FILE *dst)
{
  char buffer[BUFSIZ];
  size_t n, copied = 0;

  if (src == NULL || dst == NULL)
    return -1;

  while ((n = fread (buffer, sizeof (char), sizeof (buffer), src)) > 0)
  {
    if (fwrite (buffer, sizeof (char), n, dst) != n) {
      fprintf (stderr, "Failed to copy data");
      return -1;
    }
    fflush (dst);
    copied += n;
  }
  return copied;
}

int
rmfile(const char *fpath,
       const struct stat *sb,
       int typeflag,
       struct FTW *ftwbuf)
{
  int err = remove (fpath);
  if (err < 0) return err;
  return 0;
}

int
rrmdir(char *pathname)
{
  int err = nftw (pathname, rmfile, 10, FTW_DEPTH|FTW_MOUNT|FTW_PHYS);
  if (err < 0) return err;
  return 0;
}