Add custom indentation config for vim.
[libguestfs.git] / fish / tilde.c
1 /* guestfish - the filesystem interactive shell
2  * Copyright (C) 2009 Red Hat Inc.
3  *
4  * This program is free software; you can redistribute it and/or modify
5  * it under the terms of the GNU General Public License as published by
6  * the Free Software Foundation; either version 2 of the License, or
7  * (at your option) any later version.
8  *
9  * This program is distributed in the hope that it will be useful,
10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
12  * GNU General Public License for more details.
13  *
14  * You should have received a copy of the GNU General Public License
15  * along with this program; if not, write to the Free Software
16  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
17  */
18
19 #include <config.h>
20
21 #include <stdio.h>
22 #include <stdlib.h>
23 #include <string.h>
24 #include <unistd.h>
25 #include <assert.h>
26 #include <pwd.h>
27 #include <sys/types.h>
28
29 #include "fish.h"
30
31 static char *expand_home (char *orig, const char *append);
32 static const char *find_home_for_username (const char *, size_t);
33 static const char *find_home_for_current_user (void);
34
35 /* This is called from the script loop if we find a candidate for
36  * ~username (tilde-expansion).
37  */
38 char *
39 try_tilde_expansion (char *str)
40 {
41   assert (str[0] == '~');
42
43   /* Expand "~" to current user's home directory. */
44   if (str[1] == '\0')           /* ~ */
45     return expand_home (str, NULL);
46   else if (str[1] == '/')       /* ~/... */
47     return expand_home (str, &str[1]);
48
49   /* Try expanding the part up to the following '\0' or '/' as a
50    * username from the password file.
51    */
52   else {
53     const char *home, *rest;
54     size_t len = strcspn (&str[1], "/");
55     rest = &str[1+len];
56
57     home = find_home_for_username (&str[1], len);
58
59     if (home) {
60       len = strlen (home) + strlen (rest) + 1;
61       str = malloc (len);
62       if (str == NULL) {
63         perror ("malloc");
64         exit (EXIT_FAILURE);
65       }
66       strcpy (str, home);
67       strcat (str, rest);
68       return str;
69     }
70   }
71
72   /* No match, return the orignal string. */
73   return str;
74 }
75
76 /* Return $HOME + append string. */
77 static char *
78 expand_home (char *orig, const char *append)
79 {
80   const char *home;
81   int len;
82   char *str;
83
84   home = getenv ("HOME");
85   if (!home) {
86     /* $HOME not set, bash can look up the current user in the
87      * password file and find their home that way.  (RHBZ#617440).
88      */
89     home = find_home_for_current_user ();
90     if (!home)
91       return orig;
92   }
93
94   len = strlen (home) + (append ? strlen (append) : 0) + 1;
95   str = malloc (len);
96   if (str == NULL) {
97     perror ("malloc");
98     exit (EXIT_FAILURE);
99   }
100
101   strcpy (str, home);
102   if (append)
103     strcat (str, append);
104
105   return str;
106 }
107
108 /* Lookup username (of length ulen), return home directory if found,
109  * or NULL if not found.
110  */
111 static const char *
112 find_home_for_username (const char *username, size_t ulen)
113 {
114   struct passwd *pw;
115
116   setpwent ();
117   while ((pw = getpwent ()) != NULL) {
118     if (strlen (pw->pw_name) == ulen &&
119         STREQLEN (username, pw->pw_name, ulen))
120       return pw->pw_dir;
121   }
122
123   return NULL;
124 }
125
126 static const char *
127 find_home_for_current_user (void)
128 {
129   struct passwd *pw;
130   uid_t euid = geteuid ();
131
132   setpwent ();
133   while ((pw = getpwent ()) != NULL) {
134     if (pw->pw_uid == euid)
135       return pw->pw_dir;
136   }
137
138   return NULL;
139 }