1*0d601874SDavid du Colombier /* 2*0d601874SDavid du Colombier * The authors of this software are Rob Pike and Ken Thompson. 3*0d601874SDavid du Colombier * Copyright (c) 2002 by Lucent Technologies. 4*0d601874SDavid du Colombier * Permission to use, copy, modify, and distribute this software for any 5*0d601874SDavid du Colombier * purpose without fee is hereby granted, provided that this entire notice 6*0d601874SDavid du Colombier * is included in all copies of any software which is or includes a copy 7*0d601874SDavid du Colombier * or modification of this software and in all copies of the supporting 8*0d601874SDavid du Colombier * documentation for such software. 9*0d601874SDavid du Colombier * THIS SOFTWARE IS BEING PROVIDED "AS IS", WITHOUT ANY EXPRESS OR IMPLIED 10*0d601874SDavid du Colombier * WARRANTY. IN PARTICULAR, NEITHER THE AUTHORS NOR LUCENT TECHNOLOGIES MAKE 11*0d601874SDavid du Colombier * ANY REPRESENTATION OR WARRANTY OF ANY KIND CONCERNING THE MERCHANTABILITY 12*0d601874SDavid du Colombier * OF THIS SOFTWARE OR ITS FITNESS FOR ANY PARTICULAR PURPOSE. 13*0d601874SDavid du Colombier */ 14*0d601874SDavid du Colombier 158ccd4a63SDavid du Colombier #include <u.h> 168ccd4a63SDavid du Colombier #include <libc.h> 178ccd4a63SDavid du Colombier #include "fmtdef.h" 188ccd4a63SDavid du Colombier 198ccd4a63SDavid du Colombier static int 208ccd4a63SDavid du Colombier fmtStrFlush(Fmt *f) 218ccd4a63SDavid du Colombier { 228ccd4a63SDavid du Colombier char *s; 238ccd4a63SDavid du Colombier int n; 248ccd4a63SDavid du Colombier 258ccd4a63SDavid du Colombier if(f->start == nil) 268ccd4a63SDavid du Colombier return 0; 27*0d601874SDavid du Colombier n = (uintptr)f->farg; 288ccd4a63SDavid du Colombier n *= 2; 29*0d601874SDavid du Colombier s = (char*)f->start; 308ccd4a63SDavid du Colombier f->start = realloc(s, n); 318ccd4a63SDavid du Colombier if(f->start == nil){ 328ccd4a63SDavid du Colombier f->farg = nil; 338ccd4a63SDavid du Colombier f->to = nil; 348ccd4a63SDavid du Colombier f->stop = nil; 358ccd4a63SDavid du Colombier free(s); 368ccd4a63SDavid du Colombier return 0; 378ccd4a63SDavid du Colombier } 38*0d601874SDavid du Colombier f->farg = (void*)(uintptr)n; 398ccd4a63SDavid du Colombier f->to = (char*)f->start + ((char*)f->to - s); 408ccd4a63SDavid du Colombier f->stop = (char*)f->start + n - 1; 418ccd4a63SDavid du Colombier return 1; 428ccd4a63SDavid du Colombier } 438ccd4a63SDavid du Colombier 448ccd4a63SDavid du Colombier int 458ccd4a63SDavid du Colombier fmtstrinit(Fmt *f) 468ccd4a63SDavid du Colombier { 478ccd4a63SDavid du Colombier int n; 488ccd4a63SDavid du Colombier 498ccd4a63SDavid du Colombier memset(f, 0, sizeof *f); 508ccd4a63SDavid du Colombier f->runes = 0; 518ccd4a63SDavid du Colombier n = 32; 528ccd4a63SDavid du Colombier f->start = malloc(n); 538ccd4a63SDavid du Colombier if(f->start == nil) 548ccd4a63SDavid du Colombier return -1; 558ccd4a63SDavid du Colombier f->to = f->start; 568ccd4a63SDavid du Colombier f->stop = (char*)f->start + n - 1; 578ccd4a63SDavid du Colombier f->flush = fmtStrFlush; 58*0d601874SDavid du Colombier f->farg = (void*)(uintptr)n; 598ccd4a63SDavid du Colombier f->nfmt = 0; 608ccd4a63SDavid du Colombier return 0; 618ccd4a63SDavid du Colombier } 628ccd4a63SDavid du Colombier 638ccd4a63SDavid du Colombier /* 648ccd4a63SDavid du Colombier * print into an allocated string buffer 658ccd4a63SDavid du Colombier */ 668ccd4a63SDavid du Colombier char* 678ccd4a63SDavid du Colombier vsmprint(char *fmt, va_list args) 688ccd4a63SDavid du Colombier { 698ccd4a63SDavid du Colombier Fmt f; 708ccd4a63SDavid du Colombier int n; 718ccd4a63SDavid du Colombier 728ccd4a63SDavid du Colombier if(fmtstrinit(&f) < 0) 738ccd4a63SDavid du Colombier return nil; 74*0d601874SDavid du Colombier VA_COPY(f.args,args); 758ccd4a63SDavid du Colombier n = dofmt(&f, fmt); 76*0d601874SDavid du Colombier VA_END(f.args); 778ccd4a63SDavid du Colombier if(n < 0){ 788ccd4a63SDavid du Colombier free(f.start); 798ccd4a63SDavid du Colombier return nil; 808ccd4a63SDavid du Colombier } 81*0d601874SDavid du Colombier return fmtstrflush(&f); 828ccd4a63SDavid du Colombier } 83