libft/ft_split.c at master · fausrguez/libft · 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
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
/* ************************************************************************** */
/*                                                                            */
/*                                                        ::::::::            */
/*   ft_split.c                                         :+:    :+:            */
/*                                                     +:+                    */
/*   By: farodrig <farodrig@student.codam.nl>         +#+                     */
/*                                                   +#+                      */
/*   Created: 2020/12/05 18:58:33 by farodrig      #+#    #+#                 */
/*   Updated: 2021/02/28 22:05:31 by farodrig      ########   odam.nl         */
/*                                                                            */
/* ************************************************************************** */

#include "libft.h"
#include <stdlib.h>

static int	cpy_word(
	char **arr,
	size_t dimension,
	char *str,
	size_t w_len)
{
	arr[dimension] = ft_calloc((w_len + 1), sizeof(char));
	if (!arr[dimension])
	{
		return (0);
	}
	ft_strlcpy(arr[dimension], str, w_len + 1);
	return (1);
}

static size_t	count_words(char *str, char c)
{
	char	*last_c;
	size_t	words;

	last_c = str - 1;
	words = 0;
	while (str)
	{
		if (*str == c || *str == 0)
		{
			if (str - last_c > 1)
			{
				words++;
			}
			last_c = str;
			if (*str == 0)
			{
				break ;
			}
		}
		str++;
	}
	return (words);
}

static void	free_all(char **arr, size_t dimension)
{
	while (dimension)
	{
		free(arr[dimension]);
		dimension--;
	}
	free(arr);
}

static int	str_split(char **arr, char *str, char c, size_t n_words)
{
	char	*last_c;
	size_t	dimension;

	last_c = str - 1;
	dimension = 0;
	while (dimension < n_words)
	{
		if (*str == c)
		{
			if (str - last_c > 1)
			{
				if (!cpy_word(arr, dimension, last_c + 1, (str - last_c) - 1))
				{
					free_all(arr, dimension);
					return (0);
				}
				dimension++;
			}
			last_c = str;
		}
		str++;
	}
	arr[dimension] = 0;
	return (1);
}

/*
** Allocates (with malloc(3)) and returns an array of strings obtained by
** splitting 's' using the character 'c' as a delimiter. The array must be
** ended by a NULL pointer. Returns NULL if the allocation fails
*/

char	**ft_split(char const *str, char c)
{
	char	**arr;
	size_t	n_words;

	n_words = count_words((char *)str, c);
	arr = ft_calloc(n_words + 1, sizeof(char *));
	if (!arr)
	{
		return (0);
	}
	if (str_split(arr, (char *)str, c, n_words))
		return (arr);
	return (0);
}