Objective-C可變參數(shù)的函數(shù)實(shí)現(xiàn)

ciscopuke

1.前言
相信接觸過OC的對(duì)NSLog都很熟悉，細(xì)心查看NSLog的原始定義，會(huì)發(fā)現(xiàn)，他的原型如下：

FOUNDATION_EXPORT void NSLog(NSString *format, ...) NS_FORMAT_FUNCTION(1,2);
路徑在：OS X version/Frameworks/Foundation/NSObjCRuntime.h

注意到參數(shù)最后的...，這里是可變參數(shù)。這樣，在調(diào)用時(shí)就可以根據(jù)需要傳入相應(yīng)個(gè)數(shù)的參數(shù)了。

PS：其實(shí)在C#中也有params指定可變參數(shù)，跟OC這個(gè)很類似。

那么，如何在自己寫的函數(shù)中實(shí)現(xiàn)可變參數(shù)呢？

2.實(shí)現(xiàn)
要實(shí)現(xiàn)OC中的可變參數(shù)，需要幾個(gè)宏定義va_list、va_start、va_arg、va_end，先實(shí)現(xiàn)效果，以無限個(gè)整數(shù)相加為例：

RandomArgs.h

1. #import <Foundation/Foundation.h>
   
2. 
   
3. @interface RandomArgs : NSObject
   
4. -(int)add:(int)item,...;
   
5. @end
   
6. RandomArgs.m
   
7. 
   
8. #import "RandomArgs.h"
   
9. 
   
10. @implementation RandomArgs
    
11. -(int)add:(int)item,...{
    
12.     va_list list;
    
13.     va_start(list, item);
    
14. 
    
15.     int result=0;
    
16.      NSLog(@"第一個(gè)參數(shù)：%d",item);
    
17.     result+=item;
    
18.     int arg;
    
19.     while ((arg=va_arg(list,int))) {
    
20.          NSLog(@"當(dāng)前參數(shù)：%d",arg);
    
21.         result+=arg;
    
22.     }
    
23.     va_end(list);
    
24.     return result;
    
25. }
    
26. @end
    
27. main.m
    
28. 
    
29. #import <Foundation/Foundation.h>
    
30. #import "RandomArgs.h"
    
31. 
    
32. int main(int argc, const char * argv[]) {
    
33.     @autoreleasepool {
    
34.         RandomArgs* rand=[[RandomArgs alloc]init];
    
35.         int result=[rand add:4,5,6,nil];
    
36.         NSLog(@"結(jié)果：%d",result);
    
37. 
    
38.     }
    
39.     return 0;
    
40. }

復(fù)制代碼

效果

3.總結(jié)
主要是通過循環(huán)va_arg來獲取，但是要注意的是，第一個(gè)參數(shù)必須是固定的，循環(huán)里面只能獲取第二個(gè)參數(shù)以后的參數(shù)。

4.原理
參數(shù)在堆棧中分布,位置

在進(jìn)程中,堆棧地址是從高到低分配的.當(dāng)執(zhí)行一個(gè)函數(shù)的時(shí)候,將參數(shù)列表入棧,壓入堆棧的高地址部分,然后入棧函數(shù)的返回地址,接著入棧函數(shù)的執(zhí)行代碼,這個(gè)入棧過程,堆棧地址不斷遞減,一些黑客就是在堆棧中修改函數(shù)返回地址,執(zhí)行自己的代碼來達(dá)到執(zhí)行自己插入的代碼段的目的。

總之,函數(shù)在堆棧中的分布情況是:地址從高到低,依次是:函數(shù)參數(shù)列表,函數(shù)返回地址,函數(shù)執(zhí)行代碼段。

堆棧中,各個(gè)函數(shù)的分布情況是倒序的.即最后一個(gè)參數(shù)在列表中地址最高部分,第一個(gè)參數(shù)在列表地址的最低部分.參數(shù)在堆棧中的分布情況如下:

最后一個(gè)參數(shù)

倒數(shù)第二個(gè)參數(shù)

...

第一個(gè)參數(shù)

函數(shù)返回地址

函數(shù)代碼段