12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685168616871688168916901691169216931694169516961697169816991700170117021703170417051706170717081709171017111712171317141715171617171718171917201721172217231724172517261727172817291730173117321733173417351736173717381739174017411742174317441745174617471748174917501751175217531754175517561757175817591760176117621763176417651766176717681769177017711772177317741775177617771778177917801781178217831784178517861787178817891790179117921793179417951796179717981799180018011802180318041805180618071808180918101811181218131814181518161817181818191820182118221823182418251826182718281829183018311832183318341835183618371838183918401841184218431844184518461847184818491850185118521853185418551856185718581859186018611862186318641865186618671868186918701871187218731874187518761877187818791880188118821883188418851886188718881889189018911892189318941895189618971898189919001901190219031904190519061907190819091910191119121913191419151916191719181919192019211922192319241925192619271928192919301931193219331934193519361937193819391940194119421943194419451946194719481949195019511952195319541955195619571958195919601961196219631964196519661967196819691970197119721973197419751976197719781979198019811982198319841985198619871988198919901991199219931994199519961997199819992000200120022003200420052006200720082009201020112012201320142015201620172018201920202021202220232024202520262027202820292030203120322033203420352036203720382039204020412042204320442045204620472048204920502051205220532054205520562057205820592060206120622063206420652066206720682069207020712072207320742075207620772078207920802081208220832084208520862087208820892090209120922093209420952096209720982099210021012102210321042105210621072108210921102111211221132114211521162117211821192120212121222123212421252126212721282129213021312132213321342135213621372138213921402141214221432144214521462147214821492150215121522153215421552156215721582159216021612162216321642165216621672168216921702171217221732174217521762177217821792180218121822183218421852186218721882189219021912192219321942195219621972198219922002201220222032204220522062207220822092210221122122213221422152216221722182219222022212222222322242225222622272228222922302231223222332234223522362237223822392240224122422243224422452246224722482249225022512252225322542255225622572258225922602261226222632264226522662267226822692270227122722273227422752276227722782279228022812282228322842285228622872288228922902291229222932294229522962297229822992300230123022303230423052306230723082309231023112312231323142315231623172318231923202321232223232324232523262327232823292330233123322333233423352336233723382339234023412342234323442345234623472348234923502351235223532354235523562357235823592360236123622363236423652366236723682369237023712372237323742375237623772378237923802381238223832384238523862387238823892390239123922393239423952396239723982399240024012402240324042405240624072408240924102411241224132414241524162417241824192420242124222423242424252426242724282429243024312432243324342435243624372438243924402441244224432444244524462447244824492450245124522453245424552456245724582459246024612462246324642465246624672468246924702471247224732474247524762477247824792480248124822483248424852486248724882489249024912492249324942495249624972498249925002501250225032504250525062507250825092510251125122513251425152516251725182519252025212522252325242525252625272528252925302531253225332534253525362537253825392540254125422543254425452546254725482549255025512552255325542555255625572558255925602561256225632564256525662567256825692570257125722573257425752576257725782579258025812582258325842585258625872588258925902591259225932594259525962597259825992600260126022603260426052606260726082609261026112612261326142615261626172618261926202621262226232624262526262627262826292630263126322633263426352636263726382639264026412642264326442645264626472648264926502651265226532654265526562657265826592660266126622663266426652666266726682669267026712672267326742675267626772678267926802681268226832684268526862687268826892690269126922693269426952696269726982699270027012702270327042705270627072708270927102711271227132714271527162717271827192720272127222723272427252726272727282729273027312732273327342735273627372738273927402741274227432744274527462747274827492750275127522753275427552756275727582759276027612762276327642765276627672768276927702771277227732774277527762777277827792780278127822783278427852786278727882789279027912792279327942795279627972798279928002801280228032804280528062807280828092810281128122813281428152816281728182819282028212822282328242825282628272828282928302831283228332834283528362837283828392840284128422843284428452846284728482849285028512852285328542855285628572858285928602861286228632864286528662867286828692870287128722873287428752876287728782879288028812882288328842885288628872888288928902891289228932894289528962897289828992900290129022903290429052906290729082909291029112912291329142915291629172918291929202921292229232924292529262927292829292930293129322933293429352936293729382939294029412942294329442945294629472948294929502951295229532954295529562957295829592960296129622963296429652966296729682969297029712972297329742975297629772978297929802981298229832984298529862987298829892990299129922993299429952996299729982999300030013002300330043005300630073008300930103011301230133014301530163017301830193020302130223023302430253026302730283029303030313032303330343035303630373038303930403041304230433044304530463047304830493050305130523053305430553056305730583059306030613062306330643065306630673068306930703071307230733074307530763077307830793080308130823083308430853086308730883089309030913092309330943095309630973098309931003101310231033104310531063107310831093110311131123113311431153116311731183119312031213122312331243125312631273128312931303131313231333134313531363137313831393140314131423143314431453146314731483149315031513152315331543155315631573158315931603161316231633164316531663167316831693170317131723173317431753176317731783179318031813182318331843185318631873188318931903191319231933194319531963197319831993200320132023203320432053206320732083209321032113212321332143215321632173218321932203221322232233224322532263227322832293230323132323233323432353236323732383239324032413242324332443245324632473248324932503251325232533254325532563257325832593260326132623263326432653266326732683269327032713272327332743275327632773278327932803281328232833284328532863287328832893290329132923293329432953296329732983299330033013302330333043305330633073308330933103311331233133314331533163317331833193320332133223323332433253326332733283329333033313332333333343335333633373338333933403341334233433344334533463347334833493350335133523353335433553356335733583359336033613362336333643365336633673368336933703371337233733374337533763377337833793380338133823383338433853386338733883389339033913392339333943395339633973398339934003401340234033404340534063407340834093410341134123413341434153416341734183419342034213422342334243425342634273428342934303431343234333434343534363437343834393440344134423443344434453446344734483449345034513452345334543455345634573458345934603461346234633464346534663467346834693470347134723473347434753476347734783479348034813482348334843485348634873488348934903491349234933494349534963497349834993500350135023503350435053506350735083509351035113512351335143515351635173518351935203521352235233524352535263527352835293530353135323533353435353536353735383539354035413542354335443545354635473548354935503551355235533554355535563557355835593560356135623563356435653566356735683569357035713572357335743575357635773578357935803581358235833584358535863587358835893590359135923593359435953596359735983599360036013602360336043605360636073608360936103611361236133614361536163617361836193620362136223623362436253626362736283629363036313632363336343635363636373638363936403641364236433644364536463647364836493650365136523653365436553656365736583659366036613662366336643665366636673668366936703671367236733674367536763677367836793680368136823683368436853686368736883689369036913692369336943695369636973698369937003701370237033704370537063707370837093710371137123713371437153716371737183719372037213722372337243725372637273728372937303731373237333734373537363737373837393740374137423743374437453746374737483749375037513752375337543755375637573758375937603761376237633764376537663767376837693770377137723773377437753776377737783779378037813782378337843785378637873788378937903791379237933794379537963797379837993800380138023803380438053806380738083809381038113812381338143815381638173818381938203821382238233824382538263827382838293830383138323833383438353836383738383839384038413842384338443845384638473848384938503851385238533854385538563857385838593860386138623863386438653866386738683869387038713872387338743875387638773878387938803881388238833884388538863887388838893890389138923893389438953896389738983899390039013902390339043905390639073908390939103911391239133914391539163917391839193920392139223923392439253926392739283929393039313932 |
- /* Copyright (c) 2013, Vsevolod Stakhov
- * Copyright (c) 2015 Allan Jude <allanjude@freebsd.org>
- * All rights reserved.
- *
- * Redistribution and use in source and binary forms, with or without
- * modification, are permitted provided that the following conditions are met:
- * * Redistributions of source code must retain the above copyright
- * notice, this list of conditions and the following disclaimer.
- * * Redistributions in binary form must reproduce the above copyright
- * notice, this list of conditions and the following disclaimer in the
- * documentation and/or other materials provided with the distribution.
- *
- * THIS SOFTWARE IS PROVIDED ''AS IS'' AND ANY
- * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
- * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
- * DISCLAIMED. IN NO EVENT SHALL AUTHOR BE LIABLE FOR ANY
- * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
- * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
- * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
- * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
- * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
- * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
- */
-
- #include "ucl.h"
- #include "ucl_internal.h"
- #include "ucl_chartable.h"
- #include "kvec.h"
- #include <limits.h>
- #include <stdarg.h>
- #include <stdio.h> /* for snprintf */
-
- #ifndef _WIN32
- #include <glob.h>
- #include <sys/param.h>
- #else
- #ifndef NBBY
- #define NBBY 8
- #endif
- #endif
-
- #ifdef HAVE_LIBGEN_H
- #ifndef _WIN32
- # include <libgen.h> /* For dirname */
- #endif
- #endif
-
- typedef kvec_t(ucl_object_t *) ucl_array_t;
-
- #define UCL_ARRAY_GET(ar, obj) ucl_array_t *ar = \
- (ucl_array_t *)((obj) != NULL ? (obj)->value.av : NULL)
-
- #ifdef HAVE_OPENSSL
- #include <openssl/err.h>
- #include <openssl/sha.h>
- #include <openssl/rsa.h>
- #include <openssl/ssl.h>
- #include <openssl/evp.h>
- #endif
-
- #ifdef CURL_FOUND
- /* Seems to be broken */
- #define CURL_DISABLE_TYPECHECK 1
- #include <curl/curl.h>
- #endif
- #ifdef HAVE_FETCH_H
- #include <fetch.h>
- #endif
-
- #if defined(_MSC_VER)
- #include <windows.h>
- #include <io.h>
- #include <direct.h>
-
- #ifndef PROT_READ
- #define PROT_READ 1
- #endif
- #ifndef PROT_WRITE
- #define PROT_WRITE 2
- #endif
- #ifndef PROT_READWRITE
- #define PROT_READWRITE 3
- #endif
- #ifndef MAP_SHARED
- #define MAP_SHARED 1
- #endif
- #ifndef MAP_PRIVATE
- #define MAP_PRIVATE 2
- #endif
- #ifndef MAP_FAILED
- #define MAP_FAILED ((void *) -1)
- #endif
-
- #define getcwd _getcwd
- #define open _open
- #define close _close
-
- static void *ucl_mmap(char *addr, size_t length, int prot, int access, int fd, off_t offset)
- {
- void *map = NULL;
- HANDLE handle = INVALID_HANDLE_VALUE;
-
- switch (prot) {
- default:
- case PROT_READ:
- {
- handle = CreateFileMapping((HANDLE) _get_osfhandle(fd), 0, PAGE_READONLY, 0, length, 0);
- if (!handle) break;
- map = (void *) MapViewOfFile(handle, FILE_MAP_READ, 0, 0, length);
- CloseHandle(handle);
- break;
- }
- case PROT_WRITE:
- {
- handle = CreateFileMapping((HANDLE) _get_osfhandle(fd), 0, PAGE_READWRITE, 0, length, 0);
- if (!handle) break;
- map = (void *) MapViewOfFile(handle, FILE_MAP_WRITE, 0, 0, length);
- CloseHandle(handle);
- break;
- }
- case PROT_READWRITE:
- {
- handle = CreateFileMapping((HANDLE) _get_osfhandle(fd), 0, PAGE_READWRITE, 0, length, 0);
- if (!handle) break;
- map = (void *) MapViewOfFile(handle, FILE_MAP_ALL_ACCESS, 0, 0, length);
- CloseHandle(handle);
- break;
- }
- }
- if (map == (void *) NULL) {
- return (void *) MAP_FAILED;
- }
- return (void *) ((char *) map + offset);
- }
-
- static int ucl_munmap(void *map,size_t length)
- {
- if (!UnmapViewOfFile(map)) {
- return(-1);
- }
- return(0);
- }
-
- static char* ucl_realpath(const char *path, char *resolved_path)
- {
- char *p;
- char tmp[MAX_PATH + 1];
- strncpy(tmp, path, sizeof(tmp)-1);
- p = tmp;
- while(*p) {
- if (*p == '/') *p = '\\';
- p++;
- }
- return _fullpath(resolved_path, tmp, MAX_PATH);
- }
-
-
- char *dirname(char *path)
- {
- static char path_buffer[_MAX_PATH];
- char drive[_MAX_DRIVE];
- char dir[_MAX_DIR];
- char fname[_MAX_FNAME];
- char ext[_MAX_EXT];
-
- _splitpath (path, drive, dir, fname, ext);
- _makepath(path_buffer, drive, dir, NULL, NULL);
-
- return path_buffer;
- }
-
- char *basename(char *path)
- {
- static char path_buffer[_MAX_PATH];
- char drive[_MAX_DRIVE];
- char dir[_MAX_DIR];
- char fname[_MAX_FNAME];
- char ext[_MAX_EXT];
-
- _splitpath(path, drive, dir, fname, ext);
- _makepath(path_buffer, NULL, NULL, fname, ext);
-
- return path_buffer;
- }
- #else
- #define ucl_mmap mmap
- #define ucl_munmap munmap
- #define ucl_realpath realpath
- #endif
-
- typedef void (*ucl_object_dtor) (ucl_object_t *obj);
- static void ucl_object_free_internal (ucl_object_t *obj, bool allow_rec,
- ucl_object_dtor dtor);
- static void ucl_object_dtor_unref (ucl_object_t *obj);
-
- static void
- ucl_object_dtor_free (ucl_object_t *obj)
- {
- if (obj->trash_stack[UCL_TRASH_KEY] != NULL) {
- UCL_FREE (obj->hh.keylen, obj->trash_stack[UCL_TRASH_KEY]);
- }
- if (obj->trash_stack[UCL_TRASH_VALUE] != NULL) {
- UCL_FREE (obj->len, obj->trash_stack[UCL_TRASH_VALUE]);
- }
- /* Do not free ephemeral objects */
- if ((obj->flags & UCL_OBJECT_EPHEMERAL) == 0) {
- if (obj->type != UCL_USERDATA) {
- UCL_FREE (sizeof (ucl_object_t), obj);
- }
- else {
- struct ucl_object_userdata *ud = (struct ucl_object_userdata *)obj;
- if (ud->dtor) {
- ud->dtor (obj->value.ud);
- }
- UCL_FREE (sizeof (*ud), obj);
- }
- }
- }
-
- /*
- * This is a helper function that performs exactly the same as
- * `ucl_object_unref` but it doesn't iterate over elements allowing
- * to use it for individual elements of arrays and multiple values
- */
- static void
- ucl_object_dtor_unref_single (ucl_object_t *obj)
- {
- if (obj != NULL) {
- #ifdef HAVE_ATOMIC_BUILTINS
- unsigned int rc = __sync_sub_and_fetch (&obj->ref, 1);
- if (rc == 0) {
- #else
- if (--obj->ref == 0) {
- #endif
- ucl_object_free_internal (obj, false, ucl_object_dtor_unref);
- }
- }
- }
-
- static void
- ucl_object_dtor_unref (ucl_object_t *obj)
- {
- if (obj->ref == 0) {
- ucl_object_dtor_free (obj);
- }
- else {
- /* This may cause dtor unref being called one more time */
- ucl_object_dtor_unref_single (obj);
- }
- }
-
- static void
- ucl_object_free_internal (ucl_object_t *obj, bool allow_rec, ucl_object_dtor dtor)
- {
- ucl_object_t *tmp, *sub;
-
- while (obj != NULL) {
- if (obj->type == UCL_ARRAY) {
- UCL_ARRAY_GET (vec, obj);
- unsigned int i;
-
- if (vec != NULL) {
- for (i = 0; i < vec->n; i ++) {
- sub = kv_A (*vec, i);
- if (sub != NULL) {
- tmp = sub;
- while (sub) {
- tmp = sub->next;
- dtor (sub);
- sub = tmp;
- }
- }
- }
- kv_destroy (*vec);
- UCL_FREE (sizeof (*vec), vec);
- }
- obj->value.av = NULL;
- }
- else if (obj->type == UCL_OBJECT) {
- if (obj->value.ov != NULL) {
- ucl_hash_destroy (obj->value.ov, (ucl_hash_free_func)dtor);
- }
- obj->value.ov = NULL;
- }
- tmp = obj->next;
- dtor (obj);
- obj = tmp;
-
- if (!allow_rec) {
- break;
- }
- }
- }
-
- void
- ucl_object_free (ucl_object_t *obj)
- {
- ucl_object_free_internal (obj, true, ucl_object_dtor_free);
- }
-
- size_t
- ucl_unescape_json_string (char *str, size_t len)
- {
- char *t = str, *h = str;
- int i, uval;
-
- if (len <= 1) {
- return len;
- }
- /* t is target (tortoise), h is source (hare) */
-
- while (len) {
- if (*h == '\\') {
- h ++;
-
- if (len == 1) {
- /*
- * If \ is last, then do not try to go further
- * Issue: #74
- */
- len --;
- *t++ = '\\';
- continue;
- }
-
- switch (*h) {
- case 'n':
- *t++ = '\n';
- break;
- case 'r':
- *t++ = '\r';
- break;
- case 'b':
- *t++ = '\b';
- break;
- case 't':
- *t++ = '\t';
- break;
- case 'f':
- *t++ = '\f';
- break;
- case '\\':
- *t++ = '\\';
- break;
- case '"':
- *t++ = '"';
- break;
- case 'u':
- /* Unicode escape */
- uval = 0;
- h ++; /* u character */
- len --;
-
- if (len > 3) {
- for (i = 0; i < 4; i++) {
- uval <<= 4;
- if (isdigit (h[i])) {
- uval += h[i] - '0';
- }
- else if (h[i] >= 'a' && h[i] <= 'f') {
- uval += h[i] - 'a' + 10;
- }
- else if (h[i] >= 'A' && h[i] <= 'F') {
- uval += h[i] - 'A' + 10;
- }
- else {
- break;
- }
- }
-
- /* Encode */
- if(uval < 0x80) {
- t[0] = (char)uval;
- t ++;
- }
- else if(uval < 0x800) {
- t[0] = 0xC0 + ((uval & 0x7C0) >> 6);
- t[1] = 0x80 + ((uval & 0x03F));
- t += 2;
- }
- else if(uval < 0x10000) {
- t[0] = 0xE0 + ((uval & 0xF000) >> 12);
- t[1] = 0x80 + ((uval & 0x0FC0) >> 6);
- t[2] = 0x80 + ((uval & 0x003F));
- t += 3;
- }
- #if 0
- /* It's not actually supported now */
- else if(uval <= 0x10FFFF) {
- t[0] = 0xF0 + ((uval & 0x1C0000) >> 18);
- t[1] = 0x80 + ((uval & 0x03F000) >> 12);
- t[2] = 0x80 + ((uval & 0x000FC0) >> 6);
- t[3] = 0x80 + ((uval & 0x00003F));
- t += 4;
- }
- #endif
- else {
- *t++ = '?';
- }
-
- /* Consume 4 characters of source */
- h += 4;
- len -= 4;
-
- if (len > 0) {
- len --; /* for '\' character */
- }
- continue;
- }
- else {
- *t++ = 'u';
- }
- break;
- default:
- *t++ = *h;
- break;
- }
- h ++;
- len --;
- }
- else {
- *t++ = *h++;
- }
-
- if (len > 0) {
- len --;
- }
- }
- *t = '\0';
-
- return (t - str);
- }
-
- size_t
- ucl_unescape_squoted_string (char *str, size_t len)
- {
- char *t = str, *h = str;
-
- if (len <= 1) {
- return len;
- }
-
- /* t is target (tortoise), h is source (hare) */
-
- while (len) {
- if (*h == '\\') {
- h ++;
-
- if (len == 1) {
- /*
- * If \ is last, then do not try to go further
- * Issue: #74
- */
- len --;
- *t++ = '\\';
- continue;
- }
-
- switch (*h) {
- case '\'':
- *t++ = '\'';
- break;
- case '\n':
- /* Ignore \<newline> style stuff */
- break;
- case '\r':
- /* Ignore \r and the following \n if needed */
- if (len > 1 && h[1] == '\n') {
- h ++;
- len --;
- }
- break;
- default:
- /* Ignore \ */
- *t++ = '\\';
- *t++ = *h;
- break;
- }
-
- h ++;
- len --;
- }
- else {
- *t++ = *h++;
- }
-
- if (len > 0) {
- len --;
- }
- }
-
- *t = '\0';
-
- return (t - str);
- }
-
- char *
- ucl_copy_key_trash (const ucl_object_t *obj)
- {
- ucl_object_t *deconst;
-
- if (obj == NULL) {
- return NULL;
- }
- if (obj->trash_stack[UCL_TRASH_KEY] == NULL && obj->key != NULL) {
- deconst = __DECONST (ucl_object_t *, obj);
- deconst->trash_stack[UCL_TRASH_KEY] = malloc (obj->keylen + 1);
- if (deconst->trash_stack[UCL_TRASH_KEY] != NULL) {
- memcpy (deconst->trash_stack[UCL_TRASH_KEY], obj->key, obj->keylen);
- deconst->trash_stack[UCL_TRASH_KEY][obj->keylen] = '\0';
- }
- deconst->key = obj->trash_stack[UCL_TRASH_KEY];
- deconst->flags |= UCL_OBJECT_ALLOCATED_KEY;
- }
-
- return obj->trash_stack[UCL_TRASH_KEY];
- }
-
- void
- ucl_chunk_free (struct ucl_chunk *chunk)
- {
- if (chunk) {
- struct ucl_parser_special_handler_chain *chain, *tmp;
-
- LL_FOREACH_SAFE (chunk->special_handlers, chain, tmp) {
- if (chain->special_handler->free_function) {
- chain->special_handler->free_function (
- chain->begin,
- chain->len,
- chain->special_handler->user_data);
- } else {
- UCL_FREE (chain->len, chain->begin);
- }
-
- UCL_FREE (sizeof (*chain), chain);
- }
-
- chunk->special_handlers = NULL;
-
- if (chunk->fname) {
- free (chunk->fname);
- }
-
- UCL_FREE (sizeof (*chunk), chunk);
- }
- }
-
- char *
- ucl_copy_value_trash (const ucl_object_t *obj)
- {
- ucl_object_t *deconst;
-
- if (obj == NULL) {
- return NULL;
- }
- if (obj->trash_stack[UCL_TRASH_VALUE] == NULL) {
- deconst = __DECONST (ucl_object_t *, obj);
- if (obj->type == UCL_STRING) {
-
- /* Special case for strings */
- if (obj->flags & UCL_OBJECT_BINARY) {
- deconst->trash_stack[UCL_TRASH_VALUE] = malloc (obj->len);
- if (deconst->trash_stack[UCL_TRASH_VALUE] != NULL) {
- memcpy (deconst->trash_stack[UCL_TRASH_VALUE],
- obj->value.sv,
- obj->len);
- deconst->value.sv = obj->trash_stack[UCL_TRASH_VALUE];
- }
- }
- else {
- deconst->trash_stack[UCL_TRASH_VALUE] = malloc (obj->len + 1);
- if (deconst->trash_stack[UCL_TRASH_VALUE] != NULL) {
- memcpy (deconst->trash_stack[UCL_TRASH_VALUE],
- obj->value.sv,
- obj->len);
- deconst->trash_stack[UCL_TRASH_VALUE][obj->len] = '\0';
- deconst->value.sv = obj->trash_stack[UCL_TRASH_VALUE];
- }
- }
- }
- else {
- /* Just emit value in json notation */
- deconst->trash_stack[UCL_TRASH_VALUE] = ucl_object_emit_single_json (obj);
- deconst->len = strlen (obj->trash_stack[UCL_TRASH_VALUE]);
- }
- deconst->flags |= UCL_OBJECT_ALLOCATED_VALUE;
- }
-
- return obj->trash_stack[UCL_TRASH_VALUE];
- }
-
- ucl_object_t*
- ucl_parser_get_object (struct ucl_parser *parser)
- {
- if (parser->state != UCL_STATE_ERROR && parser->top_obj != NULL) {
- return ucl_object_ref (parser->top_obj);
- }
-
- return NULL;
- }
-
- void
- ucl_parser_free (struct ucl_parser *parser)
- {
- struct ucl_stack *stack, *stmp;
- struct ucl_macro *macro, *mtmp;
- struct ucl_chunk *chunk, *ctmp;
- struct ucl_pubkey *key, *ktmp;
- struct ucl_variable *var, *vtmp;
- ucl_object_t *tr, *trtmp;
-
- if (parser == NULL) {
- return;
- }
-
- if (parser->top_obj != NULL) {
- ucl_object_unref (parser->top_obj);
- }
-
- if (parser->includepaths != NULL) {
- ucl_object_unref (parser->includepaths);
- }
-
- LL_FOREACH_SAFE (parser->stack, stack, stmp) {
- free (stack);
- }
- HASH_ITER (hh, parser->macroes, macro, mtmp) {
- free (macro->name);
- HASH_DEL (parser->macroes, macro);
- UCL_FREE (sizeof (struct ucl_macro), macro);
- }
- LL_FOREACH_SAFE (parser->chunks, chunk, ctmp) {
- ucl_chunk_free (chunk);
- }
- LL_FOREACH_SAFE (parser->keys, key, ktmp) {
- UCL_FREE (sizeof (struct ucl_pubkey), key);
- }
- LL_FOREACH_SAFE (parser->variables, var, vtmp) {
- free (var->value);
- free (var->var);
- UCL_FREE (sizeof (struct ucl_variable), var);
- }
- LL_FOREACH_SAFE (parser->trash_objs, tr, trtmp) {
- ucl_object_free_internal (tr, false, ucl_object_dtor_free);
- }
-
- if (parser->err != NULL) {
- utstring_free (parser->err);
- }
-
- if (parser->cur_file) {
- free (parser->cur_file);
- }
-
- if (parser->comments) {
- ucl_object_unref (parser->comments);
- }
-
- UCL_FREE (sizeof (struct ucl_parser), parser);
- }
-
- const char *
- ucl_parser_get_error(struct ucl_parser *parser)
- {
- if (parser == NULL) {
- return NULL;
- }
-
- if (parser->err == NULL) {
- return NULL;
- }
-
- return utstring_body (parser->err);
- }
-
- int
- ucl_parser_get_error_code(struct ucl_parser *parser)
- {
- if (parser == NULL) {
- return 0;
- }
-
- return parser->err_code;
- }
-
- unsigned
- ucl_parser_get_column(struct ucl_parser *parser)
- {
- if (parser == NULL || parser->chunks == NULL) {
- return 0;
- }
-
- return parser->chunks->column;
- }
-
- unsigned
- ucl_parser_get_linenum(struct ucl_parser *parser)
- {
- if (parser == NULL || parser->chunks == NULL) {
- return 0;
- }
-
- return parser->chunks->line;
- }
-
- void
- ucl_parser_clear_error(struct ucl_parser *parser)
- {
- if (parser != NULL && parser->err != NULL) {
- utstring_free(parser->err);
- parser->err = NULL;
- parser->err_code = 0;
- }
- }
-
- bool
- ucl_pubkey_add (struct ucl_parser *parser, const unsigned char *key, size_t len)
- {
- #ifndef HAVE_OPENSSL
- ucl_create_err (&parser->err, "cannot check signatures without openssl");
- return false;
- #else
- # if (OPENSSL_VERSION_NUMBER < 0x10000000L)
- ucl_create_err (&parser->err, "cannot check signatures, openssl version is unsupported");
- return EXIT_FAILURE;
- # else
- struct ucl_pubkey *nkey;
- BIO *mem;
-
- mem = BIO_new_mem_buf ((void *)key, len);
- nkey = UCL_ALLOC (sizeof (struct ucl_pubkey));
- if (nkey == NULL) {
- ucl_create_err (&parser->err, "cannot allocate memory for key");
- return false;
- }
- nkey->key = PEM_read_bio_PUBKEY (mem, &nkey->key, NULL, NULL);
- BIO_free (mem);
- if (nkey->key == NULL) {
- UCL_FREE (sizeof (struct ucl_pubkey), nkey);
- ucl_create_err (&parser->err, "%s",
- ERR_error_string (ERR_get_error (), NULL));
- return false;
- }
- LL_PREPEND (parser->keys, nkey);
- # endif
- #endif
- return true;
- }
-
- void ucl_parser_add_special_handler (struct ucl_parser *parser,
- struct ucl_parser_special_handler *handler)
- {
- LL_APPEND (parser->special_handlers, handler);
- }
-
- #ifdef CURL_FOUND
- struct ucl_curl_cbdata {
- unsigned char *buf;
- size_t buflen;
- };
-
- static size_t
- ucl_curl_write_callback (void* contents, size_t size, size_t nmemb, void* ud)
- {
- struct ucl_curl_cbdata *cbdata = ud;
- size_t realsize = size * nmemb;
-
- cbdata->buf = realloc (cbdata->buf, cbdata->buflen + realsize + 1);
- if (cbdata->buf == NULL) {
- return 0;
- }
-
- memcpy (&(cbdata->buf[cbdata->buflen]), contents, realsize);
- cbdata->buflen += realsize;
- cbdata->buf[cbdata->buflen] = 0;
-
- return realsize;
- }
- #endif
-
- /**
- * Fetch a url and save results to the memory buffer
- * @param url url to fetch
- * @param len length of url
- * @param buf target buffer
- * @param buflen target length
- * @return
- */
- bool
- ucl_fetch_url (const unsigned char *url, unsigned char **buf, size_t *buflen,
- UT_string **err, bool must_exist)
- {
-
- #ifdef HAVE_FETCH_H
- struct url *fetch_url;
- struct url_stat us;
- FILE *in;
-
- fetch_url = fetchParseURL (url);
- if (fetch_url == NULL) {
- ucl_create_err (err, "invalid URL %s: %s",
- url, strerror (errno));
- return false;
- }
- if ((in = fetchXGet (fetch_url, &us, "")) == NULL) {
- if (!must_exist) {
- ucl_create_err (err, "cannot fetch URL %s: %s",
- url, strerror (errno));
- }
- fetchFreeURL (fetch_url);
- return false;
- }
-
- *buflen = us.size;
- *buf = malloc (*buflen);
- if (*buf == NULL) {
- ucl_create_err (err, "cannot allocate buffer for URL %s: %s",
- url, strerror (errno));
- fclose (in);
- fetchFreeURL (fetch_url);
- return false;
- }
-
- if (fread (*buf, *buflen, 1, in) != 1) {
- ucl_create_err (err, "cannot read URL %s: %s",
- url, strerror (errno));
- fclose (in);
- fetchFreeURL (fetch_url);
- return false;
- }
-
- fetchFreeURL (fetch_url);
- return true;
- #elif defined(CURL_FOUND)
- CURL *curl;
- int r;
- struct ucl_curl_cbdata cbdata;
-
- curl = curl_easy_init ();
- if (curl == NULL) {
- ucl_create_err (err, "CURL interface is broken");
- return false;
- }
- if ((r = curl_easy_setopt (curl, CURLOPT_URL, url)) != CURLE_OK) {
- ucl_create_err (err, "invalid URL %s: %s",
- url, curl_easy_strerror (r));
- curl_easy_cleanup (curl);
- return false;
- }
- curl_easy_setopt (curl, CURLOPT_WRITEFUNCTION, ucl_curl_write_callback);
- cbdata.buf = NULL;
- cbdata.buflen = 0;
- curl_easy_setopt (curl, CURLOPT_WRITEDATA, &cbdata);
-
- if ((r = curl_easy_perform (curl)) != CURLE_OK) {
- if (!must_exist) {
- ucl_create_err (err, "error fetching URL %s: %s",
- url, curl_easy_strerror (r));
- }
- curl_easy_cleanup (curl);
- if (cbdata.buf) {
- free (cbdata.buf);
- }
- return false;
- }
- *buf = cbdata.buf;
- *buflen = cbdata.buflen;
-
- return true;
- #else
- ucl_create_err (err, "URL support is disabled");
- return false;
- #endif
- }
-
- /**
- * Fetch a file and save results to the memory buffer
- * @param filename filename to fetch
- * @param len length of filename
- * @param buf target buffer
- * @param buflen target length
- * @return
- */
- bool
- ucl_fetch_file (const unsigned char *filename, unsigned char **buf, size_t *buflen,
- UT_string **err, bool must_exist)
- {
- int fd;
- struct stat st;
-
- if (stat (filename, &st) == -1) {
- if (must_exist || errno == EPERM) {
- ucl_create_err (err, "cannot stat file %s: %s",
- filename, strerror (errno));
- }
- return false;
- }
- if (!S_ISREG (st.st_mode)) {
- if (must_exist) {
- ucl_create_err (err, "file %s is not a regular file", filename);
- }
-
- return false;
- }
- if (st.st_size == 0) {
- /* Do not map empty files */
- *buf = NULL;
- *buflen = 0;
- }
- else {
- if ((fd = open (filename, O_RDONLY)) == -1) {
- ucl_create_err (err, "cannot open file %s: %s",
- filename, strerror (errno));
- return false;
- }
- if ((*buf = ucl_mmap (NULL, st.st_size, PROT_READ, MAP_SHARED, fd, 0)) == MAP_FAILED) {
- close (fd);
- ucl_create_err (err, "cannot mmap file %s: %s",
- filename, strerror (errno));
- *buf = NULL;
-
- return false;
- }
- *buflen = st.st_size;
- close (fd);
- }
-
- return true;
- }
-
-
- #if (defined(HAVE_OPENSSL) && OPENSSL_VERSION_NUMBER >= 0x10000000L)
- static inline bool
- ucl_sig_check (const unsigned char *data, size_t datalen,
- const unsigned char *sig, size_t siglen, struct ucl_parser *parser)
- {
- struct ucl_pubkey *key;
- char dig[EVP_MAX_MD_SIZE];
- unsigned int diglen;
- EVP_PKEY_CTX *key_ctx;
- EVP_MD_CTX *sign_ctx = NULL;
-
- sign_ctx = EVP_MD_CTX_create ();
-
- LL_FOREACH (parser->keys, key) {
- key_ctx = EVP_PKEY_CTX_new (key->key, NULL);
- if (key_ctx != NULL) {
- if (EVP_PKEY_verify_init (key_ctx) <= 0) {
- EVP_PKEY_CTX_free (key_ctx);
- continue;
- }
- if (EVP_PKEY_CTX_set_rsa_padding (key_ctx, RSA_PKCS1_PADDING) <= 0) {
- EVP_PKEY_CTX_free (key_ctx);
- continue;
- }
- if (EVP_PKEY_CTX_set_signature_md (key_ctx, EVP_sha256 ()) <= 0) {
- EVP_PKEY_CTX_free (key_ctx);
- continue;
- }
- EVP_DigestInit (sign_ctx, EVP_sha256 ());
- EVP_DigestUpdate (sign_ctx, data, datalen);
- EVP_DigestFinal (sign_ctx, dig, &diglen);
-
- if (EVP_PKEY_verify (key_ctx, sig, siglen, dig, diglen) == 1) {
- EVP_MD_CTX_destroy (sign_ctx);
- EVP_PKEY_CTX_free (key_ctx);
- return true;
- }
-
- EVP_PKEY_CTX_free (key_ctx);
- }
- }
-
- EVP_MD_CTX_destroy (sign_ctx);
-
- return false;
- }
- #endif
-
- struct ucl_include_params {
- bool check_signature;
- bool must_exist;
- bool use_glob;
- bool use_prefix;
- bool soft_fail;
- bool allow_glob;
- unsigned priority;
- enum ucl_duplicate_strategy strat;
- enum ucl_parse_type parse_type;
- const char *prefix;
- const char *target;
- };
-
- /**
- * Include an url to configuration
- * @param data
- * @param len
- * @param parser
- * @param err
- * @return
- */
- static bool
- ucl_include_url (const unsigned char *data, size_t len,
- struct ucl_parser *parser,
- struct ucl_include_params *params)
- {
-
- bool res;
- unsigned char *buf = NULL;
- size_t buflen = 0;
- struct ucl_chunk *chunk;
- char urlbuf[PATH_MAX];
- int prev_state;
-
- snprintf (urlbuf, sizeof (urlbuf), "%.*s", (int)len, data);
-
- if (!ucl_fetch_url (urlbuf, &buf, &buflen, &parser->err, params->must_exist)) {
- return !params->must_exist;
- }
-
- if (params->check_signature) {
- #if (defined(HAVE_OPENSSL) && OPENSSL_VERSION_NUMBER >= 0x10000000L)
- unsigned char *sigbuf = NULL;
- size_t siglen = 0;
- /* We need to check signature first */
- snprintf (urlbuf, sizeof (urlbuf), "%.*s.sig", (int)len, data);
- if (!ucl_fetch_url (urlbuf, &sigbuf, &siglen, &parser->err, true)) {
- return false;
- }
- if (!ucl_sig_check (buf, buflen, sigbuf, siglen, parser)) {
- ucl_create_err (&parser->err, "cannot verify url %s: %s",
- urlbuf,
- ERR_error_string (ERR_get_error (), NULL));
- if (siglen > 0) {
- ucl_munmap (sigbuf, siglen);
- }
- return false;
- }
- if (siglen > 0) {
- ucl_munmap (sigbuf, siglen);
- }
- #endif
- }
-
- prev_state = parser->state;
- parser->state = UCL_STATE_INIT;
-
- res = ucl_parser_add_chunk_full (parser, buf, buflen, params->priority,
- params->strat, params->parse_type);
- if (res == true) {
- /* Remove chunk from the stack */
- chunk = parser->chunks;
- if (chunk != NULL) {
- parser->chunks = chunk->next;
- ucl_chunk_free (chunk);
- }
- }
-
- parser->state = prev_state;
- free (buf);
-
- return res;
- }
-
- /**
- * Include a single file to the parser
- * @param data
- * @param len
- * @param parser
- * @param check_signature
- * @param must_exist
- * @param allow_glob
- * @param priority
- * @return
- */
- static bool
- ucl_include_file_single (const unsigned char *data, size_t len,
- struct ucl_parser *parser, struct ucl_include_params *params)
- {
- bool res;
- struct ucl_chunk *chunk;
- unsigned char *buf = NULL;
- char *old_curfile, *ext;
- size_t buflen = 0;
- char filebuf[PATH_MAX], realbuf[PATH_MAX];
- int prev_state;
- struct ucl_variable *cur_var, *tmp_var, *old_curdir = NULL,
- *old_filename = NULL;
- ucl_object_t *nest_obj = NULL, *old_obj = NULL, *new_obj = NULL;
- ucl_hash_t *container = NULL;
- struct ucl_stack *st = NULL;
-
- snprintf (filebuf, sizeof (filebuf), "%.*s", (int)len, data);
- if (ucl_realpath (filebuf, realbuf) == NULL) {
- if (params->soft_fail) {
- return false;
- }
- if (!params->must_exist && errno != EPERM) {
- return true;
- }
-
- ucl_create_err (&parser->err, "cannot open file %s: %s",
- filebuf,
- strerror (errno));
- return false;
- }
-
- if (parser->cur_file && strcmp (realbuf, parser->cur_file) == 0) {
- /* We are likely including the file itself */
- if (params->soft_fail) {
- return false;
- }
-
- ucl_create_err (&parser->err, "trying to include the file %s from itself",
- realbuf);
- return false;
- }
-
- if (!ucl_fetch_file (realbuf, &buf, &buflen, &parser->err, params->must_exist)) {
- if (params->soft_fail) {
- return false;
- }
-
- if (params->must_exist || parser->err != NULL) {
- /* The case of fatal errors */
- return false;
- }
-
- return true;
- }
-
- if (params->check_signature) {
- #if (defined(HAVE_OPENSSL) && OPENSSL_VERSION_NUMBER >= 0x10000000L)
- unsigned char *sigbuf = NULL;
- size_t siglen = 0;
- /* We need to check signature first */
- snprintf (filebuf, sizeof (filebuf), "%s.sig", realbuf);
- if (!ucl_fetch_file (filebuf, &sigbuf, &siglen, &parser->err, true)) {
- return false;
- }
- if (!ucl_sig_check (buf, buflen, sigbuf, siglen, parser)) {
- ucl_create_err (&parser->err, "cannot verify file %s: %s",
- filebuf,
- ERR_error_string (ERR_get_error (), NULL));
- if (sigbuf) {
- ucl_munmap (sigbuf, siglen);
- }
- return false;
- }
- if (sigbuf) {
- ucl_munmap (sigbuf, siglen);
- }
- #endif
- }
-
- old_curfile = parser->cur_file;
- parser->cur_file = NULL;
-
- /* Store old file vars */
- DL_FOREACH_SAFE (parser->variables, cur_var, tmp_var) {
- if (strcmp (cur_var->var, "CURDIR") == 0) {
- old_curdir = cur_var;
- DL_DELETE (parser->variables, cur_var);
- }
- else if (strcmp (cur_var->var, "FILENAME") == 0) {
- old_filename = cur_var;
- DL_DELETE (parser->variables, cur_var);
- }
- }
-
- ucl_parser_set_filevars (parser, realbuf, false);
-
- prev_state = parser->state;
- parser->state = UCL_STATE_INIT;
-
- if (params->use_prefix && params->prefix == NULL) {
- /* Auto generate a key name based on the included filename */
- params->prefix = basename (realbuf);
- ext = strrchr (params->prefix, '.');
- if (ext != NULL && (strcmp (ext, ".conf") == 0 || strcmp (ext, ".ucl") == 0)) {
- /* Strip off .conf or .ucl */
- *ext = '\0';
- }
- }
- if (params->prefix != NULL) {
- /* This is a prefixed include */
- container = parser->stack->obj->value.ov;
-
- old_obj = __DECONST (ucl_object_t *, ucl_hash_search (container,
- params->prefix, strlen (params->prefix)));
-
- if (strcasecmp (params->target, "array") == 0) {
- if (old_obj == NULL) {
- /* Create an array with key: prefix */
- old_obj = ucl_object_new_full (UCL_ARRAY, params->priority);
- old_obj->key = params->prefix;
- old_obj->keylen = strlen (params->prefix);
- ucl_copy_key_trash (old_obj);
- old_obj->prev = old_obj;
- old_obj->next = NULL;
-
- container = ucl_hash_insert_object (container, old_obj,
- parser->flags & UCL_PARSER_KEY_LOWERCASE);
- parser->stack->obj->len++;
-
- nest_obj = ucl_object_new_full (UCL_OBJECT, params->priority);
- nest_obj->prev = nest_obj;
- nest_obj->next = NULL;
-
- ucl_array_append (old_obj, nest_obj);
- }
- else {
- if (ucl_object_type (old_obj) == UCL_ARRAY) {
- /* Append to the existing array */
- nest_obj = ucl_object_new_full (UCL_OBJECT,
- params->priority);
- if (nest_obj == NULL) {
- ucl_create_err (&parser->err,
- "cannot allocate memory for an object");
- if (buf) {
- ucl_munmap (buf, buflen);
- }
-
- return false;
- }
- nest_obj->prev = nest_obj;
- nest_obj->next = NULL;
-
- ucl_array_append (old_obj, nest_obj);
- }
- else {
- /* Convert the object to an array */
- new_obj = ucl_object_typed_new (UCL_ARRAY);
- if (new_obj == NULL) {
- ucl_create_err (&parser->err,
- "cannot allocate memory for an object");
- if (buf) {
- ucl_munmap (buf, buflen);
- }
-
- return false;
- }
- new_obj->key = old_obj->key;
- new_obj->keylen = old_obj->keylen;
- new_obj->flags |= UCL_OBJECT_MULTIVALUE;
- new_obj->prev = new_obj;
- new_obj->next = NULL;
-
- nest_obj = ucl_object_new_full (UCL_OBJECT,
- params->priority);
- if (nest_obj == NULL) {
- ucl_create_err (&parser->err,
- "cannot allocate memory for an object");
- if (buf) {
- ucl_munmap (buf, buflen);
- }
-
- return false;
- }
- nest_obj->prev = nest_obj;
- nest_obj->next = NULL;
-
- ucl_array_append (new_obj, old_obj);
- ucl_array_append (new_obj, nest_obj);
- ucl_hash_replace (container, old_obj, new_obj);
- }
- }
- }
- else {
- /* Case of object */
- if (old_obj == NULL) {
- /* Create an object with key: prefix */
- nest_obj = ucl_object_new_full (UCL_OBJECT, params->priority);
-
- if (nest_obj == NULL) {
- ucl_create_err (&parser->err, "cannot allocate memory for an object");
- if (buf) {
- ucl_munmap (buf, buflen);
- }
-
- return false;
- }
-
- nest_obj->key = params->prefix;
- nest_obj->keylen = strlen (params->prefix);
- ucl_copy_key_trash(nest_obj);
- nest_obj->prev = nest_obj;
- nest_obj->next = NULL;
-
- container = ucl_hash_insert_object (container, nest_obj,
- parser->flags & UCL_PARSER_KEY_LOWERCASE);
- parser->stack->obj->len ++;
- }
- else {
- if (ucl_object_type (old_obj) == UCL_OBJECT) {
- /* Append to existing Object*/
- nest_obj = old_obj;
- }
- else {
- /* The key is not an object */
- ucl_create_err (&parser->err,
- "Conflicting type for key: %s, asked %s, has %s",
- params->prefix, params->target,
- ucl_object_type_to_string (ucl_object_type (old_obj)));
- if (buf) {
- ucl_munmap (buf, buflen);
- }
-
- return false;
- }
- }
- }
-
-
- /* Put all of the content of the include inside that object */
- parser->stack->obj->value.ov = container;
-
- st = UCL_ALLOC (sizeof (struct ucl_stack));
- if (st == NULL) {
- ucl_create_err (&parser->err, "cannot allocate memory for an object");
- ucl_object_unref (nest_obj);
-
- if (buf) {
- ucl_munmap (buf, buflen);
- }
-
- return false;
- }
- st->obj = nest_obj;
- st->e.params.level = parser->stack->e.params.level;
- st->e.params.flags = parser->stack->e.params.flags;
- st->e.params.line = parser->stack->e.params.line;
- st->chunk = parser->chunks;
- LL_PREPEND (parser->stack, st);
- parser->cur_obj = nest_obj;
- }
-
- res = ucl_parser_add_chunk_full (parser, buf, buflen, params->priority,
- params->strat, params->parse_type);
-
- if (res) {
- /* Stop nesting the include, take 1 level off the stack */
- if (params->prefix != NULL && nest_obj != NULL) {
- parser->stack = st->next;
- UCL_FREE (sizeof (struct ucl_stack), st);
- }
-
- /* Remove chunk from the stack */
- chunk = parser->chunks;
- if (chunk != NULL) {
- parser->chunks = chunk->next;
- ucl_chunk_free (chunk);
- parser->recursion--;
- }
-
- /* Restore old file vars */
- if (parser->cur_file) {
- free (parser->cur_file);
- }
-
- parser->cur_file = old_curfile;
- DL_FOREACH_SAFE (parser->variables, cur_var, tmp_var) {
- if (strcmp (cur_var->var, "CURDIR") == 0 && old_curdir) {
- DL_DELETE (parser->variables, cur_var);
- free (cur_var->var);
- free (cur_var->value);
- UCL_FREE (sizeof (struct ucl_variable), cur_var);
- } else if (strcmp (cur_var->var, "FILENAME") == 0 && old_filename) {
- DL_DELETE (parser->variables, cur_var);
- free (cur_var->var);
- free (cur_var->value);
- UCL_FREE (sizeof (struct ucl_variable), cur_var);
- }
- }
- if (old_filename) {
- DL_APPEND (parser->variables, old_filename);
- }
- if (old_curdir) {
- DL_APPEND (parser->variables, old_curdir);
- }
-
- parser->state = prev_state;
- }
-
- if (buflen > 0) {
- ucl_munmap (buf, buflen);
- }
-
- return res;
- }
-
- /**
- * Include a file to configuration
- * @param data
- * @param len
- * @param parser
- * @param err
- * @return
- */
- static bool
- ucl_include_file (const unsigned char *data, size_t len,
- struct ucl_parser *parser,
- struct ucl_include_params *params,
- const ucl_object_t *args)
- {
- const unsigned char *p = data, *end = data + len;
- bool need_glob = false;
- int cnt = 0;
- char glob_pattern[PATH_MAX];
- size_t i;
-
- #ifndef _WIN32
- if (!params->allow_glob) {
- return ucl_include_file_single (data, len, parser, params);
- }
- else {
- /* Check for special symbols in a filename */
- while (p != end) {
- if (*p == '*' || *p == '?') {
- need_glob = true;
- break;
- }
- p ++;
- }
- if (need_glob) {
- glob_t globbuf;
- memset (&globbuf, 0, sizeof (globbuf));
- ucl_strlcpy (glob_pattern, (const char *)data,
- (len + 1 < sizeof (glob_pattern) ? len + 1 : sizeof (glob_pattern)));
- if (glob (glob_pattern, 0, NULL, &globbuf) != 0) {
- return (!params->must_exist || false);
- }
- for (i = 0; i < globbuf.gl_pathc; i ++) {
-
- if (parser->include_trace_func) {
- const ucl_object_t *parent = NULL;
-
- if (parser->stack) {
- parent = parser->stack->obj;
- }
-
- parser->include_trace_func (parser, parent, NULL,
- globbuf.gl_pathv[i],
- strlen (globbuf.gl_pathv[i]),
- parser->include_trace_ud);
- }
-
- if (!ucl_include_file_single ((unsigned char *)globbuf.gl_pathv[i],
- strlen (globbuf.gl_pathv[i]), parser, params)) {
- if (params->soft_fail) {
- continue;
- }
- globfree (&globbuf);
- return false;
- }
- cnt ++;
- }
- globfree (&globbuf);
-
- if (cnt == 0 && params->must_exist) {
- ucl_create_err (&parser->err, "cannot match any files for pattern %s",
- glob_pattern);
- return false;
- }
- }
- else {
- return ucl_include_file_single (data, len, parser, params);
- }
- }
- #else
- /* Win32 compilers do not support globbing. Therefore, for Win32,
- treat allow_glob/need_glob as a NOOP and just return */
- return ucl_include_file_single (data, len, parser, params);
- #endif
-
- return true;
- }
-
- /**
- * Common function to handle .*include* macros
- * @param data
- * @param len
- * @param args
- * @param parser
- * @param default_try
- * @param default_sign
- * @return
- */
- static bool
- ucl_include_common (const unsigned char *data, size_t len,
- const ucl_object_t *args, struct ucl_parser *parser,
- bool default_try,
- bool default_sign)
- {
- bool allow_url = false, search = false;
- const char *duplicate;
- const ucl_object_t *param;
- ucl_object_iter_t it = NULL, ip = NULL;
- char ipath[PATH_MAX];
- struct ucl_include_params params;
-
- /* Default values */
- params.soft_fail = default_try;
- params.allow_glob = false;
- params.check_signature = default_sign;
- params.use_prefix = false;
- params.target = "object";
- params.prefix = NULL;
- params.priority = 0;
- params.parse_type = UCL_PARSE_UCL;
- params.strat = UCL_DUPLICATE_APPEND;
- params.must_exist = !default_try;
-
- if (parser->include_trace_func) {
- const ucl_object_t *parent = NULL;
-
- if (parser->stack) {
- parent = parser->stack->obj;
- }
-
- parser->include_trace_func (parser, parent, args,
- data, len, parser->include_trace_ud);
- }
-
- /* Process arguments */
- if (args != NULL && args->type == UCL_OBJECT) {
- while ((param = ucl_object_iterate (args, &it, true)) != NULL) {
- if (param->type == UCL_BOOLEAN) {
- if (strncmp (param->key, "try", param->keylen) == 0) {
- params.must_exist = !ucl_object_toboolean (param);
- }
- else if (strncmp (param->key, "sign", param->keylen) == 0) {
- params.check_signature = ucl_object_toboolean (param);
- }
- else if (strncmp (param->key, "glob", param->keylen) == 0) {
- params.allow_glob = ucl_object_toboolean (param);
- }
- else if (strncmp (param->key, "url", param->keylen) == 0) {
- allow_url = ucl_object_toboolean (param);
- }
- else if (strncmp (param->key, "prefix", param->keylen) == 0) {
- params.use_prefix = ucl_object_toboolean (param);
- }
- }
- else if (param->type == UCL_STRING) {
- if (strncmp (param->key, "key", param->keylen) == 0) {
- params.prefix = ucl_object_tostring (param);
- }
- else if (strncmp (param->key, "target", param->keylen) == 0) {
- params.target = ucl_object_tostring (param);
- }
- else if (strncmp (param->key, "duplicate", param->keylen) == 0) {
- duplicate = ucl_object_tostring (param);
-
- if (strcmp (duplicate, "append") == 0) {
- params.strat = UCL_DUPLICATE_APPEND;
- }
- else if (strcmp (duplicate, "merge") == 0) {
- params.strat = UCL_DUPLICATE_MERGE;
- }
- else if (strcmp (duplicate, "rewrite") == 0) {
- params.strat = UCL_DUPLICATE_REWRITE;
- }
- else if (strcmp (duplicate, "error") == 0) {
- params.strat = UCL_DUPLICATE_ERROR;
- }
- }
- }
- else if (param->type == UCL_ARRAY) {
- if (strncmp (param->key, "path", param->keylen) == 0) {
- ucl_set_include_path (parser, __DECONST(ucl_object_t *, param));
- }
- }
- else if (param->type == UCL_INT) {
- if (strncmp (param->key, "priority", param->keylen) == 0) {
- params.priority = ucl_object_toint (param);
- }
- }
- }
- }
-
- if (parser->includepaths == NULL) {
- if (allow_url && ucl_strnstr (data, "://", len) != NULL) {
- /* Globbing is not used for URL's */
- return ucl_include_url (data, len, parser, ¶ms);
- }
- else if (data != NULL) {
- /* Try to load a file */
- return ucl_include_file (data, len, parser, ¶ms, args);
- }
- }
- else {
- if (allow_url && ucl_strnstr (data, "://", len) != NULL) {
- /* Globbing is not used for URL's */
- return ucl_include_url (data, len, parser, ¶ms);
- }
-
- ip = ucl_object_iterate_new (parser->includepaths);
- while ((param = ucl_object_iterate_safe (ip, true)) != NULL) {
- if (ucl_object_type(param) == UCL_STRING) {
- snprintf (ipath, sizeof (ipath), "%s/%.*s", ucl_object_tostring(param),
- (int)len, data);
- if ((search = ucl_include_file (ipath, strlen (ipath),
- parser, ¶ms, args))) {
- if (!params.allow_glob) {
- break;
- }
- }
- }
- }
- ucl_object_iterate_free (ip);
- if (search == true) {
- return true;
- }
- else {
- ucl_create_err (&parser->err,
- "cannot find file: %.*s in search path",
- (int)len, data);
- return false;
- }
- }
-
- return false;
- }
-
- /**
- * Handle include macro
- * @param data include data
- * @param len length of data
- * @param args UCL object representing arguments to the macro
- * @param ud user data
- * @return
- */
- bool
- ucl_include_handler (const unsigned char *data, size_t len,
- const ucl_object_t *args, void* ud)
- {
- struct ucl_parser *parser = ud;
-
- return ucl_include_common (data, len, args, parser, false, false);
- }
-
- /**
- * Handle includes macro
- * @param data include data
- * @param len length of data
- * @param args UCL object representing arguments to the macro
- * @param ud user data
- * @return
- */
- bool
- ucl_includes_handler (const unsigned char *data, size_t len,
- const ucl_object_t *args, void* ud)
- {
- struct ucl_parser *parser = ud;
-
- return ucl_include_common (data, len, args, parser, false, true);
- }
-
- /**
- * Handle tryinclude macro
- * @param data include data
- * @param len length of data
- * @param args UCL object representing arguments to the macro
- * @param ud user data
- * @return
- */
- bool
- ucl_try_include_handler (const unsigned char *data, size_t len,
- const ucl_object_t *args, void* ud)
- {
- struct ucl_parser *parser = ud;
-
- return ucl_include_common (data, len, args, parser, true, false);
- }
-
- /**
- * Handle priority macro
- * @param data include data
- * @param len length of data
- * @param args UCL object representing arguments to the macro
- * @param ud user data
- * @return
- */
- bool
- ucl_priority_handler (const unsigned char *data, size_t len,
- const ucl_object_t *args, void* ud)
- {
- struct ucl_parser *parser = ud;
- unsigned priority = 255;
- const ucl_object_t *param;
- bool found = false;
- char *value = NULL, *leftover = NULL;
- ucl_object_iter_t it = NULL;
-
- if (parser == NULL) {
- return false;
- }
-
- /* Process arguments */
- if (args != NULL && args->type == UCL_OBJECT) {
- while ((param = ucl_object_iterate (args, &it, true)) != NULL) {
- if (param->type == UCL_INT) {
- if (strncmp (param->key, "priority", param->keylen) == 0) {
- priority = ucl_object_toint (param);
- found = true;
- }
- }
- }
- }
-
- if (len > 0) {
- value = malloc(len + 1);
- ucl_strlcpy(value, (const char *)data, len + 1);
- priority = strtol(value, &leftover, 10);
- if (*leftover != '\0') {
- ucl_create_err (&parser->err, "Invalid priority value in macro: %s",
- value);
- free(value);
- return false;
- }
- free(value);
- found = true;
- }
-
- if (found == true) {
- parser->chunks->priority = priority;
- return true;
- }
-
- ucl_create_err (&parser->err, "Unable to parse priority macro");
- return false;
- }
-
- /**
- * Handle load macro
- * @param data include data
- * @param len length of data
- * @param args UCL object representing arguments to the macro
- * @param ud user data
- * @return
- */
- bool
- ucl_load_handler (const unsigned char *data, size_t len,
- const ucl_object_t *args, void* ud)
- {
- struct ucl_parser *parser = ud;
- const ucl_object_t *param;
- ucl_object_t *obj, *old_obj;
- ucl_object_iter_t it = NULL;
- bool try_load, multiline, test;
- const char *target, *prefix;
- char *load_file, *tmp;
- unsigned char *buf;
- size_t buflen;
- unsigned priority;
- int64_t iv;
- ucl_object_t *container = NULL;
- enum ucl_string_flags flags;
-
- /* Default values */
- try_load = false;
- multiline = false;
- test = false;
- target = "string";
- prefix = NULL;
- load_file = NULL;
- buf = NULL;
- buflen = 0;
- priority = 0;
- obj = NULL;
- old_obj = NULL;
- flags = 0;
-
- if (parser == NULL) {
- return false;
- }
-
- /* Process arguments */
- if (args != NULL && args->type == UCL_OBJECT) {
- while ((param = ucl_object_iterate (args, &it, true)) != NULL) {
- if (param->type == UCL_BOOLEAN) {
- if (strncmp (param->key, "try", param->keylen) == 0) {
- try_load = ucl_object_toboolean (param);
- }
- else if (strncmp (param->key, "multiline", param->keylen) == 0) {
- multiline = ucl_object_toboolean (param);
- }
- else if (strncmp (param->key, "escape", param->keylen) == 0) {
- test = ucl_object_toboolean (param);
- if (test) {
- flags |= UCL_STRING_ESCAPE;
- }
- }
- else if (strncmp (param->key, "trim", param->keylen) == 0) {
- test = ucl_object_toboolean (param);
- if (test) {
- flags |= UCL_STRING_TRIM;
- }
- }
- }
- else if (param->type == UCL_STRING) {
- if (strncmp (param->key, "key", param->keylen) == 0) {
- prefix = ucl_object_tostring (param);
- }
- else if (strncmp (param->key, "target", param->keylen) == 0) {
- target = ucl_object_tostring (param);
- }
- }
- else if (param->type == UCL_INT) {
- if (strncmp (param->key, "priority", param->keylen) == 0) {
- priority = ucl_object_toint (param);
- }
- }
- }
- }
-
- if (prefix == NULL || strlen (prefix) == 0) {
- ucl_create_err (&parser->err, "No Key specified in load macro");
- return false;
- }
-
- if (len > 0) {
- load_file = malloc (len + 1);
- if (!load_file) {
- ucl_create_err (&parser->err, "cannot allocate memory for suffix");
-
- return false;
- }
-
- snprintf (load_file, len + 1, "%.*s", (int)len, data);
-
- if (!ucl_fetch_file (load_file, &buf, &buflen, &parser->err,
- !try_load)) {
- free (load_file);
-
- return (try_load || false);
- }
-
- free (load_file);
- container = parser->stack->obj;
- old_obj = __DECONST (ucl_object_t *, ucl_object_lookup (container,
- prefix));
-
- if (old_obj != NULL) {
- ucl_create_err (&parser->err, "Key %s already exists", prefix);
- if (buf) {
- ucl_munmap (buf, buflen);
- }
-
- return false;
- }
-
- if (strcasecmp (target, "string") == 0) {
- obj = ucl_object_fromstring_common (buf, buflen, flags);
- ucl_copy_value_trash (obj);
- if (multiline) {
- obj->flags |= UCL_OBJECT_MULTILINE;
- }
- }
- else if (strcasecmp (target, "int") == 0) {
- tmp = malloc (buflen + 1);
-
- if (tmp == NULL) {
- ucl_create_err (&parser->err, "Memory allocation failed");
- if (buf) {
- ucl_munmap (buf, buflen);
- }
-
- return false;
- }
-
- snprintf (tmp, buflen + 1, "%.*s", (int)buflen, buf);
- iv = strtoll (tmp, NULL, 10);
- obj = ucl_object_fromint (iv);
- free (tmp);
- }
-
- if (buf) {
- ucl_munmap (buf, buflen);
- }
-
- if (obj != NULL) {
- obj->key = prefix;
- obj->keylen = strlen (prefix);
- ucl_copy_key_trash (obj);
- obj->prev = obj;
- obj->next = NULL;
- ucl_object_set_priority (obj, priority);
- ucl_object_insert_key (container, obj, obj->key, obj->keylen, false);
- }
-
- return true;
- }
-
- ucl_create_err (&parser->err, "Unable to parse load macro");
- return false;
- }
-
- bool
- ucl_inherit_handler (const unsigned char *data, size_t len,
- const ucl_object_t *args, const ucl_object_t *ctx, void* ud)
- {
- const ucl_object_t *parent, *cur;
- ucl_object_t *target, *copy;
- ucl_object_iter_t it = NULL;
- bool replace = false;
- struct ucl_parser *parser = ud;
-
- parent = ucl_object_lookup_len (ctx, data, len);
-
- /* Some sanity checks */
- if (parent == NULL || ucl_object_type (parent) != UCL_OBJECT) {
- ucl_create_err (&parser->err, "Unable to find inherited object %*.s",
- (int)len, data);
- return false;
- }
-
- if (parser->stack == NULL || parser->stack->obj == NULL ||
- ucl_object_type (parser->stack->obj) != UCL_OBJECT) {
- ucl_create_err (&parser->err, "Invalid inherit context");
- return false;
- }
-
- target = parser->stack->obj;
-
- if (args && (cur = ucl_object_lookup (args, "replace")) != NULL) {
- replace = ucl_object_toboolean (cur);
- }
-
- while ((cur = ucl_object_iterate (parent, &it, true))) {
- /* We do not replace existing keys */
- if (!replace && ucl_object_lookup_len (target, cur->key, cur->keylen)) {
- continue;
- }
-
- copy = ucl_object_copy (cur);
-
- if (!replace) {
- copy->flags |= UCL_OBJECT_INHERITED;
- }
-
- ucl_object_insert_key (target, copy, copy->key,
- copy->keylen, false);
- }
-
- return true;
- }
-
- bool
- ucl_parser_set_filevars (struct ucl_parser *parser, const char *filename, bool need_expand)
- {
- char realbuf[PATH_MAX], *curdir;
-
- if (filename != NULL) {
- if (need_expand) {
- if (ucl_realpath (filename, realbuf) == NULL) {
- return false;
- }
- }
- else {
- ucl_strlcpy (realbuf, filename, sizeof (realbuf));
- }
-
- if (parser->cur_file) {
- free (parser->cur_file);
- }
-
- parser->cur_file = strdup (realbuf);
-
- /* Define variables */
- ucl_parser_register_variable (parser, "FILENAME", realbuf);
- curdir = dirname (realbuf);
- ucl_parser_register_variable (parser, "CURDIR", curdir);
- }
- else {
- /* Set everything from the current dir */
- curdir = getcwd (realbuf, sizeof (realbuf));
- ucl_parser_register_variable (parser, "FILENAME", "undef");
- ucl_parser_register_variable (parser, "CURDIR", curdir);
- }
-
- return true;
- }
-
- bool
- ucl_parser_add_file_full (struct ucl_parser *parser, const char *filename,
- unsigned priority, enum ucl_duplicate_strategy strat,
- enum ucl_parse_type parse_type)
- {
- unsigned char *buf;
- size_t len;
- bool ret;
- char realbuf[PATH_MAX];
-
- if (ucl_realpath (filename, realbuf) == NULL) {
- ucl_create_err (&parser->err, "cannot open file %s: %s",
- filename,
- strerror (errno));
- return false;
- }
-
- if (!ucl_fetch_file (realbuf, &buf, &len, &parser->err, true)) {
- return false;
- }
-
- ucl_parser_set_filevars (parser, realbuf, false);
- ret = ucl_parser_add_chunk_full (parser, buf, len, priority, strat,
- parse_type);
-
- if (len > 0) {
- ucl_munmap (buf, len);
- }
-
- return ret;
- }
-
- bool
- ucl_parser_add_file_priority (struct ucl_parser *parser, const char *filename,
- unsigned priority)
- {
- if (parser == NULL) {
- return false;
- }
-
- return ucl_parser_add_file_full(parser, filename, priority,
- UCL_DUPLICATE_APPEND, UCL_PARSE_UCL);
- }
-
- bool
- ucl_parser_add_file (struct ucl_parser *parser, const char *filename)
- {
- if (parser == NULL) {
- return false;
- }
-
- return ucl_parser_add_file_full(parser, filename,
- parser->default_priority, UCL_DUPLICATE_APPEND,
- UCL_PARSE_UCL);
- }
-
-
- bool
- ucl_parser_add_fd_full (struct ucl_parser *parser, int fd,
- unsigned priority, enum ucl_duplicate_strategy strat,
- enum ucl_parse_type parse_type)
- {
- unsigned char *buf;
- size_t len;
- bool ret;
- struct stat st;
-
- if (fstat (fd, &st) == -1) {
- ucl_create_err (&parser->err, "cannot stat fd %d: %s",
- fd, strerror (errno));
- return false;
- }
- if (st.st_size == 0) {
- return true;
- }
- if ((buf = ucl_mmap (NULL, st.st_size, PROT_READ, MAP_SHARED, fd, 0)) == MAP_FAILED) {
- ucl_create_err (&parser->err, "cannot mmap fd %d: %s",
- fd, strerror (errno));
- return false;
- }
-
- if (parser->cur_file) {
- free (parser->cur_file);
- }
- parser->cur_file = NULL;
- len = st.st_size;
- ret = ucl_parser_add_chunk_full (parser, buf, len, priority, strat,
- parse_type);
-
- if (len > 0) {
- ucl_munmap (buf, len);
- }
-
- return ret;
- }
-
- bool
- ucl_parser_add_fd_priority (struct ucl_parser *parser, int fd,
- unsigned priority)
- {
- if (parser == NULL) {
- return false;
- }
-
- return ucl_parser_add_fd_full(parser, fd, parser->default_priority,
- UCL_DUPLICATE_APPEND, UCL_PARSE_UCL);
- }
-
- bool
- ucl_parser_add_fd (struct ucl_parser *parser, int fd)
- {
- if (parser == NULL) {
- return false;
- }
-
- return ucl_parser_add_fd_priority(parser, fd, parser->default_priority);
- }
-
- size_t
- ucl_strlcpy (char *dst, const char *src, size_t siz)
- {
- char *d = dst;
- const char *s = src;
- size_t n = siz;
-
- /* Copy as many bytes as will fit */
- if (n != 0) {
- while (--n != 0) {
- if ((*d++ = *s++) == '\0') {
- break;
- }
- }
- }
-
- if (n == 0 && siz != 0) {
- *d = '\0';
- }
-
- return (s - src - 1); /* count does not include NUL */
- }
-
- size_t
- ucl_strlcpy_unsafe (char *dst, const char *src, size_t siz)
- {
- memcpy (dst, src, siz - 1);
- dst[siz - 1] = '\0';
-
- return siz - 1;
- }
-
- size_t
- ucl_strlcpy_tolower (char *dst, const char *src, size_t siz)
- {
- char *d = dst;
- const char *s = src;
- size_t n = siz;
-
- /* Copy as many bytes as will fit */
- if (n != 0) {
- while (--n != 0) {
- if ((*d++ = tolower (*s++)) == '\0') {
- break;
- }
- }
- }
-
- if (n == 0 && siz != 0) {
- *d = '\0';
- }
-
- return (s - src); /* count does not include NUL */
- }
-
- /*
- * Find the first occurrence of find in s
- */
- char *
- ucl_strnstr (const char *s, const char *find, int len)
- {
- char c, sc;
- int mlen;
-
- if ((c = *find++) != 0) {
- mlen = strlen (find);
- do {
- do {
- if ((sc = *s++) == 0 || len-- == 0)
- return (NULL);
- } while (sc != c);
- } while (strncmp (s, find, mlen) != 0);
- s--;
- }
- return ((char *)s);
- }
-
- /*
- * Find the first occurrence of find in s, ignore case.
- */
- char *
- ucl_strncasestr (const char *s, const char *find, int len)
- {
- char c, sc;
- int mlen;
-
- if ((c = *find++) != 0) {
- c = tolower (c);
- mlen = strlen (find);
- do {
- do {
- if ((sc = *s++) == 0 || len-- == 0)
- return (NULL);
- } while (tolower (sc) != c);
- } while (strncasecmp (s, find, mlen) != 0);
- s--;
- }
- return ((char *)s);
- }
-
- ucl_object_t *
- ucl_object_fromstring_common (const char *str, size_t len, enum ucl_string_flags flags)
- {
- ucl_object_t *obj;
- const char *start, *end, *p, *pos;
- char *dst, *d;
- size_t escaped_len;
-
- if (str == NULL) {
- return NULL;
- }
-
- obj = ucl_object_new ();
- if (obj) {
- if (len == 0) {
- len = strlen (str);
- }
- if (flags & UCL_STRING_TRIM) {
- /* Skip leading spaces */
- for (start = str; (size_t)(start - str) < len; start ++) {
- if (!ucl_test_character (*start, UCL_CHARACTER_WHITESPACE_UNSAFE)) {
- break;
- }
- }
- /* Skip trailing spaces */
- for (end = str + len - 1; end > start; end --) {
- if (!ucl_test_character (*end, UCL_CHARACTER_WHITESPACE_UNSAFE)) {
- break;
- }
- }
- end ++;
- }
- else {
- start = str;
- end = str + len;
- }
-
- obj->type = UCL_STRING;
- if (flags & UCL_STRING_ESCAPE) {
- for (p = start, escaped_len = 0; p < end; p ++, escaped_len ++) {
- if (ucl_test_character (*p, UCL_CHARACTER_JSON_UNSAFE | UCL_CHARACTER_WHITESPACE_UNSAFE)) {
- switch (*p) {
- case '\v':
- case '\0':
- escaped_len += 5;
- break;
- case ' ':
- break;
- default:
- escaped_len ++;
- break;
- }
- }
- }
- dst = malloc (escaped_len + 1);
- if (dst != NULL) {
- for (p = start, d = dst; p < end; p ++, d ++) {
- if (ucl_test_character (*p, UCL_CHARACTER_JSON_UNSAFE | UCL_CHARACTER_WHITESPACE_UNSAFE)) {
- switch (*p) {
- case '\n':
- *d++ = '\\';
- *d = 'n';
- break;
- case '\r':
- *d++ = '\\';
- *d = 'r';
- break;
- case '\b':
- *d++ = '\\';
- *d = 'b';
- break;
- case '\t':
- *d++ = '\\';
- *d = 't';
- break;
- case '\f':
- *d++ = '\\';
- *d = 'f';
- break;
- case '\0':
- *d++ = '\\';
- *d++ = 'u';
- *d++ = '0';
- *d++ = '0';
- *d++ = '0';
- *d = '0';
- break;
- case '\v':
- *d++ = '\\';
- *d++ = 'u';
- *d++ = '0';
- *d++ = '0';
- *d++ = '0';
- *d = 'B';
- break;
- case '\\':
- *d++ = '\\';
- *d = '\\';
- break;
- case ' ':
- *d = ' ';
- break;
- case '"':
- *d++ = '\\';
- *d = '"';
- break;
- }
- }
- else {
- *d = *p;
- }
- }
- *d = '\0';
- obj->value.sv = dst;
- obj->trash_stack[UCL_TRASH_VALUE] = dst;
- obj->len = escaped_len;
- }
- }
- else {
- dst = malloc (end - start + 1);
- if (dst != NULL) {
- ucl_strlcpy_unsafe (dst, start, end - start + 1);
- obj->value.sv = dst;
- obj->trash_stack[UCL_TRASH_VALUE] = dst;
- obj->len = end - start;
- }
- }
- if ((flags & UCL_STRING_PARSE) && dst != NULL) {
- /* Parse what we have */
- if (flags & UCL_STRING_PARSE_BOOLEAN) {
- if (!ucl_maybe_parse_boolean (obj, dst, obj->len) && (flags & UCL_STRING_PARSE_NUMBER)) {
- ucl_maybe_parse_number (obj, dst, dst + obj->len, &pos,
- flags & UCL_STRING_PARSE_DOUBLE,
- flags & UCL_STRING_PARSE_BYTES,
- flags & UCL_STRING_PARSE_TIME);
- }
- }
- else {
- ucl_maybe_parse_number (obj, dst, dst + obj->len, &pos,
- flags & UCL_STRING_PARSE_DOUBLE,
- flags & UCL_STRING_PARSE_BYTES,
- flags & UCL_STRING_PARSE_TIME);
- }
- }
- }
-
- return obj;
- }
-
- static bool
- ucl_object_insert_key_common (ucl_object_t *top, ucl_object_t *elt,
- const char *key, size_t keylen, bool copy_key, bool merge, bool replace)
- {
- ucl_object_t *found, *tmp;
- const ucl_object_t *cur;
- ucl_object_iter_t it = NULL;
- const char *p;
- int ret = true;
-
- if (elt == NULL || key == NULL) {
- return false;
- }
-
- if (top == NULL) {
- return false;
- }
-
- if (top->type != UCL_OBJECT) {
- /* It is possible to convert NULL type to an object */
- if (top->type == UCL_NULL) {
- top->type = UCL_OBJECT;
- }
- else {
- /* Refuse converting of other object types */
- return false;
- }
- }
-
- if (top->value.ov == NULL) {
- top->value.ov = ucl_hash_create (false);
- }
-
- if (keylen == 0) {
- keylen = strlen (key);
- }
-
- for (p = key; p < key + keylen; p ++) {
- if (ucl_test_character (*p, UCL_CHARACTER_UCL_UNSAFE)) {
- elt->flags |= UCL_OBJECT_NEED_KEY_ESCAPE;
- break;
- }
- }
-
- /* workaround for some use cases */
- if (elt->trash_stack[UCL_TRASH_KEY] != NULL &&
- key != (const char *)elt->trash_stack[UCL_TRASH_KEY]) {
- /* Remove copied key */
- free (elt->trash_stack[UCL_TRASH_KEY]);
- elt->trash_stack[UCL_TRASH_KEY] = NULL;
- elt->flags &= ~UCL_OBJECT_ALLOCATED_KEY;
- }
-
- elt->key = key;
- elt->keylen = keylen;
-
- if (copy_key) {
- ucl_copy_key_trash (elt);
- }
-
- found = __DECONST (ucl_object_t *, ucl_hash_search_obj (top->value.ov, elt));
-
- if (found == NULL) {
- top->value.ov = ucl_hash_insert_object (top->value.ov, elt, false);
- top->len ++;
- if (replace) {
- ret = false;
- }
- }
- else {
- if (replace) {
- ucl_hash_replace (top->value.ov, found, elt);
- ucl_object_unref (found);
- }
- else if (merge) {
- if (found->type != UCL_OBJECT && elt->type == UCL_OBJECT) {
- /* Insert old elt to new one */
- ucl_object_insert_key_common (elt, found, found->key,
- found->keylen, copy_key, false, false);
- ucl_hash_delete (top->value.ov, found);
- top->value.ov = ucl_hash_insert_object (top->value.ov, elt, false);
- }
- else if (found->type == UCL_OBJECT && elt->type != UCL_OBJECT) {
- /* Insert new to old */
- ucl_object_insert_key_common (found, elt, elt->key,
- elt->keylen, copy_key, false, false);
- }
- else if (found->type == UCL_OBJECT && elt->type == UCL_OBJECT) {
- /* Mix two hashes */
- while ((cur = ucl_object_iterate (elt, &it, true)) != NULL) {
- tmp = ucl_object_ref (cur);
- ucl_object_insert_key_common (found, tmp, cur->key,
- cur->keylen, copy_key, true, false);
- }
- ucl_object_unref (elt);
- }
- else {
- /* Just make a list of scalars */
- DL_CONCAT (found, elt);
- }
- }
- else {
- DL_CONCAT (found, elt);
- }
- }
-
- return ret;
- }
-
- bool
- ucl_object_delete_keyl (ucl_object_t *top, const char *key, size_t keylen)
- {
- ucl_object_t *found;
-
- if (top == NULL || key == NULL) {
- return false;
- }
-
- found = __DECONST (ucl_object_t *, ucl_object_lookup_len (top, key, keylen));
-
- if (found == NULL) {
- return false;
- }
-
- ucl_hash_delete (top->value.ov, found);
- ucl_object_unref (found);
- top->len --;
-
- return true;
- }
-
- bool
- ucl_object_delete_key (ucl_object_t *top, const char *key)
- {
- return ucl_object_delete_keyl (top, key, strlen (key));
- }
-
- ucl_object_t*
- ucl_object_pop_keyl (ucl_object_t *top, const char *key, size_t keylen)
- {
- const ucl_object_t *found;
-
- if (top == NULL || key == NULL) {
- return false;
- }
- found = ucl_object_lookup_len (top, key, keylen);
-
- if (found == NULL) {
- return NULL;
- }
- ucl_hash_delete (top->value.ov, found);
- top->len --;
-
- return __DECONST (ucl_object_t *, found);
- }
-
- ucl_object_t*
- ucl_object_pop_key (ucl_object_t *top, const char *key)
- {
- return ucl_object_pop_keyl (top, key, strlen (key));
- }
-
- bool
- ucl_object_insert_key (ucl_object_t *top, ucl_object_t *elt,
- const char *key, size_t keylen, bool copy_key)
- {
- return ucl_object_insert_key_common (top, elt, key, keylen, copy_key, false, false);
- }
-
- bool
- ucl_object_insert_key_merged (ucl_object_t *top, ucl_object_t *elt,
- const char *key, size_t keylen, bool copy_key)
- {
- return ucl_object_insert_key_common (top, elt, key, keylen, copy_key, true, false);
- }
-
- bool
- ucl_object_replace_key (ucl_object_t *top, ucl_object_t *elt,
- const char *key, size_t keylen, bool copy_key)
- {
- return ucl_object_insert_key_common (top, elt, key, keylen, copy_key, false, true);
- }
-
- bool
- ucl_object_merge (ucl_object_t *top, ucl_object_t *elt, bool copy)
- {
- ucl_object_t *cur = NULL, *cp = NULL, *found = NULL;
- ucl_object_iter_t iter = NULL;
-
- if (top == NULL || elt == NULL) {
- return false;
- }
-
- if (top->type == UCL_ARRAY) {
- if (elt->type == UCL_ARRAY) {
- /* Merge two arrays */
- return ucl_array_merge (top, elt, copy);
- }
- else {
- if (copy) {
- ucl_array_append (top, ucl_object_copy (elt));
-
- return true;
- }
- else {
- ucl_array_append (top, ucl_object_ref (elt));
-
- return true;
- }
- }
- }
- else if (top->type == UCL_OBJECT) {
- if (elt->type == UCL_OBJECT) {
- /* Mix two hashes */
- while ((cur = (ucl_object_t *) ucl_hash_iterate (elt->value.ov,
- &iter))) {
-
- if (copy) {
- cp = ucl_object_copy (cur);
- } else {
- cp = ucl_object_ref (cur);
- }
-
- found = __DECONST(ucl_object_t *,
- ucl_hash_search (top->value.ov, cp->key, cp->keylen));
-
- if (found == NULL) {
- /* The key does not exist */
- top->value.ov = ucl_hash_insert_object (top->value.ov, cp,
- false);
- top->len++;
- }
- else {
- /* The key already exists, merge it recursively */
- if (found->type == UCL_OBJECT || found->type == UCL_ARRAY) {
- if (!ucl_object_merge (found, cp, copy)) {
- return false;
- }
- }
- else {
- ucl_hash_replace (top->value.ov, found, cp);
- ucl_object_unref (found);
- }
- }
- }
- }
- else {
- if (copy) {
- cp = ucl_object_copy (elt);
- }
- else {
- cp = ucl_object_ref (elt);
- }
-
- found = __DECONST(ucl_object_t *,
- ucl_hash_search (top->value.ov, cp->key, cp->keylen));
-
- if (found == NULL) {
- /* The key does not exist */
- top->value.ov = ucl_hash_insert_object (top->value.ov, cp,
- false);
- top->len++;
- }
- else {
- /* The key already exists, merge it recursively */
- if (found->type == UCL_OBJECT || found->type == UCL_ARRAY) {
- if (!ucl_object_merge (found, cp, copy)) {
- return false;
- }
- }
- else {
- ucl_hash_replace (top->value.ov, found, cp);
- ucl_object_unref (found);
- }
- }
- }
- }
- else {
- /* Cannot merge trivial objects */
- return false;
- }
-
- return true;
- }
-
- const ucl_object_t *
- ucl_object_lookup_len (const ucl_object_t *obj, const char *key, size_t klen)
- {
- const ucl_object_t *ret;
- ucl_object_t srch;
-
- if (obj == NULL || obj->type != UCL_OBJECT || key == NULL) {
- return NULL;
- }
-
- srch.key = key;
- srch.keylen = klen;
- ret = ucl_hash_search_obj (obj->value.ov, &srch);
-
- return ret;
- }
-
- const ucl_object_t *
- ucl_object_lookup (const ucl_object_t *obj, const char *key)
- {
- if (key == NULL) {
- return NULL;
- }
-
- return ucl_object_lookup_len (obj, key, strlen (key));
- }
-
- const ucl_object_t*
- ucl_object_lookup_any (const ucl_object_t *obj,
- const char *key, ...)
- {
- va_list ap;
- const ucl_object_t *ret = NULL;
- const char *nk = NULL;
-
- if (obj == NULL || key == NULL) {
- return NULL;
- }
-
- ret = ucl_object_lookup_len (obj, key, strlen (key));
-
- if (ret == NULL) {
- va_start (ap, key);
-
- while (ret == NULL) {
- nk = va_arg (ap, const char *);
-
- if (nk == NULL) {
- break;
- }
- else {
- ret = ucl_object_lookup_len (obj, nk, strlen (nk));
- }
- }
-
- va_end (ap);
- }
-
- return ret;
- }
-
- const ucl_object_t*
- ucl_object_iterate_with_error (const ucl_object_t *obj, ucl_object_iter_t *iter, bool expand_values,
- int *ep)
- {
- const ucl_object_t *elt = NULL;
-
- if (obj == NULL || iter == NULL) {
- return NULL;
- }
-
- if (expand_values) {
- switch (obj->type) {
- case UCL_OBJECT:
- return (const ucl_object_t*)ucl_hash_iterate2 (obj->value.ov, iter, ep);
- break;
- case UCL_ARRAY: {
- unsigned int idx;
- UCL_ARRAY_GET (vec, obj);
- idx = (unsigned int)(uintptr_t)(*iter);
-
- if (vec != NULL) {
- while (idx < kv_size (*vec)) {
- if ((elt = kv_A (*vec, idx)) != NULL) {
- idx ++;
- break;
- }
- idx ++;
- }
- *iter = (void *)(uintptr_t)idx;
- }
-
- return elt;
- break;
- }
- default:
- /* Go to linear iteration */
- break;
- }
- }
- /* Treat everything as a linear list */
- elt = *iter;
- if (elt == NULL) {
- elt = obj;
- }
- else if (elt == obj) {
- return NULL;
- }
- *iter = __DECONST (void *, elt->next ? elt->next : obj);
- return elt;
-
- /* Not reached */
- return NULL;
- }
-
- enum ucl_safe_iter_flags {
- UCL_ITERATE_FLAG_UNDEFINED = 0,
- UCL_ITERATE_FLAG_INSIDE_ARRAY,
- UCL_ITERATE_FLAG_INSIDE_OBJECT,
- UCL_ITERATE_FLAG_IMPLICIT,
- UCL_ITERATE_FLAG_EXCEPTION
- };
-
- static const char safe_iter_magic[4] = {'u', 'i', 't', 'e'};
- struct ucl_object_safe_iter {
- char magic[4]; /* safety check */
- uint32_t flags;
- const ucl_object_t *impl_it; /* implicit object iteration */
- ucl_object_iter_t expl_it; /* explicit iteration */
- };
-
- #define UCL_SAFE_ITER(ptr) (struct ucl_object_safe_iter *)(ptr)
- #define UCL_SAFE_ITER_CHECK(it) do { \
- assert (it != NULL); \
- assert (memcmp (it->magic, safe_iter_magic, sizeof (it->magic)) == 0); \
- } while (0)
-
- ucl_object_iter_t
- ucl_object_iterate_new (const ucl_object_t *obj)
- {
- struct ucl_object_safe_iter *it;
-
- it = UCL_ALLOC (sizeof (*it));
- if (it != NULL) {
- memcpy (it->magic, safe_iter_magic, sizeof (it->magic));
- it->flags = UCL_ITERATE_FLAG_UNDEFINED;
- it->expl_it = NULL;
- it->impl_it = obj;
- }
-
- return (ucl_object_iter_t)it;
- }
-
- bool
- ucl_object_iter_chk_excpn(ucl_object_iter_t *it)
- {
- struct ucl_object_safe_iter *rit = UCL_SAFE_ITER (it);
-
- UCL_SAFE_ITER_CHECK (rit);
-
- return (rit->flags == UCL_ITERATE_FLAG_EXCEPTION);
- }
-
- ucl_object_iter_t
- ucl_object_iterate_reset (ucl_object_iter_t it, const ucl_object_t *obj)
- {
- struct ucl_object_safe_iter *rit = UCL_SAFE_ITER (it);
-
- UCL_SAFE_ITER_CHECK (rit);
-
- if (rit->expl_it != NULL) {
- if (rit->flags == UCL_ITERATE_FLAG_INSIDE_OBJECT) {
- UCL_FREE (sizeof (*rit->expl_it), rit->expl_it);
- }
- }
-
- rit->impl_it = obj;
- rit->expl_it = NULL;
- rit->flags = UCL_ITERATE_FLAG_UNDEFINED;
-
- return it;
- }
-
- const ucl_object_t*
- ucl_object_iterate_safe (ucl_object_iter_t it, bool expand_values)
- {
- return ucl_object_iterate_full (it, expand_values ? UCL_ITERATE_BOTH :
- UCL_ITERATE_IMPLICIT);
- }
-
- const ucl_object_t*
- ucl_object_iterate_full (ucl_object_iter_t it, enum ucl_iterate_type type)
- {
- struct ucl_object_safe_iter *rit = UCL_SAFE_ITER (it);
- const ucl_object_t *ret = NULL;
- int ern;
-
- UCL_SAFE_ITER_CHECK (rit);
-
- if (rit->impl_it == NULL) {
- return NULL;
- }
-
- if (rit->impl_it->type == UCL_OBJECT) {
- rit->flags = UCL_ITERATE_FLAG_INSIDE_OBJECT;
- ret = ucl_object_iterate_with_error (rit->impl_it, &rit->expl_it, true, &ern);
-
- if (ret == NULL && ern != 0) {
- rit->flags = UCL_ITERATE_FLAG_EXCEPTION;
- return NULL;
- }
-
- if (ret == NULL && (type & UCL_ITERATE_IMPLICIT)) {
- /* Need to switch to another implicit object in chain */
- rit->impl_it = rit->impl_it->next;
- rit->expl_it = NULL;
-
- return ucl_object_iterate_safe (it, type);
- }
- }
- else if (rit->impl_it->type == UCL_ARRAY) {
- rit->flags = UCL_ITERATE_FLAG_INSIDE_ARRAY;
- ret = ucl_object_iterate (rit->impl_it, &rit->expl_it, true);
-
- if (ret == NULL && (type & UCL_ITERATE_IMPLICIT)) {
- /* Need to switch to another implicit object in chain */
- rit->impl_it = rit->impl_it->next;
- rit->expl_it = NULL;
-
- return ucl_object_iterate_safe (it, type);
- }
- }
- else {
- /* Just iterate over the implicit array */
- rit->flags = UCL_ITERATE_FLAG_IMPLICIT;
- ret = rit->impl_it;
- rit->impl_it = rit->impl_it->next;
-
- if (type & UCL_ITERATE_EXPLICIT) {
- /* We flatten objects if need to expand values */
- if (ret->type == UCL_OBJECT || ret->type == UCL_ARRAY) {
- return ucl_object_iterate_safe (it, type);
- }
- }
- }
-
- return ret;
- }
-
- void
- ucl_object_iterate_free (ucl_object_iter_t it)
- {
- struct ucl_object_safe_iter *rit = UCL_SAFE_ITER (it);
-
- UCL_SAFE_ITER_CHECK (rit);
-
- if (rit->expl_it != NULL) {
- if (rit->flags == UCL_ITERATE_FLAG_INSIDE_OBJECT) {
- UCL_FREE (sizeof (*rit->expl_it), rit->expl_it);
- }
- }
-
- UCL_FREE (sizeof (*rit), it);
- }
-
- const ucl_object_t *
- ucl_object_lookup_path (const ucl_object_t *top, const char *path_in) {
- return ucl_object_lookup_path_char (top, path_in, '.');
- }
-
-
- const ucl_object_t *
- ucl_object_lookup_path_char (const ucl_object_t *top, const char *path_in, const char sep) {
- const ucl_object_t *o = NULL, *found;
- const char *p, *c;
- char *err_str;
- unsigned index;
-
- if (path_in == NULL || top == NULL) {
- return NULL;
- }
-
- found = NULL;
- p = path_in;
-
- /* Skip leading dots */
- while (*p == sep) {
- p ++;
- }
-
- c = p;
- while (*p != '\0') {
- p ++;
- if (*p == sep || *p == '\0') {
- if (p > c) {
- switch (top->type) {
- case UCL_ARRAY:
- /* Key should be an int */
- index = strtoul (c, &err_str, 10);
- if (err_str != NULL && (*err_str != sep && *err_str != '\0')) {
- return NULL;
- }
- o = ucl_array_find_index (top, index);
- break;
- default:
- o = ucl_object_lookup_len (top, c, p - c);
- break;
- }
- if (o == NULL) {
- return NULL;
- }
- top = o;
- }
- if (*p != '\0') {
- c = p + 1;
- }
- }
- }
- found = o;
-
- return found;
- }
-
-
- ucl_object_t *
- ucl_object_new (void)
- {
- return ucl_object_typed_new (UCL_NULL);
- }
-
- ucl_object_t *
- ucl_object_typed_new (ucl_type_t type)
- {
- return ucl_object_new_full (type, 0);
- }
-
- ucl_object_t *
- ucl_object_new_full (ucl_type_t type, unsigned priority)
- {
- ucl_object_t *new;
-
- if (type != UCL_USERDATA) {
- new = UCL_ALLOC (sizeof (ucl_object_t));
- if (new != NULL) {
- memset (new, 0, sizeof (ucl_object_t));
- new->ref = 1;
- new->type = (type <= UCL_NULL ? type : UCL_NULL);
- new->next = NULL;
- new->prev = new;
- ucl_object_set_priority (new, priority);
-
- if (type == UCL_ARRAY) {
- new->value.av = UCL_ALLOC (sizeof (ucl_array_t));
- if (new->value.av) {
- memset (new->value.av, 0, sizeof (ucl_array_t));
- UCL_ARRAY_GET (vec, new);
-
- /* Preallocate some space for arrays */
- kv_resize_safe (ucl_object_t *, *vec, 8, enomem);
- }
- }
- }
- }
- else {
- new = ucl_object_new_userdata (NULL, NULL, NULL);
- ucl_object_set_priority (new, priority);
- }
- enomem:
- return new;
- }
-
- bool ucl_object_reserve (ucl_object_t *obj, size_t reserved)
- {
- if (obj->type == UCL_ARRAY) {
- UCL_ARRAY_GET (vec, obj);
-
- if (vec->m < reserved) {
- /* Preallocate some space for arrays */
- kv_resize_safe (ucl_object_t *, *vec, reserved, e0);
- }
- }
- else if (obj->type == UCL_OBJECT) {
- ucl_hash_reserve (obj->value.ov, reserved);
- }
- return true;
- e0:
- return false;
- }
-
- ucl_object_t*
- ucl_object_new_userdata (ucl_userdata_dtor dtor,
- ucl_userdata_emitter emitter,
- void *ptr)
- {
- struct ucl_object_userdata *new;
- size_t nsize = sizeof (*new);
-
- new = UCL_ALLOC (nsize);
- if (new != NULL) {
- memset (new, 0, nsize);
- new->obj.ref = 1;
- new->obj.type = UCL_USERDATA;
- new->obj.next = NULL;
- new->obj.prev = (ucl_object_t *)new;
- new->dtor = dtor;
- new->emitter = emitter;
- new->obj.value.ud = ptr;
- }
-
- return (ucl_object_t *)new;
- }
-
- ucl_type_t
- ucl_object_type (const ucl_object_t *obj)
- {
- if (obj == NULL) {
- return UCL_NULL;
- }
-
- return obj->type;
- }
-
- ucl_object_t*
- ucl_object_fromstring (const char *str)
- {
- return ucl_object_fromstring_common (str, 0, UCL_STRING_ESCAPE);
- }
-
- ucl_object_t *
- ucl_object_fromlstring (const char *str, size_t len)
- {
- return ucl_object_fromstring_common (str, len, UCL_STRING_ESCAPE);
- }
-
- ucl_object_t *
- ucl_object_fromint (int64_t iv)
- {
- ucl_object_t *obj;
-
- obj = ucl_object_new ();
- if (obj != NULL) {
- obj->type = UCL_INT;
- obj->value.iv = iv;
- }
-
- return obj;
- }
-
- ucl_object_t *
- ucl_object_fromdouble (double dv)
- {
- ucl_object_t *obj;
-
- obj = ucl_object_new ();
- if (obj != NULL) {
- obj->type = UCL_FLOAT;
- obj->value.dv = dv;
- }
-
- return obj;
- }
-
- ucl_object_t*
- ucl_object_frombool (bool bv)
- {
- ucl_object_t *obj;
-
- obj = ucl_object_new ();
- if (obj != NULL) {
- obj->type = UCL_BOOLEAN;
- obj->value.iv = bv;
- }
-
- return obj;
- }
-
- bool
- ucl_array_append (ucl_object_t *top, ucl_object_t *elt)
- {
- UCL_ARRAY_GET (vec, top);
-
- if (elt == NULL || top == NULL) {
- return false;
- }
-
- if (vec == NULL) {
- vec = UCL_ALLOC (sizeof (*vec));
-
- if (vec == NULL) {
- return false;
- }
-
- kv_init (*vec);
- top->value.av = (void *)vec;
- }
-
- kv_push_safe (ucl_object_t *, *vec, elt, e0);
-
- top->len ++;
-
- return true;
- e0:
- return false;
- }
-
- bool
- ucl_array_prepend (ucl_object_t *top, ucl_object_t *elt)
- {
- UCL_ARRAY_GET (vec, top);
-
- if (elt == NULL || top == NULL) {
- return false;
- }
-
- if (vec == NULL) {
- vec = UCL_ALLOC (sizeof (*vec));
- kv_init (*vec);
- top->value.av = (void *)vec;
- kv_push_safe (ucl_object_t *, *vec, elt, e0);
- }
- else {
- /* Slow O(n) algorithm */
- kv_prepend_safe (ucl_object_t *, *vec, elt, e0);
- }
-
- top->len ++;
-
- return true;
- e0:
- return false;
- }
-
- bool
- ucl_array_merge (ucl_object_t *top, ucl_object_t *elt, bool copy)
- {
- unsigned i;
- ucl_object_t *cp = NULL;
- ucl_object_t **obj;
-
- if (elt == NULL || top == NULL || top->type != UCL_ARRAY || elt->type != UCL_ARRAY) {
- return false;
- }
-
- if (copy) {
- cp = ucl_object_copy (elt);
- }
- else {
- cp = ucl_object_ref (elt);
- }
-
- UCL_ARRAY_GET (v1, top);
- UCL_ARRAY_GET (v2, cp);
-
- if (v1 && v2) {
- kv_concat_safe (ucl_object_t *, *v1, *v2, e0);
-
- for (i = v2->n; i < v1->n; i ++) {
- obj = &kv_A (*v1, i);
- if (*obj == NULL) {
- continue;
- }
- top->len ++;
- }
- }
-
- return true;
- e0:
- return false;
- }
-
- ucl_object_t *
- ucl_array_delete (ucl_object_t *top, ucl_object_t *elt)
- {
- UCL_ARRAY_GET (vec, top);
- ucl_object_t *ret = NULL;
- unsigned i;
-
- if (vec == NULL) {
- return NULL;
- }
-
- for (i = 0; i < vec->n; i ++) {
- if (kv_A (*vec, i) == elt) {
- kv_del (ucl_object_t *, *vec, i);
- ret = elt;
- top->len --;
- break;
- }
- }
-
- return ret;
- }
-
- const ucl_object_t *
- ucl_array_head (const ucl_object_t *top)
- {
- UCL_ARRAY_GET (vec, top);
-
- if (vec == NULL || top == NULL || top->type != UCL_ARRAY ||
- top->value.av == NULL) {
- return NULL;
- }
-
- return (vec->n > 0 ? vec->a[0] : NULL);
- }
-
- const ucl_object_t *
- ucl_array_tail (const ucl_object_t *top)
- {
- UCL_ARRAY_GET (vec, top);
-
- if (top == NULL || top->type != UCL_ARRAY || top->value.av == NULL) {
- return NULL;
- }
-
- return (vec->n > 0 ? vec->a[vec->n - 1] : NULL);
- }
-
- ucl_object_t *
- ucl_array_pop_last (ucl_object_t *top)
- {
- UCL_ARRAY_GET (vec, top);
- ucl_object_t **obj, *ret = NULL;
-
- if (vec != NULL && vec->n > 0) {
- obj = &kv_A (*vec, vec->n - 1);
- ret = *obj;
- kv_del (ucl_object_t *, *vec, vec->n - 1);
- top->len --;
- }
-
- return ret;
- }
-
- ucl_object_t *
- ucl_array_pop_first (ucl_object_t *top)
- {
- UCL_ARRAY_GET (vec, top);
- ucl_object_t **obj, *ret = NULL;
-
- if (vec != NULL && vec->n > 0) {
- obj = &kv_A (*vec, 0);
- ret = *obj;
- kv_del (ucl_object_t *, *vec, 0);
- top->len --;
- }
-
- return ret;
- }
-
- unsigned int
- ucl_array_size (const ucl_object_t *top)
- {
- if (top == NULL || top->type != UCL_ARRAY) {
- return 0;
- }
-
- UCL_ARRAY_GET (vec, top);
-
- if (vec != NULL) {
- return kv_size(*vec);
- }
-
- return 0;
- }
-
- const ucl_object_t *
- ucl_array_find_index (const ucl_object_t *top, unsigned int index)
- {
- UCL_ARRAY_GET (vec, top);
-
- if (vec != NULL && vec->n > 0 && index < vec->n) {
- return kv_A (*vec, index);
- }
-
- return NULL;
- }
-
- unsigned int
- ucl_array_index_of (ucl_object_t *top, ucl_object_t *elt)
- {
- UCL_ARRAY_GET (vec, top);
- unsigned i;
-
- if (vec == NULL) {
- return (unsigned int)(-1);
- }
-
- for (i = 0; i < vec->n; i ++) {
- if (kv_A (*vec, i) == elt) {
- return i;
- }
- }
-
- return (unsigned int)(-1);
- }
-
- ucl_object_t *
- ucl_array_replace_index (ucl_object_t *top, ucl_object_t *elt,
- unsigned int index)
- {
- UCL_ARRAY_GET (vec, top);
- ucl_object_t *ret = NULL;
-
- if (vec != NULL && vec->n > 0 && index < vec->n) {
- ret = kv_A (*vec, index);
- kv_A (*vec, index) = elt;
- }
-
- return ret;
- }
-
- ucl_object_t *
- ucl_elt_append (ucl_object_t *head, ucl_object_t *elt)
- {
-
- if (head == NULL) {
- elt->next = NULL;
- elt->prev = elt;
- head = elt;
- }
- else {
- elt->prev = head->prev;
- head->prev->next = elt;
- head->prev = elt;
- elt->next = NULL;
- }
-
- return head;
- }
-
- bool
- ucl_object_todouble_safe (const ucl_object_t *obj, double *target)
- {
- if (obj == NULL || target == NULL) {
- return false;
- }
- switch (obj->type) {
- case UCL_INT:
- *target = obj->value.iv; /* Probably could cause overflow */
- break;
- case UCL_FLOAT:
- case UCL_TIME:
- *target = obj->value.dv;
- break;
- default:
- return false;
- }
-
- return true;
- }
-
- double
- ucl_object_todouble (const ucl_object_t *obj)
- {
- double result = 0.;
-
- ucl_object_todouble_safe (obj, &result);
- return result;
- }
-
- bool
- ucl_object_toint_safe (const ucl_object_t *obj, int64_t *target)
- {
- if (obj == NULL || target == NULL) {
- return false;
- }
- switch (obj->type) {
- case UCL_INT:
- *target = obj->value.iv;
- break;
- case UCL_FLOAT:
- case UCL_TIME:
- *target = obj->value.dv; /* Losing of decimal points */
- break;
- default:
- return false;
- }
-
- return true;
- }
-
- int64_t
- ucl_object_toint (const ucl_object_t *obj)
- {
- int64_t result = 0;
-
- ucl_object_toint_safe (obj, &result);
- return result;
- }
-
- bool
- ucl_object_toboolean_safe (const ucl_object_t *obj, bool *target)
- {
- if (obj == NULL || target == NULL) {
- return false;
- }
- switch (obj->type) {
- case UCL_BOOLEAN:
- *target = (obj->value.iv == true);
- break;
- default:
- return false;
- }
-
- return true;
- }
-
- bool
- ucl_object_toboolean (const ucl_object_t *obj)
- {
- bool result = false;
-
- ucl_object_toboolean_safe (obj, &result);
- return result;
- }
-
- bool
- ucl_object_tostring_safe (const ucl_object_t *obj, const char **target)
- {
- if (obj == NULL || target == NULL) {
- return false;
- }
-
- switch (obj->type) {
- case UCL_STRING:
- if (!(obj->flags & UCL_OBJECT_BINARY)) {
- *target = ucl_copy_value_trash (obj);
- }
- break;
- default:
- return false;
- }
-
- return true;
- }
-
- const char *
- ucl_object_tostring (const ucl_object_t *obj)
- {
- const char *result = NULL;
-
- ucl_object_tostring_safe (obj, &result);
- return result;
- }
-
- const char *
- ucl_object_tostring_forced (const ucl_object_t *obj)
- {
- /* TODO: For binary strings we might encode string here */
- if (!(obj->flags & UCL_OBJECT_BINARY)) {
- return ucl_copy_value_trash (obj);
- }
-
- return NULL;
- }
-
- bool
- ucl_object_tolstring_safe (const ucl_object_t *obj, const char **target, size_t *tlen)
- {
- if (obj == NULL || target == NULL) {
- return false;
- }
- switch (obj->type) {
- case UCL_STRING:
- *target = obj->value.sv;
- if (tlen != NULL) {
- *tlen = obj->len;
- }
- break;
- default:
- return false;
- }
-
- return true;
- }
-
- const char *
- ucl_object_tolstring (const ucl_object_t *obj, size_t *tlen)
- {
- const char *result = NULL;
-
- ucl_object_tolstring_safe (obj, &result, tlen);
- return result;
- }
-
- const char *
- ucl_object_key (const ucl_object_t *obj)
- {
- return ucl_copy_key_trash (obj);
- }
-
- const char *
- ucl_object_keyl (const ucl_object_t *obj, size_t *len)
- {
- if (len == NULL || obj == NULL) {
- return NULL;
- }
- *len = obj->keylen;
- return obj->key;
- }
-
- ucl_object_t *
- ucl_object_ref (const ucl_object_t *obj)
- {
- ucl_object_t *res = NULL;
-
- if (obj != NULL) {
- if (obj->flags & UCL_OBJECT_EPHEMERAL) {
- /*
- * Use deep copy for ephemeral objects, note that its refcount
- * is NOT increased, since ephemeral objects does not need refcount
- * at all
- */
- res = ucl_object_copy (obj);
- }
- else {
- res = __DECONST (ucl_object_t *, obj);
- #ifdef HAVE_ATOMIC_BUILTINS
- (void)__sync_add_and_fetch (&res->ref, 1);
- #else
- res->ref ++;
- #endif
- }
- }
- return res;
- }
-
- static ucl_object_t *
- ucl_object_copy_internal (const ucl_object_t *other, bool allow_array)
- {
-
- ucl_object_t *new;
- ucl_object_iter_t it = NULL;
- const ucl_object_t *cur;
-
- new = malloc (sizeof (*new));
-
- if (new != NULL) {
- memcpy (new, other, sizeof (*new));
- if (other->flags & UCL_OBJECT_EPHEMERAL) {
- /* Copied object is always non ephemeral */
- new->flags &= ~UCL_OBJECT_EPHEMERAL;
- }
- new->ref = 1;
- /* Unlink from others */
- new->next = NULL;
- new->prev = new;
-
- /* deep copy of values stored */
- if (other->trash_stack[UCL_TRASH_KEY] != NULL) {
- new->trash_stack[UCL_TRASH_KEY] =
- strdup (other->trash_stack[UCL_TRASH_KEY]);
- if (other->key == (const char *)other->trash_stack[UCL_TRASH_KEY]) {
- new->key = new->trash_stack[UCL_TRASH_KEY];
- }
- }
- if (other->trash_stack[UCL_TRASH_VALUE] != NULL) {
- new->trash_stack[UCL_TRASH_VALUE] =
- strdup (other->trash_stack[UCL_TRASH_VALUE]);
- if (new->type == UCL_STRING) {
- new->value.sv = new->trash_stack[UCL_TRASH_VALUE];
- }
- }
-
- if (other->type == UCL_ARRAY || other->type == UCL_OBJECT) {
- /* reset old value */
- memset (&new->value, 0, sizeof (new->value));
-
- while ((cur = ucl_object_iterate (other, &it, true)) != NULL) {
- if (other->type == UCL_ARRAY) {
- ucl_array_append (new, ucl_object_copy_internal (cur, false));
- }
- else {
- ucl_object_t *cp = ucl_object_copy_internal (cur, true);
- if (cp != NULL) {
- ucl_object_insert_key (new, cp, cp->key, cp->keylen,
- false);
- }
- }
- }
- }
- else if (allow_array && other->next != NULL) {
- LL_FOREACH (other->next, cur) {
- ucl_object_t *cp = ucl_object_copy_internal (cur, false);
- if (cp != NULL) {
- DL_APPEND (new, cp);
- }
- }
- }
- }
-
- return new;
- }
-
- ucl_object_t *
- ucl_object_copy (const ucl_object_t *other)
- {
- return ucl_object_copy_internal (other, true);
- }
-
- void
- ucl_object_unref (ucl_object_t *obj)
- {
- if (obj != NULL) {
- #ifdef HAVE_ATOMIC_BUILTINS
- unsigned int rc = __sync_sub_and_fetch (&obj->ref, 1);
- if (rc == 0) {
- #else
- if (--obj->ref == 0) {
- #endif
- ucl_object_free_internal (obj, true, ucl_object_dtor_unref);
- }
- }
- }
-
- int
- ucl_object_compare (const ucl_object_t *o1, const ucl_object_t *o2)
- {
- const ucl_object_t *it1, *it2;
- ucl_object_iter_t iter = NULL;
- int ret = 0;
-
- if (o1->type != o2->type) {
- return (o1->type) - (o2->type);
- }
-
- switch (o1->type) {
- case UCL_STRING:
- if (o1->len == o2->len && o1->len > 0) {
- ret = strcmp (ucl_object_tostring(o1), ucl_object_tostring(o2));
- }
- else {
- ret = o1->len - o2->len;
- }
- break;
- case UCL_FLOAT:
- case UCL_INT:
- case UCL_TIME:
- ret = ucl_object_todouble (o1) - ucl_object_todouble (o2);
- break;
- case UCL_BOOLEAN:
- ret = ucl_object_toboolean (o1) - ucl_object_toboolean (o2);
- break;
- case UCL_ARRAY:
- if (o1->len == o2->len && o1->len > 0) {
- UCL_ARRAY_GET (vec1, o1);
- UCL_ARRAY_GET (vec2, o2);
- unsigned i;
-
- /* Compare all elements in both arrays */
- for (i = 0; i < vec1->n; i ++) {
- it1 = kv_A (*vec1, i);
- it2 = kv_A (*vec2, i);
-
- if (it1 == NULL && it2 != NULL) {
- return -1;
- }
- else if (it2 == NULL && it1 != NULL) {
- return 1;
- }
- else if (it1 != NULL && it2 != NULL) {
- ret = ucl_object_compare (it1, it2);
- if (ret != 0) {
- break;
- }
- }
- }
- }
- else {
- ret = o1->len - o2->len;
- }
- break;
- case UCL_OBJECT:
- if (o1->len == o2->len && o1->len > 0) {
- while ((it1 = ucl_object_iterate (o1, &iter, true)) != NULL) {
- it2 = ucl_object_lookup (o2, ucl_object_key (it1));
- if (it2 == NULL) {
- ret = 1;
- break;
- }
- ret = ucl_object_compare (it1, it2);
- if (ret != 0) {
- break;
- }
- }
- }
- else {
- ret = o1->len - o2->len;
- }
- break;
- default:
- ret = 0;
- break;
- }
-
- return ret;
- }
-
- int
- ucl_object_compare_qsort (const ucl_object_t **o1,
- const ucl_object_t **o2)
- {
- return ucl_object_compare (*o1, *o2);
- }
-
- void
- ucl_object_array_sort (ucl_object_t *ar,
- int (*cmp)(const ucl_object_t **o1, const ucl_object_t **o2))
- {
- UCL_ARRAY_GET (vec, ar);
-
- if (cmp == NULL || ar == NULL || ar->type != UCL_ARRAY) {
- return;
- }
-
- qsort (vec->a, vec->n, sizeof (ucl_object_t *),
- (int (*)(const void *, const void *))cmp);
- }
-
- void ucl_object_sort_keys (ucl_object_t *obj,
- enum ucl_object_keys_sort_flags how)
- {
- if (obj != NULL && obj->type == UCL_OBJECT) {
- ucl_hash_sort (obj->value.ov, how);
- }
- }
-
- #define PRIOBITS 4
-
- unsigned int
- ucl_object_get_priority (const ucl_object_t *obj)
- {
- if (obj == NULL) {
- return 0;
- }
-
- return (obj->flags >> ((sizeof (obj->flags) * NBBY) - PRIOBITS));
- }
-
- void
- ucl_object_set_priority (ucl_object_t *obj,
- unsigned int priority)
- {
- if (obj != NULL) {
- priority &= (0x1 << PRIOBITS) - 1;
- priority <<= ((sizeof (obj->flags) * NBBY) - PRIOBITS);
- priority |= obj->flags & ((1 << ((sizeof (obj->flags) * NBBY) -
- PRIOBITS)) - 1);
- obj->flags = priority;
- }
- }
-
- bool
- ucl_object_string_to_type (const char *input, ucl_type_t *res)
- {
- if (strcasecmp (input, "object") == 0) {
- *res = UCL_OBJECT;
- }
- else if (strcasecmp (input, "array") == 0) {
- *res = UCL_ARRAY;
- }
- else if (strcasecmp (input, "integer") == 0) {
- *res = UCL_INT;
- }
- else if (strcasecmp (input, "number") == 0) {
- *res = UCL_FLOAT;
- }
- else if (strcasecmp (input, "string") == 0) {
- *res = UCL_STRING;
- }
- else if (strcasecmp (input, "boolean") == 0) {
- *res = UCL_BOOLEAN;
- }
- else if (strcasecmp (input, "null") == 0) {
- *res = UCL_NULL;
- }
- else if (strcasecmp (input, "userdata") == 0) {
- *res = UCL_USERDATA;
- }
- else {
- return false;
- }
-
- return true;
- }
-
- const char *
- ucl_object_type_to_string (ucl_type_t type)
- {
- const char *res = "unknown";
-
- switch (type) {
- case UCL_OBJECT:
- res = "object";
- break;
- case UCL_ARRAY:
- res = "array";
- break;
- case UCL_INT:
- res = "integer";
- break;
- case UCL_FLOAT:
- case UCL_TIME:
- res = "number";
- break;
- case UCL_STRING:
- res = "string";
- break;
- case UCL_BOOLEAN:
- res = "boolean";
- break;
- case UCL_USERDATA:
- res = "userdata";
- break;
- case UCL_NULL:
- res = "null";
- break;
- }
-
- return res;
- }
-
- const ucl_object_t *
- ucl_parser_get_comments (struct ucl_parser *parser)
- {
- if (parser && parser->comments) {
- return parser->comments;
- }
-
- return NULL;
- }
-
- const ucl_object_t *
- ucl_comments_find (const ucl_object_t *comments,
- const ucl_object_t *srch)
- {
- if (comments && srch) {
- return ucl_object_lookup_len (comments, (const char *)&srch,
- sizeof (void *));
- }
-
- return NULL;
- }
-
- bool
- ucl_comments_move (ucl_object_t *comments,
- const ucl_object_t *from, const ucl_object_t *to)
- {
- const ucl_object_t *found;
- ucl_object_t *obj;
-
- if (comments && from && to) {
- found = ucl_object_lookup_len (comments,
- (const char *)&from, sizeof (void *));
-
- if (found) {
- /* Replace key */
- obj = ucl_object_ref (found);
- ucl_object_delete_keyl (comments, (const char *)&from,
- sizeof (void *));
- ucl_object_insert_key (comments, obj, (const char *)&to,
- sizeof (void *), true);
-
- return true;
- }
- }
-
- return false;
- }
-
- void
- ucl_comments_add (ucl_object_t *comments, const ucl_object_t *obj,
- const char *comment)
- {
- if (comments && obj && comment) {
- ucl_object_insert_key (comments, ucl_object_fromstring (comment),
- (const char *)&obj, sizeof (void *), true);
- }
- }
-
- void
- ucl_parser_set_include_tracer (struct ucl_parser *parser,
- ucl_include_trace_func_t func,
- void *user_data)
- {
- parser->include_trace_func = func;
- parser->include_trace_ud = user_data;
- }
-
- const char *
- ucl_parser_get_cur_file (struct ucl_parser *parser)
- {
- return parser->cur_file;
- }
|