Linux守护进程是Linux的后台服务进程,相当于Windows服务,对于为Linux开发服务程序的朋友来说,Linux守护进程相关技术是必不可少的,因为这个技术不仅仅是为了开发守护进程,还可以拓展到多进程,父子进程文件描述符共享,父子进程通讯、控制等方面,是实现Linux大型服务的基础技术之一。
去年我也曾写了一篇关于守护进程的帖子,名字叫《.NET跨平台实践:用C#开发Linux守护进程》,这篇文章的的确确实现了一个Daemon,不过,它有一个弱点,不能运行多线程!
这篇帖子的目的就是进一步完善,让我们写出一个功能完整,可以用于生产环节的基本的守护进程。
先帖代码(假设项目名是daemon):
1 using System;2 using System.Threading;3 using System.Timers;4 using System.Runtime.InteropServices;5 using System.IO;6 using System.Text;7 8 9 /********************************************10 * 一个完整的linux daemon示例,作者宇内流云 *11 ********************************************/12 13 namespace daemon14 {15 class Program16 {17 18 const string DaemonTag = "--daemon.";19 static void Main(string[] args)20 {21 // 判断是否已经进入Daemon状态,如果是,就直接执行后台主函数22 if (string.IsNullOrEmpty(Environment.GetEnvironmentVariable(DaemonTag)) == false)23 {24 Environment.SetEnvironmentVariable(DaemonTag, null);25 DaemonMain(args);26 return;27 }28 29 30 // 如果还没有进入daemon状态,就作daemon处理31 /32 33 int pid = fork();34 if (pid != 0) exit(0);35 setsid();36 pid = fork();37 if (pid != 0) exit(0);38 umask(0);39 40 41 // 这儿已经进入“守护进程”工作状态了!42 43 // 关闭所有打开的文件描述符44 int max = open("/dev/null", 0);45 for (var i = 0; i <= max; i++) { close(i); }46 47 48 // 设置标记,防止重复运行进入49 Environment.SetEnvironmentVariable(DaemonTag,"yes");50 51 52 //为execp参数重组参数53 var args1 = args == null ? new string[2] : new string[args.Length + 2];54 55 args1[0] = "MyDaemon";56 args1[1] = Path.Combine(Environment.CurrentDirectory, Thread.GetDomain().FriendlyName);57 58 if (args1.Length > 2)59 {60 for (var i = 0; i < args.Length; i++)61 { args1[i + 2] = args[i]; }62 }63 64 65 //守护状态下重新加载和运行本程序66 execvp("mono", args1);67 68 }69 70 71 /// <summary>72 /// Daemon工作状态的主方法73 /// </summary>74 /// <param name="aargs"></param>75 static void DaemonMain(string[] aargs)76 {77 //启动一个线程去处理一些事情78 (new Thread(DaemonWorkFunct) { IsBackground = true }).Start();79 80 81 //daemon时,控制台输入、输出流已经关闭82 //请不要再用Console.Write/Read等方法83 84 //阻止daemon进程退出85 (new AutoResetEvent(false)).WaitOne();86 87 }88 89 90 static FileStream fs;91 static int count = 0;92 static void DaemonWorkFunct() {93 fs = File.Open("/tmp/daemon.txt", FileMode.OpenOrCreate);94 var t = new System.Timers.Timer() { Interval = 1000 };95 t.Elapsed += OnElapsed;96 t.Start();97 }98 private static void OnElapsed(object sender, ElapsedEventArgs e)99 { 100 var s = DateTime.Now.ToString("yyy-MM-dd HH:mm:ss") + "\n"; 101 var b = Encoding.ASCII.GetBytes(s); 102 fs.Write(b, 0, b.Length); 103 fs.Flush(); 104 105 count++; 106 if (count > 100) { 107 fs.Close(); 108 fs.Dispose(); 109 exit(0); 110 } 111 112 } 113 114 115 116 [DllImport("libc", SetLastError = true)] 117 static extern int fork(); 118 119 [DllImport("libc", SetLastError = true)] 120 static extern int setsid(); 121 122 [DllImport("libc", SetLastError = true)] 123 static extern int umask(int mask); 124 125 [DllImport("libc", SetLastError = true)] 126 static extern int open([MarshalAs(UnmanagedType.LPStr)]string pathname, int flags); 127 128 [DllImport("libc", SetLastError = true)] 129 static extern int close(int fd); 130 131 [DllImport("libc", SetLastError = true)] 132 static extern int exit(int code); 133 134 [DllImport("libc", SetLastError = true)] 135 static extern int execvp([MarshalAs(UnmanagedType.LPStr)]string file, string[] argv); 136 137 138 } 139 140 }
以上代码的工作过程是:判断程序自身是否已经处于daemon(后台服务)状态,如果是,就直接开始具体的服务工作(开启一个线程,每秒向 /tmp/daemon.txt中打印一行字符,100次后退出),如果不是daemon状态,就进入Daemon处理,使之进入daemon工作状态。
以上代码编译后,会生成一个叫 daemon.exe 的程序,当然,这个程序是为linux开发的,不能在windows上运行。现在,我把它放到linux上面,用mono daemon.exe命令启动它。
这时我们可以看到这个程序启动后,控制台上没有任何输出,也没有阻塞控制台,那么,在哪儿能找到它呢?用 ps -ef命令看看,原来它真的已经在后台运行起来了。
再看看这个后台进程是否完成了它的工作:cat /tmp/daemon.txt 查看文件内容:
从生成的文件内容看,这个Daemon服务程序的确按我们的设计意图,每秒钟向/tmp/daemon.txt打印了一行字符。