gets_chk.c revision 1.5.8.2
1 1.5.8.2 martin /* $NetBSD: gets_chk.c,v 1.5.8.2 2008/04/28 20:23:01 martin Exp $ */ 2 1.5.8.2 martin 3 1.5.8.2 martin /*- 4 1.5.8.2 martin * Copyright (c) 2006 The NetBSD Foundation, Inc. 5 1.5.8.2 martin * All rights reserved. 6 1.5.8.2 martin * 7 1.5.8.2 martin * This code is derived from software contributed to The NetBSD Foundation 8 1.5.8.2 martin * by Christos Zoulas. 9 1.5.8.2 martin * 10 1.5.8.2 martin * Redistribution and use in source and binary forms, with or without 11 1.5.8.2 martin * modification, are permitted provided that the following conditions 12 1.5.8.2 martin * are met: 13 1.5.8.2 martin * 1. Redistributions of source code must retain the above copyright 14 1.5.8.2 martin * notice, this list of conditions and the following disclaimer. 15 1.5.8.2 martin * 2. Redistributions in binary form must reproduce the above copyright 16 1.5.8.2 martin * notice, this list of conditions and the following disclaimer in the 17 1.5.8.2 martin * documentation and/or other materials provided with the distribution. 18 1.5.8.2 martin * 19 1.5.8.2 martin * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS 20 1.5.8.2 martin * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED 21 1.5.8.2 martin * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR 22 1.5.8.2 martin * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS 23 1.5.8.2 martin * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR 24 1.5.8.2 martin * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF 25 1.5.8.2 martin * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS 26 1.5.8.2 martin * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN 27 1.5.8.2 martin * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) 28 1.5.8.2 martin * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE 29 1.5.8.2 martin * POSSIBILITY OF SUCH DAMAGE. 30 1.5.8.2 martin */ 31 1.5.8.2 martin #include <sys/cdefs.h> 32 1.5.8.2 martin __RCSID("$NetBSD: gets_chk.c,v 1.5.8.2 2008/04/28 20:23:01 martin Exp $"); 33 1.5.8.2 martin 34 1.5.8.2 martin /*LINTLIBRARY*/ 35 1.5.8.2 martin 36 1.5.8.2 martin #include <ssp/ssp.h> 37 1.5.8.2 martin #include <stdio.h> 38 1.5.8.2 martin #include <string.h> 39 1.5.8.2 martin #include <limits.h> 40 1.5.8.2 martin #include <stdlib.h> 41 1.5.8.2 martin #include <ssp/stdio.h> 42 1.5.8.2 martin 43 1.5.8.2 martin #undef gets 44 1.5.8.2 martin 45 1.5.8.2 martin char * 46 1.5.8.2 martin __gets_chk(char * __restrict buf, size_t slen) 47 1.5.8.2 martin { 48 1.5.8.2 martin char *abuf; 49 1.5.8.2 martin size_t len; 50 1.5.8.2 martin 51 1.5.8.2 martin if (slen >= (size_t)INT_MAX) 52 1.5.8.2 martin return gets(buf); 53 1.5.8.2 martin 54 1.5.8.2 martin if ((abuf = malloc(slen + 1)) == NULL) 55 1.5.8.2 martin return gets(buf); 56 1.5.8.2 martin 57 1.5.8.2 martin if (fgets(abuf, (int)(slen + 1), stdin) == NULL) 58 1.5.8.2 martin return NULL; 59 1.5.8.2 martin 60 1.5.8.2 martin len = strlen(abuf); 61 1.5.8.2 martin if (len > 0 && abuf[len - 1] == '\n') 62 1.5.8.2 martin --len; 63 1.5.8.2 martin 64 1.5.8.2 martin if (len >= slen) 65 1.5.8.2 martin __chk_fail(); 66 1.5.8.2 martin 67 1.5.8.2 martin (void)memcpy(buf, abuf, len); 68 1.5.8.2 martin 69 1.5.8.2 martin buf[len] = '\0'; 70 1.5.8.2 martin free(abuf); 71 1.5.8.2 martin return buf; 72 1.5.8.2 martin } 73
Indexes created Thu Oct 02 14:10:14 GMT 2025