1/*-
2 * Copyright (c) 2015 Jilles Tjoelker
3 * All rights reserved.
4 *
5 * Redistribution and use in source and binary forms, with or without
6 * modification, are permitted provided that the following conditions
7 * are met:
8 * 1. Redistributions of source code must retain the above copyright
9 *    notice, this list of conditions and the following disclaimer.
10 * 2. Redistributions in binary form must reproduce the above copyright
11 *    notice, this list of conditions and the following disclaimer in the
12 *    documentation and/or other materials provided with the distribution.
13 *
14 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
15 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
16 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
17 * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
18 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
19 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
20 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
21 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
22 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
23 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
24 * SUCH DAMAGE.
25 */
26
27#include <sys/cdefs.h>
28#include <sys/stat.h>
29
30#include <errno.h>
31#include <fcntl.h>
32#include <time.h>
33
34#ifndef	UTIME_NOW
35#define	UTIME_NOW	-1
36#define	UTIME_OMIT	-2
37#endif
38
39int
40utimensat(int fd, const char *path, const struct timespec times[2], int flag)
41{
42	struct timeval now, tv[2], *tvp;
43	struct stat sb;
44	int osreldate;
45
46	if ((flag & ~AT_SYMLINK_NOFOLLOW) != 0) {
47		errno = EINVAL;
48		return (-1);
49	}
50	if (times == NULL || (times[0].tv_nsec == UTIME_NOW &&
51	    times[1].tv_nsec == UTIME_NOW))
52		tvp = NULL;
53	else if (times[0].tv_nsec == UTIME_OMIT &&
54	    times[1].tv_nsec == UTIME_OMIT)
55		return (0);
56	else {
57		if ((times[0].tv_nsec < 0 || times[0].tv_nsec > 999999999) &&
58		    times[0].tv_nsec != UTIME_NOW &&
59		    times[0].tv_nsec != UTIME_OMIT) {
60			errno = EINVAL;
61			return (-1);
62		}
63		if ((times[1].tv_nsec < 0 || times[1].tv_nsec > 999999999) &&
64		    times[1].tv_nsec != UTIME_NOW &&
65		    times[1].tv_nsec != UTIME_OMIT) {
66			errno = EINVAL;
67			return (-1);
68		}
69		tv[0].tv_sec = times[0].tv_sec;
70		tv[0].tv_usec = times[0].tv_nsec / 1000;
71		tv[1].tv_sec = times[1].tv_sec;
72		tv[1].tv_usec = times[1].tv_nsec / 1000;
73		tvp = tv;
74		if (times[0].tv_nsec == UTIME_OMIT ||
75		    times[1].tv_nsec == UTIME_OMIT) {
76			if (fstatat(fd, path, &sb, flag) == -1)
77				return (-1);
78			if (times[0].tv_nsec == UTIME_OMIT) {
79				tv[0].tv_sec = sb.st_atim.tv_sec;
80				tv[0].tv_usec = sb.st_atim.tv_nsec / 1000;
81			}
82			if (times[1].tv_nsec == UTIME_OMIT) {
83				tv[1].tv_sec = sb.st_mtim.tv_sec;
84				tv[1].tv_usec = sb.st_mtim.tv_nsec / 1000;
85			}
86		}
87		if (times[0].tv_nsec == UTIME_NOW ||
88		    times[1].tv_nsec == UTIME_NOW) {
89			if (gettimeofday(&now, NULL) == -1)
90				return (-1);
91			if (times[0].tv_nsec == UTIME_NOW)
92				tv[0] = now;
93			if (times[1].tv_nsec == UTIME_NOW)
94				tv[1] = now;
95		}
96	}
97	if ((flag & AT_SYMLINK_NOFOLLOW) == 0)
98		return (futimesat(fd, path, tvp));
99	else if ((flag & AT_SYMLINK_NOFOLLOW) != 0 &&
100	    (fd == AT_FDCWD || path[0] == '/'))
101		return (lutimes(path, tvp));
102	else {
103		errno = ENOTSUP;
104		return (-1);
105	}
106}
107