• source navigation  • diff markup  • identifier search  • freetext search  • 

Sources/libubox/utils.c

  1 /*
  2  * utils - misc libubox utility functions
  3  *
  4  * Copyright (C) 2012 Felix Fietkau <nbd@openwrt.org>
  5  *
  6  * Permission to use, copy, modify, and/or distribute this software for any
  7  * purpose with or without fee is hereby granted, provided that the above
  8  * copyright notice and this permission notice appear in all copies.
  9  *
 10  * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
 11  * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
 12  * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
 13  * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
 14  * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
 15  * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
 16  * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
 17  */
 18 
 19 #include <sys/mman.h>
 20 #include <stdarg.h>
 21 #include <stdlib.h>
 22 #include <stdio.h>
 23 #include "utils.h"
 24 
 25 #define foreach_arg(_arg, _addr, _len, _first_addr, _first_len) \
 26         for (_addr = (_first_addr), _len = (_first_len); \
 27                 _addr; \
 28                 _addr = va_arg(_arg, void **), _len = _addr ? va_arg(_arg, size_t) : 0)
 29 
 30 #define C_PTR_ALIGN     (sizeof(size_t))
 31 #define C_PTR_MASK      (-C_PTR_ALIGN)
 32 
 33 void *__calloc_a(size_t len, ...)
 34 {
 35         va_list ap, ap1;
 36         void *ret;
 37         void **cur_addr;
 38         size_t cur_len;
 39         int alloc_len = 0;
 40         char *ptr;
 41 
 42         va_start(ap, len);
 43 
 44         va_copy(ap1, ap);
 45         foreach_arg(ap1, cur_addr, cur_len, &ret, len)
 46                 alloc_len += (cur_len + C_PTR_ALIGN - 1 ) & C_PTR_MASK;
 47         va_end(ap1);
 48 
 49         ptr = calloc(1, alloc_len);
 50         if (!ptr) {
 51                 va_end(ap);
 52                 return NULL;
 53         }
 54 
 55         alloc_len = 0;
 56         foreach_arg(ap, cur_addr, cur_len, &ret, len) {
 57                 *cur_addr = &ptr[alloc_len];
 58                 alloc_len += (cur_len + C_PTR_ALIGN - 1) & C_PTR_MASK;
 59         }
 60         va_end(ap);
 61 
 62         return ret;
 63 }
 64 
 65 #ifdef LIBUBOX_COMPAT_CLOCK_GETTIME
 66 #include <mach/mach_host.h>             /* host_get_clock_service() */
 67 #include <mach/mach_port.h>             /* mach_port_deallocate() */
 68 #include <mach/mach_init.h>             /* mach_host_self(), mach_task_self() */
 69 #include <mach/clock.h>                 /* clock_get_time() */
 70 
 71 static clock_serv_t clock_realtime;
 72 static clock_serv_t clock_monotonic;
 73 
 74 static void __constructor clock_name_init(void)
 75 {
 76         mach_port_t host_self = mach_host_self();
 77 
 78         host_get_clock_service(host_self, CLOCK_REALTIME, &clock_realtime);
 79         host_get_clock_service(host_self, CLOCK_MONOTONIC, &clock_monotonic);
 80 }
 81 
 82 static void __destructor clock_name_dealloc(void)
 83 {
 84         mach_port_t self = mach_task_self();
 85 
 86         mach_port_deallocate(self, clock_realtime);
 87         mach_port_deallocate(self, clock_monotonic);
 88 }
 89 
 90 int clock_gettime(int type, struct timespec *tv)
 91 {
 92         int retval = -1;
 93         mach_timespec_t mts;
 94 
 95         switch (type) {
 96                 case CLOCK_REALTIME:
 97                         retval = clock_get_time(clock_realtime, &mts);
 98                         break;
 99                 case CLOCK_MONOTONIC:
100                         retval = clock_get_time(clock_monotonic, &mts);
101                         break;
102                 default:
103                         goto out;
104         }
105 
106         tv->tv_sec = mts.tv_sec;
107         tv->tv_nsec = mts.tv_nsec;
108 out:
109         return retval;
110 }
111 
112 #endif
113 
114 void *cbuf_alloc(unsigned int order)
115 {
116         char path[] = "/tmp/cbuf-XXXXXX";
117         unsigned long size = cbuf_size(order);
118         void *ret = NULL;
119         int fd;
120 
121         fd = mkstemp(path);
122         if (fd < 0)
123                 return NULL;
124 
125         if (unlink(path))
126                 goto close;
127 
128         if (ftruncate(fd, cbuf_size(order)))
129                 goto close;
130 
131 #ifndef MAP_ANONYMOUS
132 #define MAP_ANONYMOUS MAP_ANON
133 #endif
134 
135         ret = mmap(NULL, size * 2, PROT_NONE, MAP_ANON | MAP_PRIVATE, -1, 0);
136         if (ret == MAP_FAILED) {
137                 ret = NULL;
138                 goto close;
139         }
140 
141         if (mmap(ret, size, PROT_READ | PROT_WRITE, MAP_FIXED | MAP_SHARED,
142                  fd, 0) != ret ||
143             mmap(ret + size, size, PROT_READ | PROT_WRITE,
144                  MAP_FIXED | MAP_SHARED, fd, 0) != ret + size) {
145                 munmap(ret, size * 2);
146                 ret = NULL;
147         }
148 
149 close:
150         close(fd);
151         return ret;
152 }
153 
154 void cbuf_free(void *ptr, unsigned int order)
155 {
156         munmap(ptr, cbuf_size(order) * 2);
157 }
158 

This page was automatically generated by LXR 0.3.1.  •  OpenWrt