C语言实现可变参函数

今天给大家介绍如何用C语言实现可变参数函数。

在日常的C语言编程中,相信很多人都喜欢用printf函数,不知道你是否有关注过该函数的声明呢?

int printf(const char *format, ...);

这里的...并非码哥偷懒写的省略符,而是一种特殊符号,或者说助记符,或者占位符,随你们喜欢啦。它的含义是表示在format参数后有不定个数不定类型的参数。

例如:

printf("%s %d %c\n", "abc", 10, 'A');

在本例中:

  • format"%s %d %c\n"
  • 其后不定参数部分有三个参数,分别为字符数组类型、整型、字符型。

实现可变参数函数

下面切入正题,码哥将实现一个可变参函数,直接上代码吧。

/*
 * Author:码哥比特
 */
#include 
#include 
#include 

void my_print(char *fmt, ...)
{
    va_list arg;
    va_start(arg, fmt);
    while (*fmt) {
        if (*fmt != '%') {
            write(STDOUT_FILENO, fmt, 1);
            ++fmt;
            continue;
        }
        ++fmt;
        switch (*fmt) {
            case 's': {
                char *s = va_arg(arg, char *);
                write(STDOUT_FILENO, s, strlen(s));
                break;
            }
            default:
                write(STDOUT_FILENO, fmt, 1);
                break;
        }
        ++fmt;
    }
    va_end(arg);
}

int main(void)
{
    my_print("%s %s %s\n", "abc", "bcd", "cde");
    return 0;
}

这段函数中,码哥实现了一个名为my_print的函数,这是一个可变参函数。

简短起见,我仅实现了%s输出字符串的功能,其余%x都会将x原封不动输出。

实现需要的工具函数

可变参的实现需要使用stdarg.h头文件下声明的函数和数据类型来完成。在本例中涉及1个数据类型和3个函数,他们分别是:

  • va_list类型
  • void va_start(va_list ap, last);
  • type va_arg(va_list ap, type);
  • void va_end(va_list ap);

在使用可变参部分前,首先要定义一个va_list变量,后续处理需要用到这个变量,大家也可以粗略的将其看作一个句柄(谁说只有fd能叫句柄呢~)。

下面我们对三个函数分别进行说明:

va_start

这个函数有两个参数,第一个参数是我们定义的va_list变量,第二个参数是可变参...左侧最靠近...的变量名,本例为fmt。通过这个函数就可以定位到可变参部分的位置了。

这里额外多说一点,在IA-32架构下,C调用约定中,函数参数都是由右向左依此压入栈中的。因此在获取可变参时,有一种比较hack的实现方式是,获取到可变参…前的那个参数在内存中的地址作为指针类型,然后自增获取其后的可变参数。(注意,x86架构处理器,栈增长方向是由高向低)。但是由于64位系统下,C函数参数为了性能提升,有部分是存放在寄存器中,因此这种方式不具有可移植性。

va_arg

这个函数(暂且当函数看吧)用于获取每一个可变参数,由其声明可以看到,返回值为我们需要的类型,函数参数为va_list变量和我们期望的函数类型。

可以参考上面的例子中,我们获取一个char *参数的写法:

char *s = va_arg(arg, char *);
va_end

用于销毁va_list结构所使用的资源。

到此,一个可变参函数就实现了,感兴趣的读者可以自己尝试实现intfloatchar等其他类型参数。


感谢阅读!

你可能感兴趣的:(c语言,经验分享,linux,学习,程序人生,c++,后端)