Annotation of src/usr.bin/sudo/alias.c, Revision 1.1
1.1 ! millert 1: /*
! 2: * Copyright (c) 2004-2005m, 2007-2008
! 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: * OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF
! 17: * ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
! 18: */
! 19:
! 20: #include <config.h>
! 21:
! 22: #include <sys/types.h>
! 23: #include <sys/param.h>
! 24: #include <stdio.h>
! 25: #ifdef STDC_HEADERS
! 26: # include <stdlib.h>
! 27: # include <stddef.h>
! 28: #else
! 29: # ifdef HAVE_STDLIB_H
! 30: # include <stdlib.h>
! 31: # endif
! 32: #endif /* STDC_HEADERS */
! 33: #ifdef HAVE_STRING_H
! 34: # include <string.h>
! 35: #else
! 36: # ifdef HAVE_STRINGS_H
! 37: # include <strings.h>
! 38: # endif
! 39: #endif /* HAVE_STRING_H */
! 40: #ifdef HAVE_UNISTD_H
! 41: # include <unistd.h>
! 42: #endif /* HAVE_UNISTD_H */
! 43:
! 44: #include "sudo.h"
! 45: #include "parse.h"
! 46: #include "redblack.h"
! 47: #include <gram.h>
! 48:
! 49: #ifndef lint
! 50: __unused static const char rcsid[] = "$Sudo: alias.c,v 1.13 2008/11/09 14:13:12 millert Exp $";
! 51: #endif /* lint */
! 52:
! 53: /*
! 54: * Globals
! 55: */
! 56: struct rbtree *aliases;
! 57: unsigned int alias_seqno;
! 58:
! 59: /*
! 60: * Local protoypes
! 61: */
! 62: static int alias_compare __P((const void *, const void *));
! 63: static void alias_free __P((void *));
! 64:
! 65: /*
! 66: * Comparison function for the red-black tree.
! 67: * Aliases are sorted by name with the type used as a tie-breaker.
! 68: */
! 69: static int
! 70: alias_compare(v1, v2)
! 71: const void *v1, *v2;
! 72: {
! 73: const struct alias *a1 = (const struct alias *)v1;
! 74: const struct alias *a2 = (const struct alias *)v2;
! 75: int res;
! 76:
! 77: if (v1 == NULL)
! 78: res = -1;
! 79: else if (v2 == NULL)
! 80: res = 1;
! 81: else if ((res = strcmp(a1->name, a2->name)) == 0)
! 82: res = a1->type - a2->type;
! 83: return(res);
! 84: }
! 85:
! 86: /*
! 87: * Search the tree for an alias with the specified name and type.
! 88: * Returns a pointer to the alias structure or NULL if not found.
! 89: */
! 90: struct alias *
! 91: find_alias(name, type)
! 92: char *name;
! 93: int type;
! 94: {
! 95: struct alias key;
! 96: struct rbnode *node;
! 97: struct alias *a = NULL;
! 98:
! 99: key.name = name;
! 100: key.type = type;
! 101: if ((node = rbfind(aliases, &key)) != NULL) {
! 102: /*
! 103: * Compare the global sequence number with the one stored
! 104: * in the alias. If they match then we've seen this alias
! 105: * before and found a loop.
! 106: */
! 107: a = node->data;
! 108: if (a->seqno == alias_seqno)
! 109: return(NULL);
! 110: a->seqno = alias_seqno;
! 111: }
! 112: return(a);
! 113: }
! 114:
! 115: /*
! 116: * Add an alias to the aliases redblack tree.
! 117: * Returns NULL on success and an error string on failure.
! 118: */
! 119: char *
! 120: alias_add(name, type, members)
! 121: char *name;
! 122: int type;
! 123: struct member *members;
! 124: {
! 125: static char errbuf[512];
! 126: struct alias *a;
! 127:
! 128: a = emalloc(sizeof(*a));
! 129: a->name = name;
! 130: a->type = type;
! 131: a->seqno = 0;
! 132: list2tq(&a->members, members);
! 133: if (rbinsert(aliases, a)) {
! 134: alias_free(a);
! 135: snprintf(errbuf, sizeof(errbuf), "Alias `%s' already defined", name);
! 136: return(errbuf);
! 137: }
! 138: return(NULL);
! 139: }
! 140:
! 141: /*
! 142: * Apply a function to each alias entry and pass in a cookie.
! 143: */
! 144: void
! 145: alias_apply(func, cookie)
! 146: int (*func) __P((void *, void *));
! 147: void *cookie;
! 148: {
! 149: rbapply(aliases, func, cookie, inorder);
! 150: }
! 151:
! 152: /*
! 153: * Returns TRUE if there are no aliases, else FALSE.
! 154: */
! 155: int
! 156: no_aliases()
! 157: {
! 158: return(rbisempty(aliases));
! 159: }
! 160:
! 161: /*
! 162: * Free memory used by an alias struct and its members.
! 163: */
! 164: static void
! 165: alias_free(v)
! 166: void *v;
! 167: {
! 168: struct alias *a = (struct alias *)v;
! 169: struct member *m;
! 170: struct sudo_command *c;
! 171: void *next;
! 172:
! 173: efree(a->name);
! 174: for (m = a->members.first; m != NULL; m = next) {
! 175: next = m->next;
! 176: if (m->type == COMMAND) {
! 177: c = (struct sudo_command *) m->name;
! 178: efree(c->cmnd);
! 179: efree(c->args);
! 180: }
! 181: efree(m->name);
! 182: efree(m);
! 183: }
! 184: efree(a);
! 185: }
! 186:
! 187: /*
! 188: * Find the named alias, delete it from the tree and recover its resources.
! 189: */
! 190: int
! 191: alias_remove(name, type)
! 192: char *name;
! 193: int type;
! 194: {
! 195: struct rbnode *node;
! 196: struct alias key;
! 197:
! 198: key.name = name;
! 199: key.type = type;
! 200: if ((node = rbfind(aliases, &key)) == NULL)
! 201: return(FALSE);
! 202: rbdelete(aliases, node);
! 203: alias_free(node->data);
! 204: return(TRUE);
! 205: }
! 206:
! 207: void
! 208: init_aliases()
! 209: {
! 210: if (aliases != NULL)
! 211: rbdestroy(aliases, alias_free);
! 212: aliases = rbcreate(alias_compare);
! 213: }