1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92
|
#include <string.h>
#include <stdlib.h>
#include <stdio.h>
static char *prompt (void)
{
char *invite = NULL;
char const *user = getenv("USERNAME");
if (user == NULL)
{
user = getenv("USER");
if (user == NULL)
{
user = "";
}
}
{
char const *host = getenv("HOSTNAME");
if (host == NULL)
{
host = "";
}
{
char const *aux = getenv("PWD");
if (aux == NULL)
{
aux = "";
}
{
char const *rep = aux + strlen(aux);
while (rep >= aux && *rep != '/')
{
rep--;
}
rep++;
invite = malloc(strlen(user) + strlen(host) + strlen(rep) + 6);
if (invite != NULL)
{
strcpy(invite, "[");
strcat(invite, user);
strcat(invite, "@");
strcat(invite, host);
strcat(invite, " ");
strcat(invite, rep);
strcat(invite, "]");
strcat(invite, "$");
strcat(invite, " ");
}
}
}
}
return invite;
}
int main (void)
{
int end = 0;
while (!end)
{
char *s = prompt();
if (s != NULL)
{
printf ("%s", s);
fflush (stdout);
free (s), s = NULL;
}
{
char line[128];
fgets(line, sizeof line, stdin);
end = strcmp(line, "quit\n") == 0;
}
}
return 0;
} |
Partager