203 	struct tb_dma_port *dma;  in dma_port_alloc()  local
210 	dma = kzalloc(sizeof(*dma), GFP_KERNEL);  in dma_port_alloc()
211 	if (!dma)  in dma_port_alloc()
214 	dma->buf = kmalloc_array(MAIL_DATA_DWORDS, sizeof(u32), GFP_KERNEL);  in dma_port_alloc()
215 	if (!dma->buf) {  in dma_port_alloc()
216 		kfree(dma);  in dma_port_alloc()
220 	dma->sw = sw;  in dma_port_alloc()
221 	dma->port = port;  in dma_port_alloc()
222 	dma->base = DMA_PORT_CAP;  in dma_port_alloc()
224 	return dma;  in dma_port_alloc()
231 void dma_port_free(struct tb_dma_port *dma)  in dma_port_free()  argument
233 	if (dma) {  in dma_port_free()
234 		kfree(dma->buf);  in dma_port_free()
235 		kfree(dma);  in dma_port_free()
239 static int dma_port_wait_for_completion(struct tb_dma_port *dma,  in dma_port_wait_for_completion()  argument
243 	struct tb_switch *sw = dma->sw;  in dma_port_wait_for_completion()
249 		ret = dma_port_read(sw->tb->ctl, &in, tb_route(sw), dma->port,  in dma_port_wait_for_completion()
250 				    dma->base + MAIL_IN, 1, 50);  in dma_port_wait_for_completion()
278 static int dma_port_request(struct tb_dma_port *dma, u32 in,  in dma_port_request()  argument
281 	struct tb_switch *sw = dma->sw;  in dma_port_request()
285 	ret = dma_port_write(sw->tb->ctl, &in, tb_route(sw), dma->port,  in dma_port_request()
286 			     dma->base + MAIL_IN, 1, DMA_PORT_TIMEOUT);  in dma_port_request()
290 	ret = dma_port_wait_for_completion(dma, timeout);  in dma_port_request()
294 	ret = dma_port_read(sw->tb->ctl, &out, tb_route(sw), dma->port,  in dma_port_request()
295 			    dma->base + MAIL_OUT, 1, DMA_PORT_TIMEOUT);  in dma_port_request()
302 static int dma_port_flash_read_block(struct tb_dma_port *dma, u32 address,  in dma_port_flash_read_block()  argument
305 	struct tb_switch *sw = dma->sw;  in dma_port_flash_read_block()
318 	ret = dma_port_request(dma, in, DMA_PORT_TIMEOUT);  in dma_port_flash_read_block()
322 	return dma_port_read(sw->tb->ctl, buf, tb_route(sw), dma->port,  in dma_port_flash_read_block()
323 			     dma->base + MAIL_DATA, dwords, DMA_PORT_TIMEOUT);  in dma_port_flash_read_block()
326 static int dma_port_flash_write_block(struct tb_dma_port *dma, u32 address,  in dma_port_flash_write_block()  argument
329 	struct tb_switch *sw = dma->sw;  in dma_port_flash_write_block()
336 	ret = dma_port_write(sw->tb->ctl, buf, tb_route(sw), dma->port,  in dma_port_flash_write_block()
337 			    dma->base + MAIL_DATA, dwords, DMA_PORT_TIMEOUT);  in dma_port_flash_write_block()
353 	return dma_port_request(dma, in, DMA_PORT_TIMEOUT);  in dma_port_flash_write_block()
363 int dma_port_flash_read(struct tb_dma_port *dma, unsigned int address,  in dma_port_flash_read()  argument
376 		ret = dma_port_flash_read_block(dma, address, dma->buf,  in dma_port_flash_read()
387 		memcpy(buf, dma->buf + offset, nbytes);  in dma_port_flash_read()
408 int dma_port_flash_write(struct tb_dma_port *dma, unsigned int address,  in dma_port_flash_write()  argument
427 		memcpy(dma->buf + offset, buf, nbytes);  in dma_port_flash_write()
429 		ret = dma_port_flash_write_block(dma, address, buf, nbytes);  in dma_port_flash_write()
458 int dma_port_flash_update_auth(struct tb_dma_port *dma)  in dma_port_flash_update_auth()  argument
465 	return dma_port_request(dma, in, 150);  in dma_port_flash_update_auth()
481 int dma_port_flash_update_auth_status(struct tb_dma_port *dma, u32 *status)  in dma_port_flash_update_auth_status()  argument
483 	struct tb_switch *sw = dma->sw;  in dma_port_flash_update_auth_status()
487 	ret = dma_port_read(sw->tb->ctl, &out, tb_route(sw), dma->port,  in dma_port_flash_update_auth_status()
488 			    dma->base + MAIL_OUT, 1, DMA_PORT_TIMEOUT);  in dma_port_flash_update_auth_status()
511 int dma_port_power_cycle(struct tb_dma_port *dma)  in dma_port_power_cycle()  argument
518 	return dma_port_request(dma, in, 150);  in dma_port_power_cycle()