summaryrefslogtreecommitdiffstats
path: root/tinyusb/src/tusb.c
blob: 6494287d2dea8ad414a1495888ed6299c48c01f2 (plain)
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
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
/*
 * The MIT License (MIT)
 *
 * Copyright (c) 2019 Ha Thach (tinyusb.org)
 *
 * Permission is hereby granted, free of charge, to any person obtaining a copy
 * of this software and associated documentation files (the "Software"), to deal
 * in the Software without restriction, including without limitation the rights
 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
 * copies of the Software, and to permit persons to whom the Software is
 * furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included in
 * all copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
 * THE SOFTWARE.
 *
 * This file is part of the TinyUSB stack.
 */

#include "tusb_option.h"

#if TUSB_OPT_HOST_ENABLED || TUSB_OPT_DEVICE_ENABLED

#include "tusb.h"

// TODO clean up
#if TUSB_OPT_DEVICE_ENABLED
#include "device/usbd_pvt.h"
#endif

bool tusb_init(void)
{
#if TUSB_OPT_DEVICE_ENABLED
  TU_ASSERT ( tud_init(TUD_OPT_RHPORT) ); // init device stack
#endif

#if TUSB_OPT_HOST_ENABLED
  TU_ASSERT( tuh_init(TUH_OPT_RHPORT) ); // init host stack
#endif

  return true;
}

bool tusb_inited(void)
{
  bool ret = false;

#if TUSB_OPT_DEVICE_ENABLED
  ret = ret || tud_inited();
#endif

#if TUSB_OPT_HOST_ENABLED
  ret = ret || tuh_inited();
#endif

  return ret;
}

//--------------------------------------------------------------------+
// Internal Helper for both Host and Device stack
//--------------------------------------------------------------------+

bool tu_edpt_validate(tusb_desc_endpoint_t const * desc_ep, tusb_speed_t speed)
{
  uint16_t const max_packet_size = tu_le16toh(desc_ep->wMaxPacketSize.size);
  TU_LOG2("  Open EP %02X with Size = %u\r\n", desc_ep->bEndpointAddress, max_packet_size);

  switch (desc_ep->bmAttributes.xfer)
  {
    case TUSB_XFER_ISOCHRONOUS:
    {
      uint16_t const spec_size = (speed == TUSB_SPEED_HIGH ? 1024 : 1023);
      TU_ASSERT(max_packet_size <= spec_size);
    }
    break;

    case TUSB_XFER_BULK:
      if (speed == TUSB_SPEED_HIGH)
      {
        // Bulk highspeed must be EXACTLY 512
        TU_ASSERT(max_packet_size == 512);
      }else
      {
        // TODO Bulk fullspeed can only be 8, 16, 32, 64
        TU_ASSERT(max_packet_size <= 64);
      }
    break;

    case TUSB_XFER_INTERRUPT:
    {
      uint16_t const spec_size = (speed == TUSB_SPEED_HIGH ? 1024 : 64);
      TU_ASSERT(max_packet_size <= spec_size);
    }
    break;

    default: return false;
  }

  return true;
}

void tu_edpt_bind_driver(uint8_t ep2drv[][2], tusb_desc_interface_t const* desc_itf, uint16_t desc_len, uint8_t driver_id)
{
  uint8_t const* p_desc = (uint8_t const*) desc_itf;
  uint16_t len = 0;

  while( len < desc_len )
  {
    if ( TUSB_DESC_ENDPOINT == tu_desc_type(p_desc) )
    {
      uint8_t const ep_addr = ((tusb_desc_endpoint_t const*) p_desc)->bEndpointAddress;

      TU_LOG(2, "  Bind EP %02x to driver id %u\r\n", ep_addr, driver_id);
      ep2drv[tu_edpt_number(ep_addr)][tu_edpt_dir(ep_addr)] = driver_id;
    }

    len    = (uint16_t)(len + tu_desc_len(p_desc));
    p_desc = tu_desc_next(p_desc);
  }
}

uint16_t tu_desc_get_interface_total_len(tusb_desc_interface_t const* desc_itf, uint8_t itf_count, uint16_t max_len)
{
  uint8_t const* p_desc = (uint8_t const*) desc_itf;
  uint16_t len = 0;

  while (itf_count--)
  {
    // Next on interface desc
    len += tu_desc_len(desc_itf);
    p_desc = tu_desc_next(p_desc);

    while (len < max_len)
    {
      // return on IAD regardless of itf count
      if ( tu_desc_type(p_desc) == TUSB_DESC_INTERFACE_ASSOCIATION ) return len;

      if ( (tu_desc_type(p_desc) == TUSB_DESC_INTERFACE) &&
           ((tusb_desc_interface_t const*) p_desc)->bAlternateSetting == 0 )
      {
        break;
      }

      len += tu_desc_len(p_desc);
      p_desc = tu_desc_next(p_desc);
    }
  }

  return len;
}

/*------------------------------------------------------------------*/
/* Debug
 *------------------------------------------------------------------*/
#if CFG_TUSB_DEBUG
#include <ctype.h>

char const* const tusb_strerr[TUSB_ERROR_COUNT] = { ERROR_TABLE(ERROR_STRING) };

static void dump_str_line(uint8_t const* buf, uint16_t count)
{
  tu_printf("  |");

  // each line is 16 bytes
  for(uint16_t i=0; i<count; i++)
  {
    const char ch = buf[i];
    tu_printf("%c", isprint(ch) ? ch : '.');
  }

  tu_printf("|\r\n");
}

/* Print out memory contents
 *  - buf   : buffer
 *  - count : number of item
 *  - indent: prefix spaces on every line
 */
void tu_print_mem(void const *buf, uint32_t count, uint8_t indent)
{
  uint8_t const size = 1; // fixed 1 byte for now

  if ( !buf || !count )
  {
    tu_printf("NULL\r\n");
    return;
  }

  uint8_t const *buf8 = (uint8_t const *) buf;

  char format[] = "%00X";
  format[2] += 2*size;

  const uint8_t item_per_line  = 16 / size;

  for(unsigned int i=0; i<count; i++)
  {
    unsigned int value=0;

    if ( i%item_per_line == 0 )
    {
      // Print Ascii
      if ( i != 0 )
      {
        dump_str_line(buf8-16, 16);
      }

      for(uint8_t s=0; s < indent; s++) tu_printf(" ");

      // print offset or absolute address
      tu_printf("%04X: ", 16*i/item_per_line);
    }

    memcpy(&value, buf8, size);
    buf8 += size;

    tu_printf(" ");
    tu_printf(format, value);
  }

  // fill up last row to 16 for printing ascii
  const uint32_t remain = count%16;
  uint8_t nback = (remain ? remain : 16);

  if ( remain )
  {
    for(uint32_t i=0; i< 16-remain; i++)
    {
      tu_printf(" ");
      for(int j=0; j<2*size; j++) tu_printf(" ");
    }
  }

  dump_str_line(buf8-nback, nback);
}

#endif

#endif // host or device enabled
l { color: #000000; background-color: #ffffc0; padding: 0 5px 0 5px; } .highlight .hll { background-color: #ffffcc } .highlight { background: #ffffff; } .highlight .c { color: #888888 } /* Comment */ .highlight .err { color: #a61717; background-color: #e3d2d2 } /* Error */ .highlight .k { color: #008800; font-weight: bold } /* Keyword */ .highlight .ch { color: #888888 } /* Comment.Hashbang */ .highlight .cm { color: #888888 } /* Comment.Multiline */ .highlight .cp { color: #cc0000; font-weight: bold } /* Comment.Preproc */ .highlight .cpf { color: #888888 } /* Comment.PreprocFile */ .highlight .c1 { color: #888888 } /* Comment.Single */ .highlight .cs { color: #cc0000; font-weight: bold; background-color: #fff0f0 } /* Comment.Special */ .highlight .gd { color: #000000; background-color: #ffdddd } /* Generic.Deleted */ .highlight .ge { font-style: italic } /* Generic.Emph */ .highlight .gr { color: #aa0000 } /* Generic.Error */ .highlight .gh { color: #333333 } /* Generic.Heading */ .highlight .gi { color: #000000; background-color: #ddffdd } /* Generic.Inserted */ .highlight .go { color: #888888 } /* Generic.Output */ .highlight .gp { color: #555555 } /* Generic.Prompt */ .highlight .gs { font-weight: bold } /* Generic.Strong */ .highlight .gu { color: #666666 } /* Generic.Subheading */ .highlight .gt { color: #aa0000 } /* Generic.Traceback */ .highlight .kc { color: #008800; font-weight: bold } /* Keyword.Constant */ .highlight .kd { color: #008800; font-weight: bold } /* Keyword.Declaration */ .highlight .kn { color: #008800; font-weight: bold } /* Keyword.Namespace */ .highlight .kp { color: #008800 } /* Keyword.Pseudo */ .highlight .kr { color: #008800; font-weight: bold } /* Keyword.Reserved */ .highlight .kt { color: #888888; font-weight: bold } /* Keyword.Type */ .highlight .m { color: #0000DD; font-weight: bold } /* Literal.Number */ .highlight .s { color: #dd2200; background-color: #fff0f0 } /* Literal.String */ .highlight .na { color: #336699 } /* Name.Attribute */ .highlight .nb { color: #003388 } /* Name.Builtin */ .highlight .nc { color: #bb0066; font-weight: bold } /* Name.Class */ .highlight .no { color: #003366; font-weight: bold } /* Name.Constant */ .highlight .nd { color: #555555 } /* Name.Decorator */ .highlight .ne { color: #bb0066; font-weight: bold } /* Name.Exception */ .highlight .nf { color: #0066bb; font-weight: bold } /* Name.Function */ .highlight .nl { color: #336699; font-style: italic } /* Name.Label */ .highlight .nn { color: #bb0066; font-weight: bold } /* Name.Namespace */ .highlight .py { color: #336699; font-weight: bold } /* Name.Property */ .highlight .nt { color: #bb0066; font-weight: bold } /* Name.Tag */ .highlight .nv { color: #336699 } /* Name.Variable */ .highlight .ow { color: #008800 } /* Operator.Word */ .highlight .w { color: #bbbbbb } /* Text.Whitespace */ .highlight .mb { color: #0000DD; font-weight: bold } /* Literal.Number.Bin */ .highlight .mf { color: #0000DD; font-weight: bold } /* Literal.Number.Float */ .highlight .mh { color: #0000DD; font-weight: bold } /* Literal.Number.Hex */ .highlight .mi { color: #0000DD; font-weight: bold } /* Literal.Number.Integer */ .highlight .mo { color: #0000DD; font-weight: bold } /* Literal.Number.Oct */ .highlight .sa { color: #dd2200; background-color: #fff0f0 } /* Literal.String.Affix */ .highlight .sb { color: #dd2200; background-color: #fff0f0 } /* Literal.String.Backtick */ .highlight .sc { color: #dd2200; background-color: #fff0f0 } /* Literal.String.Char */ .highlight .dl { color: #dd2200; background-color: #fff0f0 } /* Literal.String.Delimiter */ .highlight .sd { color: #dd2200; background-color: #fff0f0 } /* Literal.String.Doc */ .highlight .s2 { color: #dd2200; background-color: #fff0f0 } /* Literal.String.Double */ .highlight .se { color: #0044dd; background-color: #fff0f0 } /* Literal.String.Escape */ .highlight .sh { color: #dd2200; background-color: #fff0f0 } /* Literal.String.Heredoc */ .highlight .si { color: #3333bb; background-color: #fff0f0 } /* Literal.String.Interpol */ .highlight .sx { color: #22bb22; background-color: #f0fff0 } /* Literal.String.Other */ .highlight .sr { color: #008800; background-color: #fff0ff } /* Literal.String.Regex */ .highlight .s1 { color: #dd2200; background-color: #fff0f0 } /* Literal.String.Single */ .highlight .ss { color: #aa6600; background-color: #fff0f0 } /* Literal.String.Symbol */ .highlight .bp { color: #003388 } /* Name.Builtin.Pseudo */ .highlight .fm { color: #0066bb; font-weight: bold } /* Name.Function.Magic */ .highlight .vc { color: #336699 } /* Name.Variable.Class */ .highlight .vg { color: #dd7700 } /* Name.Variable.Global */ .highlight .vi { color: #3333bb } /* Name.Variable.Instance */ .highlight .vm { color: #336699 } /* Name.Variable.Magic */ .highlight .il { color: #0000DD; font-weight: bold } /* Literal.Number.Integer.Long */
%module pyabc

// -------------------------------------------------------------------
// SWIG typemap allowing us to grab a Python callable object
// -------------------------------------------------------------------

#ifdef SWIG<Python>

%typemap(in) PyObject *PyFunc 
{
  
  if ( !PyCallable_Check($source) ) 
  {
      PyErr_SetString(PyExc_TypeError, "Need a callable object!");
      return NULL;
  }
  
  $target = $source;
}

#endif /* #ifdef SWIG<Python> */

%{
    
#include <base/main/main.h>
#include <misc/util/utilCex.h>

#include <stdlib.h>
#include <signal.h>

#include <sys/prctl.h>
#include <sys/types.h>
#include <sys/time.h>
#include <sys/resource.h>
#include <sys/wait.h>
#include <sys/stat.h>
#include <fcntl.h>
    
int n_ands()
{
    Abc_Frame_t* pAbc = Abc_FrameGetGlobalFrame();
    Abc_Ntk_t * pNtk = Abc_FrameReadNtk(pAbc);

    if ( pNtk && Abc_NtkIsStrash(pNtk) )
    {        
        return Abc_NtkNodeNum(pNtk);
    }

    return -1;
}
    
int n_nodes()
{
    Abc_Frame_t* pAbc = Abc_FrameGetGlobalFrame();
    Abc_Ntk_t * pNtk = Abc_FrameReadNtk(pAbc);

    if ( pNtk )
    {        
        return Abc_NtkNodeNum(pNtk);
    }

    return -1;
}

int n_pis()
{
    Abc_Frame_t* pAbc = Abc_FrameGetGlobalFrame();
    Abc_Ntk_t * pNtk = Abc_FrameReadNtk(pAbc);

    if ( pNtk )
    {        
        return Abc_NtkPiNum(pNtk);
    }

    return -1;
}


int n_pos()
{
    Abc_Frame_t* pAbc = Abc_FrameGetGlobalFrame();
    Abc_Ntk_t * pNtk = Abc_FrameReadNtk(pAbc);

    if ( pNtk )
    {        
        return Abc_NtkPoNum(pNtk);
    }

    return -1;
}

int n_latches()
{
    Abc_Frame_t* pAbc = Abc_FrameGetGlobalFrame();
    Abc_Ntk_t * pNtk = Abc_FrameReadNtk(pAbc);

    if ( pNtk )
    {        
        return Abc_NtkLatchNum(pNtk);
    }

    return -1;
}

int n_levels()
{
    Abc_Frame_t* pAbc = Abc_FrameGetGlobalFrame();
    Abc_Ntk_t * pNtk = Abc_FrameReadNtk(pAbc);

    if ( pNtk )
    {        
        return Abc_NtkLevel(pNtk);
    }

    return -1;
}

int has_comb_model()
{
    Abc_Frame_t* pAbc = Abc_FrameGetGlobalFrame();
    Abc_Ntk_t * pNtk = Abc_FrameReadNtk(pAbc);

    return pNtk && pNtk->pModel;
}

int has_seq_model()
{
    Abc_Frame_t* pAbc = Abc_FrameGetGlobalFrame();
    Abc_Ntk_t * pNtk = Abc_FrameReadNtk(pAbc);

    return pNtk && pNtk->pSeqModel;
}

int n_bmc_frames()
{
    Abc_Frame_t* pAbc = Abc_FrameGetGlobalFrame();
    return Abc_FrameReadBmcFrames(pAbc);
}

int prob_status()
{
    Abc_Frame_t* pAbc = Abc_FrameGetGlobalFrame();
    return Abc_FrameReadProbStatus(pAbc);
}

int is_valid_cex()
{
    Abc_Frame_t* pAbc = Abc_FrameGetGlobalFrame();
    Abc_Ntk_t * pNtk = Abc_FrameReadNtk(pAbc);

    return pNtk && Abc_FrameReadCex(pAbc) && Abc_NtkIsValidCex( pNtk, Abc_FrameReadCex(pAbc) );
}

int is_true_cex()
{
    Abc_Frame_t* pAbc = Abc_FrameGetGlobalFrame();
    Abc_Ntk_t * pNtk = Abc_FrameReadNtk(pAbc);

    return pNtk && Abc_FrameReadCex(pAbc) && Abc_NtkIsTrueCex( pNtk, Abc_FrameReadCex(pAbc) );
}

int n_cex_pis()
{
    Abc_Frame_t* pAbc = Abc_FrameGetGlobalFrame();

    return Abc_FrameReadCex(pAbc) ? Abc_FrameReadCexPiNum( pAbc ) : -1;
}

int n_cex_regs()
{
    Abc_Frame_t* pAbc = Abc_FrameGetGlobalFrame();

    return Abc_FrameReadCex(pAbc) ? Abc_FrameReadCexRegNum( pAbc ) : -1;
}

int cex_po()
{
    Abc_Frame_t* pAbc = Abc_FrameGetGlobalFrame();

    return Abc_FrameReadCex(pAbc) ? Abc_FrameReadCexPo( pAbc ) : -1;
}

int cex_frame()
{
    Abc_Frame_t* pAbc = Abc_FrameGetGlobalFrame();

    return Abc_FrameReadCex(pAbc) ? Abc_FrameReadCexFrame( pAbc ) : -1;
}

int n_phases()
{
    Abc_Frame_t* pAbc = Abc_FrameGetGlobalFrame();
    Abc_Ntk_t * pNtk = Abc_FrameReadNtk(pAbc);

    return pNtk ? Abc_NtkPhaseFrameNum(pNtk) : 1;
}

Abc_Cex_t* _cex_get()
{
    Abc_Frame_t* pAbc = Abc_FrameGetGlobalFrame();
    Abc_Cex_t* pCex = Abc_FrameReadCex(pAbc);
    
    if ( ! pCex )
    {
        return NULL;
    }

    return Abc_CexDup( pCex, -1 ); 
}

int _cex_get_vec_len()
{
    Abc_Frame_t* pAbc = Abc_FrameGetGlobalFrame();
	Vec_Ptr_t* vCexVec = Abc_FrameReadCexVec(pAbc);
	
	if( ! vCexVec )
	{
		return 0;
	}
	
	return Vec_PtrSize(vCexVec);
}

Abc_Cex_t* _cex_get_vec(int i)
{
    Abc_Frame_t* pAbc = Abc_FrameGetGlobalFrame();
	Vec_Ptr_t* vCexVec = Abc_FrameReadCexVec(pAbc);
    
	if( ! vCexVec )
	{
		return NULL;
	}

	Abc_Cex_t* pCex = Vec_PtrEntry( vCexVec, i );

	if ( ! pCex )
	{
		return NULL;
	}

    return Abc_CexDup( pCex, -1 ); 
}

void _cex_put(Abc_Cex_t* pCex)
{
    if ( pCex )
    {
        pCex = Abc_CexDup(pCex, -1);
    }
    
    Abc_FrameSetCex( Abc_CexDup(pCex, -1) );
}

void _cex_free(Abc_Cex_t* pCex)
{
    Abc_CexFree(pCex);
}

int _cex_n_regs(Abc_Cex_t* pCex)
{
    return pCex->nRegs;
}

int _cex_n_pis(Abc_Cex_t* pCex)
{
    return pCex->nPis;
}

int _cex_get_po(Abc_Cex_t* pCex)
{
    return pCex->iPo;
}

int _cex_get_frame(Abc_Cex_t* pCex)
{
    return pCex->iFrame;
}

static PyObject* VecInt_To_PyList(Vec_Int_t* v)
{
    PyObject* pylist = PyList_New( Vec_IntSize(v) );
    
    int elem, i;
    
    Vec_IntForEachEntry( v, elem, i)
    {
        PyList_SetItem( pylist, i, PyInt_FromLong(elem) );
    }
    
    return pylist;
}

PyObject* iso_eq_classes()
{
    Abc_Frame_t* pAbc = Abc_FrameGetGlobalFrame();
    Vec_Ptr_t *vPoEquivs = Abc_FrameReadPoEquivs(pAbc);
    
    PyObject* eq_classes;
    Vec_Int_t* pEntry;
    int i;
    
    if( ! vPoEquivs )
    {
        Py_RETURN_NONE;
    }
    
    eq_classes = PyList_New( Vec_PtrSize(vPoEquivs) );
    
    Vec_PtrForEachEntry( Vec_Int_t*, vPoEquivs, pEntry, i )
    {
        PyList_SetItem( eq_classes, i, VecInt_To_PyList(pEntry) );
    }

    return eq_classes;    
}

static PyObject* pyabc_internal_python_command_callback = 0;

void pyabc_internal_set_command_callback( PyObject* callback )
{
    Py_XINCREF(callback);
    Py_XDECREF(pyabc_internal_python_command_callback);

    pyabc_internal_python_command_callback = callback;
}

static int pyabc_internal_abc_command_callback(Abc_Frame_t * pAbc, int argc, char ** argv)
{
    int i;
    
    PyObject* args;
    PyObject* arglist;
    PyObject* res;
    
    PyGILState_STATE gstate;
    
    long lres;
    
    if ( !pyabc_internal_python_command_callback )
            return 0;
            
    gstate = PyGILState_Ensure();
            
    args = PyList_New(argc);
    
    for( i=0 ; i<argc ; i++ )
            PyList_SetItem(args, i, PyString_FromString(argv[i]) );

    arglist = Py_BuildValue("(O)", args);
    Py_INCREF(arglist);

    res = PyEval_CallObject( pyabc_internal_python_command_callback, arglist );
    Py_DECREF(arglist);

    if ( !res )
    {
        PyGILState_Release(gstate);
        return -1;
    }
    
    lres = PyInt_AsLong(res);
    Py_DECREF(res);

    PyGILState_Release(gstate);
    
    return lres;
}

int run_command(char* cmd)
{
    Abc_Frame_t* pAbc = Abc_FrameGetGlobalFrame();
    int rc;
    
    Py_BEGIN_ALLOW_THREADS
    
    rc = Cmd_CommandExecute(pAbc, cmd);
    
    Py_END_ALLOW_THREADS
    
    return rc;
}

void pyabc_internal_register_command( char * sGroup, char * sName, int fChanges )
{
    Abc_Frame_t* pAbc = Abc_FrameGetGlobalFrame();

    Cmd_CommandAdd( pAbc, sGroup, sName, (void*)pyabc_internal_abc_command_callback, fChanges);
}

static int sigchld_pipe_fd = -1;
    
static void sigchld_handler(int signum)
{
    while( write(sigchld_pipe_fd, "", 1) == -1 && errno==EINTR )
        ;
}

static void install_sigchld_handler(int sigchld_fd)
{
    sigchld_pipe_fd = sigchld_fd;
    signal(SIGCHLD, sigchld_handler);
}

static int sigint_pipe_fd = -1;

static void sigint_handler(int signum)
{
    unsigned char tmp = (unsigned char)signum;
    while( write(sigint_pipe_fd, &tmp, 1) == -1 && errno==EINTR )
        ;
}

static void install_sigint_handler(int sigint_fd)
{
    sigint_pipe_fd = sigint_fd;
    
    signal(SIGINT, sigint_handler);

    // try to catch other signals that ask the process to terminate    
    signal(SIGABRT, sigint_handler);
    signal(SIGQUIT, sigint_handler);
    signal(SIGTERM, sigint_handler);

    // try to ensure cleanup on exceptional conditions
    signal(SIGBUS, sigint_handler);
    signal(SIGILL, sigint_handler);
    signal(SIGSEGV, sigint_handler);
    
    // try to ensure cleanup before being killed due to resource limit
    signal(SIGXCPU, sigint_handler);
    signal(SIGXFSZ, sigint_handler);
}

sigset_t old_procmask;
static int nblocks = 0;

void block_sigint()
{
    sigset_t procmask;

    assert(nblocks==0);
    nblocks ++ ;
    
    sigemptyset(&procmask);
    sigaddset(&procmask, SIGINT);
    
    sigprocmask(SIG_BLOCK, &procmask, &old_procmask);
}

void unblock_sigint()
{
    assert( nblocks==1);
    nblocks--;
    
    sigprocmask(SIG_SETMASK, &old_procmask, NULL);
}

static PyObject* pyabc_internal_system_callback = 0;
static PyObject* pyabc_internal_tmpfile_callback = 0;
static PyObject* pyabc_internal_tmpfile_remove_callback = 0;

int Util_SignalSystem(const char* cmd)
{
    PyObject* arglist;
    PyObject* res;
    
    PyGILState_STATE gstate;

    long lres;
    
    if ( !pyabc_internal_system_callback )
            return -1;
            
    gstate = PyGILState_Ensure();
            
    arglist = Py_BuildValue("(O)", PyString_FromString(cmd));
    Py_INCREF(arglist);

    res = PyEval_CallObject( pyabc_internal_system_callback, arglist );
    Py_DECREF(arglist);

    if ( !res )
    {
        PyGILState_Release(gstate);
        return -1;
    }
    
    lres = PyInt_AsLong(res);
    Py_DECREF(res);

    PyGILState_Release(gstate);
    
    return lres;
}

int Util_SignalTmpFile(const char* prefix, const char* suffix, char** out_name)
{
    char* str;
    Py_ssize_t size;
    
    PyObject* arglist;
    PyObject* res;
    
    PyGILState_STATE gstate;
    
    *out_name = NULL;

    if ( !pyabc_internal_tmpfile_callback )
            return 0;
            
    gstate = PyGILState_Ensure();
            
    arglist = Py_BuildValue("(ss)", prefix, suffix);
    Py_INCREF(arglist);

    res = PyEval_CallObject( pyabc_internal_tmpfile_callback, arglist );
    Py_DECREF(arglist);

    if ( !res )
    {
        PyGILState_Release(gstate);
        return -1;
    }
    
    PyString_AsStringAndSize(res, &str, &size);
    
    *out_name = ABC_ALLOC(char, size+1);
    strcpy(*out_name, str);
    
    Py_DECREF(res);

    PyGILState_Release(gstate);
    
    return open(*out_name, O_WRONLY);
}

void Util_SignalTmpFileRemove(const char* fname, int fLeave)
{
    PyObject* arglist;
    PyObject* res;
    
    PyGILState_STATE gstate;

    if ( !pyabc_internal_tmpfile_remove_callback )
            return;
            
    gstate = PyGILState_Ensure();
            
    arglist = Py_BuildValue("(si)", fname, fLeave);
    Py_INCREF(arglist);

    res = PyEval_CallObject( pyabc_internal_tmpfile_remove_callback, arglist );
    Py_DECREF(arglist);
    Py_XDECREF(res);

    PyGILState_Release(gstate);
}

void pyabc_internal_set_util_callbacks( PyObject* system_callback, PyObject* tmpfile_callback, PyObject* tmpfile_remove_callback )
{
    Py_XINCREF(system_callback);
    Py_XDECREF(pyabc_internal_system_callback);

    pyabc_internal_system_callback = system_callback;

    Py_XINCREF(tmpfile_callback);
    Py_XDECREF(pyabc_internal_tmpfile_callback);

    pyabc_internal_tmpfile_callback = tmpfile_callback;

    Py_XINCREF(tmpfile_remove_callback);
    Py_XDECREF(pyabc_internal_tmpfile_remove_callback);

    pyabc_internal_tmpfile_remove_callback = tmpfile_remove_callback;
}

PyObject* _wait_no_hang()
{
    int status;
    int pid;
    
    pid = wait3(&status, WNOHANG, NULL);
    
    return Py_BuildValue("(iii)", pid, status, errno);
}

int _posix_kill(int pid, int signum)
{
    return kill(pid, signum);
}

void _set_death_signal()
{
    prctl(PR_SET_PDEATHSIG, SIGINT);
}

%}

%init 
%{
    Abc_Start();
%}

int n_ands();
int n_nodes();
int n_pis();
int n_pos();
int n_latches();
int n_levels();

int run_command(char* cmd);

int has_comb_model();
int has_seq_model();

int  n_bmc_frames();
int  prob_status();

int is_valid_cex();
int is_true_cex();
int  n_cex_pis();
int  n_cex_regs();
int  cex_po();
int  cex_frame();

int  n_phases();

Abc_Cex_t* _cex_get();
int _cex_get_vec_len();
Abc_Cex_t* _cex_get_vec(int i);
void _cex_put(Abc_Cex_t* pCex);
void _cex_free(Abc_Cex_t* pCex);
int _cex_n_regs(Abc_Cex_t* pCex);
int _cex_n_pis(Abc_Cex_t* pCex);
int _cex_get_po(Abc_Cex_t* pCex);
int _cex_get_frame(Abc_Cex_t* pCex);

PyObject* iso_eq_classes();

void pyabc_internal_set_command_callback( PyObject* callback );
void pyabc_internal_register_command( char * sGroup, char * sName, int fChanges );

void install_sigchld_handler(int sigint_fd);
void install_sigint_handler(int sigint_fd);

void block_sigint();
void unblock_sigint();

void pyabc_internal_set_util_callbacks( PyObject* system_callback, PyObject* tmpfile_callback, PyObject* tmpfile_remove_callback );

PyObject* _wait_no_hang();

void _set_death_signal();

int _posix_kill(int pid, int signum);
void _set_death_signal();

%pythoncode 
%{

class _Cex(object):

    def __init__(self, pCex):
        self.pCex = pCex
        
    def __del__(self):
        _cex_free(self.pCex)

    def n_regs(self):
        return _cex_n_regs(self.pCex)
        
    def n_pis(self):
        return _cex_n_pis(self.pCex)
    
    def get_po(self):
        return _cex_get_po(self.pCex)

    def get_frame(self):
        return _cex_get_frame(self.pCex)
       
def cex_get_vector():
    
    res = []
    
    for i in xrange(_cex_get_vec_len()):
        cex = _cex_get_vec(i)

        if cex is None:
            res.append(None)
        else:
            res.append(_Cex(cex))
    
    return res
       
def cex_get():

    cex = _cex_get()
    
    if cex is None:
        return None
        
    return _Cex(cex)

def cex_put(cex):

    assert cex is not None
    assert cex.pCex is not None
    
    return _cex_put(cex.pCex)
        
import threading
import select
import signal
import tempfile
import os
import errno
import sys, traceback
import subprocess

_active_lock = threading.Lock()
_die_flag = False

_active_pids = set()
_active_temp_files = set()

_terminated_pids_cond = threading.Condition(_active_lock)
_terminated_pids = {}

def add_temp_file(fname):
    with _active_lock:
        _active_temp_files.add(fname)
            
def remove_temp_file(fname):
    with _active_lock:
        _active_temp_files.remove(fname)
            
_old_os_wait3 = os.wait3
_select_select = select.select

def _retry_select(fd):
    while True:
        try:
            rrdy,_,_ = _select_select([fd],[],[])
            if fd in rrdy:
                return
        except select.error as e:
            if e[0] == errno.EINTR:
                continue
            raise

def _retry_read(fd):

    while True:
        try:
            return fd.read(1)
        except OSError as e:
            if e.errno == errno.EINTR:
                continue
            raise

def _retry_wait():

    while True:
        
        pid, status, e = _wait_no_hang()
        
        if pid>0:
            return pid, status
            
        elif pid==0:
            return 0,0
            
        elif pid == -1 and e == errno.ECHILD:
            return 0,0
            
        elif pid==-1 and e != errno.EINTR:
            raise OSError(e, 'unknown error in wait3()')

def _sigint_wait_thread_func(fd):

    global _die_flag
    
    while True:
        
        _retry_select(fd)
        _retry_read(fd) 

        with _active_lock:
        
            if _die_flag:
                os._exit(-1)
            
            _die_flag = True
        
            for pid in _active_pids:
                rc = _posix_kill(pid, signal.SIGINT)
    
            for fname in _active_temp_files:
                os.remove(fname)

            os._exit(-1)

def _child_wait_thread_func(fd):

    while True:
    
        _retry_select(fd)
        rc = _retry_read(fd)
                
        with _active_lock:
        
            while True:
            
                pid, status = _retry_wait()

                if pid==0:
                    break
                    
                if pid in _active_pids:
                    _active_pids.remove(pid)
                    
                _terminated_pids[pid] = status
                _terminated_pids_cond.notifyAll()

_sigint_pipe_read_fd = -1
_sigint_pipe_write_fd = -1

_sigchld_pipe_read_fd = -1
_sigchld_pipe_write_fd = -1

def _start_threads():

    global _sigint_pipe_read_fd, _sigint_pipe_write_fd
    
    _sigint_pipe_read_fd, _sigint_pipe_write_fd = os.pipe()
    sigint_read = os.fdopen(_sigint_pipe_read_fd, "r", 0 )
    
    sigint_wait_thread = threading.Thread(target=_sigint_wait_thread_func, name="SIGINT wait thread", args=(sigint_read,))
    sigint_wait_thread.setDaemon(True)
    sigint_wait_thread.start()
    
    install_sigint_handler(_sigint_pipe_write_fd)
    
    global _sigchld_pipe_read_fd, _sigchld_pipe_write_fd
  
    _sigchld_pipe_read_fd, _sigchld_pipe_write_fd = os.pipe()
    sigchld_read = os.fdopen(_sigchld_pipe_read_fd, "r", 0 )
    
    child_wait_thread = threading.Thread(target=_child_wait_thread_func, name="child process wait thread", args=(sigchld_read,))
    child_wait_thread.setDaemon(True)
    child_wait_thread.start()
    
    install_sigchld_handler(_sigchld_pipe_write_fd)

_close_on_fork = []

def close_on_fork(fd):
    _close_on_fork.append(fd)

def after_fork():

    _set_death_signal()

    global _close_on_fork

    for fd in _close_on_fork:
        os.close(fd)
        
    _close_on_fork = []

    os.close(_sigint_pipe_read_fd)
    os.close(_sigint_pipe_write_fd)
    
    os.close(_sigchld_pipe_read_fd)
    os.close(_sigchld_pipe_write_fd)
   
    global _active_lock
    _active_lock = threading.Lock()

    global _terminated_pids_cond
    _terminated_pids_cond = threading.Condition(_active_lock)
    
    global _terminated_pids
    _terminated_pids = {}
    
    global _active_pids
    _active_pids = set()
    
    global _active_temp_files
    _active_temp_files = set()
    
    _start_threads()

class _sigint_block_section(object):
    def __init__(self):
        self.blocked = False
        
    def __enter__(self):
        block_sigint()
        self.blocked = True
        
    def __exit__(self, type, value, traceback):
        self.release()
        
    def release(self):
        if self.blocked:
            self.blocked = False
            unblock_sigint()

_old_os_fork = os.fork

def _fork():

    ppid = os.getpid()

    with _sigint_block_section() as cs:
    
        with _active_lock:
        
            if _die_flag:
                os._exit(-1)
    
            pid = _old_os_fork()
    
            if pid == 0:
                after_fork()
            
            if pid > 0:
                _active_pids.add(pid)  
    
            return pid
    
def _waitpid(pid, options=0):
    
    while True:
    
        with _active_lock:
        
            if pid in _terminated_pids:
                status = _terminated_pids[pid]
                del _terminated_pids[pid]
                return pid, status
            
            if options==os.WNOHANG:
                return 0, 0
                
            _terminated_pids_cond.wait()

def _wait(options=0):

    while True:
        
        with _active_lock:
        
            for pid, status in _terminated_pids.iteritems():
                del _terminated_pids[pid]
                return pid, status
            
            if options==os.WNOHANG:
                return 0, 0
                
            _terminated_pids_cond.wait()

_old_os_kill = os.kill

def _kill(pid, sig):

    with _active_lock:
    
        if pid in _terminated_pids:
            return None
            
        return _old_os_kill(pid,sig)
    
os.kill = _kill
os.fork = _fork
os.wait = _wait
os.waitpid = _waitpid

def _split_command_line(cmd):
    
    args = []
    
    i=0
    
    while i<len(cmd):
        
        while i<len(cmd) and cmd[i] in [' ','\t','\f']:
            i += 1
            
        if i >= len(cmd):
            break
            
        arg = []
        
        in_quotes = None

        while i<len(cmd):
            
            if not in_quotes and cmd[i] in ['\'','\"','\'']:
                in_quotes = cmd[i]
                
            elif in_quotes and cmd[i]==in_quotes:
                in_quotes = None
                
            elif cmd[i] == '\\' and i<(len(cmd)+1):
                
                i += 1
                
                if cmd[i]=='\\':
                    arg.append('\\')
                elif cmd[i]=='\'':
                    arg.append('\'')
                elif cmd[i]=='\"':
                    arg.append('\'')
                elif cmd[i]=='\"':
                    arg.append('\"')
                elif cmd[i]=='a':
                    arg.append('\a')
                elif cmd[i]=='b':
                    arg.append('\b')
                elif cmd[i]=='n':
                    arg.append('\n')
                elif cmd[i]=='f':
                    arg.append('\f')
                elif cmd[i]=='r':
                    arg.append('\r')
                elif cmd[i]=='t':
                    arg.append('\t')
                elif cmd[i]=='v':
                    arg.append('\v')
                else:
                    arg.append(cmd[i])
                    
            elif not in_quotes and cmd[i] in [' ','\t','\f']:
                break
                
            else:
                arg.append(cmd[i])
                
            i += 1
        
        args.append( "".join(arg) )
    
    return args


def system(cmd):

    args = _split_command_line(cmd)
    
    if args[-2] == '>':
    
        with open(args[-1],'w') as fout:
            p = subprocess.Popen(args[:-2], stdout=fout)
            rc = p.wait()
            return rc
         
    else:
        p = subprocess.Popen(args)
        return p.wait()

def tmpfile(prefix, suffix):

    with _active_lock:
        with tempfile.NamedTemporaryFile(delete=False, prefix=prefix, suffix=suffix) as file:
            _active_temp_files.add(file.name)
            return file.name
       
def tmpfile_remove(fname, leave):

    with _active_lock:
        os.remove(fname)
        _active_temp_files.remove(fname) 

pyabc_internal_set_util_callbacks( system, tmpfile,tmpfile_remove )


_start_threads()


_registered_commands = {}

def _cmd_callback(args):
    try:
        assert len(args) > 0
        
        cmd = args[0]
        assert cmd in _registered_commands
    
        res = _registered_commands[cmd](args)
        
        assert type(res) == int, "User-defined Python command must return an integer."
        
        return res
    
    except Exception, e:
        import traceback
        traceback.print_exc()

    except SystemExit, se:
        pass
            
    return 0
    
pyabc_internal_set_command_callback( _cmd_callback )
    
def add_abc_command(fcmd, group, cmd, change):
    _registered_commands[ cmd ] = fcmd
    pyabc_internal_register_command( group, cmd, change)

import optparse

xxx = {} 

def cmd_python(cmd_args):
    
    usage = "usage: %prog [options] <Python files>"
    
    parser = optparse.OptionParser(usage, prog="python")
    
    parser.add_option("-c", "--cmd", dest="cmd", help="Execute Python command directly")
    parser.add_option("-v", "--version", action="store_true", dest="version", help="Display Python Version")

    options, args = parser.parse_args(cmd_args)
    
    if options.version:
        print sys.version
        return 0
    
    if options.cmd:
        exec options.cmd in xxx
        return 0
    
    scripts_dir = os.getenv('ABC_PYTHON_SCRIPTS', ".")
    scripts_dirs = scripts_dir.split(':')
    
    for fname in args[1:]:
        if os.path.isabs(fname):
            execfile(fname, xxx)
        else:
            for d in scripts_dirs:
                fname = os.path.join(scripts_dir, fname)
                if os.path.exists(fname):
                    execfile(fname, xxx)
                    break
    
    return 0
    
add_abc_command(cmd_python, "Python", "python", 0) 

%}