roken: Add mkdtemp()

This commit is contained in:
Nicolas Williams
2020-01-24 00:24:52 -06:00
parent 77619f245d
commit a7359d6898
6 changed files with 96 additions and 0 deletions

View File

@@ -195,6 +195,7 @@ AC_CHECK_FUNCS([ \
getspnam \
issetugid \
memmem \
mkdtemp \
mkostemp \
on_exit \
poll \

View File

@@ -116,6 +116,7 @@ libroken_la_SOURCES = \
memmem.c \
mini_inetd.c \
mkdir.c \
mkdtemp.c \
mkostemp.c \
net_read.c \
net_write.c \

View File

@@ -78,6 +78,7 @@ libroken_la_OBJS = \
$(OBJ)\memset_s.obj \
$(OBJ)\mkdir.obj \
$(OBJ)\mini_inetd.obj \
$(OBJ)\mkdtemp.obj \
$(OBJ)\mkstemp.obj \
$(OBJ)\mkostemp.obj \
$(OBJ)\net_read.obj \

87
lib/roken/mkdtemp.c Normal file
View File

@@ -0,0 +1,87 @@
/*
* Copyright (c) 2020 Kungliga Tekniska Högskolan
* (Royal Institute of Technology, Stockholm, Sweden).
* All rights reserved.
*
* Redistribution and use in source and binary forms, with or without
* modification, are permitted provided that the following conditions
* are met:
*
* 1. Redistributions of source code must retain the above copyright
* notice, this list of conditions and the following disclaimer.
*
* 2. Redistributions in binary form must reproduce the above copyright
* notice, this list of conditions and the following disclaimer in the
* documentation and/or other materials provided with the distribution.
*
* 3. Neither the name of the Institute nor the names of its contributors
* may be used to endorse or promote products derived from this software
* without specific prior written permission.
*
* THIS SOFTWARE IS PROVIDED BY THE INSTITUTE AND CONTRIBUTORS ``AS IS'' AND
* ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
* IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
* ARE DISCLAIMED. IN NO EVENT SHALL THE INSTITUTE OR CONTRIBUTORS BE LIABLE
* FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
* DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
* OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
* HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
* LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
* OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
* SUCH DAMAGE.
*/
#include <config.h>
#include <sys/stat.h>
#include <sys/types.h>
#include <string.h>
#include <errno.h>
#include <roken.h>
#ifndef O_APPEND
#define O_APPEND 0
#endif
#ifndef O_CLOEXEC
#define O_CLOEXEC 0
#endif
#ifndef O_SYNC
#define O_SYNC 0
#endif
#ifndef HAVE_MKDTEMP
ROKEN_LIB_FUNCTION char * ROKEN_LIB_CALL
mkdtemp(char *template)
{
size_t len = strlen(template) - 1;
size_t start, i;
pid_t val = getpid();
for (i = 0; i < len && i < 7 && template[len - i] == 'X'; i++) {
template[len] = '0' + val % 10;
val /= 10;
len--;
if (!val)
val = getpid();
}
if (i < 6) {
errno = EINVAL;
return NULL;
}
start = len - i;
do {
if (mkdir(template, 0700) == 0)
return template;
for (i = start + 1; i < len; i++) {
if (++(template[i]) == '9' + 1)
template[i] = 'a';
if (template[i] <= 'z')
break;
template[i] = 'a';
}
} while(1);
}
#endif

View File

@@ -708,6 +708,11 @@ ROKEN_LIB_FUNCTION int ROKEN_LIB_CALL mkstemp(char *);
ROKEN_LIB_FUNCTION int ROKEN_LIB_CALL mkostemp(char *, int);
#endif
#ifndef HAVE_MKDTEMP
#define mkdtemp rk_mkdtemp
ROKEN_LIB_FUNCTION char * ROKEN_LIB_CALL mkdtemp(char *);
#endif
#ifndef HAVE_CGETENT
#define cgetent rk_cgetent
#define cgetstr rk_cgetstr

View File

@@ -88,6 +88,7 @@ HEIMDAL_ROKEN_2.0 {
rk_memmem;
rk_memset_s;
rk_mkdir;
rk_mkdtemp;
rk_mkostemp;
rk_mkstemp;
rk_parse_flags;