int main(int argc, char *argv[])
{
char *sUsername, *sPassword;
- int pid, uid;
+ int pid, uid = 0;
int status = 0;
tUserInfo *uinfo;
- putchar('\n');
+ printf("\x1B[2J"); // Clear Screen
+
for(;;)
{
// Validate User
- do {
+ for(;;)
+ {
sUsername = GetUsername();
sPassword = GetPassword();
- } while( (uid = ValidateUser(sUsername, sPassword)) == -1 );
- putchar('\n');
-
- // Get user information
- uinfo = GetUserInfo(uid);
+ if( (uid = ValidateUser(sUsername, sPassword)) == -1 )
+ {
+ printf("\nInvalid username or password for '%s'\n", sUsername);
+ free(sUsername);
+ free(sPassword);
+ }
+ else
+ break;
+ }
+ printf("\n");
// Create child process
pid = clone(CLONE_VM, 0);
// Spawn shell in a child process
if(pid == 0)
{
- char *argv[2] = {uinfo->Shell, 0};
- char **envp = NULL;
+ char *child_argv[2] = {NULL, 0};
+ char **child_envp = NULL;
+
+ // Get user information
+ uinfo = GetUserInfo(uid);
+
+ child_argv[0] = uinfo->Shell;
+ // Set Environment
setgid(uinfo->GID);
- setuid(uid);
+ //setuid(uid);
+ setuid(uinfo->UID);
- execve(uinfo->Shell, argv, envp);
+ execve(uinfo->Shell, child_argv, child_envp);
exit(-1);
}
*/
char *GetUsername()
{
- char ret[BUFLEN];
+ char ret[BUFLEN] = {0};
int pos = 0;
char ch;
while( (ch = fgetc(stdin)) != -1 && ch != '\n' )
{
if(ch == '\b') {
+ if( pos <= 0 ) continue;
pos --;
ret[pos] = '\0';
}
while( (ch = fgetc(stdin)) != -1 && ch != '\n' )
{
if(ch == '\b') {
+ if( pos <= 0 ) continue;
pos --;
ret[pos] = '\0';
}