C语言-自定义函数

// 输入任意字符串，以某个字符结束，并返回字符串长度
int get_str( char *s, int size, const char ch='\n' ) //如果不输入第三个参数，默认为回车符
{
    int i;
    for( i=0;i<size-1;i++ )
    {
        s[i]=getchar(); //读入一个字符
        if ( s[i] == ch ) //遇到指定字符，结束输入
            break;
    }
    s[i]='\0'; //字符串结束符
    return i; //返回读入字符串数据的长度
}
 
void main()
{
    char str[10];
    get_str( str, sizeof(str), '#' ); //以#结束
    printf("input is :%s\n", str );
    fflush(stdin); //清除缓存中的残余数据
     
    get_str( str, sizeof(str) ); //以回车结束
    printf("input is :%s\n", str );
}