[bus] Add WvBusGetNodeCount function
[people/sha0/winvblock.git] / src / include / bus.h
1 /**
2  * Copyright (C) 2009-2010, Shao Miller <shao.miller@yrdsb.edu.on.ca>.
3  * Copyright 2006-2008, V.
4  * For WinAoE contact information, see http://winaoe.org/
5  *
6  * This file is part of WinVBlock, derived from WinAoE.
7  *
8  * WinVBlock is free software: you can redistribute it and/or modify
9  * it under the terms of the GNU General Public License as published by
10  * the Free Software Foundation, either version 3 of the License, or
11  * (at your option) any later version.
12  *
13  * WinVBlock is distributed in the hope that it will be useful,
14  * but WITHOUT ANY WARRANTY; without even the implied warranty of
15  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
16  * GNU General Public License for more details.
17  *
18  * You should have received a copy of the GNU General Public License
19  * along with WinVBlock.  If not, see <http://www.gnu.org/licenses/>.
20  */
21 #ifndef WV_M_BUS_H_
22 #  define WV_M_BUS_H_
23
24 /**
25  * @file
26  *
27  * Bus specifics.
28  */
29
30 /**
31  * A bus thread routine.
32  *
33  * @v bus       The bus to be used in the thread routine.
34  *
35  * If you implement your own bus thread routine, you should call
36  * WvBusProcessWorkItems() within its loop.
37  */
38 typedef void STDCALL WV_F_BUS_THREAD(IN WV_SP_BUS_T);
39 typedef WV_F_BUS_THREAD * WV_FP_BUS_THREAD;
40
41 /**
42  * A bus PnP routine.
43  *
44  * @v bus               The bus to receive the PnP IRP.
45  * @v irp               The IRP to process.
46  * @ret NTSTATUS        The status of the operation.
47  */
48 typedef NTSTATUS STDCALL WV_F_BUS_PNP(IN WV_SP_BUS_T, IN PIRP);
49 typedef WV_F_BUS_PNP * WV_FP_BUS_PNP;
50
51 /* Device state. */
52 typedef enum WV_BUS_STATE {
53     WvBusStateNotStarted,
54     WvBusStateStarted,
55     WvBusStateStopPending,
56     WvBusStateStopped,
57     WvBusStateRemovePending,
58     WvBusStateSurpriseRemovePending,
59     WvBusStateDeleted,
60     WvBusStates
61   } WV_E_BUS_STATE, * WV_EP_BUS_STATE;
62
63 /* The bus type. */
64 typedef struct WV_BUS_T {
65     PDEVICE_OBJECT LowerDeviceObject;
66     PDEVICE_OBJECT PhysicalDeviceObject;
67     PDEVICE_OBJECT Fdo;
68     winvblock__uint32 Children;
69     WV_SP_DEV_T first_child;
70     WV_FP_BUS_THREAD Thread;
71     KEVENT ThreadSignal;
72     winvblock__bool Stop;
73     KEVENT ThreadStopped;
74     WV_E_BUS_STATE OldState;
75     WV_E_BUS_STATE State;
76     WV_FP_BUS_PNP QueryDevText;
77     struct {
78         LIST_ENTRY Nodes;
79         USHORT NodeCount;
80         LIST_ENTRY WorkItems;
81         KSPIN_LOCK WorkItemsLock;
82       } BusPrivate_;
83   } WV_S_BUS_T, * WV_SP_BUS_T;
84
85 /* A child PDO node on a bus.  Treat this as an opaque type. */
86 typedef struct WV_BUS_NODE {
87     struct {
88         LIST_ENTRY Link;
89         PDEVICE_OBJECT Pdo;
90         WV_SP_BUS_T Bus;
91         /* The child's unit number relative to the parent bus. */
92         winvblock__uint32 Num;
93       } BusPrivate_;
94   } WV_S_BUS_NODE, * WV_SP_BUS_NODE;
95
96 /* Exports. */
97 extern winvblock__lib_func void WvBusInit(WV_SP_BUS_T);
98 extern winvblock__lib_func WV_SP_BUS_T WvBusCreate(void);
99 extern winvblock__lib_func void WvBusProcessWorkItems(WV_SP_BUS_T);
100 extern winvblock__lib_func void WvBusCancelWorkItems(WV_SP_BUS_T);
101 extern winvblock__lib_func NTSTATUS WvBusStartThread(WV_SP_BUS_T);
102 extern winvblock__lib_func winvblock__bool STDCALL WvBusInitNode(
103     OUT WV_SP_BUS_NODE,
104     IN PDEVICE_OBJECT
105   );
106 extern winvblock__lib_func NTSTATUS STDCALL WvBusAddNode(
107     WV_SP_BUS_T,
108     WV_SP_BUS_NODE
109   );
110 extern winvblock__lib_func NTSTATUS STDCALL WvBusRemoveNode(WV_SP_BUS_NODE);
111 extern winvblock__lib_func NTSTATUS STDCALL WvBusEnqueueIrp(WV_SP_BUS_T, PIRP);
112 extern winvblock__lib_func NTSTATUS STDCALL WvBusSysCtl(
113     IN WV_SP_BUS_T,
114     IN PIRP
115   );
116 extern winvblock__lib_func NTSTATUS STDCALL WvBusPower(
117     IN WV_SP_BUS_T,
118     IN PIRP
119   );
120 /* IRP_MJ_PNP dispatcher in bus/pnp.c */
121 extern winvblock__lib_func NTSTATUS STDCALL WvBusPnp(
122     IN WV_SP_BUS_T,
123     IN PIRP,
124     IN UCHAR
125   );
126 extern winvblock__lib_func winvblock__uint32 STDCALL WvBusGetNodeNum(
127     IN WV_SP_BUS_NODE
128   );
129 extern winvblock__lib_func WV_SP_BUS_NODE STDCALL WvBusGetNextNode(
130     IN WV_SP_BUS_T,
131     IN WV_SP_BUS_NODE
132   );
133 extern winvblock__lib_func PDEVICE_OBJECT STDCALL WvBusGetNodePdo(
134     IN WV_SP_BUS_NODE
135   );
136 extern winvblock__lib_func winvblock__uint32 STDCALL WvBusGetNodeCount(
137     WV_SP_BUS_T
138   );
139
140 #endif  /* WV_M_BUS_H_ */