[C언어 소스] 부분 문자열 복사하는 함수 만들기(버퍼 오버플로우 방지)

알고리즘

부분 문자열 복사(dest: 버퍼, dsize:버퍼 길이, src: 원본 문자열, n:복사할 문자 개수)

origin:= dest

반복(dsize와 n이 참이고 dest 가리키는 곳에 src가 가리키는 문자 대입)

dest와 src를 다음 위치로 증가, dsize 1감소, n 1감소

origin 반환

부분 문자열 복사 실행 화면

소스 코드

//부분 문자열 복사하는 함수 만들기 - 버퍼 오버플로우 방지
#include <stdio.h>
#include <string.h>
char *mystrncpy_s(char *dest, size_t dsize, const char *src, size_t n);
int main(void)
{
    char source[100] = "Hello World";
    char dest[100] = "abcde Yahoo";
    
    printf("원본 문자열:%s 의 5개 문자 복사\n", source);
    printf("복사 전 현재 결과 문자열:%s\n", dest);
    mystrncpy_s(dest, sizeof(dest), source, 5);
    printf("복사 후 결과 문자열: %s\n", dest);
    return 0;
}

char *mystrncpy_s(char *dest, size_t dsize, const char *src, size_t n)
{
    char *origin;

    dsize--;//맨 마지막에 종료 문자('\0')를 추가해야 하므로 버퍼 길이를 1 감소    

    //dsize와 n이 참이고 복사한 문자가 참이면 반복
    for (origin = dest; dsize && n && (*dest = *src); dest++, src++, n--, dsize--);
    return origin;
}

C언어 소스