simple_shell/custom_setenv.c at main · ImStillBlessed/simple_shell · GitHub

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
93
94
95
96
97
98
99
#include "main.h"

/**
 *_cdSetenv - sets specific variable
 *@data: shell data
 *@var: variable
 *@value: its value
 *Return: should return 0 on success
*/

int _cdSetenv(shell_data *data, char *var, char *value)
{
	int i = 0;
	char *new_env, *env, *name;

	for (; data->_environ[i]; i++)
	{
		env = _strdup(data->_environ[i]);
		name = _strtok(env, "=");
		if (strcmp(name, var) == 0)
		{
			new_env = malloc(_strlen(var) + _strlen(value) + 2);
			_strcpy(new_env, var);
			_strcat(new_env, "=");
			_strcat(new_env, value);
			free(data->_environ[i]);
			data->_environ[i] = new_env;
			environ[i] = data->_environ[i];
			freeInputArgsData(data, 1, 0, 1);
			free(env);
			return (0);
		}
		free(env);
	}
	return (-1);
}
/**
 *_setenv - sets environment variable
 *@data: shell data
 *Return: should return 0 on success
*/

int _setenv(shell_data *data)
{
	int i = 0;
	char *new_env, *env, *name;

	if (data->args[2] == NULL)
		data->args[2] = "";
	for (; data->_environ[i]; i++)
	{
		env = _strdup(data->_environ[i]);
		name = _strtok(env, "=");
		if (strcmp(name, data->args[1]) == 0)
		{
			new_env = malloc(_strlen(data->args[1]) + _strlen(data->args[2]) + 2);
			_strcpy(new_env, data->args[1]);
			_strcat(new_env, "=");
			_strcat(new_env, data->args[2]);
			free(data->_environ[i]);
			data->_environ[i] = new_env;
			environ[i] = data->_environ[i];
			freeInputArgsData(data, 1, 0, 1);
			free(env);
			return (0);
		}
		free(env);
	}
	new_env = malloc(_strlen(data->args[1]) + _strlen(data->args[2]) + 2);
	_strcpy(new_env, data->args[1]);
	_strcat(new_env, "=");
	_strcat(new_env, data->args[2]);
	data->_environ[i] = new_env;
	environ[i] = data->_environ[i];
	freeInputArgsData(data, 1, 0, 1);
	return (0);
}

/**
 *setenvCommand - command for setenv
 *@data: shell data
*/

void setenvCommand(shell_data *data)
{
	if (data->args[3] != NULL)
	{
		printError(data, "./shell", ": too many arguments\n", 127);
		freeInputArgsData(data, 1, 0, 1);
		return;
	}
	if (data->args[1] == NULL && data->args[2] == NULL)
	{
		printError(data, "./shell", ": not enough arguments\n", 127);
		freeInputArgsData(data, 1, 0, 1);
		return;
	}
	_setenv(data);
}