C++ 函数中返回字符串的一个陷阱

Posted 2020-10-18 左百工

tags:

篇首语：本文由小常识网(cha138.com)小编为大家整理，主要介绍了C++ 函数中返回字符串的一个陷阱相关的知识，希望对你有一定的参考价值。

 1 #include<iostream>
 2 using namespace std;
 3 char * getname(void);
 4 int main()
 5 {
 6     char * name;
 7     name = getname();
 8     cout << "My name is : " << name << endl;
 9     cout << "---------华丽的分割线----------" << endl;
10     return 0;
11 }
12 char * getname()
13 {
14     char get_name[30];
15     cout << "Enter your name :";// a word
16     cin >> get_name;
17     return get_name;
18 }

可能第一眼看上去没什么毛病，BUT getname()里面的get_name是一个字符串数组。在函数return之后这个get_name会释放内存（因为她在栈中，函数执行玩会弹栈）。所以main函数中的name变成了一个野指针，这是一个很危险的操作。

解决办法：返回一个在堆中的地址。

 1 #include<iostream>
 2 #include<cstring>
 3 using namespace std;
 4 char * getname(void);
 5 int main()
 6 {
 7     char * name;
 8     name = getname();
 9     cout << "My name is : " << name << endl;
10     cout << "---------华丽的分割线----------" << endl;
11     return 0;
12 }
13 char * getname()
14 {
15     char get_name[30];
16     cout << "Enter your name :";// a word
17     cin >> get_name;
18     char * name = new char[strlen(get_name)+1];
19     strcpy(name, get_name);// do not use name = get_name
20     //because name will get get_name address it\'s in the stack
21     //it is stupid idea.
22     return name;
23 }

考虑到内存的问题记得要在new之后不用了要delete，释放内存资源。

delete [] name；

没学过C语言，照着文档用malloc-free版本的。

 1 #include<iostream>
 2 #include<cstring>
 3 #include<cstdlib>
 4 using namespace std;
 5 char * getname(void);
 6 int main()
 7 {
 8     char * name;
 9     name = getname();
10     cout << "My name is : " << name << endl;
11 //    delete [] name;
12     free(name);
13     cout << "---------华丽的分割线----------" << endl;
14     return 0;
15 }
16 
17 char * getname()
18 {
19     char get_name[30];
20     cout << "Enter your name :";// read a word
21     cin >> get_name;
22     char * name = (char*)malloc(strlen(get_name)+1);
23     strcpy(name, get_name);
24     return name;
25 }