1/*
2 * Copyright (c) 2023 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#ifndef PRINT_SECURITY_GUARD_UTIL_H
17#define PRINT_SECURITY_GUARD_UTIL_H
18
19#include "print_util.h"
20
21namespace OHOS::Print {
22enum PrinterFoundType {
23    FROM_P2P = 0,
24    FROM_EPRINT = 1,
25    FROM_LOCAL_NET = 2,
26    FROM_USB = 3
27};
28
29class PrintSecurityGuardUtil {
30public:
31    static int32_t GetPrinterType(const std::string& des);
32};
33
34inline int32_t PrintSecurityGuardUtil::GetPrinterType(const std::string& des)
35{
36    std::string upper = PrintUtil::ToUpper(des);
37    std::map<std::string, int32_t> typeMap;
38    typeMap.insert({"P2P", FROM_P2P});
39    typeMap.insert({"EPRINT", FROM_EPRINT});
40    typeMap.insert({"USB", FROM_USB});
41
42    for (auto iter = typeMap.begin(); iter != typeMap.end(); iter++) {
43        std::string key = iter->first;
44        if (PrintUtil::CheckContains(upper, key)) {
45            return iter->second;
46        }
47    }
48    return FROM_LOCAL_NET;
49}
50} // namespace OHOS::Print
51
52#endif // PRINT_SECURITY_GUARD_UTIL_H