LED-Cube/CubeControl/helper/unixSerial.c

/*
 *
 * unixSerial.c
 *
 * POSIX compatible serial port library
 * Uses 8 databits, no parity, 1 stop bit, no handshaking
 *
 * Copyright 2011 Thomas Buck <xythobuz@me.com>
 * Copyright 2011 Max Nuding <max.nuding@gmail.com>
 * Copyright 2011 Felix Bäder <baeder.felix@gmail.com>
 *
 * This file is part of LED-Cube.
 *
 * LED-Cube is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * LED-Cube 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 General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with LED-Cube.  If not, see <http://www.gnu.org/licenses/>.
 */

#include <stdio.h>
#include <string.h>
#include <unistd.h>
#include <fcntl.h>
#include <termios.h>
#include <dirent.h>

#define SEARCH "tty."

#define BAUD B19200

int fd = -1;

// Open the serial port
int serialOpen(char *port) {
	struct termios options;

	if ((strcmp(ttyname(fileno(stdin)), port) == 0) || (strcmp(ttyname(fileno(stdout)), port) == 0)) {
		return -1;
	}

	if (fd != -1) {
		close(fd);
	}
	fd = open(port, O_RDWR | O_NOCTTY | O_NDELAY);
	if (fd == -1) {
		return -1;
	}
	
	if (fcntl(fd, F_SETFL, FNDELAY) == -1) { // read() isn't blocking'
		close(fd);
		return -1;
	}
	if (tcgetattr(fd, &options) == -1) {
		close(fd);
		return -1;
	}
	if (cfsetispeed(&options, BAUD) == -1) { // Set speed
		close(fd);
		return -1;
	}
	if (cfsetospeed(&options, BAUD) == -1) {
		close(fd);
		return -1;
	}
	options.c_cflag |= (CLOCAL | CREAD);
	
	options.c_cflag &= ~PARENB; // 8N1
	options.c_cflag &= ~CSTOPB;
	options.c_cflag &= ~CSIZE;
	options.c_cflag |= CS8;
	
	options.c_lflag &= ~(ICANON | ECHO | ECHOE | ISIG); // Raw input
	options.c_oflag &= ~OPOST; // Raw output
	options.c_iflag &= ~(IXON | IXOFF | IXANY); // No flow control

	if (tcsetattr(fd, TCSANOW, &options) == -1) {
		close(fd);
		return -1;
	}

	return 0;
}

// Write to port. Returns number of characters sent, -1 on error
ssize_t serialWrite(char *data, size_t length) {
	return write(fd, data, length);
}

// Read from port. Return number of characters read, 0 if none available, -1 on error
ssize_t serialRead(char *data, size_t length) {
	return read(fd, data, length);
}

// Close the serial Port
void serialClose(void) {
	close(fd);
}

char** namesInDev(int *siz) {
	DIR *dir;
	struct dirent *ent;
	int i = 0, size = 0;
	char **files = NULL;
	dir = opendir("/dev/");
	while ((ent = readdir(dir)) != NULL) {
		size++;
	}
	files = (char **)malloc((size + 1) * sizeof(char *));
	files[size] = NULL;
	closedir(dir);
	dir = opendir("/dev/");
	while ((ent = readdir(dir)) != NULL) {
		files[i] = (char *)malloc((strlen(ent->d_name) + 1) * sizeof(char));
		files[i] = strcpy(files[i], ent->d_name);
		i++;
	}
	closedir(dir);

	char *tmp = NULL;
	// Fix every string, addin /dev/ in front of it...
	for (i = 0; i < size; i++) {
		tmp = (char *)malloc((strlen(files[i]) + 6) * sizeof(char));
		tmp[0] = '/';
		tmp[1] = 'd';
		tmp[2] = 'e';
		tmp[3] = 'v';
		tmp[4] = '/';
		files[i] = strncat(tmp, files[i], strlen(files[i]));
	}

	*siz = size;
	return files;
}

char** getSerialPorts(void) {
	int size;
	char** files = namesInDev(&size);
	char** fin = NULL;
	int i = 0, j = 0, f, g;

	fin = (char **)malloc((size + 1) * sizeof(char *));
	fin[size] = NULL;
	while (files[i] != NULL) {
		if (strstr(files[i], SEARCH) != NULL) {
			f = serialOpen(files[i]);
			if (fd != -1) {
				fin[j++] = files[i];
				serialClose();
			}
		}
		i++;
	}
	free(files);

	char** finish = (char **)malloc((j + 1) * sizeof(char *));
	for (i = 0; i < (j + 1); i++) {
		finish[i] = fin[i];
	}
	free(fin);

	return finish;
}