1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78
|
/*
* chatter-socket.c -- do some read/writes on a given Unix stream socket, for
* testing socket r/w recording
*
* Copyright (C) 2013 Canonical Ltd.
* Author: Martin Pitt <martin.pitt@ubuntu.com>
*
* umockdev is free software; you can redistribute it and/or
* modify it under the terms of the GNU Lesser General Public
* License as published by the Free Software Foundation; either
* version 2.1 of the License, or (at your option) any later version.
*
* umockdev is distributed in the hope that it will be useful, but
* WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
* Lesser General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public License
* along with this program; If not, see <http://www.gnu.org/licenses/>.
*/
#include <stdio.h>
#include <assert.h>
#include <fcntl.h>
#include <unistd.h>
#include <string.h>
#include <sys/socket.h>
#include <sys/un.h>
#define writestr(s) assert(write(fd, s, strlen(s)) >= 0)
int
main(int argc, char **argv)
{
int fd;
struct sockaddr_un addr;
char buf[100];
int len;
if (argc != 2) {
fprintf(stderr, "Usage: %s socket\n", argv[0]);
return 1;
}
fd = socket(AF_UNIX, SOCK_STREAM, 0);
if (fd < 0) {
perror("socket");
return -1;
}
memset(&addr, 0, sizeof(addr));
addr.sun_family = AF_UNIX;
strncpy(addr.sun_path, argv[1], sizeof(addr.sun_path)-1);
if (connect(fd, (struct sockaddr*)&addr, sizeof(addr)) == -1) {
perror("connect");
return 1;
}
writestr("What is your name?\n");
len = read(fd, buf, sizeof(buf) - 1);
assert(len >= 0);
buf[len] = 0;
printf("Got name: %s\n", buf);
writestr("hello ");
writestr(buf);
/* test send/recv, too */
usleep(20000);
assert(send(fd, "send()", 6, 0) == 6);
len = recv(fd, buf, sizeof(buf) - 1, 0);
assert(len > 0);
buf[len] = 0;
printf("Got recv: %s\n", buf);
close(fd);
return 0;
}
|