背景：
在我们开发C/C++项目中，经常会使用到系统提供的可变参数函数，例如printf,scanf等。根据需求我们也可能使用到可变参数函数，虽然真正使用的机会不是很多，但是我还是比较好奇如何去实现可变参数函数。这篇博文主要是我记录学习可变参数的一个demo，具体功能是模仿scanf，目前只实现简单的功能，对可变参数函数的一种简单理解。其中使用到这篇博文使用宏实现日志信息以及异常处理。除此之外，我还搜索了几遍博文感觉讲的不错，大家也可以仔细读一读。
Demo:
1 /** WARNING: this code is fresh and potentially isn't correct yet. */ 2 #include <stdio.h> 3 #include <stdlib.h> 4 #include <stdarg.h> 5 #include "dbg.h" 6 7 #define MAX_DATA 100 8 9 int read_string(char **out_string, int max_buffer) 10 { 11 *out_string = calloc(1, max_buffer + 1); 12 13 check_mem(*out_string); 14 char *result = fgets(*out_string, max_buffer, stdin); 15 check(result != NULL, "Input error."); 16 return 0; 17 error: 18 if(*out_string) free(*out_string); 19 *out_string = NULL; 20 return -1; 21 } 22 23 int read_int(int *out_int) 24 { 25 char *input = NULL; 26 int rc = read_string(&input, MAX_DATA); 27 check(rc == 0, "Failed to read number."); 28 29 *out_int = atoi(input); 30 free(input); 31 return 0; 32 error: 33 if(input) free(input); 34 return -1; 35 } 36 37 int read_scan(const char *fmt, ...) 38 { 39 int i = 0; 40 int rc = 0; 41 int *out_int = NULL; 42 char *out_char = NULL; 43 char *out_string = NULL; 44 int max_buffer = 0; 45 va_list argp; 46 va_start(argp, fmt); 47 48 for (i = 0; fmt[i] != '\0'; i++) { 49 if (fmt[i] == '%') { 50 i++; 51 switch(fmt[i]) 52 { 53 case '\0': 54 sentinel("Invalid format, you ended with %%."); 55 break; 56 case 'd': 57 out_int = va_arg(argp, int *); 58 rc = read_int(out_int); 59 check(rc == 0, "Failed to read int."); 60 break; 61 case 'c': 62 out_char = va_arg(argp, char *); 63 *out_char = fgetc(stdin); 64 break; 65 case 's': 66 max_buffer = va_arg(argp, int); 67 out_string = va_arg(argp, char **); 68 rc = read_string(out_string, max_buffer); 69 check(rc == 0, "Failed to read string."); 70 break; 71 default: 72 sentinel("Invalid format."); 73 } 74 } else { 75 fgetc(stdin); 76 } 77 check(!feof(stdin) && ! ferror(stdin), "Input error."); 78 } 79 va_end(argp); 80 return 0; 81 error: 82 va_end(argp); 83 return -1; 84 } 85 86 int main(int argc, char* argv[]) 87 { 88 char *first_name = NULL; 89 char initial = ' '; 90 char *last_name = NULL; 91 int age = 0; 92 int age1 = 0; 93 int age2 = 0; 94 printf("What's your first name?"); 95 int rc = read_scan("%s", MAX_DATA, &first_name); 96 check(rc == 0, "Failed first name."); 97 98 printf("What's your initial?"); 99 rc = read_scan("%c\n", &initial); 100 check(rc == 0, "Failed initial."); 101 102 printf("What's your last name?"); 103 rc = read_scan("%s", MAX_DATA, &last_name); 104 check(rc == 0, "Failed last name."); 105 106 printf("How old are you?"); 107 rc = read_scan("%d", &age); 108 check(rc == 0, "Failed age."); 109 110 rc = read_scan("%d%s", &age1, MAX_DATA, &first_name); 111 check(rc == 0, "Failed more age."); 112 113 printf("-------- RESULT -------\n"); 114 printf("First name: %s", first_name); 115 printf("Initial: %c\n", initial); 116 printf("Last name : %s", last_name); 117 printf("Age: %d\n", age1); 118 119 free(first_name); 120 free(last_name); 121 return 0; 122 error: 123 return -1; 124 }
这个demo主要是简单模拟scanf函数功能。
看看结果：
1 zhaoscmatoMacBook-Pro:c zhaosc$ ./ex25 2 What's your first name?LI 3 What's your initial?T 4 What's your last name?Shuchao 5 How old are you?26 6 23 7 Zhaosc 8 -------- RESULT ------- 9 First name: Zhaosc 10 Initial: T 11 Last name : Shuchao 12 Age: 23
简单说说：
首先必须引入头文件stdarg.h，这样才能使用结构体va_list,和宏va_start,va_arg,va_start,va_end,具体每个宏在“读一读”中有详细的说明。这个demo只支持%s,%c和%d,他们可以单独使用也可以组合使用。其中主要处理在read_scan函数中。