X-Git-Url: https://git.saurik.com/apple/xnu.git/blobdiff_plain/2d21ac55c334faf3a56e5634905ed6987fc787d4..c18c124eaa464aaaa5549e99e5a70fc9cbb50944:/bsd/kern/kern_time.c?ds=sidebyside

diff --git a/bsd/kern/kern_time.c b/bsd/kern/kern_time.c
index 30b0a651c..cfbd3c99f 100644
--- a/bsd/kern/kern_time.c
+++ b/bsd/kern/kern_time.c
@@ -1,5 +1,5 @@
 /*
- * Copyright (c) 2000-2007 Apple Inc. All rights reserved.
+ * Copyright (c) 2000-2008 Apple Inc. All rights reserved.
  *
  * @APPLE_OSREFERENCE_LICENSE_HEADER_START@
  * 
@@ -74,10 +74,13 @@
 #include <sys/proc_internal.h>
 #include <sys/kauth.h>
 #include <sys/vnode.h>
+#include <sys/time.h>
+#include <sys/priv.h>
 
 #include <sys/mount_internal.h>
 #include <sys/sysproto.h>
 #include <sys/signalvar.h>
+#include <sys/protosw.h> /* for net_uptime2timeval() */
 
 #include <kern/clock.h>
 #include <kern/task.h>
@@ -97,7 +100,7 @@ lck_grp_attr_t	*tz_slock_grp_attr;
 static void		setthetime(
 					struct timeval	*tv);
 
-void time_zone_slock_init(void) __attribute__((section("__TEXT, initcode")));
+void time_zone_slock_init(void);
 
 /* 
  * Time of day and interval timer support.
@@ -113,13 +116,19 @@ int
 gettimeofday(
 __unused	struct proc	*p,
 			struct gettimeofday_args *uap, 
-			register_t *retval)
+			int32_t *retval)
 {
 	int error = 0;
 	struct timezone ltz; /* local copy */
 
-	if (uap->tp)
-		clock_gettimeofday((uint32_t *)&retval[0], (uint32_t *)&retval[1]);
+	if (uap->tp) {
+		clock_sec_t		secs;
+		clock_usec_t	usecs;
+
+		clock_gettimeofday(&secs, &usecs);
+		retval[0] = secs;
+		retval[1] = usecs;
+	}
 	
 	if (uap->tzp) {
 		lck_spin_lock(tz_slock);
@@ -137,30 +146,33 @@ __unused	struct proc	*p,
  */
 /* ARGSUSED */
 int
-settimeofday(__unused struct proc *p, struct settimeofday_args  *uap, __unused register_t *retval)
+settimeofday(__unused struct proc *p, struct settimeofday_args  *uap, __unused int32_t *retval)
 {
 	struct timeval atv;
 	struct timezone atz;
 	int error;
 
+	bzero(&atv, sizeof(atv));
+
 #if CONFIG_MACF
 	error = mac_system_check_settime(kauth_cred_get());
 	if (error)
 		return (error);
 #endif
-#ifndef CONFIG_EMBEDDED
 	if ((error = suser(kauth_cred_get(), &p->p_acflag)))
 		return (error);
-#endif
 	/* Verify all parameters before changing time */
 	if (uap->tv) {
 		if (IS_64BIT_PROCESS(p)) {
-			struct user_timeval user_atv;
-			error = copyin(uap->tv, &user_atv, sizeof(struct user_timeval));
+			struct user64_timeval user_atv;
+			error = copyin(uap->tv, &user_atv, sizeof(user_atv));
 			atv.tv_sec = user_atv.tv_sec;
 			atv.tv_usec = user_atv.tv_usec;
 		} else {
-			error = copyin(uap->tv, &atv, sizeof(struct timeval));
+			struct user32_timeval user_atv;
+			error = copyin(uap->tv, &user_atv, sizeof(user_atv));
+			atv.tv_sec = user_atv.tv_sec;
+			atv.tv_usec = user_atv.tv_usec;
 		}
 		if (error)
 			return (error);
@@ -193,7 +205,7 @@ setthetime(
  */
 /* ARGSUSED */
 int
-adjtime(struct proc *p, struct adjtime_args *uap, __unused register_t *retval)
+adjtime(struct proc *p, struct adjtime_args *uap, __unused int32_t *retval)
 {
 	struct timeval atv;
 	int error;
@@ -203,15 +215,18 @@ adjtime(struct proc *p, struct adjtime_args *uap, __unused register_t *retval)
 	if (error)
 		return (error);
 #endif
-	if ((error = suser(kauth_cred_get(), &p->p_acflag)))
+	if ((error = priv_check_cred(kauth_cred_get(), PRIV_ADJTIME, 0)))
 		return (error);
 	if (IS_64BIT_PROCESS(p)) {
-		struct user_timeval user_atv;
-		error = copyin(uap->delta, &user_atv, sizeof(struct user_timeval));
+		struct user64_timeval user_atv;
+		error = copyin(uap->delta, &user_atv, sizeof(user_atv));
 		atv.tv_sec = user_atv.tv_sec;
 		atv.tv_usec = user_atv.tv_usec;
 	} else {
-		error = copyin(uap->delta, &atv, sizeof(struct timeval));
+		struct user32_timeval user_atv;
+		error = copyin(uap->delta, &user_atv, sizeof(user_atv));
+		atv.tv_sec = user_atv.tv_sec;
+		atv.tv_usec = user_atv.tv_usec;
 	}
 	if (error)
 		return (error);
@@ -219,16 +234,19 @@ adjtime(struct proc *p, struct adjtime_args *uap, __unused register_t *retval)
 	/*
 	 * Compute the total correction and the rate at which to apply it.
 	 */
-	clock_adjtime((int32_t *)&atv.tv_sec, &atv.tv_usec);
+	clock_adjtime(&atv.tv_sec, &atv.tv_usec);
 
 	if (uap->olddelta) {
 		if (IS_64BIT_PROCESS(p)) {
-			struct user_timeval user_atv;
+			struct user64_timeval user_atv;
 			user_atv.tv_sec = atv.tv_sec;
 			user_atv.tv_usec = atv.tv_usec;
-			error = copyout(&user_atv, uap->olddelta, sizeof(struct user_timeval));
+			error = copyout(&user_atv, uap->olddelta, sizeof(user_atv));
 		} else {
-			error = copyout(&atv, uap->olddelta, sizeof(struct timeval));
+			struct user32_timeval user_atv;
+			user_atv.tv_sec = atv.tv_sec;
+			user_atv.tv_usec = atv.tv_usec;
+			error = copyout(&user_atv, uap->olddelta, sizeof(user_atv));
 		}
 	}
 
@@ -267,12 +285,12 @@ inittodr(
 time_t
 boottime_sec(void)
 {
-	uint32_t	sec, nanosec;
-	clock_get_boottime_nanotime(&sec, &nanosec);
-	return (sec);
-}
+	clock_sec_t		secs;
+	clock_nsec_t	nanosecs;
 
-uint64_t tvtoabstime(struct timeval *tvp);
+	clock_get_boottime_nanotime(&secs, &nanosecs);
+	return (secs);
+}
 
 /*
  * Get value of an interval timer.  The process virtual and
@@ -298,13 +316,15 @@ uint64_t tvtoabstime(struct timeval *tvp);
  */
 /* ARGSUSED */
 int
-getitimer(struct proc *p, struct getitimer_args *uap, __unused register_t *retval)
+getitimer(struct proc *p, struct getitimer_args *uap, __unused int32_t *retval)
 {
 	struct itimerval aitv;
 
 	if (uap->which > ITIMER_PROF)
 		return(EINVAL);
 
+	bzero(&aitv, sizeof(aitv));
+
 	proc_spinlock(p);
 	switch (uap->which) {
 
@@ -342,14 +362,19 @@ getitimer(struct proc *p, struct getitimer_args *uap, __unused register_t *retva
 	proc_spinunlock(p);
 
 	if (IS_64BIT_PROCESS(p)) {
-		struct user_itimerval user_itv;
+		struct user64_itimerval user_itv;
 		user_itv.it_interval.tv_sec = aitv.it_interval.tv_sec;
 		user_itv.it_interval.tv_usec = aitv.it_interval.tv_usec;
 		user_itv.it_value.tv_sec = aitv.it_value.tv_sec;
 		user_itv.it_value.tv_usec = aitv.it_value.tv_usec;
-		return (copyout((caddr_t)&user_itv, uap->itv, sizeof (struct user_itimerval)));
+		return (copyout((caddr_t)&user_itv, uap->itv, sizeof (user_itv)));
 	} else {
-		return (copyout((caddr_t)&aitv, uap->itv, sizeof (struct itimerval)));
+		struct user32_itimerval user_itv;
+		user_itv.it_interval.tv_sec = aitv.it_interval.tv_sec;
+		user_itv.it_interval.tv_usec = aitv.it_interval.tv_usec;
+		user_itv.it_value.tv_sec = aitv.it_value.tv_sec;
+		user_itv.it_value.tv_usec = aitv.it_value.tv_usec;
+		return (copyout((caddr_t)&user_itv, uap->itv, sizeof (user_itv)));
 	}
 }
 
@@ -362,26 +387,33 @@ getitimer(struct proc *p, struct getitimer_args *uap, __unused register_t *retva
  */
 /* ARGSUSED */
 int
-setitimer(struct proc *p, struct setitimer_args *uap, register_t *retval)
+setitimer(struct proc *p, struct setitimer_args *uap, int32_t *retval)
 {
 	struct itimerval aitv;
 	user_addr_t itvp;
 	int error;
 
+	bzero(&aitv, sizeof(aitv));
+
 	if (uap->which > ITIMER_PROF)
 		return (EINVAL);
 	if ((itvp = uap->itv)) {
 		if (IS_64BIT_PROCESS(p)) {
-			struct user_itimerval user_itv;
-			if ((error = copyin(itvp, (caddr_t)&user_itv, sizeof (struct user_itimerval))))
+			struct user64_itimerval user_itv;
+			if ((error = copyin(itvp, (caddr_t)&user_itv, sizeof (user_itv))))
 				return (error);
 			aitv.it_interval.tv_sec = user_itv.it_interval.tv_sec;
 			aitv.it_interval.tv_usec = user_itv.it_interval.tv_usec;
 			aitv.it_value.tv_sec = user_itv.it_value.tv_sec;
 			aitv.it_value.tv_usec = user_itv.it_value.tv_usec;
 		} else { 
-			if ((error = copyin(itvp, (caddr_t)&aitv, sizeof (struct itimerval))))
+			struct user32_itimerval user_itv;
+			if ((error = copyin(itvp, (caddr_t)&user_itv, sizeof (user_itv))))
 				return (error);
+			aitv.it_interval.tv_sec = user_itv.it_interval.tv_sec;
+			aitv.it_interval.tv_usec = user_itv.it_interval.tv_usec;
+			aitv.it_value.tv_sec = user_itv.it_value.tv_sec;
+			aitv.it_value.tv_usec = user_itv.it_value.tv_usec;
 		}
 	}
 	if ((uap->itv = uap->oitv) && (error = getitimer(p, (struct getitimer_args *)uap, retval)))
@@ -399,7 +431,8 @@ setitimer(struct proc *p, struct setitimer_args *uap, register_t *retval)
 			microuptime(&p->p_rtime);
 			timevaladd(&p->p_rtime, &aitv.it_value);
 			p->p_realtimer = aitv;
-			if (!thread_call_enter_delayed(p->p_rcall, tvtoabstime(&p->p_rtime)))
+			if (!thread_call_enter_delayed_with_leeway(p->p_rcall, NULL,
+					         tvtoabstime(&p->p_rtime), 0, THREAD_CALL_DELAY_USER_NORMAL))
 				p->p_ractive++;
 		} else  {
 			timerclear(&p->p_rtime);
@@ -612,14 +645,39 @@ void
 microtime(
 	struct timeval	*tvp)
 {
-	clock_get_calendar_microtime((uint32_t *)&tvp->tv_sec, (uint32_t *)&tvp->tv_usec);
+	clock_sec_t		tv_sec;
+	clock_usec_t	tv_usec;
+
+	clock_get_calendar_microtime(&tv_sec, &tv_usec);
+
+	tvp->tv_sec = tv_sec;
+	tvp->tv_usec = tv_usec;
+}
+
+void
+microtime_with_abstime(
+	struct timeval	*tvp, uint64_t *abstime)
+{
+	clock_sec_t		tv_sec;
+	clock_usec_t	tv_usec;
+
+	clock_get_calendar_absolute_and_microtime(&tv_sec, &tv_usec, abstime);
+
+	tvp->tv_sec = tv_sec;
+	tvp->tv_usec = tv_usec;
 }
 
 void
 microuptime(
 	struct timeval	*tvp)
 {
-	clock_get_system_microtime((uint32_t *)&tvp->tv_sec, (uint32_t *)&tvp->tv_usec);
+	clock_sec_t		tv_sec;
+	clock_usec_t	tv_usec;
+
+	clock_get_system_microtime(&tv_sec, &tv_usec);
+
+	tvp->tv_sec = tv_sec;
+	tvp->tv_usec = tv_usec;
 }
 
 /*
@@ -629,14 +687,26 @@ void
 nanotime(
 	struct timespec *tsp)
 {
-	clock_get_calendar_nanotime((uint32_t *)&tsp->tv_sec, (uint32_t *)&tsp->tv_nsec);
+	clock_sec_t		tv_sec;
+	clock_nsec_t	tv_nsec;
+
+	clock_get_calendar_nanotime(&tv_sec, &tv_nsec);
+
+	tsp->tv_sec = tv_sec;
+	tsp->tv_nsec = tv_nsec;
 }
 
 void
 nanouptime(
 	struct timespec *tsp)
 {
-	clock_get_system_nanotime((uint32_t *)&tsp->tv_sec, (uint32_t *)&tsp->tv_nsec);
+	clock_sec_t		tv_sec;
+	clock_nsec_t	tv_nsec;
+
+	clock_get_system_nanotime(&tv_sec, &tv_nsec);
+
+	tsp->tv_sec = tv_sec;
+	tsp->tv_nsec = tv_nsec;
 }
 
 uint64_t
@@ -652,6 +722,86 @@ tvtoabstime(
 
 	return (result + usresult);
 }
+
+#if NETWORKING
+/*
+ * ratecheck(): simple time-based rate-limit checking.
+ */
+int
+ratecheck(struct timeval *lasttime, const struct timeval *mininterval)
+{
+	struct timeval tv, delta;
+	int rv = 0;
+
+	net_uptime2timeval(&tv);
+	delta = tv;
+	timevalsub(&delta, lasttime);
+
+	/*
+	 * check for 0,0 is so that the message will be seen at least once,
+	 * even if interval is huge.
+	 */
+	if (timevalcmp(&delta, mininterval, >=) ||
+	    (lasttime->tv_sec == 0 && lasttime->tv_usec == 0)) {
+		*lasttime = tv;
+		rv = 1;
+	}
+
+	return (rv);
+}
+
+/*
+ * ppsratecheck(): packets (or events) per second limitation.
+ */
+int
+ppsratecheck(struct timeval *lasttime, int *curpps, int maxpps)
+{
+	struct timeval tv, delta;
+	int rv;
+
+	net_uptime2timeval(&tv);
+
+	timersub(&tv, lasttime, &delta);
+
+	/*
+	 * Check for 0,0 so that the message will be seen at least once.
+	 * If more than one second has passed since the last update of
+	 * lasttime, reset the counter.
+	 *
+	 * we do increment *curpps even in *curpps < maxpps case, as some may
+	 * try to use *curpps for stat purposes as well.
+	 */
+	if ((lasttime->tv_sec == 0 && lasttime->tv_usec == 0) ||
+	    delta.tv_sec >= 1) {
+		*lasttime = tv;
+		*curpps = 0;
+		rv = 1;
+	} else if (maxpps < 0)
+		rv = 1;
+	else if (*curpps < maxpps)
+		rv = 1;
+	else
+		rv = 0;
+
+#if 1 /* DIAGNOSTIC? */
+	/* be careful about wrap-around */
+	if (*curpps + 1 > 0)
+		*curpps = *curpps + 1;
+#else
+	/*
+	 * assume that there's not too many calls to this function.
+	 * not sure if the assumption holds, as it depends on *caller's*
+	 * behavior, not the behavior of this function.
+	 * IMHO it is wrong to make assumption on the caller's behavior,
+	 * so the above #if is #if 1, not #ifdef DIAGNOSTIC.
+	 */
+	*curpps = *curpps + 1;
+#endif
+
+	return (rv);
+}
+#endif /* NETWORKING */
+
 void
 time_zone_slock_init(void)
 {