1fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry/* 2fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry * Copyright (C) 2008 The Android Open Source Project 3fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry * All rights reserved. 4fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry * 5fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry * Redistribution and use in source and binary forms, with or without 6fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry * modification, are permitted provided that the following conditions 7fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry * are met: 8fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry * * Redistributions of source code must retain the above copyright 9fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry * notice, this list of conditions and the following disclaimer. 10fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry * * Redistributions in binary form must reproduce the above copyright 11fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry * notice, this list of conditions and the following disclaimer in 12fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry * the documentation and/or other materials provided with the 13fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry * distribution. 14fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry * 15fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS 16fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT 17fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS 18fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry * FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE 19fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry * COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, 20fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, 21fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS 22fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry * OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED 23fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry * AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, 24fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT 25fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry * OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 26fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry * SUCH DAMAGE. 27fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry */ 28fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry 29e275d6d72db1868056a204e8483a7346a28cb8d6Tom Cherry#include "system_properties/prop_area.h" 30e275d6d72db1868056a204e8483a7346a28cb8d6Tom Cherry 313429c090f50a3114369482c550c382f8fa070c2eDan Albert#include <errno.h> 32fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry#include <fcntl.h> 33fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry#include <stdlib.h> 34fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry#include <sys/cdefs.h> 35fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry#include <sys/stat.h> 36fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry#include <sys/types.h> 37fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry#include <sys/xattr.h> 38fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry#include <unistd.h> 39fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry 40fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry#include <new> 41fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry 42fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry#include <async_safe/log.h> 43fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry 44fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherryconstexpr size_t PA_SIZE = 128 * 1024; 45fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherryconstexpr uint32_t PROP_AREA_MAGIC = 0x504f5250; 46fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherryconstexpr uint32_t PROP_AREA_VERSION = 0xfc6ed0ab; 47fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry 48f76bbf5384c0504b165168f51ccadfb605ccb2afTom Cherrysize_t prop_area::pa_size_ = 0; 49f76bbf5384c0504b165168f51ccadfb605ccb2afTom Cherrysize_t prop_area::pa_data_size_ = 0; 50fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry 51fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherryprop_area* prop_area::map_prop_area_rw(const char* filename, const char* context, 52fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry bool* fsetxattr_failed) { 53fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry /* dev is a tmpfs that we can use to carve a shared workspace 54fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry * out of, so let's do that... 55fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry */ 56fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry const int fd = open(filename, O_RDWR | O_CREAT | O_NOFOLLOW | O_CLOEXEC | O_EXCL, 0444); 57fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry 58fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry if (fd < 0) { 59fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry if (errno == EACCES) { 60fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry /* for consistency with the case where the process has already 61fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry * mapped the page in and segfaults when trying to write to it 62fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry */ 63fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry abort(); 64fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry } 65fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry return nullptr; 66fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry } 67fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry 68fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry if (context) { 69fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry if (fsetxattr(fd, XATTR_NAME_SELINUX, context, strlen(context) + 1, 0) != 0) { 70fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry async_safe_format_log(ANDROID_LOG_ERROR, "libc", 71fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry "fsetxattr failed to set context (%s) for \"%s\"", context, filename); 72fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry /* 73fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry * fsetxattr() will fail during system properties tests due to selinux policy. 74fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry * We do not want to create a custom policy for the tester, so we will continue in 75fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry * this function but set a flag that an error has occurred. 76fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry * Init, which is the only daemon that should ever call this function will abort 77fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry * when this error occurs. 78fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry * Otherwise, the tester will ignore it and continue, albeit without any selinux 79fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry * property separation. 80fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry */ 81fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry if (fsetxattr_failed) { 82fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry *fsetxattr_failed = true; 83fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry } 84fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry } 85fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry } 86fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry 87fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry if (ftruncate(fd, PA_SIZE) < 0) { 88fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry close(fd); 89fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry return nullptr; 90fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry } 91fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry 92f76bbf5384c0504b165168f51ccadfb605ccb2afTom Cherry pa_size_ = PA_SIZE; 93f76bbf5384c0504b165168f51ccadfb605ccb2afTom Cherry pa_data_size_ = pa_size_ - sizeof(prop_area); 94fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry 95f76bbf5384c0504b165168f51ccadfb605ccb2afTom Cherry void* const memory_area = mmap(nullptr, pa_size_, PROT_READ | PROT_WRITE, MAP_SHARED, fd, 0); 96fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry if (memory_area == MAP_FAILED) { 97fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry close(fd); 98fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry return nullptr; 99fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry } 100fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry 101fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry prop_area* pa = new (memory_area) prop_area(PROP_AREA_MAGIC, PROP_AREA_VERSION); 102fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry 103fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry close(fd); 104fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry return pa; 105fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry} 106fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry 107fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherryprop_area* prop_area::map_fd_ro(const int fd) { 108fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry struct stat fd_stat; 109fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry if (fstat(fd, &fd_stat) < 0) { 110fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry return nullptr; 111fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry } 112fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry 113fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry if ((fd_stat.st_uid != 0) || (fd_stat.st_gid != 0) || 114fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry ((fd_stat.st_mode & (S_IWGRP | S_IWOTH)) != 0) || 115fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry (fd_stat.st_size < static_cast<off_t>(sizeof(prop_area)))) { 116fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry return nullptr; 117fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry } 118fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry 119f76bbf5384c0504b165168f51ccadfb605ccb2afTom Cherry pa_size_ = fd_stat.st_size; 120f76bbf5384c0504b165168f51ccadfb605ccb2afTom Cherry pa_data_size_ = pa_size_ - sizeof(prop_area); 121fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry 122f76bbf5384c0504b165168f51ccadfb605ccb2afTom Cherry void* const map_result = mmap(nullptr, pa_size_, PROT_READ, MAP_SHARED, fd, 0); 123fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry if (map_result == MAP_FAILED) { 124fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry return nullptr; 125fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry } 126fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry 127fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry prop_area* pa = reinterpret_cast<prop_area*>(map_result); 128fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry if ((pa->magic() != PROP_AREA_MAGIC) || (pa->version() != PROP_AREA_VERSION)) { 129f76bbf5384c0504b165168f51ccadfb605ccb2afTom Cherry munmap(pa, pa_size_); 130fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry return nullptr; 131fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry } 132fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry 133fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry return pa; 134fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry} 135fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry 136fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherryprop_area* prop_area::map_prop_area(const char* filename) { 137fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry int fd = open(filename, O_CLOEXEC | O_NOFOLLOW | O_RDONLY); 138fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry if (fd == -1) return nullptr; 139fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry 140fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry prop_area* map_result = map_fd_ro(fd); 141fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry close(fd); 142fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry 143fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry return map_result; 144fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry} 145fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry 146fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherryvoid* prop_area::allocate_obj(const size_t size, uint_least32_t* const off) { 147fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry const size_t aligned = __BIONIC_ALIGN(size, sizeof(uint_least32_t)); 148f76bbf5384c0504b165168f51ccadfb605ccb2afTom Cherry if (bytes_used_ + aligned > pa_data_size_) { 149fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry return nullptr; 150fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry } 151fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry 152fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry *off = bytes_used_; 153fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry bytes_used_ += aligned; 154fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry return data_ + *off; 155fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry} 156fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry 157fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherryprop_bt* prop_area::new_prop_bt(const char* name, uint32_t namelen, uint_least32_t* const off) { 158fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry uint_least32_t new_offset; 159fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry void* const p = allocate_obj(sizeof(prop_bt) + namelen + 1, &new_offset); 160fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry if (p != nullptr) { 161fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry prop_bt* bt = new (p) prop_bt(name, namelen); 162fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry *off = new_offset; 163fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry return bt; 164fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry } 165fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry 166fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry return nullptr; 167fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry} 168fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry 169fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherryprop_info* prop_area::new_prop_info(const char* name, uint32_t namelen, const char* value, 170fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry uint32_t valuelen, uint_least32_t* const off) { 171fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry uint_least32_t new_offset; 172fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry void* const p = allocate_obj(sizeof(prop_info) + namelen + 1, &new_offset); 173fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry if (p == nullptr) return nullptr; 174fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry 175fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry prop_info* info; 176fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry if (valuelen >= PROP_VALUE_MAX) { 177fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry uint32_t long_value_offset = 0; 178fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry char* long_location = reinterpret_cast<char*>(allocate_obj(valuelen + 1, &long_value_offset)); 179fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry if (!long_location) return nullptr; 180fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry 181fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry memcpy(long_location, value, valuelen); 182fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry long_location[valuelen] = '\0'; 183fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry 184fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry // Both new_offset and long_value_offset are offsets based off of data_, however prop_info 185fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry // does not know what data_ is, so we change this offset to be an offset from the prop_info 186fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry // pointer that contains it. 187fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry long_value_offset -= new_offset; 188fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry 189fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry info = new (p) prop_info(name, namelen, long_value_offset); 190fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry } else { 191fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry info = new (p) prop_info(name, namelen, value, valuelen); 192fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry } 193fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry *off = new_offset; 194fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry return info; 195fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry} 196fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry 197fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherryvoid* prop_area::to_prop_obj(uint_least32_t off) { 198f76bbf5384c0504b165168f51ccadfb605ccb2afTom Cherry if (off > pa_data_size_) return nullptr; 199fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry 200fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry return (data_ + off); 201fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry} 202fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry 203fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherryinline prop_bt* prop_area::to_prop_bt(atomic_uint_least32_t* off_p) { 204fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry uint_least32_t off = atomic_load_explicit(off_p, memory_order_consume); 205fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry return reinterpret_cast<prop_bt*>(to_prop_obj(off)); 206fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry} 207fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry 208fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherryinline prop_info* prop_area::to_prop_info(atomic_uint_least32_t* off_p) { 209fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry uint_least32_t off = atomic_load_explicit(off_p, memory_order_consume); 210fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry return reinterpret_cast<prop_info*>(to_prop_obj(off)); 211fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry} 212fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry 213fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherryinline prop_bt* prop_area::root_node() { 214fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry return reinterpret_cast<prop_bt*>(to_prop_obj(0)); 215fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry} 216fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry 217fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherrystatic int cmp_prop_name(const char* one, uint32_t one_len, const char* two, uint32_t two_len) { 218fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry if (one_len < two_len) 219fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry return -1; 220fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry else if (one_len > two_len) 221fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry return 1; 222fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry else 223fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry return strncmp(one, two, one_len); 224fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry} 225fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry 226fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherryprop_bt* prop_area::find_prop_bt(prop_bt* const bt, const char* name, uint32_t namelen, 227fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry bool alloc_if_needed) { 228fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry prop_bt* current = bt; 229fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry while (true) { 230fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry if (!current) { 231fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry return nullptr; 232fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry } 233fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry 234fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry const int ret = cmp_prop_name(name, namelen, current->name, current->namelen); 235fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry if (ret == 0) { 236fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry return current; 237fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry } 238fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry 239fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry if (ret < 0) { 240fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry uint_least32_t left_offset = atomic_load_explicit(¤t->left, memory_order_relaxed); 241fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry if (left_offset != 0) { 242fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry current = to_prop_bt(¤t->left); 243fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry } else { 244fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry if (!alloc_if_needed) { 245fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry return nullptr; 246fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry } 247fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry 248fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry uint_least32_t new_offset; 249fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry prop_bt* new_bt = new_prop_bt(name, namelen, &new_offset); 250fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry if (new_bt) { 251fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry atomic_store_explicit(¤t->left, new_offset, memory_order_release); 252fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry } 253fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry return new_bt; 254fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry } 255fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry } else { 256fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry uint_least32_t right_offset = atomic_load_explicit(¤t->right, memory_order_relaxed); 257fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry if (right_offset != 0) { 258fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry current = to_prop_bt(¤t->right); 259fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry } else { 260fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry if (!alloc_if_needed) { 261fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry return nullptr; 262fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry } 263fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry 264fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry uint_least32_t new_offset; 265fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry prop_bt* new_bt = new_prop_bt(name, namelen, &new_offset); 266fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry if (new_bt) { 267fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry atomic_store_explicit(¤t->right, new_offset, memory_order_release); 268fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry } 269fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry return new_bt; 270fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry } 271fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry } 272fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry } 273fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry} 274fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry 275fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherryconst prop_info* prop_area::find_property(prop_bt* const trie, const char* name, uint32_t namelen, 276fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry const char* value, uint32_t valuelen, 277fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry bool alloc_if_needed) { 278fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry if (!trie) return nullptr; 279fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry 280fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry const char* remaining_name = name; 281fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry prop_bt* current = trie; 282fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry while (true) { 283fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry const char* sep = strchr(remaining_name, '.'); 284fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry const bool want_subtree = (sep != nullptr); 285fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry const uint32_t substr_size = (want_subtree) ? sep - remaining_name : strlen(remaining_name); 286fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry 287fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry if (!substr_size) { 288fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry return nullptr; 289fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry } 290fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry 291fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry prop_bt* root = nullptr; 292fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry uint_least32_t children_offset = atomic_load_explicit(¤t->children, memory_order_relaxed); 293fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry if (children_offset != 0) { 294fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry root = to_prop_bt(¤t->children); 295fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry } else if (alloc_if_needed) { 296fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry uint_least32_t new_offset; 297fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry root = new_prop_bt(remaining_name, substr_size, &new_offset); 298fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry if (root) { 299fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry atomic_store_explicit(¤t->children, new_offset, memory_order_release); 300fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry } 301fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry } 302fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry 303fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry if (!root) { 304fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry return nullptr; 305fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry } 306fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry 307fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry current = find_prop_bt(root, remaining_name, substr_size, alloc_if_needed); 308fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry if (!current) { 309fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry return nullptr; 310fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry } 311fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry 312fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry if (!want_subtree) break; 313fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry 314fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry remaining_name = sep + 1; 315fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry } 316fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry 317fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry uint_least32_t prop_offset = atomic_load_explicit(¤t->prop, memory_order_relaxed); 318fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry if (prop_offset != 0) { 319fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry return to_prop_info(¤t->prop); 320fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry } else if (alloc_if_needed) { 321fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry uint_least32_t new_offset; 322fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry prop_info* new_info = new_prop_info(name, namelen, value, valuelen, &new_offset); 323fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry if (new_info) { 324fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry atomic_store_explicit(¤t->prop, new_offset, memory_order_release); 325fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry } 326fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry 327fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry return new_info; 328fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry } else { 329fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry return nullptr; 330fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry } 331fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry} 332fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry 333fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherrybool prop_area::foreach_property(prop_bt* const trie, 334fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry void (*propfn)(const prop_info* pi, void* cookie), void* cookie) { 335fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry if (!trie) return false; 336fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry 337fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry uint_least32_t left_offset = atomic_load_explicit(&trie->left, memory_order_relaxed); 338fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry if (left_offset != 0) { 339fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry const int err = foreach_property(to_prop_bt(&trie->left), propfn, cookie); 340fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry if (err < 0) return false; 341fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry } 342fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry uint_least32_t prop_offset = atomic_load_explicit(&trie->prop, memory_order_relaxed); 343fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry if (prop_offset != 0) { 344fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry prop_info* info = to_prop_info(&trie->prop); 345fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry if (!info) return false; 346fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry propfn(info, cookie); 347fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry } 348fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry uint_least32_t children_offset = atomic_load_explicit(&trie->children, memory_order_relaxed); 349fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry if (children_offset != 0) { 350fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry const int err = foreach_property(to_prop_bt(&trie->children), propfn, cookie); 351fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry if (err < 0) return false; 352fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry } 353fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry uint_least32_t right_offset = atomic_load_explicit(&trie->right, memory_order_relaxed); 354fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry if (right_offset != 0) { 355fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry const int err = foreach_property(to_prop_bt(&trie->right), propfn, cookie); 356fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry if (err < 0) return false; 357fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry } 358fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry 359fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry return true; 360fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry} 361fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry 362fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherryconst prop_info* prop_area::find(const char* name) { 363fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry return find_property(root_node(), name, strlen(name), nullptr, 0, false); 364fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry} 365fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry 366fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherrybool prop_area::add(const char* name, unsigned int namelen, const char* value, 367fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry unsigned int valuelen) { 368fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry return find_property(root_node(), name, namelen, value, valuelen, true); 369fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry} 370fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry 371fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherrybool prop_area::foreach (void (*propfn)(const prop_info* pi, void* cookie), void* cookie) { 372fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry return foreach_property(root_node(), propfn, cookie); 373fd44b9f8d89ce4f33ff39d7f340a5ed08029d21cTom Cherry} 374