Пример #1
0
int
main (int argc, char *argv[])
{
    const char *progname;
    char *output_filename;
    FILE *peer_file = NULL, *output = NULL;
    int sock = -1, peer, err;
    struct sockaddr_in addr, peer_addr;
    struct hostent *hostinfo;
    socklen_t peer_addr_len;
    int reuse;
    int background;
    int ret = 0;

    progname = argv[0];

    background = 0;
    for (; argc >= 2; argc--, argv++) {
	if (argv[1][0] != '-')
	    break;
	if (strcmp (argv[1], "--") == 0) {
	    argc--;
	    argv++;
	    break;
	}
	if (strcmp (argv[1], "--background") == 0) {
	    background = 1;
	    continue;
	}
	fprintf (stderr, "%s: unregognized option '%s'\n",
		 progname, argv[1]);
	return 1;
    }

    if (argc != 2) {
	fprintf (stderr,
		 "Usage: %s [--background] <output-file>\n", progname);
	return 1;
    }

    output_filename = argv[1];
    output = fopen (output_filename, "w");
    if (output == NULL) {
	fprintf (stderr, "Failed to open %s for writing: %s\n",
		 output_filename, strerror (errno));
	ret = 1;
	goto DONE;
    }

    sock = socket (AF_INET, SOCK_STREAM, 0);
    if (sock == -1) {
	fprintf (stderr, "Error: socket() failed: %s\n",
		 strerror (errno));
	ret = 1;
	goto DONE;
    }

    reuse = 1;
    err = setsockopt (sock, SOL_SOCKET, SO_REUSEADDR, &reuse, sizeof (reuse));
    if (err) {
	fprintf (stderr, "Error: setsockopt() failed: %s\n",
		 strerror (errno));
	ret = 1;
	goto DONE;
    }

    hostinfo = gethostbyname ("localhost");
    if (hostinfo == NULL) {
	fprintf (stderr, "Unknown host: localhost\n");
	ret = 1;
	goto DONE;
    }

    memset (&addr, 0, sizeof (addr));
    addr.sin_family = AF_INET;
    addr.sin_port = htons (25025);
    addr.sin_addr = *(struct in_addr *) hostinfo->h_addr;
    err = bind (sock, (struct sockaddr *) &addr, sizeof (addr));
    if (err) {
	fprintf (stderr, "Error: bind() failed: %s\n",
		 strerror (errno));
	close (sock);
	ret = 1;
	goto DONE;
    }

    err = listen (sock, 1);
    if (err) {
	fprintf (stderr, "Error: listen() failed: %s\n",
		 strerror (errno));
	close (sock);
	ret = 1;
	goto DONE;
    }

    if (background) {
	int pid = fork ();
	if (pid > 0) {
	    printf ("smtp_dummy_pid='%d'\n", pid);
	    fflush (stdout);
	    close (sock);
	    ret = 0;
	    goto DONE;
	}
	if (pid < 0) {
	    fprintf (stderr, "Error: fork() failed: %s\n",
		     strerror (errno));
	    close (sock);
	    ret = 1;
	    goto DONE;
	}
	/* Reached if pid == 0 (the child process). */
	/* Close stdout so that the one interested in pid value will
	 * also get EOF. */
	close (STDOUT_FILENO);
	/* dup2() will re-reserve fd of stdout (1) (opportunistically),
	 * in case fd of stderr (2) is open. If that was not open we
	 * don't care fd of stdout (1) either. */
	dup2 (STDERR_FILENO, STDOUT_FILENO);

	/* This process is now out of reach of shell's job control.
	 * To resolve the rare but possible condition where this
	 * "daemon" is started but never connected this process will
	 * (only) have 30 seconds to exist. */
	alarm (30);
    }

    peer_addr_len = sizeof (peer_addr);
    peer = accept (sock, (struct sockaddr *) &peer_addr, &peer_addr_len);
    if (peer == -1) {
	fprintf (stderr, "Error: accept() failed: %s\n",
		 strerror (errno));
	ret = 1;
	goto DONE;
    }

    peer_file = fdopen (peer, "w+");
    if (peer_file == NULL) {
	fprintf (stderr, "Error: fdopen() failed: %s\n",
		 strerror (errno));
	ret = 1;
	goto DONE;
    }

    do_smtp_to_file (peer_file, output);

 DONE:
    if (output)
	fclose (output);
    if (peer_file)
	fclose (peer_file);
    if (sock >= 0)
	close (sock);

    return ret;
}
Пример #2
0
int
main (int argc, char *argv[])
{
	char *output_filename;
	FILE *peer_file, *output;
	int sock, peer, err;
	struct sockaddr_in addr, peer_addr;
	struct hostent *hostinfo;
	socklen_t peer_addr_len;
	int reuse;

	if (argc != 2) {
		fprintf (stderr, "Usage: %s <output-file>\n", argv[0]);
		exit (1);
	}

	output_filename = argv[1];
	output = fopen (output_filename, "w");
	if (output == NULL) {
		fprintf (stderr, "Failed to open %s for writing: %s\n",
			 output_filename, strerror (errno));
		exit (1);
	}

	sock = socket (AF_INET, SOCK_STREAM, 0);
	if (sock == -1) {
		fprintf (stderr, "Error: socket() failed: %s\n",
			 strerror (errno));
		exit (1);
	}

	reuse = 1;
	err = setsockopt (sock, SOL_SOCKET, SO_REUSEADDR, &reuse, sizeof (reuse));
	if (err) {
		fprintf (stderr, "Error: setsockopt() failed: %s\n",
			 strerror (errno));
		exit (1);
	}

	hostinfo = gethostbyname ("localhost");
	if (hostinfo == NULL) {
		fprintf (stderr, "Unknown host: localhost\n");
		exit (1);
	}

	addr.sin_family = AF_INET;
	addr.sin_port = htons (25025);
	addr.sin_addr = *(struct in_addr *) hostinfo->h_addr;
	err = bind (sock, (struct sockaddr *) &addr, sizeof(addr));
	if (err) {
		fprintf (stderr, "Error: bind() failed: %s\n",
			 strerror (errno));
		close (sock);
		exit (1);
	}

	err = listen (sock, 1);
	if (err) {
		fprintf (stderr, "Error: listen() failed: %s\n",
			 strerror (errno));
		close (sock);
		exit (1);
	}

	peer_addr_len = sizeof (peer_addr);
	peer = accept (sock, (struct sockaddr *) &peer_addr, &peer_addr_len);
	if (peer == -1) {
		fprintf (stderr, "Error: accept() failed: %s\n",
			 strerror (errno));
		exit (1);
	}

	peer_file = fdopen (peer, "w+");
	if (peer_file == NULL) {
		fprintf (stderr, "Error: fdopen() failed: %s\n",
			 strerror (errno));
		return;
	}

	do_smtp_to_file (peer_file, output);

	fclose (output);
	fclose (peer_file);
	close (sock);

	return 0;
}