Home | History | Annotate | Line # | Download | only in roken
      1  1.1     elric /*	$NetBSD: dumpdata.c,v 1.2 2017/01/28 21:31:50 christos Exp $	*/
      2  1.1     elric 
      3  1.1     elric /*
      4  1.1     elric  * Copyright (c) 2005 Kungliga Tekniska Hgskolan
      5  1.1     elric  * (Royal Institute of Technology, Stockholm, Sweden).
      6  1.1     elric  * All rights reserved.
      7  1.1     elric  *
      8  1.1     elric  * Redistribution and use in source and binary forms, with or without
      9  1.1     elric  * modification, are permitted provided that the following conditions
     10  1.1     elric  * are met:
     11  1.1     elric  *
     12  1.1     elric  * 1. Redistributions of source code must retain the above copyright
     13  1.1     elric  *    notice, this list of conditions and the following disclaimer.
     14  1.1     elric  *
     15  1.1     elric  * 2. Redistributions in binary form must reproduce the above copyright
     16  1.1     elric  *    notice, this list of conditions and the following disclaimer in the
     17  1.1     elric  *    documentation and/or other materials provided with the distribution.
     18  1.1     elric  *
     19  1.1     elric  * 3. Neither the name of the Institute nor the names of its contributors
     20  1.1     elric  *    may be used to endorse or promote products derived from this software
     21  1.1     elric  *    without specific prior written permission.
     22  1.1     elric  *
     23  1.1     elric  * THIS SOFTWARE IS PROVIDED BY THE INSTITUTE AND CONTRIBUTORS ``AS IS'' AND
     24  1.1     elric  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
     25  1.1     elric  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
     26  1.1     elric  * ARE DISCLAIMED.  IN NO EVENT SHALL THE INSTITUTE OR CONTRIBUTORS BE LIABLE
     27  1.1     elric  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
     28  1.1     elric  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
     29  1.1     elric  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
     30  1.1     elric  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
     31  1.1     elric  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
     32  1.1     elric  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
     33  1.1     elric  * SUCH DAMAGE.
     34  1.1     elric  */
     35  1.1     elric 
     36  1.1     elric #include <config.h>
     37  1.1     elric 
     38  1.1     elric #include <krb5/roken.h>
     39  1.1     elric 
     40  1.1     elric /*
     41  1.1     elric  * Write datablob to a filename, don't care about errors.
     42  1.1     elric  */
     43  1.1     elric 
     44  1.1     elric ROKEN_LIB_FUNCTION void ROKEN_LIB_CALL
     45  1.1     elric rk_dumpdata (const char *filename, const void *buf, size_t size)
     46  1.1     elric {
     47  1.1     elric     int fd;
     48  1.1     elric 
     49  1.1     elric     fd = open(filename, O_WRONLY|O_TRUNC|O_CREAT, 0640);
     50  1.1     elric     if (fd < 0)
     51  1.1     elric 	return;
     52  1.1     elric     net_write(fd, buf, size);
     53  1.1     elric     close(fd);
     54  1.1     elric }
     55  1.1     elric 
     56  1.1     elric /*
     57  1.1     elric  * Read all data from a filename, care about errors.
     58  1.1     elric  */
     59  1.1     elric 
     60  1.1     elric ROKEN_LIB_FUNCTION int ROKEN_LIB_CALL
     61  1.1     elric rk_undumpdata(const char *filename, void **buf, size_t *size)
     62  1.1     elric {
     63  1.1     elric     struct stat sb;
     64  1.1     elric     int fd, ret;
     65  1.1     elric     ssize_t sret;
     66  1.1     elric 
     67  1.1     elric     *buf = NULL;
     68  1.1     elric 
     69  1.1     elric     fd = open(filename, O_RDONLY, 0);
     70  1.1     elric     if (fd < 0)
     71  1.1     elric 	return errno;
     72  1.1     elric     if (fstat(fd, &sb) != 0){
     73  1.1     elric 	ret = errno;
     74  1.1     elric 	goto out;
     75  1.1     elric     }
     76  1.1     elric     *buf = malloc(sb.st_size);
     77  1.1     elric     if (*buf == NULL) {
     78  1.1     elric 	ret = ENOMEM;
     79  1.1     elric 	goto out;
     80  1.1     elric     }
     81  1.1     elric     *size = sb.st_size;
     82  1.1     elric 
     83  1.1     elric     sret = net_read(fd, *buf, *size);
     84  1.1     elric     if (sret < 0)
     85  1.1     elric 	ret = errno;
     86  1.2  christos     else if (sret != (ssize_t)*size)
     87  1.1     elric 	ret = EINVAL;
     88  1.2  christos     else
     89  1.2  christos 	ret = 0;
     90  1.2  christos 
     91  1.2  christos   out:
     92  1.2  christos     if (ret) {
     93  1.1     elric 	free(*buf);
     94  1.1     elric 	*buf = NULL;
     95  1.2  christos     }
     96  1.1     elric     close(fd);
     97  1.1     elric     return ret;
     98  1.1     elric }
     99