graphs/piscine/my_strtok_r/my_strtok_r.c


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

#include "my_strtok_r.h"

#include <stddef.h>

static int is_delim(char c, const char *delims)
{
    for (const char *d = delims; *d; d++)
    {
        if (*d == c)
            return 1;
    }
    return 0;
}

char *my_strtok_r(char *str, const char *delim, char **saveptr)
{
    if (str == NULL)
    {
        if (*saveptr == NULL)
        {
            return NULL;
        }
        str = *saveptr;
    }

    size_t i = 0;
    while (str[i] != '\0' && is_delim(str[i], delim))
    {
        i++;
    }
    if (str[i] == '\0')
    {
        *saveptr = NULL;
        return NULL;
    }

    char *res = str + i;

    while (str[i] != '\0' && !is_delim(str[i], delim))
    {
        i++;
    }
    if (str[i] == '\0')
    {
        *saveptr = NULL;
        return res;
    }
    *saveptr = str + i + 1;
    str[i] = '\0';
    return res;
}