ZCS-11191: [Backend] Integrate postsrsd

Zimbra · May 25, 2022 · d7f4e71 · d7f4e71
1 parent c63deb5
commit d7f4e71
Showing 1 changed file with 122 additions and 6 deletions.
diff --git a/store/src/java/com/zimbra/cs/service/admin/ModifyServer.java b/store/src/java/com/zimbra/cs/service/admin/ModifyServer.java
@@ -16,19 +16,30 @@
  */
 package com.zimbra.cs.service.admin;
 
+import java.io.BufferedReader;
+import java.io.IOException;
+import java.io.InputStream;
+import java.io.InputStreamReader;
+import java.util.ArrayList;
+import java.util.Arrays;
+import java.util.Collections;
 import java.util.List;
 import java.util.Map;
+import java.util.stream.Collectors;
+import java.util.stream.Stream;
 
-import com.zimbra.cs.account.AccountServiceException;
-import com.zimbra.cs.account.Server;
-import com.zimbra.cs.account.Provisioning;
-import com.zimbra.cs.account.accesscontrol.AdminRight;
-import com.zimbra.cs.account.accesscontrol.Rights.Admin;
 import com.zimbra.common.account.Key;
+import com.zimbra.common.localconfig.LC;
 import com.zimbra.common.service.ServiceException;
-import com.zimbra.common.util.ZimbraLog;
 import com.zimbra.common.soap.AdminConstants;
 import com.zimbra.common.soap.Element;
+import com.zimbra.common.util.ZimbraLog;
+import com.zimbra.cs.account.AccountServiceException;
+import com.zimbra.cs.account.Provisioning;
+import com.zimbra.cs.account.Server;
+import com.zimbra.cs.account.ZAttrServer;
+import com.zimbra.cs.account.accesscontrol.AdminRight;
+import com.zimbra.cs.account.accesscontrol.Rights.Admin;
 import com.zimbra.soap.ZimbraSoapContext;
 
 /**
@@ -58,6 +69,8 @@ public Element handle(Element request, Map<String, Object> context) throws Servi
         }
         checkRight(zsc, context, server, attrs);
 
+        startOrStopPostSRSd(server, attrs);
+
         // pass in true to checkImmutable
         prov.modifyAttrs(server, attrs, true);
 
@@ -74,4 +87,107 @@ public void docRights(List<AdminRight> relatedRights, List<String> notes) {
         notes.add(String.format(AdminRightCheckPoint.Notes.MODIFY_ENTRY,
                 Admin.R_modifyServer.getName(), "server"));
     }
+
+    public void startOrStopPostSRSd(Server server, Map<String, Object> attrs) throws ServiceException {
+
+    	ZimbraLog.mailbox.info("==============================");
+
+    	List<String> command = new ArrayList<>();
+    	List<String> response = new ArrayList<>();
+    	List<String> attrsUI = new ArrayList<>();
+    	List<String> attrsLDAP = new ArrayList<>();
+    	boolean UIWantsToEnablePostsrs = false;
+    	boolean isPostsrsEnabledInLDAP = false;
+    	final String POSTSRSD_EXE = LC.zimbra_home.value() + "/common/sbin/postsrsd";
+    	final String POSTSRSD_SECRET = "/opt/zimbra/common/etc/postsrsd.secret";
+
+    	try {
+
+    		if(!attrs.isEmpty()) {
+    			Collections.addAll(attrsUI, (String[]) attrs.get("zimbraServiceEnabled"));
+    			ZimbraLog.mailbox.info("attrsUI: " + attrsUI);
+    			UIWantsToEnablePostsrs = attrsUI.contains("postsrs");
+    		}
+
+    		if(!server.getAttrs().isEmpty()) {
+    			Collections.addAll(attrsLDAP, server.getServiceEnabled());
+    			ZimbraLog.mailbox.info("attrsLDAP: " + attrsLDAP);
+    			isPostsrsEnabledInLDAP = attrsLDAP.contains("postsrs");
+    		}
+
+			if(UIWantsToEnablePostsrs && !isPostsrsEnabledInLDAP) {
+				command = Stream.of(POSTSRSD_EXE, "-s", POSTSRSD_SECRET, "-d", "mydomain", "-D").collect(Collectors.toList());
+				response = executeLinuxCommand(command);
+				ZimbraLog.mailbox.info(response);
+				ZimbraLog.mailbox.info("postsrsd has been enabled");
+			}
+			else if(UIWantsToEnablePostsrs && isPostsrsEnabledInLDAP) {
+				ZimbraLog.mailbox.info("postsrsd is already enabled");
+			}
+			else if(!UIWantsToEnablePostsrs && isPostsrsEnabledInLDAP) {
+				// There is no command to disable SRS so far. The only way is killing the process.
+				command = Stream.of("pgrep", "-f", "postsrsd").collect(Collectors.toList());
+				response = executeLinuxCommand(command);
+				ZimbraLog.mailbox.info("response: " + response);
+				if(response.isEmpty()) {
+					ZimbraLog.mailbox.info("postsrsd is already disabled");
+				}
+				else {
+					String postSrsdPID = response.get(0);
+					command.clear();
+					response.clear();
+					command = Stream.of("kill", "-9", postSrsdPID).collect(Collectors.toList());
+					response = executeLinuxCommand(command);
+					ZimbraLog.mailbox.info("response: " + response);
+					ZimbraLog.mailbox.info("postsrsd has been disabled");
+				}
+			}
+			else if(!UIWantsToEnablePostsrs && !isPostsrsEnabledInLDAP) {
+				ZimbraLog.mailbox.info("postsrsd is already disabled");
+			}
+
+		} catch (IOException e) {
+			ZimbraLog.mailbox.warn(e);
+		} catch (InterruptedException e) {
+			ZimbraLog.mailbox.warn(e);
+		}
+
+
+    	ZimbraLog.mailbox.info("==============================");
+
+    }
+
+    public List<String> executeLinuxCommand(List<String> command) throws IOException, InterruptedException  {
+
+    	InputStream is = null;
+    	ProcessBuilder pb = null;
+		Process ps = null;
+		List<String> lines = new ArrayList<>();
+
+    	ZimbraLog.mailbox.info("command: " + command);
+
+		// Executing the linux command
+		pb = new ProcessBuilder(command);
+		ps = pb.start();
+		int exitValue = ps.waitFor();
+		ZimbraLog.mailbox.info("command executed");
+		// Getting executed command response as List
+		if(exitValue == 0) {
+			is = ps.getInputStream();
+		}
+		else {
+			is = ps.getErrorStream();
+		}
+		InputStreamReader isr = new InputStreamReader(is);
+		BufferedReader br = new BufferedReader(isr);
+		String line;
+		while((line = br.readLine()) != null) {
+			lines.add(line);
+		}
+		ps.destroy();
+
+		return lines;
+
+	}
+
 }