Add sessionConfigSetKeepRunning and shutdown API function

Setting sessionConfigSetKeepRunning to true makes aria2 core keep
running even if there is no download to perform, just like --enable-rpc
option.

1 files changed, 19 insertions, 0 deletions

diff --git a/src/aria2api.cc b/src/aria2api.cc
index 9f46e957..72352266 100644
--- a/src/aria2api.cc
+++ b/src/aria2api.cc
@@ -106,6 +106,13 @@ int sessionFinal(Session* session)
   return rv;
 }
 
+int sessionConfigSetKeepRunning(Session* session, bool flag)
+{
+  session->context->reqinfo->getDownloadEngine()->getRequestGroupMan()
+    ->setKeepRunning(flag);
+  return 0;
+}
+
 int run(Session* session, RUN_MODE mode)
 {
   const SharedHandle<DownloadEngine>& e =
@@ -113,6 +120,18 @@ int run(Session* session, RUN_MODE mode)
   return e->run(mode == RUN_ONCE);
 }
 
+int shutdown(Session* session, bool force)
+{
+  const SharedHandle<DownloadEngine>& e =
+    session->context->reqinfo->getDownloadEngine();
+  if(force) {
+    e->requestForceHalt();
+  } else {
+    e->requestHalt();
+  }
+  return 0;
+}
+
 std::string gidToHex(const A2Gid& gid)
 {
   return GroupId::toHex(gid);