Annotation of embedaddon/sudo/plugins/sudoers/prompt.c, revision 1.1
1.1 ! misho 1: /*
! 2: * Copyright (c) 1993-1996,1998-2005, 2007-2013
! 3: * Todd C. Miller <Todd.Miller@courtesan.com>
! 4: *
! 5: * Permission to use, copy, modify, and distribute this software for any
! 6: * purpose with or without fee is hereby granted, provided that the above
! 7: * copyright notice and this permission notice appear in all copies.
! 8: *
! 9: * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
! 10: * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
! 11: * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
! 12: * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
! 13: * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
! 14: * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
! 15: * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
! 16: *
! 17: * Sponsored in part by the Defense Advanced Research Projects
! 18: * Agency (DARPA) and Air Force Research Laboratory, Air Force
! 19: * Materiel Command, USAF, under agreement number F39502-99-1-0512.
! 20: */
! 21:
! 22: #include <config.h>
! 23:
! 24: #include <sys/types.h>
! 25: #include <stdio.h>
! 26: #ifdef STDC_HEADERS
! 27: # include <stdlib.h>
! 28: # include <stddef.h>
! 29: #else
! 30: # ifdef HAVE_STDLIB_H
! 31: # include <stdlib.h>
! 32: # endif
! 33: #endif /* STDC_HEADERS */
! 34: #ifdef HAVE_STRING_H
! 35: # include <string.h>
! 36: #endif /* HAVE_STRING_H */
! 37: #ifdef HAVE_STRINGS_H
! 38: # include <strings.h>
! 39: #endif /* HAVE_STRINGS_H */
! 40: #include <pwd.h>
! 41: #include <grp.h>
! 42:
! 43: #include "sudoers.h"
! 44:
! 45: /*
! 46: * Expand %h and %u escapes in the prompt and pass back the dynamically
! 47: * allocated result. Returns the same string if there are no escapes.
! 48: */
! 49: char *
! 50: expand_prompt(const char *old_prompt, const char *user, const char *host)
! 51: {
! 52: size_t len, n;
! 53: int subst;
! 54: const char *p;
! 55: char *np, *new_prompt, *endp;
! 56: debug_decl(expand_prompt, SUDO_DEBUG_AUTH)
! 57:
! 58: /* How much space do we need to malloc for the prompt? */
! 59: subst = 0;
! 60: for (p = old_prompt, len = strlen(old_prompt); *p; p++) {
! 61: if (p[0] =='%') {
! 62: switch (p[1]) {
! 63: case 'h':
! 64: p++;
! 65: len += strlen(user_shost) - 2;
! 66: subst = 1;
! 67: break;
! 68: case 'H':
! 69: p++;
! 70: len += strlen(user_host) - 2;
! 71: subst = 1;
! 72: break;
! 73: case 'p':
! 74: p++;
! 75: if (def_rootpw)
! 76: len += 2;
! 77: else if (def_targetpw || def_runaspw)
! 78: len += strlen(runas_pw->pw_name) - 2;
! 79: else
! 80: len += strlen(user_name) - 2;
! 81: subst = 1;
! 82: break;
! 83: case 'u':
! 84: p++;
! 85: len += strlen(user_name) - 2;
! 86: subst = 1;
! 87: break;
! 88: case 'U':
! 89: p++;
! 90: len += strlen(runas_pw->pw_name) - 2;
! 91: subst = 1;
! 92: break;
! 93: case '%':
! 94: p++;
! 95: len--;
! 96: subst = 1;
! 97: break;
! 98: default:
! 99: break;
! 100: }
! 101: }
! 102: }
! 103:
! 104: if (subst) {
! 105: new_prompt = emalloc(++len);
! 106: endp = new_prompt + len;
! 107: for (p = old_prompt, np = new_prompt; *p; p++) {
! 108: if (p[0] =='%') {
! 109: switch (p[1]) {
! 110: case 'h':
! 111: p++;
! 112: n = strlcpy(np, user_shost, np - endp);
! 113: if (n >= np - endp)
! 114: goto oflow;
! 115: np += n;
! 116: continue;
! 117: case 'H':
! 118: p++;
! 119: n = strlcpy(np, user_host, np - endp);
! 120: if (n >= np - endp)
! 121: goto oflow;
! 122: np += n;
! 123: continue;
! 124: case 'p':
! 125: p++;
! 126: if (def_rootpw)
! 127: n = strlcpy(np, "root", np - endp);
! 128: else if (def_targetpw || def_runaspw)
! 129: n = strlcpy(np, runas_pw->pw_name, np - endp);
! 130: else
! 131: n = strlcpy(np, user_name, np - endp);
! 132: if (n >= np - endp)
! 133: goto oflow;
! 134: np += n;
! 135: continue;
! 136: case 'u':
! 137: p++;
! 138: n = strlcpy(np, user_name, np - endp);
! 139: if (n >= np - endp)
! 140: goto oflow;
! 141: np += n;
! 142: continue;
! 143: case 'U':
! 144: p++;
! 145: n = strlcpy(np, runas_pw->pw_name, np - endp);
! 146: if (n >= np - endp)
! 147: goto oflow;
! 148: np += n;
! 149: continue;
! 150: case '%':
! 151: /* convert %% -> % */
! 152: p++;
! 153: break;
! 154: default:
! 155: /* no conversion */
! 156: break;
! 157: }
! 158: }
! 159: *np++ = *p;
! 160: if (np >= endp)
! 161: goto oflow;
! 162: }
! 163: *np = '\0';
! 164: } else
! 165: new_prompt = estrdup(old_prompt);
! 166:
! 167: debug_return_str(new_prompt);
! 168:
! 169: oflow:
! 170: /* We pre-allocate enough space, so this should never happen. */
! 171: fatalx(_("internal error, %s overflow"), "expand_prompt()");
! 172: }
FreeBSD-CVSweb <freebsd-cvsweb@FreeBSD.org>
![[BACK]](/icons/cvsweb/back.gif){kind=icon}
Return to prompt.c CVS log ![[TXT]](/icons/cvsweb/text.gif){kind=icon}
![[DIR]](/icons/cvsweb/dir.gif){kind=icon}
Up to [ELWIX - Embedded LightWeight unIX -] / embedaddon / sudo / plugins / sudoers