C语言getopt函数

一直都在用这个函数，今天再详细的复习一下

0x01 C语言getopt函数介绍

getopt函数主要用于命令行的参数解析，如果我们想对一个程序输入多个数据，就需要配置参数，而传递的参数那么多，就需要程序去逐个取得对应的参数值。

getopt函数在 #include <unistd.h> 头文件中

定义原型

int getopt(int argc,char * const argv[ ],const char * optstring);

函数说明

getopt()用来分析命令行参数。参数argc和argv是由main()传递的参数个数和内容。参数optstring 则代 表欲处理的选项字符串。此函数会返回在argv 中下一个的选项字母,此字母会对应参数optstring 中的字 母。如果选项字符串里的字母后接着冒号“:”,则表示还有相关的参数,全域变量optarg 即会指向此额 外参数。如果getopt()找不到符合的参数则会印出错信息,并将全域变量optopt设为“?”字符,如果不 希望getopt()印出错信息,则只要将全域变量opterr设为0即可。

返回值

如果找到符合的参数则返回此参数字母,如果参数不包含在参数optstring 的选项字母则返回“?”字符, 分析结束则返回-1。

0x02 C语言getopt函数范例

需求：需要一个程序能对用户传入用户名、密码进行加密、解密。

那么现在，我们要先设置好参数名；

例如：

• 用户名：u，usernme
• 密码：p，password

先写一个传入用户名和密码的程序，再考虑加密、解密。

#include <stdio.h>
#include <unistd.h>
#include <string.h>
#include <stdlib.h>

struct user{
    char username[100]; // 用户名
    char password[100]; // 密码

};

int main(int argc,char **argv){
    struct user info;
    memset(info.username,0, sizeof(info.username)); // 清空垃圾数据
    memset(info.password,0, sizeof(info.password)); // 清空垃圾数据
    int ch;
    while((ch = getopt(argc,argv,"u:p:e::"))!= -1){  // u后面跟一个“:”代表是必须要参数值 e后面跟两个“:” 代表参数可有可无，不需要参数
        switch(ch){
            case 'u':
                if(strlen(optarg)>99){  //读取用户名，当ch为u的时候，optarg 就代表当前参数值了。
                    printf("username to long ... \n");
                    exit(-1);
                }
                strcpy(info.username,optarg);
                break;
            case 'p':
                if(strlen(optarg)>99){  //读取密码，当ch为p的时候，optarg 就代表当前参数值了。
                    printf("password to long ... \n");
                    exit(-1);
                }
                strcpy(info.password,optarg);
                break;
            default:
                printf("argv %c \n",ch);
        }
    }
    printf("[*] 用户名 ：%s ，密码：%s \n",info.username,info.password);
    return 0;
}

执行结果：

./untitled -u admin -p admin
[*] 用户名 ：admin ，密码：admin

0x03 加密函数

既然我们得到了用户名和密码，接下来就是加密了。

#include <stdio.h>
#include <unistd.h>
#include <string.h>
#include <stdlib.h>
#include <crypt.h>


struct user{
    char username[9]; // 用户名
    char password[9]; // 密码
    char salt[3];
};

void  encryption(struct user * info){
   strcpy(info->username,crypt(info->username,info->salt));
   strcpy(info->password,crypt(info->password,info->salt));
}


int main(int argc,char **argv){
    struct user info;
    memset(info.username,0, sizeof(info.username)); // 清空垃圾数据
    memset(info.password,0, sizeof(info.password)); // 清空垃圾数据
    int ch;
    while((ch = getopt(argc,argv,"u:p:s:d::"))!= -1){  // u后面跟一个“:”代表是必须要参数值 e后面跟两个“:” 代表参数可有可无，不需要参数
        switch(ch){
            case 'u':
                strncpy(info.username,optarg,8);
                break;
            case 'p':
                strncpy(info.password,optarg,8);
                break;
            case 's':
                strncpy(info.salt,optarg,2);
                break;
            default:
                printf("argv %c \n",ch);
        }
    }
    encryption(&info);
    printf("[*] 用户名 ：%s ，密码：%s \n",info.username,info.password);
    return 0;
}

输出：

liyingzhe@thunderobot:~$ ./untitled -u admin -p admin -s 1o
[*] 用户名 ：1oAQMMMCT1orzL7NjOD/1g ，密码：1orzL7NjOD/1g