Initial commit into public repository0.0

Signed-off-by: Michael Krelin <hacker@klever.net>

6 files changed, 318 insertions, 0 deletions

diff --git a/lib/.gitignore b/lib/.gitignore
new file mode 100644
index 0000000..b7b1941
--- a/dev/null
+++ b/lib/.gitignore
@@ -0,0 +1,11 @@
+/.deps
+/.libs
+/st-decode.lo
+/st-decode.o
+/libnapkin.la
+/st-download.lo
+/st-download.o
+/hypnodata.lo
+/hypnodata.o
+/util.lo
+/util.o
diff --git a/lib/Makefile.am b/lib/Makefile.am
new file mode 100644
index 0000000..bf55849
--- a/dev/null
+++ b/lib/Makefile.am
@@ -0,0 +1,11 @@
+lib_LTLIBRARIES = libnapkin.la
+
+INCLUDES = -I${top_builddir}/include/ -I${top_srcdir}/include/ \
+           ${MODULES_CFLAGS}
+LIBS = ${MODULES_CFLAGS}
+
+libnapkin_la_SOURCES = \
+        st-decode.cc st-download.cc \
+        hypnodata.cc \
+        util.cc
+libnapkin_la_LDFLAGS = -version-info 0:0:0
diff --git a/lib/hypnodata.cc b/lib/hypnodata.cc
new file mode 100644
index 0000000..977fb76
--- a/dev/null
+++ b/lib/hypnodata.cc
@@ -0,0 +1,92 @@
+#include <napkin/exception.h>
+#include <napkin/util.h>
+#include <napkin/types.h>
+
+namespace napkin {
+
+    void hypnodata_t::clear() {
+        to_bed = alarm = 0;
+        data_a = window = 0;
+        almost_awakes.clear();
+    }
+
+    static time_t from_minute_w3c(const string& w3c) {
+        struct tm t; memset(&t,0,sizeof(t)); t.tm_isdst=-1;
+        if(sscanf(w3c.c_str(),"%04d-%02d-%02dT%02d:%02d",
+                &t.tm_year,&t.tm_mon,&t.tm_mday,&t.tm_hour,&t.tm_min)!=5)
+            throw exception("failed to parse w3c time");
+        --t.tm_mon;t.tm_year-=1900;
+        time_t rv = mktime(&t);
+        if(rv==(time_t)-1)
+            throw exception("failed to mktime()");
+        return rv;
+    }
+
+    void hypnodata_t::set_to_bed(const string& w3c) {
+        to_bed = from_minute_w3c(w3c); }
+    void hypnodata_t::set_alarm(const string& w3c) {
+        alarm = from_minute_w3c(w3c); }
+    void hypnodata_t::set_window(const string& str) {
+        window = strtol(str.c_str(),0,10); /* TODO: check for error */
+    }
+    void hypnodata_t::set_data_a(const string& str) {
+        data_a = strtol(str.c_str(),0,10); /* TODO: check for error */
+    }
+    void hypnodata_t::set_almost_awakes(const string& str) {
+        almost_awakes.clear();
+        static const char *significants = "0123456789-T:Z";
+        string::size_type p = str.find_first_of(significants);
+        struct tm t; memset(&t,0,sizeof(t)); t.tm_isdst=-1;
+        while(p!=string::npos) {
+            string::size_type ns = str.find_first_not_of(significants,p);
+            string w3c;
+            if(ns==string::npos) {
+                w3c = str.substr(p);
+                p = string::npos;
+            }else{
+                w3c = str.substr(p,ns-p);
+                p = str.find_first_of(significants,ns);
+            }
+            if(w3c.empty()) continue;
+            if(sscanf(w3c.c_str(),"%04d-%02d-%02dT%02d:%02d:%02d",
+                    &t.tm_year,&t.tm_mon,&t.tm_mday,&t.tm_hour,&t.tm_min,&t.tm_sec)!=6)
+                throw exception("failed to parse w3c time");
+            --t.tm_mon;t.tm_year-=1900;
+            time_t aa = mktime(&t);
+            if(aa==(time_t)-1)
+                throw exception("failed to mktime()");
+            almost_awakes.push_back(aa);
+        }
+    }
+
+    const string hypnodata_t::w3c_to_bed() const {
+        return strftime("%Y-%m-%dT%H:%M",to_bed); }
+    const string hypnodata_t::w3c_alarm() const {
+        return strftime("%Y-%m-%dT%H:%M",alarm); }
+    const string hypnodata_t::w3c_almostawakes() const {
+        string rv;
+        for(vector<time_t>::const_iterator i=almost_awakes.begin();i!=almost_awakes.end();++i) {
+            if(!rv.empty())
+                rv += ',';
+            rv += strftime("%Y-%m-%dT%H:%M:%S",*i);
+        }
+        return rv;
+    }
+
+    const string hypnodata_t::str_to_bed() const {
+        return strftime("%H:%M",to_bed); }
+    const string hypnodata_t::str_alarm() const {
+        return strftime("%H:%M",alarm); }
+    const string hypnodata_t::str_date() const {
+        return strftime("%Y-%m-%d, %a",alarm); }
+    const string hypnodata_t::str_data_a() const {
+        char tmp[16];
+        snprintf(tmp,sizeof(tmp),"%d:%02d:%02d",
+                data_a/3600, (data_a%3600)/60,
+                data_a % 60 );
+        return tmp; }
+
+    time_t hypnodata_t::aligned_start() const {
+        return alarm - (alarm % (24*60*60)) - 24*60*60; }
+
+}
diff --git a/lib/st-decode.cc b/lib/st-decode.cc
new file mode 100644
index 0000000..f8459ac
--- a/dev/null
+++ b/lib/st-decode.cc
@@ -0,0 +1,121 @@
+#include <stdexcept>
+#include <numeric>
+#include <napkin/exception.h>
+#include <napkin/st/decode.h>
+
+namespace napkin {
+    namespace sleeptracker {
+        using std::invalid_argument;
+        using std::runtime_error;
+
+        struct st_time_t {
+            uint8_t hour;
+            uint8_t min;
+        };
+        struct st_date_t {
+            uint8_t month;
+            uint8_t day;
+        };
+        struct st_fulltime_t {
+            uint8_t hour;
+            uint8_t min;
+            uint8_t sec;
+        };
+        struct st_data_header_t {
+            char magic;
+            st_date_t today;
+            uint8_t unknown;
+            uint8_t window;
+            st_time_t to_bed;
+            st_time_t alarm;
+            uint8_t nawakes;
+        };
+        struct st_data_footer_t {
+            uint16_t data_a;
+            uint8_t checksum;
+            uint8_t eof_mark;
+        };
+
+        static void back_a_day(struct tm& t) {
+            time_t ts = mktime(&t);
+            if(ts==(time_t)-1)
+                throw exception_st_data("failed to make up time to step back a day");
+            ts -= 60*60*24;
+            if(!localtime_r(&ts,&t))
+                throw exception_st_data("failed to localtime_r() while stepping back a day");
+        }
+
+        hypnodata_t& decode(hypnodata_t& rv,const void *data,size_t data_length) {
+            if(data_length < (sizeof(st_data_header_t)+sizeof(st_data_footer_t)))
+                throw exception_st_data_envelope("not enough sleeptracker data to decode");
+            st_data_header_t *h = (st_data_header_t*)data;
+            if(h->magic != 'V')
+                throw exception_st_data_envelope("invalid magic in the data");
+            st_data_footer_t *f = (st_data_footer_t*)(static_cast<const char *>(data)+data_length-sizeof(st_data_footer_t));
+            if( (std::accumulate((uint8_t*)&h->today,(uint8_t*)&f->checksum,0)&0xFF) != f->checksum )
+                throw exception_st_data_integrity("checksum mismatch");
+            st_fulltime_t *aawake = (st_fulltime_t*)&h[1];
+            if((void*)&aawake[h->nawakes] != (void*)f)
+                throw exception_st_data_envelope("unbelievably screwed up data");
+            rv.clear();
+            time_t now = time(0);
+            struct tm t;
+            if(!localtime_r(&now,&t))
+                throw exception_st_data("failed to localtime_r()");
+            t.tm_mon = h->today.month-1;
+            t.tm_mday = h->today.day;
+            time_t mkt = mktime(&t);
+            if(mkt == (time_t)-1)
+                throw exception_st_data("failed to mktime() for a timestamp");
+            if(mkt > now) {
+                --t.tm_year;
+            }
+            struct tm ta;
+            memmove(&ta,&t,sizeof(ta));
+            ta.tm_sec = 0;
+            ta.tm_hour = h->alarm.hour; ta.tm_min = h->alarm.min;
+            rv.alarm = mktime(&ta);
+            if(rv.alarm == (time_t)-1)
+                throw exception_st_data("failed to mktime() for alarm");
+            struct tm tb;
+            memmove(&tb,&ta,sizeof(tb));
+            tb.tm_hour = h->to_bed.hour; tb.tm_min = h->to_bed.min;
+            rv.to_bed = mktime(&tb);
+            if(rv.to_bed == (time_t)-1)
+                throw exception_st_data("failed to mktime() for 'to bed'");
+            if(rv.to_bed > rv.alarm) {
+                back_a_day(tb);
+                rv.to_bed -= 24*60*60;
+            }
+            struct tm taaw;
+            memmove(&taaw,&tb,sizeof(taaw));
+            for(int rest=h->nawakes;rest;--rest,++aawake) {
+                if(
+                        taaw.tm_mday!=ta.tm_mday
+                        && (
+                            aawake->hour < tb.tm_hour
+                            || (
+                                aawake->hour==tb.tm_hour
+                                && aawake->min < tb.tm_min )
+                           ) )
+                    memmove(&taaw,&ta,sizeof(taaw));
+                taaw.tm_hour = aawake->hour;
+                taaw.tm_min = aawake->min;
+                taaw.tm_sec = aawake->sec;
+                rv.almost_awakes.push_back( mktime(&taaw) );
+                if(rv.almost_awakes.back() == (time_t)-1)
+                    throw exception_st_data("failed to mktime() for almost awake moment");
+            }
+            rv.window = h->window;
+            rv.data_a = f->data_a;
+            return rv;
+        }
+
+        hypnodata_ptr_t decode(const void *data,size_t data_length) {
+            hypnodata_ptr_t rv( new hypnodata_t );
+            decode(*rv,data,data_length);
+            return rv;
+        }
+
+    }
+}
diff --git a/lib/st-download.cc b/lib/st-download.cc
new file mode 100644
index 0000000..b56e52d
--- a/dev/null
+++ b/lib/st-download.cc
@@ -0,0 +1,70 @@
+#include <sys/types.h>
+#include <sys/stat.h>
+#include <fcntl.h>
+#include <unistd.h>
+#include <termios.h>
+#include <stdexcept>
+#include <napkin/exception.h>
+#include <napkin/st/download.h>
+#include <napkin/st/decode.h>
+
+namespace napkin {
+    namespace sleeptracker {
+        using std::runtime_error;
+
+                intdownload_initiate(const char *port) {
+            int fd = open(port?port:"/dev/sleeptracker",
+                    O_RDWR|O_NOCTTY|O_NONBLOCK);
+            if(fd<0)
+                throw exception_st_port("failed to open() sleeptracker port");
+
+            if(tcflush(fd,TCIOFLUSH)) {
+                close(fd);
+                throw exception_st_port("failed to tcflush()");
+            }
+            struct termios ts;
+            ts.c_cflag = CS8|CREAD;
+            cfsetispeed(&ts,B2400); cfsetospeed(&ts,B2400);
+            ts.c_iflag = IGNPAR;
+            ts.c_oflag = ts.c_lflag = 0;
+            ts.c_cc[VMIN]=1; ts.c_cc[VTIME]=0;
+            if(tcsetattr(fd,TCSANOW,&ts)) {
+                close(fd);
+                throw exception_st_port("failed to tcsetattr()");
+            }
+
+            if(write(fd,"V",1)!=1) {
+                close(fd);
+                throw exception_st_port("failed to write() to sleeptracker");
+            }
+            return fd;
+        }
+        size_t download_finish(int fd,void *buffer,size_t buffer_size) {
+            size_t rv = read(fd,buffer,buffer_size);
+            close(fd);
+
+            if(rv==(size_t)-1)
+                throw exception_st_port("failed to read() from sleeptracker");
+            return rv;
+        }
+
+        size_t download(
+                void *buffer,size_t buffer_size,
+                const char *port) {
+            int fd = download_initiate(port);
+            /* this is not the best way to wait for data, but
+             * after all it's a sleeptracker! */
+            sleep(1);
+            return download_finish(fd,buffer,buffer_size);
+        }
+
+        hypnodata_ptr_t download(const char *port) {
+            char buffer[2048];
+            size_t rb = download(buffer,sizeof(buffer),port);
+            hypnodata_ptr_t rv( new hypnodata_t );
+            decode(*rv,buffer,rb);
+            return rv;
+        }
+
+    }
+}
diff --git a/lib/util.cc b/lib/util.cc
new file mode 100644
index 0000000..350cbac
--- a/dev/null
+++ b/lib/util.cc
@@ -0,0 +1,13 @@
+#include <napkin/util.h>
+
+namespace napkin {
+
+    string strftime(const char *fmt,time_t t) {
+        struct tm tt;
+        localtime_r(&t,&tt);// TODO: check res
+        char rv[1024];
+        strftime(rv,sizeof(rv),fmt,&tt); // TODO: check res
+        return rv;
+    }
+
+}