1 /*
2 * Copyright (C) 2021 Huawei Device Co., Ltd.
3 * Licensed under the Apache License, Version 2.0 (the "License");
4 * you may not use this file except in compliance with the License.
5 * You may obtain a copy of the License at
6 *
7 * http://www.apache.org/licenses/LICENSE-2.0
8 *
9 * Unless required by applicable law or agreed to in writing, software
10 * distributed under the License is distributed on an "AS IS" BASIS,
11 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12 * See the License for the specific language governing permissions and
13 * limitations under the License.
14 */
15
16 #include "nstackx_getopt.h"
17 #include "nstackx_error.h"
18 #include "nstackx_log.h"
19
20 #define TAG "nStackXGetOpt"
21 #define NSTACK_GETOPT_MAX_ARGC 100
22
NstackInitGetOptMsg(NstackGetOptMsg * optMsg)23 int32_t NstackInitGetOptMsg(NstackGetOptMsg *optMsg)
24 {
25 if (optMsg == NULL) {
26 LOGE(TAG, "optMsg is NULL");
27 return NSTACKX_EFAILED;
28 }
29 optMsg->argvIdx = 1;
30 optMsg->argvOffset = 1;
31 optMsg->attachArg = NULL;
32 return NSTACKX_EOK;
33 }
34
GetOptCheckInputArg(NstackGetOptMsg * optMsg,int argc,const char * const * argv)35 static int32_t GetOptCheckInputArg(NstackGetOptMsg *optMsg, int argc, const char *const *argv)
36 {
37 if (optMsg->argvIdx >= argc || argv[optMsg->argvIdx][0] != '-' || argv[optMsg->argvIdx][1] == '\0') {
38 return NSTACKX_EFAILED;
39 } else if (!strcmp(argv[optMsg->argvIdx], "--")) {
40 optMsg->argvIdx++;
41 return NSTACKX_EFAILED;
42 }
43 return NSTACKX_EOK;
44 }
45
GetOptCheckCurrentOpt(NstackGetOptMsg * optMsg,int32_t currentOpt,char * currentOptIdxInOpts,char nextChar)46 static int32_t GetOptCheckCurrentOpt(NstackGetOptMsg *optMsg, int32_t currentOpt,
47 char *currentOptIdxInOpts, char nextChar)
48 {
49 if (currentOpt == ':' || currentOptIdxInOpts == NULL) {
50 LOGE(TAG, ": illegal option -- %c", currentOpt);
51 optMsg->argvOffset++;
52 if (nextChar == '\0') {
53 optMsg->argvIdx++;
54 optMsg->argvOffset = 1;
55 }
56 return NSTACKX_EFAILED;
57 }
58 return NSTACKX_EOK;
59 }
60
GetOptParseAttachArg(NstackGetOptMsg * optMsg,int32_t argc,const char * const * argv,char currentOpt,const char * currentOptIdxInOpts)61 static int32_t GetOptParseAttachArg(NstackGetOptMsg *optMsg, int32_t argc, const char *const *argv,
62 char currentOpt, const char *currentOptIdxInOpts)
63 {
64 if (*(currentOptIdxInOpts + 1) == ':') {
65 if (argv[optMsg->argvIdx][optMsg->argvOffset + 1] != '\0') {
66 optMsg->attachArg = &argv[optMsg->argvIdx++][optMsg->argvOffset + 1];
67 } else if (++optMsg->argvIdx >= argc) {
68 LOGE(TAG, ": option requires an argument -- %c", currentOpt);
69 optMsg->argvOffset = 1;
70 return NSTACKX_EFAILED;
71 } else {
72 optMsg->attachArg = argv[optMsg->argvIdx++];
73 }
74 optMsg->argvOffset = 1;
75 } else {
76 if (argv[optMsg->argvIdx][++(optMsg->argvOffset)] == '\0') {
77 optMsg->argvOffset = 1;
78 optMsg->argvIdx++;
79 }
80 optMsg->attachArg = NULL;
81 }
82 return NSTACKX_EOK;
83 }
84
NstackCheckArg(const NstackGetOptMsg * optMsg,int32_t argc,const char * const * argv)85 static int32_t NstackCheckArg(const NstackGetOptMsg *optMsg, int32_t argc, const char *const *argv)
86 {
87 if (optMsg == NULL) {
88 LOGE(TAG, "optMsg is NULL");
89 return NSTACKX_EFAILED;
90 }
91 if (argc <= 1 || argc > NSTACK_GETOPT_MAX_ARGC) {
92 LOGE(TAG, "argc is invalid %u", argc);
93 return NSTACKX_EFAILED;
94 }
95 if (argv == NULL) {
96 LOGE(TAG, "argv is NULL");
97 return NSTACKX_EFAILED;
98 }
99 int32_t i;
100 for (i = 0; i < argc; i++) {
101 if (argv[i] == NULL) {
102 LOGE(TAG, "argv[%d] is NULL", i);
103 return NSTACKX_EFAILED;
104 }
105 }
106 return NSTACKX_EOK;
107 }
108
NstackGetOpt(NstackGetOptMsg * optMsg,int32_t argc,const char * const * argv,const char * opts)109 int32_t NstackGetOpt(NstackGetOptMsg *optMsg, int32_t argc, const char *const *argv, const char *opts)
110 {
111 if (NstackCheckArg(optMsg, argc, argv) != NSTACKX_EOK) {
112 return NSTACK_GETOPT_END_OF_STR;
113 }
114 int32_t currentOpt;
115 if (optMsg->argvOffset == 1 && GetOptCheckInputArg(optMsg, argc, argv) != NSTACKX_EOK) {
116 return NSTACK_GETOPT_END_OF_STR;
117 }
118 currentOpt = argv[optMsg->argvIdx][optMsg->argvOffset];
119 char *currentOptIdxInOpts = strchr(opts, currentOpt);
120 if (GetOptCheckCurrentOpt(optMsg, currentOpt, currentOptIdxInOpts,
121 argv[optMsg->argvIdx][optMsg->argvOffset + 1]) != NSTACKX_EOK) {
122 return NSTACK_GETOPT_UNKNOW_OPT;
123 }
124 if (GetOptParseAttachArg(optMsg, argc, argv, currentOpt, currentOptIdxInOpts) != NSTACKX_EOK) {
125 return NSTACK_GETOPT_UNKNOW_OPT;
126 }
127 return currentOpt;
128 }
129
NstackGetOptArgs(const NstackGetOptMsg * optMsg)130 const char *NstackGetOptArgs(const NstackGetOptMsg *optMsg)
131 {
132 if (optMsg == NULL) {
133 LOGE(TAG, "optMsg is NULL");
134 return NULL;
135 }
136 return optMsg->attachArg;
137 }
138
139