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
|
/* ISC license. */
#include <string.h>
#include <unistd.h>
#include <errno.h>
#include <skalibs/uint64.h>
#include <skalibs/posixplz.h>
#include <skalibs/envexec.h>
#include <skalibs/tai.h>
#include <skalibs/djbunix.h>
#include <skalibs/unix-transactional.h>
#include <s6-rc/config.h>
#include <s6-rc/s6rc.h>
#define USAGE "s6-rc-set-copy [ -v verbosity ] [ -r repo ] [ -f ] srcset dstset"
#define dieusage() strerr_dieusage(100, USAGE)
enum golb_e
{
GOLB_FORCE = 0x01
} ;
enum gola_e
{
GOLA_VERBOSITY,
GOLA_REPODIR,
GOLA_N
} ;
static gol_bool const rgolb[] =
{
{ .so = 'f', .lo = "force", .clear = 0, .set = GOLB_FORCE }
} ;
static gol_arg const rgola[] =
{
{ .so = 'v', .lo = "verbosity", .i = GOLA_VERBOSITY },
{ .so = 'r', .lo = "repodir", .i = GOLA_REPODIR }
} ;
static uint64_t wgolb = 0 ;
static void cleanup (char const *fn)
{
int e = errno ;
rm_rf(fn) ;
errno = e ;
}
static inline void docopy (char const *repo, char const *srcname, char const *dstname)
{
size_t repolen = strlen(repo) ;
size_t srclen = strlen(srcname) ;
size_t dstlen = strlen(dstname) ;
size_t r ;
char src[repolen + srclen + 10] ;
char dst[repolen + dstlen + 10] ;
char realsrc[repolen + srclen + 18] ;
char realdst[repolen + dstlen + 18] ;
char olddst[repolen + dstlen + 18] ;
memcpy(src, repo, repolen) ;
memcpy(src + repolen, "/sources/", 9) ;
memcpy(src + repolen + 9, srcname, srclen + 1) ;
memcpy(dst, src, repolen + 9) ;
memcpy(dst + repolen + 9, dstname, dstlen + 1) ;
memcpy(olddst, dst, repolen + 9) ;
olddst[repolen + 9] = 0 ;
if (access(dst, F_OK) == -1)
{
if (errno != ENOENT) strerr_diefu2sys(111, "access ", dst) ;
}
else if (!(wgolb & GOLB_FORCE))
strerr_dief4x(1, "set ", dstname, " already exists in repository ", repo) ;
memcpy(realsrc, src, repolen + 9) ;
r = readlink(src, realsrc + repolen + 9, srclen + 9) ;
if (r == -1) strerr_diefu2sys(111, "readlink ", src) ;
if (r != srclen + 8) strerr_dief3x(102, "symlink ", src, "doesn't point to a valid name") ;
realsrc[repolen + srclen + 17] = 0 ;
memcpy(realdst, dst, repolen + 9) ;
realdst[repolen + 9] = '.' ;
memcpy(realdst + repolen + 10, dstname, dstlen) ;
memcpy(realdst + repolen + 10 + dstlen, ":XXXXXX", 8) ;
if (mkntemp(realdst) == -1) strerr_diefu2sys(111, "mkntemp ", realdst) ;
if (!hiercopy(realsrc, realdst))
{
cleanup(realdst) ;
strerr_diefu4sys(111, "copy ", realsrc, " to ", realdst) ;
}
if (!atomic_symlink4(realdst + repolen + 9, dst, olddst + repolen + 9, dstlen + 9))
{
cleanup(realdst) ;
strerr_diefu4sys(111, "symlink ", realdst + repolen + 9, " to ", dst) ;
}
if (olddst[repolen + 9]) cleanup(olddst) ;
}
int main (int argc, char const *const *argv)
{
int fdlock ;
unsigned int verbosity = 1 ;
char const *wgola[GOLA_N] = { 0 } ;
unsigned int golc ;
PROG = "s6-rc-set-copy" ;
wgola[GOLA_REPODIR] = S6RC_REPODIR ;
golc = GOL_main(argc, argv, rgolb, rgola, &wgolb, wgola) ;
argc -= golc ; argv += golc ;
if (wgola[GOLA_VERBOSITY] && !uint0_scan(wgola[GOLA_VERBOSITY], &verbosity))
strerr_dief1x(100, "verbosity needs to be an unsigned integer") ;
if (argc < 2) dieusage() ;
if (!strcmp(argv[0], argv[1]))
strerr_dief1x(100, "source and destination sets are the same") ;
s6rc_repo_sanitize_setname(argv[0]) ;
s6rc_repo_sanitize_setname(argv[1]) ;
tain_now_g() ;
fdlock = s6rc_repo_lock(wgola[GOLA_REPODIR], 1) ;
if (fdlock == -1) strerr_diefu2sys(111, "lock ", wgola[GOLA_REPODIR]) ;
docopy(wgola[GOLA_REPODIR], argv[0], argv[1]) ;
_exit(0) ;
}
|
