C 字符串指针函数 strdel

Question

谁能解释一下为什么我会收到“Segmentation fault...”以及如何修复这段代码？

#include<stdio.h>

int str_length(char *s) {
    int length = 0, i;
    for(i = 0; *s; i++) {
        s++;
    }
    return i;
}

char *strdel(char *s, int pos, int n) {
    int i;
    char *p, str[] = "";
    p = str;
    for(i = 0; i < str_length(s) - n + 1; i++)  {
        if(i >= pos) {
            *(p + i) = *(s + i + n);
        }
        else {
            *(p + i) = *(s + i);
        }
    }
    s = str;
    return s;
}

int main() {
    char *str = "abcdef";
    printf("str_lengh: %d\n", str_length(str));
    printf("strdel: %s\n", strdel(str, 1, 2));
    return 0;
}

我得到了这个输出:

str_lengh: 6
strdel: adef
Segmentation fault (core dumped)

另外，有没有更好的方法来创建一个函数: char *strdel(char *s, int pos, int n); 从位置 pos 删除的 n 个字符比我做的那个删除了 n 个字符？

Answer 1

我认为你在这里写满了堆栈......

char *strdel(char *s, int pos, int n) {
    int i;
    char *p, str[] = "";
    p = str; // p points to str which is "" and is on the stack with length 0.
    for(i = 0; i < str_length(s) - n + 1; i++)  {
        if(i >= pos) {
            *(p + i) = *(s + i + n); // now you are writing onto the stack past p
        }
        else {
            *(p + i) = *(s + i);// now you are writing onto the stack past p
        }
    }
    s = str; // now s points to space on stack
    return s; // now you return a pointer to the stack which is about to disapear 
}

每当你写过去的 p 时，你就会遇到未定义的行为。 UB 您正在写入尚未在堆或堆栈上分配的空间。

您可以编写一个仅适用于 s 的 strdel 版本。如果我对 strdel 的理解是正确的，那么就像这样:(粗略地说，未经测试!需要对 pos 和 n 进行边界检查)

char *strdel(char *s, int pos, int n) {
    char *dst = s + pos, *src = s + pos + n;
    while(*src) {
        *dst++ = *src++;
    }
    *dst = 0;
    return s;
}