1. 输入参数选项处理
在生成 elf 可执行文件时，对输入的参数做解析。
解析代码：
int main(int argc, char *argv[]) {
std::string source, output;
int isencode = 0;
int c;
while((c = getopt_long(argc, argv, short_options, long_options, nullptr)) != -1) {
printf("[+]get options : %d \n", c);
switch (c) {
case 'e':
source = optarg;
isencode = 1;
break;
case 'd':
source = optarg;
isencode = 2;
break;
case 'o':
output = optarg;
break;
case 'h':
useage();
break;
default:
break;
}
}
}
以上就是对参数命令的解析。同时需要设置命令选项：
const char* short_options = "hdm:d:e:o:"; // 带参数的，要使用的命令，需要在这里声明
const struct option long_options[] = {
{"help", 0, NULL, 'h'},
{"decode", 1, NULL, 'd'},
{"encode", 1, NULL, 'e'},
{"output", 1, NULL, 'o'},
{nullptr, 0, nullptr, 0}
};
void useage();
void useage() {
printf(" Options are:\n");
printf(" -d --decode DecryptionAES \n");
printf(" -e --memso memBaseAddr(16bit format) EncryptionAES \n");
printf(" -o --output generateFilePath Generate file path\n");
printf(" -h --help Display this information\n");
}
long_options[] 对应 getopt_long 函数，这个标准函数如下：
参数longopts，其实是一个结构的实例：
struct option {
const char *name; //name表示的是长参数名
int has_arg； //has_arg有3个值，no_argument(或者是0)，表示该参数后面不跟参数值
// required_argument(或者是1),表示该参数后面一定要跟个参数值
// optional_argument(或者是2),表示该参数后面可以跟，也可以不跟参数值
int *flag;
//用来决定，getopt_long()的返回值到底是什么。如果flag是null(通常情况)，则函数会返回与该项option匹配的val值;如果flag不是NULL，则将val值赋予flag所指向的内存，并且返回值设置为0。
int val; //和flag联合决定返回值
}
就是为 1 时，后面可跟参数。
2. 完整示例如下：
#include <iostream>
#include "AES.h"
#include "Base64.h"
#include <getopt.h>
#include <sys/stat.h>
#include <sys/mman.h>
#include <fcntl.h>
#include <sys/types.h>
#include <unistd.h>
using namespace std;
const char* short_options = "hdm:d:e:o:";
const struct option long_options[] = {
{"help", 0, NULL, 'h'},
{"decode", 1, NULL, 'd'},
{"encode", 1, NULL, 'e'},
{"output", 1, NULL, 'o'},
{nullptr, 0, nullptr, 0}
};
void useage();
int main(int argc, char *argv[])
{
/* string str1;
cin >> str1;
cout << "加密前:" << str1 << endl;
string str2 = EncryptionAES(str1);
cout << "加密后:" << str2 << endl;
string str3 = DecryptionAES(str2);
cout << "解密后:" << str3 << endl; */
std::string source, output;
int isencode = 0;
int c;
while((c = getopt_long(argc, argv, short_options, long_options, nullptr)) != -1) {
printf("[+]get options : %d \n", c);
switch (c) {
case 'e':
source = optarg;
isencode = 1;
break;
case 'd':
source = optarg;
isencode = 2;
break;
case 'o':
output = optarg;
break;
case 'h':
useage();
break;
default:
break;
}
}
printf("[-]open %s gooooo, cnum : %d \n",source.c_str(), c);
int fd=open(source.c_str(), O_RDWR);
if(!fd) {
printf("source so file cannot found!!!\n");
return -1;
}
struct stat buf={0};
printf("step 1 \n");
int status=stat(source.c_str(),&buf); // 获取文件信息
printf("[+]fstat %s path, size : %d \n",source.c_str(), buf.st_size);
if (status==-1) {
printf("[-]fstat %s failed \n",source.c_str());
return -1;
}
printf("step 2 \n");
int g_dexSize = buf.st_size;
printf("step 3 \n");
char *readbuf = new char[g_dexSize];
printf("step 4 \n");
int ret = read(fd, readbuf, g_dexSize);
printf("step 5 ret: %d \n", ret);
if (ret != g_dexSize) {
printf("[-]read %d failed",g_dexSize);
}
printf("step 6 \n");
char *rbuf = readbuf;
string s = rbuf;
string rstr = "";
int lenss = s.length();
printf("step 7 lensss : %d\n", lenss);
if (isencode == 1) {
rstr = EncryptionAES(s);
printf("step 8 rstr\n");
}
if (isencode == 2) {
rstr = DecryptionAES(s);
}
printf("step 9 \n");
const char *t = rstr.data();
printf("step 10 t : %c \n", t);
int len = rstr.length();
printf("[+]encode %d length \n", len);
auto wfd=fopen(output.c_str(), "wb+");
if(nullptr == wfd) {
printf("output so file cannot write !!!\n");
return -1;
}
printf("step 11 wfd\n");
//int wsize = write(wfd, t, len);
int wsize = fwrite(t, len, 1, wfd);
printf("step 12 wfd : %d \n", wsize);
if (wsize != 1) {
printf("[-]encode %d error", wsize);
}
close(fd);
fclose(wfd);
printf("step 13 end \n");
return 0;
}
void useage() {
printf(" Options are:\n");
printf(" -d --decode DecryptionAES \n");
printf(" -e --memso memBaseAddr(16bit format) EncryptionAES \n");
printf(" -o --output generateFilePath Generate file path\n");
printf(" -h --help Display this information\n");
}
网友评论