Implement Custom strlen Function

#include <stdio.h>

int custom_strlen(const char *str) {
    int len = 0;
    while(str[len]!='\0')
    len++;
    return len; 
}

int main() {
    char str[101];
     fgets(str, sizeof(str), stdin);

    // Remove newline if present
    int i = 0;
    while (str[i] != '\0') {
        if (str[i] == '\n') {
            str[i] = '\0';
            break;
        }
        i++;
    }

    printf("%d", custom_strlen(str));
    return 0;
}

#include <stdio.h>

int custom_strlen(const char *str) {
    int len = 0;
    while(str[len]!='\n')
    len++;
    return len;
}

int main() {
    char str[101];
    int str_length;
    fgets(str, sizeof(str), stdin);

    // Remove newline if present
    int i = 0;
    while (str[i] != '\0') {
        if (str[i] == '\n') {
            str[i] = '\0';
            break;
        }
        i++;
    }

    printf("%d", custom_strlen(str));
    return 0;
}