Changeset 510

Timestamp:

07/12/11 13:06:33 (10 months ago)

Author:

jet

Message:

-- Add support for --pid option

Location:

trunk

Files:

• man/crtmpserver.1 (modified) (1 diff)
• sources/crtmpserver/src/crtmpserver.cpp (modified) (4 diffs)

trunk/man/crtmpserver.1

@@ -57 +57 @@
 .B \-\-gid=<gid>
 Run the process with the specified group id.
+.TP
+.B \-\-pid=<file name>
+This option tells \fBcrtmpserver\fP to use the specified file as its pidfile. If the file exists, it will be over-written.
+.br
+Works only if \fB--daemon\fP option is specified.
 .SH FILES
 .TP

trunk/sources/crtmpserver/src/crtmpserver.cpp

@@ -61 +61 @@
 void NormalizeCommandLine(string configFile);
 bool ApplyUIDGID();
+void WritePidFile(pid_t pid);
 
 RunningStatus gRs;
@@ -170 +171 @@
 
                         if (pid > 0) {
+                                if (gRs.commandLine["arguments"].HasKey("--pid"))
+                                        WritePidFile(pid);
                                 return false;
                         }
@@ -292 +295 @@
         cout << "    --gid=<gid>" << endl;
         cout << "      Run the process with the specified group id\n" << endl;
+        cout << "    --pid=<pid_file>" << endl;
+        cout << "      Create PID file."<< endl;
+        cout << "      Works only if --daemon option is specified\n"<< endl;
 }
 
@@ -342 +348 @@
 }
 
+void WritePidFile(pid_t pid) {
+        /*!
+         * Check if PID file exists
+         * If exists -> skip pid file creation
+         * if none -> write pid file
+         */
+        string pidFile = gRs.commandLine["arguments"]["--pid"];
+        struct stat sb;
+        if (stat(pidFile.c_str(), &sb) == 0) {
+                printf("pid file exists\n");
+        } else if (errno != ENOENT) {
+                perror("stat");
+                return;
+        }
+
+        FILE * f;
+        f = fopen(pidFile.c_str(), "w");
+        if (f == NULL) {
+                perror("fopen");
+                printf("warn: can not create PID file %s\n", pidFile.c_str());
+                return;
+        }
+        fprintf(f, "%d", pid);
+        if (fclose(f) == -1) {
+                perror("fclose");
+        }
+}
+
 #ifdef COMPILE_STATIC
 #ifdef HAS_APP_ADMIN