/* * Copyright (c) 2021-2022 Cisco and/or its affiliates. * Licensed under the Apache License, Version 2.0 (the "License"); * you may not use this file except in compliance with the License. * You may obtain a copy of the License at: * * http://www.apache.org/licenses/LICENSE-2.0 * * Unless required by applicable law or agreed to in writing, software * distributed under the License is distributed on an "AS IS" BASIS, * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. * See the License for the specific language governing permissions and * limitations under the License. */ #include #include using namespace transport; int usage(char* program) { LoggerInfo() << "USAGE: " << program << "[-C|-S] [options] \n" << "Server or Client: \n" << " -P [FIRST_IPv6_WORD_HEX]\n" << " -t [number of threads]\n" << "Client Options: \n" << " -L [PROXY_LISTEN_PORT]\n" << "Server Options: \n" << " -a [ORIGIN_IP_ADDRESS]\n" << " -p [ORIGIN_PORT]\n" << " -c [CACHE_SIZE]\n" << " -m [MTU]" << " -l [DEFAULT_CONTENT_LIFETIME] (seconds)\n" << " -M (enable manifest)\n"; LoggerInfo() << "Example Server:\n" << " " << program << " -S -a example.com -p 80 -c 10000 -m 1300 -l 7200 -M -t 1 " "http://httpserver\n" << "Example Client:\n" << " " << program << " -C -L 9091 http://httpserver\n"; return -1; } struct Params : HTTPProxy::ClientParams, HTTPProxy::ServerParams { void printParams() override { if (client) { HTTPProxy::ClientParams::printParams(); } else if (server) { HTTPProxy::ServerParams::printParams(); } else { throw std::runtime_error( "Proxy configured as client and server at the same time."); } LoggerInfo() << "\tN Threads: " << n_thread; } HTTPProxy* instantiateProxyAsValue() { if (client) { HTTPProxy::ClientParams* p = dynamic_cast(this); return new transport::HTTPProxy(*p, n_thread); } else if (server) { HTTPProxy::ServerParams* p = dynamic_cast(this); return new transport::HTTPProxy(*p, n_thread); } else { throw std::runtime_error( "Proxy configured as client and server at the same time."); } } bool client = false; bool server = false; std::uint16_t n_thread = 1; }; int main(int argc, char** argv) { Params params; params.prefix = "http://hicn-http-proxy"; params.origin_address = "127.0.0.1"; params.origin_port = "80"; params.cache_size = "50000"; params.mtu = "1500"; params.first_ipv6_word = "b001"; params.content_lifetime = "7200;"; // seconds params.manifest = false; params.tcp_listen_port = 8080; int opt; while ((opt = getopt(argc, argv, "CSa:p:c:m:P:l:ML:t:")) != -1) { switch (opt) { case 'C': if (params.server) { LoggerErr() << "Cannot be both client and server (both -C anc -S " "options specified.)."; return usage(argv[0]); } params.client = true; break; case 'S': if (params.client) { LoggerErr() << "Cannot be both client and server (both -C anc -S " "options specified.)."; return usage(argv[0]); } params.server = true; break; case 'a': params.origin_address = optarg; break; case 'p': params.origin_port = optarg; break; case 'c': params.cache_size = optarg; break; case 'm': params.mtu = optarg; break; case 'P': params.first_ipv6_word = optarg; break; case 'l': params.content_lifetime = optarg; break; case 'L': params.tcp_listen_port = std::stoul(optarg); break; case 'M': params.manifest = true; break; case 't': params.n_thread = std::stoul(optarg); break; case 'h': default: return usage(argv[0]); } } if (argv[optind] == 0) { LoggerInfo() << "Using default prefix " << params.prefix; } else { params.prefix = argv[optind]; } params.printParams(); auto proxy = params.instantiateProxyAsValue(); proxy->run(); delete proxy; return 0; }