You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
116 lines
3.3 KiB
116 lines
3.3 KiB
//
|
|
// Copyright (C) 2015 Google, Inc.
|
|
//
|
|
// 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 "base.h"
|
|
#include <rapidjson/document.h>
|
|
#include <rapidjson/writer.h>
|
|
#include <rapidjson/stringbuffer.h>
|
|
#include "utils/command_receiver.h"
|
|
|
|
#include <arpa/inet.h>
|
|
#include <errno.h>
|
|
#include <iostream>
|
|
#include <netinet/in.h>
|
|
#include <pthread.h>
|
|
#include <signal.h>
|
|
#include <stdlib.h>
|
|
#include <sys/types.h>
|
|
#include <sys/socket.h>
|
|
#include <unistd.h>
|
|
|
|
const int kBacklogInt = 10;
|
|
#define PORT 8080
|
|
// TODO: Set to a lower buffer size and read socket data until termination
|
|
#define SOCK_BUF_LEN 4096
|
|
#define MEMSET_VALUE 0
|
|
|
|
int client_sock;
|
|
int socket_desc;
|
|
|
|
void SockTest() {
|
|
char str[SOCK_BUF_LEN];
|
|
int listen_fd, comm_fd, c;
|
|
struct sockaddr_in servaddr, client;
|
|
rapidjson::Document d;
|
|
rapidjson::StringBuffer buffer;
|
|
rapidjson::Writer<rapidjson::StringBuffer> writer(buffer);
|
|
CommandReceiver cr;
|
|
|
|
listen_fd = socket(AF_INET, SOCK_STREAM, 0);
|
|
memset (&servaddr, MEMSET_VALUE, sizeof(servaddr));
|
|
servaddr.sin_family = AF_INET;
|
|
servaddr.sin_addr.s_addr = INADDR_ANY;
|
|
servaddr.sin_port = htons(PORT);
|
|
|
|
int bind_result = bind(
|
|
listen_fd, (struct sockaddr *) &servaddr, sizeof(servaddr));
|
|
if (bind_result != 0) {
|
|
LOG(ERROR) << sl4n::kTagStr <<
|
|
": Failed to assign the address to the socket."
|
|
<< " Error: " << strerror(errno) << ", " << errno;
|
|
exit(1);
|
|
}
|
|
|
|
int listen_result = listen(listen_fd, kBacklogInt);
|
|
if (listen_result != 0) {
|
|
LOG(ERROR) << sl4n::kTagStr << ": Failed to setup the passive socket."
|
|
<< " Error: " << strerror(errno) << ", " << errno;
|
|
exit(1);
|
|
}
|
|
|
|
comm_fd = accept(listen_fd, (struct sockaddr*)&client, (socklen_t*)&c);
|
|
if (comm_fd == -1) {
|
|
LOG(ERROR) << sl4n::kTagStr << ": Failed to accept the socket."
|
|
<< " Error: " << strerror(errno) << ", " << errno;
|
|
exit(1);
|
|
}
|
|
|
|
while (true) {
|
|
memset(str, MEMSET_VALUE, sizeof(str));
|
|
int read_result = read(comm_fd, str, SOCK_BUF_LEN);
|
|
if (read_result < 0) {
|
|
LOG(FATAL) << sl4n::kTagStr << ": Failed to write to the socket."
|
|
<< " Error: " << strerror(errno) << ", " << errno;
|
|
exit(1);
|
|
}
|
|
|
|
d.Parse(str);
|
|
cr.Call(d);
|
|
d.Accept(writer);
|
|
std::string str2 = buffer.GetString();
|
|
str2 += '\n';
|
|
strncpy(str, str2.c_str(), sizeof(str)-1);
|
|
int result = write(comm_fd, str, strlen(str)+1);
|
|
if (result < 0) {
|
|
LOG(FATAL) << sl4n::kTagStr << ": Failed to write to the socket."
|
|
<< " Error: " << strerror(errno) << ", " << errno;
|
|
exit(1);
|
|
}
|
|
d.RemoveAllMembers(); // Remove all members from the json object
|
|
buffer.Clear();
|
|
}
|
|
}
|
|
|
|
int main(int argc, char **argv) {
|
|
logging::LoggingSettings log_settings;
|
|
if (!logging::InitLogging(log_settings)) {
|
|
LOG(ERROR) << "Failed to set up logging";
|
|
return EXIT_FAILURE;
|
|
}
|
|
SockTest();
|
|
return 0;
|
|
}
|